fix(MemberExpression): 🐛UnaryExpression

Author: vivaxy

lib/__tests__/fixtures/parse/member-expression-005/input.js

@@ -0,0 +1 @@
+++a.b

lib/__tests__/fixtures/parse/member-expression-005/output.json

@@ -0,0 +1,25 @@
+{
+  "type": "Program",
+  "body": [
+    {
+      "type": "ExpressionStatement",
+      "expression": {
+        "type": "UpdateExpression",
+        "operator": "++",
+        "argument": {
+          "type": "MemberExpression",
+          "object": {
+            "type": "Identifier",
+            "name": "a"
+          },
+          "property": {
+            "type": "Identifier",
+            "name": "b"
+          },
+          "computed": false
+        },
+        "prefix": true
+      }
+    }
+  ]
+}

lib/__tests__/parse.js

@@ -17,7 +17,7 @@ test('parse', async (t) => {
     onlyDirectories: true,
   })).filter((x) => {
     return true;
-    return x === 'conditional-expression-000';
+    return x === 'member-expression-005';
   });
 
   const tested = [];

lib/parse.js

@@ -336,6 +336,12 @@ module.exports = function parse(input) {
           );
           return (operator = null);
         }
+        if (astFactory.isUpdateExpressionOperator(operator)) {
+          expressionStatement = new astFactory.ExpressionStatement(
+            new astFactory.UpdateExpression(operator, node, true)
+          );
+          return (operator = null);
+        }
         return (expressionStatement = new astFactory.ExpressionStatement(node));
       }
 
@@ -470,6 +476,7 @@ module.exports = function parse(input) {
           );
           break;
         case astTypes.UNARY_EXPRESSION:
+        case astTypes.UPDATE_EXPRESSION:
           parent.argument = new astFactory.MemberExpression(
             parent.argument,
             expression,
@@ -607,7 +614,10 @@ module.exports = function parse(input) {
     }
 
     function checkIdentifier(node) {
-      if (node.type !== astTypes.IDENTIFIER) {
+      if (
+        node.type !== astTypes.IDENTIFIER &&
+        node.type !== astTypes.MEMBER_EXPRESSION
+      ) {
         throwError('Update operator only works on identifier nodes');
       }
     }
@@ -655,6 +665,11 @@ module.exports = function parse(input) {
           node = ast.property;
           traverse(node);
           break;
+        case astTypes.UPDATE_EXPRESSION:
+          parent = ast;
+          node = ast.argument;
+          traverse(node);
+          break;
         case astTypes.LITERAL:
         case astTypes.IDENTIFIER:
           break;