fix(transform): correct chained optional expression (#17)

input

```js
a ?. b . c
a ?. b . c ( )
a ?. b ( ) . c
a ?. b ( ) . c ( )
```

before

```json5
{ type: "MemberExpression" }
{ type: "CallExpression" }
{ type: "MemberExpression" }
{ type: "CallExpression" }
```

after

```json5
{ type: "OptionalMemberExpression" }
{ type: "OptionalCallExpression" }
{ type: "OptionalMemberExpression" }
{ type: "OptionalCallExpression" }
```

Author: ikatyang

src/transform.ts

@@ -184,6 +184,9 @@ export const transform = (node: InputNode, context: Context): OutputNode => {
         );
       } else {
         const object = _t<b.Expression>(receiver);
+        const isObjectOptionalExpression =
+          object.type === 'OptionalMemberExpression' ||
+          object.type === 'OptionalCallExpression';
         const propertyStart = _findBackChar(
           /\S/,
           _findBackChar(/\./, object.end) + 1,
@@ -196,15 +199,24 @@ export const transform = (node: InputNode, context: Context): OutputNode => {
         callExpression.callee = _c<
           b.MemberExpression | b.OptionalMemberExpression
         >(
-          isOptional ? 'OptionalMemberExpression' : 'MemberExpression',
+          isOptional || isObjectOptionalExpression
+            ? 'OptionalMemberExpression'
+            : 'MemberExpression',
           {
             computed: false,
             object,
             property,
-            ...(isOptional && { optional: true }),
+            ...(isOptional
+              ? { optional: true }
+              : isObjectOptionalExpression
+                ? { optional: false }
+                : null),
           },
           { start: object.start, end: property.end },
         );
+        if (callExpression.callee.type === 'OptionalMemberExpression') {
+          callExpression.type = 'OptionalCallExpression';
+        }
       }
       return callExpression;
     }
@@ -239,13 +251,26 @@ export const transform = (node: InputNode, context: Context): OutputNode => {
               ),
             });
       }
+      const object = _t<b.Expression>(receiver);
+      const isObjectOptionalExpression =
+        object.type === 'OptionalMemberExpression' ||
+        object.type === 'OptionalCallExpression';
       const memberExpression = _c<
         b.MemberExpression | b.OptionalMemberExpression
-      >(isOptional ? 'OptionalMemberExpression' : 'MemberExpression', {
-        computed: false,
-        object: _t<b.Expression>(receiver),
-        ...(isOptional && { optional: true }),
-      });
+      >(
+        isOptional || isObjectOptionalExpression
+          ? 'OptionalMemberExpression'
+          : 'MemberExpression',
+        {
+          computed: false,
+          object,
+          ...(isOptional
+            ? { optional: true }
+            : isObjectOptionalExpression
+              ? { optional: false }
+              : null),
+        },
+      );
       memberExpression.property = _c<b.Identifier>(
         'Identifier',
         { name },

tests/trasnform.test.ts

@@ -44,13 +44,17 @@ describe.each`
   ${'LiteralPrimitive'} | ${'StringLiteral'}            | ${' "hello" '}              | ${true}  | ${true}  | ${true}  | ${true}
   ${'MethodCall'}       | ${'CallExpression'}           | ${' a . b ( 1 , 2 ) '}      | ${true}  | ${true}  | ${true}  | ${true}
   ${'MethodCall'}       | ${'CallExpression'}           | ${' a ( 1 , 2 ) '}          | ${true}  | ${true}  | ${true}  | ${true}
+  ${'MethodCall'}       | ${'OptionalCallExpression'}   | ${' a ?. b . c ( ) '}       | ${true}  | ${true}  | ${true}  | ${true}
+  ${'MethodCall'}       | ${'OptionalCallExpression'}   | ${' a ?. b ( ) . c ( ) '}   | ${true}  | ${true}  | ${true}  | ${true}
   ${'NonNullAssert'}    | ${'TSNonNullExpression'}      | ${' x ! '}                  | ${true}  | ${true}  | ${true}  | ${true}
   ${'PrefixNot'}        | ${'UnaryExpression'}          | ${' ! x '}                  | ${true}  | ${true}  | ${true}  | ${true}
   ${'PropertyRead'}     | ${'Identifier'}               | ${' ( ( a ) ) '}            | ${true}  | ${true}  | ${true}  | ${true}
   ${'PropertyRead'}     | ${'Identifier'}               | ${' a '}                    | ${true}  | ${true}  | ${true}  | ${true}
   ${'PropertyRead'}     | ${'Identifier'}               | ${' a // hello '}           | ${true}  | ${true}  | ${true}  | ${true}
   ${'PropertyRead'}     | ${'MemberExpression'}         | ${' a . b '}                | ${true}  | ${true}  | ${true}  | ${true}
   ${'PropertyRead'}     | ${'MemberExpression'}         | ${' this . a '}             | ${true}  | ${true}  | ${true}  | ${true}
+  ${'PropertyRead'}     | ${'OptionalMemberExpression'} | ${' a ?. b . c '}           | ${true}  | ${true}  | ${true}  | ${true}
+  ${'PropertyRead'}     | ${'OptionalMemberExpression'} | ${' a ?. b ( ) . c '}       | ${true}  | ${true}  | ${true}  | ${true}
   ${'PropertyWrite'}    | ${'AssignmentExpression'}     | ${' a . b = 1 '}            | ${true}  | ${false} | ${false} | ${false}
   ${'PropertyWrite'}    | ${'AssignmentExpression'}     | ${' a = 1 '}                | ${true}  | ${false} | ${false} | ${false}
   ${'Quote'}            | ${'NGQuotedExpression'}       | ${' javascript : void(0) '} | ${false} | ${true}  | ${true}  | ${false}