feat: 解析Vue script时自动识别setup的解析

packages/gogocode-core/src/vue-core/core.js

@@ -21,7 +21,8 @@ const core = {
  if (ast.scriptAst) {
  newAst = ast.scriptAst;
  } else {
- ast.scriptAst = core.getScript(ast, { parseOptions });
+ const isSetup = ast.scriptSetup && ast.scriptSetup.setup;
+ ast.scriptAst = core.getScript(ast, { isSetup, parseOptions });
  newAst = ast.scriptAst;
  }
  } else if (selector == '<script setup></script>') {
@@ -52,7 +53,7 @@ const core = {
  return undefined;
  }
  },
- getScript(ast, { isSetup = false, parseOptions } = {} ) {
+ getScript(ast, { isSetup = false, parseOptions } = {}) {
  // 仅针对vue，取script，后续通过jscore处理
  let content;
  if (isSetup && ast.scriptSetup) {
@@ -89,7 +90,7 @@ const core = {
  } else {
  return null;
  }
- } catch(e) {
+ } catch (e) {
  console.log('buildAstByAstStr failed:' + e)
  }
  }

packages/gogocode-core/test/G.vue.test.js

@@ -635,3 +635,46 @@ test('test vue parseoptions', () => {
  const res = ast.generate()
  expect(res.match('&&')).toBeTruthy()
 })
+
+
+test('test vue3 setup by script', () => {
+ const ast = $(
+ `<template>
+ <div class="a">1</div>
+ <div class="a">2</div>
+ </template>
+
+ <script setup lang="ts">
+ const a = 1
+ const b = 2
+</script>
+ `,
+ { parseOptions: { language: 'vue' } }
+ )
+ const res = ast.find('<script></script>').generate()
+ expect(res===`
+ const a = 1
+ const b = 2
+`).toBeTruthy()
+})
+
+test('test vue3 setup by script setup', () => {
+ const ast = $(
+ `<template>
+ <div class="a">1</div>
+ <div class="a">2</div>
+ </template>
+
+ <script setup lang="ts">
+ const a = 1
+ const b = 2
+</script>
+ `,
+ { parseOptions: { language: 'vue' } }
+ )
+ const res = ast.find('<script setup></script>').generate()
+ expect(res===`
+ const a = 1
+ const b = 2
+`).toBeTruthy()
+})