feat(lang): add uspport for computed prop access e.g. `foo.(Symbol.it…

…erator)`
fink-lang · Feb 24, 2020 · 40fe6c6 · 40fe6c6
1 parent e450dcc
commit 40fe6c6
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/src/lang/prop-access/index.js b/src/lang/prop-access/index.js
@@ -6,8 +6,8 @@ import {symbol} from '../symbols';
 
 
 const member_expr = (ctx, lbp)=> {
-  // TODO: next tokens can only be a string or an identifier
-  if (next_is(ctx, '`')) {
+  // TODO: next tokens can only be a string or an identifier or a group
+  if (next_is(ctx, '`') || next_is(ctx, '(')) {
     return expression(ctx, lbp);
   }
   const key_ctx = advance(ctx);

diff --git a/src/lang/prop-access/index.test.js b/src/lang/prop-access/index.test.js
@@ -20,6 +20,7 @@ describe('prop access', ()=> {
     });
   });
 
+
   it('parses member expr: foo.`bar spam`', ()=> {
     expect(
       parse_expr('foo.`bar spam`')
@@ -34,4 +35,20 @@ describe('prop access', ()=> {
       }
     });
   });
+
+
+  it('parses member expr: foo.(Symbol.iterator)', ()=> {
+    expect(
+      parse_expr('foo.(Symbol.iterator)')
+    ).toEqual({
+      type: 'member',
+      op: '.',
+      left: parse_expr(`foo`),
+      right: parse_expr('    (Symbol.iterator)'),
+      loc: {
+        start: {pos: 0, line: 1, column: 0},
+        end: {pos: 21, line: 1, column: 21}
+      }
+    });
+  });
 });