From 5a2f9f2f8ae87064c080a48aad8d0781954a39aa Mon Sep 17 00:00:00 2001
From: ZHAO Jinxiang <xiaoxiangmoe@gmail.com>
Date: Sun, 14 Nov 2021 15:19:10 +0800
Subject: [PATCH] fix: support optional chaining operator

---
 src/core/identifiers.ts                   |  3 +++
 test/__snapshots__/transform.test.ts.snap | 12 +++++++++---
 test/fixtures/MacrosDefineExpose.vue      | 15 ++++++++++-----
 3 files changed, 22 insertions(+), 8 deletions(-)

diff --git a/src/core/identifiers.ts b/src/core/identifiers.ts
index e8e0ae3..919fe0d 100644
--- a/src/core/identifiers.ts
+++ b/src/core/identifiers.ts
@@ -70,6 +70,9 @@ export function getIdentifierUsages(node?: Expression | TSType | SpreadElement |
   else if (node.type === 'MemberExpression') {
     getIdentifierUsages(node.object, identifiers)
   }
+  else if (node.type === 'OptionalMemberExpression') {
+    getIdentifierUsages(node.object, identifiers)
+  }
   else if (node.type === 'CallExpression') {
     getIdentifierUsages(node.callee as Expression, identifiers)
     node.arguments.forEach(arg => getIdentifierUsages(arg as Expression, identifiers))
diff --git a/test/__snapshots__/transform.test.ts.snap b/test/__snapshots__/transform.test.ts.snap
index fafc642..9a03518 100644
--- a/test/__snapshots__/transform.test.ts.snap
+++ b/test/__snapshots__/transform.test.ts.snap
@@ -251,17 +251,23 @@ export default __sfc_main;
 
 exports[`transform fixtures test/fixtures/MacrosDefineExpose.vue 1`] = `
 "<template>
-  <div>{{a}}</div>
+  <div>{{ a }}</div>
+  <div>{{ text?.length ?? textLengthDefault }}</div>
 </template>
 
-<script lang=\\"js\\">
+<script lang=\\"ts\\">
+import { ref } from '@vue/composition-api';
 const __sfc_main = {};
 
 __sfc_main.setup = (__props, __ctx) => {
   const a = ref(1);
   const b = 1;
+  const text = 'hello';
+  const textLengthDefault = 0;
   return Object.assign({
-    a
+    a,
+    text,
+    textLengthDefault
   }, {
     b
   });
diff --git a/test/fixtures/MacrosDefineExpose.vue b/test/fixtures/MacrosDefineExpose.vue
index 1dbbb19..10852ee 100644
--- a/test/fixtures/MacrosDefineExpose.vue
+++ b/test/fixtures/MacrosDefineExpose.vue
@@ -1,9 +1,14 @@
 <template>
-  <div>{{a}}</div>
+  <div>{{ a }}</div>
+  <div>{{ text?.length ?? textLengthDefault }}</div>
 </template>
 
-<script setup lang="js">
-const a = ref(1);
-const b = 1;
-defineExpose({b});
+<script setup lang="ts">
+import { ref } from '@vue/composition-api'
+
+const a = ref(1)
+const b = 1
+const text = 'hello'
+const textLengthDefault = 0
+defineExpose({ b })
 </script>