From d8a9be0886a603993846222803a48a0cc93d5d4a Mon Sep 17 00:00:00 2001
From: daiwei <daiwei521@126.com>
Date: Sat, 10 Apr 2021 16:38:22 +0800
Subject: [PATCH] fix(compiler-sfc): ref sugar work with destructuring + arrow
 function expression

---
 .../__snapshots__/compileScript.spec.ts.snap      | 15 +++++++++++++++
 .../compiler-sfc/__tests__/compileScript.spec.ts  | 11 +++++++++++
 packages/compiler-sfc/src/compileScript.ts        |  7 ++++---
 3 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/packages/compiler-sfc/__tests__/__snapshots__/compileScript.spec.ts.snap b/packages/compiler-sfc/__tests__/__snapshots__/compileScript.spec.ts.snap
index 7397228fb3b..ac2b3ba810f 100644
--- a/packages/compiler-sfc/__tests__/__snapshots__/compileScript.spec.ts.snap
+++ b/packages/compiler-sfc/__tests__/__snapshots__/compileScript.spec.ts.snap
@@ -693,6 +693,21 @@ return { a, b, test }
 }"
 `;
 
+exports[`SFC compile <script setup> ref: syntax sugar work with destructuring + arrow function expression 1`] = `
+"export default {
+  expose: [],
+  setup(__props) {
+
+      const {} = () => 1;
+      const {} = useSomthing(() => 1);
+      const {} = useSomthing(computed(() => 1));
+      
+return {  }
+}
+
+}"
+`;
+
 exports[`SFC compile <script setup> should expose top level declarations 1`] = `
 "import { x } from './x'
       
diff --git a/packages/compiler-sfc/__tests__/compileScript.spec.ts b/packages/compiler-sfc/__tests__/compileScript.spec.ts
index d74d7448aa9..5d52e84e2f3 100644
--- a/packages/compiler-sfc/__tests__/compileScript.spec.ts
+++ b/packages/compiler-sfc/__tests__/compileScript.spec.ts
@@ -656,6 +656,17 @@ const emit = defineEmit(['a', 'b'])
       })
     })
 
+    test('work with destructuring + arrow function expression', () => {
+      const { content } = compile(`
+      <script setup>
+      ref: ({} = () => 1);
+      ref: ({} = useSomthing(() => 1));
+      ref: ({} = useSomthing(computed(() => 1)));
+      </script>
+      `)
+      assertCode(content)
+    })
+
     test('multi ref declarations', () => {
       const { content, bindings } = compile(`<script setup>
       ref: a = 1, b = 2, c = {
diff --git a/packages/compiler-sfc/src/compileScript.ts b/packages/compiler-sfc/src/compileScript.ts
index 5bb8a2461a9..3b8d499073a 100644
--- a/packages/compiler-sfc/src/compileScript.ts
+++ b/packages/compiler-sfc/src/compileScript.ts
@@ -348,10 +348,11 @@ export function compileScript(
             break
           }
         }
-        for (let i = left.end!; i > 0; i++) {
-          const char = source[i + startOffset]
+
+        for (let i = right.end! + startOffset; i > right.start!; i--) {
+          const char = source[i]
           if (char === ')') {
-            s.remove(i + startOffset, i + startOffset + 1)
+            s.remove(i, i + 1)
             break
           }
         }