fix: fixed an error in the vue component editor (#10293)

* fix: fixed an error in the vue component editor(https://github.com/withastro/language-tools/issues/758)

* chore: format code

* chore: change import path

2 files changed, 49 insertions, 4 deletions

diff --git a/packages/integrations/vue/src/editor.cts b/packages/integrations/vue/src/editor.cts
index 0c9642d67..d4f10aab6 100644
--- a/packages/integrations/vue/src/editor.cts
+++ b/packages/integrations/vue/src/editor.cts
@@ -23,7 +23,8 @@ export function toTSX(code: string, className: string): string {
 		const { scriptSetup } = parsedResult.descriptor;
 
 		if (scriptSetup) {
-			const definePropsType = scriptSetup.content.match(/defineProps<([\s\S]+?)>\s?\(\)/);
+			const codeWithoutComments = scriptSetup.content.replace(/\/\/.*|\/\*[\s\S]*?\*\//g, '');
+			const definePropsType = codeWithoutComments.match(/defineProps<([\s\S]+?)>\s?\(\)/);
 			const propsGeneric = scriptSetup.attrs.generic;
 			const propsGenericType = propsGeneric ? `<${propsGeneric}>` : '';
 
@@ -40,11 +41,10 @@ export function toTSX(code: string, className: string): string {
 				// TODO. Find a way to support generics when using defineProps without passing explicit types.
 				// Right now something like this `defineProps({ prop: { type: Array as PropType<T[]> } })`
 				//  won't be correctly typed in Astro.
-				const defineProps = scriptSetup.content.match(/defineProps\([\s\S]+\)/);
-
+				const defineProps = codeWithoutComments.match(/defineProps\([\s\S]+?\)/);
 				if (defineProps) {
 					result = `
-					import { defineProps } from '@vue/runtime-core';
+					import { defineProps } from 'vue';
 
 					${regularScriptBlockContent}
 
diff --git a/packages/integrations/vue/test/toTsx.test.js b/packages/integrations/vue/test/toTsx.test.js
new file mode 100644
index 000000000..bbb993788
--- /dev/null
+++ b/packages/integrations/vue/test/toTsx.test.js
@@ -0,0 +1,45 @@
+import { toTSX } from '../dist/editor.cjs';
+import {  describe, it } from 'node:test';
+import assert from 'node:assert/strict';
+
+describe('toTSX function', () => {
+	it('should correctly transform Vue code to TSX with comments', () => {
+    const vueCode = `
+      <template>
+        <div>{{ msg }}</div>
+      </template>
+
+      <script setup>
+        // This is a comment in script setup
+        // defineProps(['msg']);
+				// console.log('foo)
+      </script>
+    `;
+
+    const className = 'MyComponent';
+    const result = toTSX(vueCode, className);
+
+    // Replace the expectations below with the expected result based on your logic
+		assert.strictEqual(result, `export default function ${className}__AstroComponent_(_props: Record<string, any>): any {}`)
+  });
+	it('should correctly transform Vue code to TSX', () => {
+    const vueCode = `
+      <template>
+        <div @click="handleClick">{{ msg }}</div>
+      </template>
+
+      <script setup>
+        const props defineProps({
+					msg: String
+				});
+				const handleClick = () => {
+					console.log('foo');
+				}
+      </script>
+    `;
+
+    const className = 'MyComponent';
+    const result = toTSX(vueCode, className);
+		assert.strictEqual(result.replace(/\s/g, ''), `import{defineProps}from'vue';constProps=defineProps({msg:String})exportdefaultfunction${className}__AstroComponent_(_props:typeofProps):any{<div></div>}`)
+  });
+});