Add shorthand support to parsers

src/parser.js

···
   let currentGroup = null;
   let lastMatch;
   let currentSequence = rootSequence;
+  let capture;
 
   for (
     let quasiIndex = 0, stackIndex = 0;
···
     stackIndex++
   ) {
     if (stackIndex % 2 !== 0) {
-      currentSequence.push({
-        expression: expressions[stackIndex++ >> 1],
-      });
+      const expression = expressions[stackIndex++ >> 1];
+      currentSequence.push({ expression, capture });
+      capture = undefined;
     }
 
     const quasi = quasis[stackIndex >> 1];
···
         if (!currentSequence) syntaxError(char);
       } else if (char === '(') {
         sequenceStack.push(currentSequence);
-        currentSequence.push((currentGroup = { sequence: [] }));
+        currentSequence.push((currentGroup = { sequence: [], capture }));
         currentSequence = currentGroup.sequence;
+        capture = undefined;
+      } else if (char === ':' || char === '=' || char === '!') {
+        capture = char;
+        const nextChar = quasi[quasiIndex];
+        if (quasi[quasiIndex] && quasi[quasiIndex] !== '(') syntaxError(char);
       } else if (char === '?' && !currentSequence.length && currentGroup) {
-        const nextChar = quasi[quasiIndex++];
-        if (nextChar === ':' || nextChar === '=' || nextChar === '!') {
-          currentGroup.capture = nextChar;
+        capture = quasi[quasiIndex++];
+        if (capture === ':' || capture === '=' || capture === '!') {
+          currentGroup.capture = capture;
+          capture = undefined;
         } else {
           syntaxError(char);
         }

src/parser.test.js

···
   expect(ast).toHaveProperty('0.sequence.0.quantifier', undefined);
 });
 
-it('supports non-capturing groups', () => {
-  const ast = parseTag`(?: ${1})`;
-  expect(ast).toHaveProperty('length', 1);
-  expect(ast).toHaveProperty('0.capture', ':');
-  expect(ast).toHaveProperty('0.sequence.length', 1);
+describe('non-capturing syntax', () => {
+  it('supports regex-like syntax', () => {
+    const ast = parseTag`(?: ${1})`;
+    expect(ast).toHaveProperty('length', 1);
+    expect(ast).toHaveProperty('0.capture', ':');
+    expect(ast).toHaveProperty('0.sequence.length', 1);
+  });
+
+  it('supports shorthand', () => {
+    let ast = parseTag`:${1}`;
+    expect(ast).toHaveProperty('length', 1);
+    expect(ast).toHaveProperty('0.capture', ':');
+    expect(ast).toHaveProperty('0.expression', 1);
+    ast = parseTag`:(${1})`;
+    expect(ast).toHaveProperty('length', 1);
+    expect(ast).toHaveProperty('0.capture', ':');
+    expect(ast).toHaveProperty('0.sequence.length', 1);
+  });
+
+  it('fails on invalid usage', () => {
+    expect(() => parseTag`${1} : ${2}`).toThrow();
+    expect(() => parseTag`${1} :|${2}`).toThrow();
+  });
 });
 
-it('supports positive lookahead groups', () => {
-  const ast = parseTag`(?= ${1})`;
-  expect(ast).toHaveProperty('length', 1);
-  expect(ast).toHaveProperty('0.capture', '=');
-  expect(ast).toHaveProperty('0.sequence.length', 1);
+describe('positive lookaheads syntax', () => {
+  it('supports regex-like syntax', () => {
+    const ast = parseTag`(?= ${1})`;
+    expect(ast).toHaveProperty('length', 1);
+    expect(ast).toHaveProperty('0.capture', '=');
+    expect(ast).toHaveProperty('0.sequence.length', 1);
+  });
+
+  it('supports shorthand', () => {
+    let ast = parseTag`=${1}`;
+    expect(ast).toHaveProperty('length', 1);
+    expect(ast).toHaveProperty('0.capture', '=');
+    expect(ast).toHaveProperty('0.expression', 1);
+    ast = parseTag`=(${1})`;
+    expect(ast).toHaveProperty('length', 1);
+    expect(ast).toHaveProperty('0.capture', '=');
+    expect(ast).toHaveProperty('0.sequence.length', 1);
+  });
 });
 
-it('supports negative lookahead groups', () => {
-  const ast = parseTag`(?! ${1})`;
-  expect(ast).toHaveProperty('length', 1);
-  expect(ast).toHaveProperty('0.capture', '!');
-  expect(ast).toHaveProperty('0.sequence.length', 1);
+describe('negative lookaheads syntax', () => {
+  it('supports regex-like syntax', () => {
+    const ast = parseTag`(?! ${1})`;
+    expect(ast).toHaveProperty('length', 1);
+    expect(ast).toHaveProperty('0.capture', '!');
+    expect(ast).toHaveProperty('0.sequence.length', 1);
+  });
+
+  it('supports shorthand', () => {
+    let ast = parseTag`!${1}`;
+    expect(ast).toHaveProperty('length', 1);
+    expect(ast).toHaveProperty('0.capture', '!');
+    expect(ast).toHaveProperty('0.expression', 1);
+    ast = parseTag`!(${1})`;
+    expect(ast).toHaveProperty('length', 1);
+    expect(ast).toHaveProperty('0.capture', '!');
+    expect(ast).toHaveProperty('0.sequence.length', 1);
+  });
 });
 
 it('supports groups with alternates', () => {
   expect(parseTag`(${1} | ${2}) ${3}`).toMatchInlineSnapshot(`
     Array [
       Object {
+        "capture": undefined,
         "sequence": Array [
           Object {
+            "capture": undefined,
             "expression": 1,
           },
         ],
       },
       Object {
+        "capture": undefined,
         "expression": 3,
       },
     ]