@vue/language-core 1.8.18 → 1.8.20

package/out/generators/script.js

@@ -30,16 +30,13 @@ const muggle = __importStar(require("muggle-string"));
 const path_1 = require("path");
 const shared_1 = require("../utils/shared");
 const transform_1 = require("../utils/transform");
-function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htmlGen, compilerOptions, vueCompilerOptions, codegenStack) {
+function generate(ts, fileName, script, scriptSetup, styles, // TODO: computed it
+lang, scriptRanges, scriptSetupRanges, htmlGen, compilerOptions, vueCompilerOptions, codegenStack) {
     const [codes, codeStacks] = codegenStack ? muggle.track([]) : [[], []];
     const mirrorBehaviorMappings = [];
     //#region monkey fix: https://github.com/vuejs/language-tools/pull/2113
-    const sfc = {
-        script: _sfc.script,
-        scriptSetup: _sfc.scriptSetup,
-    };
-    if (!sfc.script && !sfc.scriptSetup) {
-        sfc.scriptSetup = {
+    if (!script && !scriptSetup) {
+        scriptSetup = {
             content: '',
             lang: 'ts',
             name: '',
@@ -50,31 +47,23 @@ function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htm
             generic: undefined,
             genericOffset: 0,
             attrs: {},
+            ast: ts.createSourceFile('', '', ts.ScriptTarget.Latest, false, ts.ScriptKind.TS),
         };
         scriptSetupRanges = {
             bindings: [],
-            emitsAssignName: undefined,
-            exposeRuntimeArg: undefined,
-            exposeTypeArg: undefined,
-            leadingCommentEndOffset: 0,
-            importSectionEndOffset: 0,
-            withDefaults: undefined,
-            defineProps: undefined,
-            defineSlots: undefined,
-            defineEmits: undefined,
-            defineExpose: undefined,
-            slotsAssignName: undefined,
-            propsAssignName: undefined,
-            propsRuntimeArg: undefined,
-            propsTypeArg: undefined,
-            withDefaultsArg: undefined,
+            props: {},
+            emits: {},
+            expose: {},
+            slots: {},
             defineProp: [],
+            importSectionEndOffset: 0,
+            leadingCommentEndOffset: 0,
         };
     }
     //#endregion
     const bindingNames = new Set([
-        ...scriptRanges?.bindings.map(range => sfc.script.content.substring(range.start, range.end)) ?? [],
-        ...scriptSetupRanges?.bindings.map(range => sfc.scriptSetup.content.substring(range.start, range.end)) ?? [],
+        ...scriptRanges?.bindings.map(range => script.content.substring(range.start, range.end)) ?? [],
+        ...scriptSetupRanges?.bindings.map(range => scriptSetup.content.substring(range.start, range.end)) ?? [],
     ]);
     const bypassDefineComponent = lang === 'js' || lang === 'jsx';
     const usedHelperTypes = {
@@ -94,12 +83,12 @@ function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htm
     if (!generatedTemplate) {
         generateTemplate(false);
     }
-    if (sfc.scriptSetup) {
+    if (scriptSetup) {
         // for code action edits
         codes.push([
             '',
             'scriptSetup',
-            sfc.scriptSetup.content.length,
+            scriptSetup.content.length,
             {},
         ]);
     }
@@ -127,7 +116,7 @@ function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htm
 				};\n`);
         }
         if (usedHelperTypes.WithTemplateSlots) {
-            codes.push(`type __VLS_WithTemplateSlots<T, S> = T & { new(): {\n`, `$slots: S;\n`);
+            codes.push(`type __VLS_WithTemplateSlots<T, S> = T & { new(): {\n`, `${(0, shared_1.getSlotsPropertyName)(vueCompilerOptions.target)}: S;\n`);
             if (vueCompilerOptions.jsxSlots) {
                 usedHelperTypes.PropsChildren = true;
                 codes.push(`$props: __VLS_PropsChildren<S>;\n`);
@@ -139,9 +128,9 @@ function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htm
         }
     }
     function generateSrc() {
-        if (!sfc.script?.src)
+        if (!script?.src)
             return;
-        let src = sfc.script.src;
+        let src = script.src;
         if (src.endsWith('.d.ts'))
             src = src.substring(0, src.length - '.d.ts'.length);
         else if (src.endsWith('.ts'))
@@ -154,23 +143,23 @@ function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htm
         codes.push([
             `'${src}'`,
             'script',
-            [sfc.script.srcOffset - 1, sfc.script.srcOffset + sfc.script.src.length + 1],
+            [script.srcOffset - 1, script.srcOffset + script.src.length + 1],
             {
                 ...language_core_1.FileRangeCapabilities.full,
-                rename: src === sfc.script.src ? true : {
+                rename: src === script.src ? true : {
                     normalize: undefined,
                     apply(newName) {
                         if (newName.endsWith('.jsx')
                             || newName.endsWith('.js')) {
                             newName = newName.split('.').slice(0, -1).join('.');
                         }
-                        if (sfc.script?.src?.endsWith('.d.ts')) {
+                        if (script?.src?.endsWith('.d.ts')) {
                             newName = newName + '.d.ts';
                         }
-                        else if (sfc.script?.src?.endsWith('.ts')) {
+                        else if (script?.src?.endsWith('.ts')) {
                             newName = newName + '.ts';
                         }
-                        else if (sfc.script?.src?.endsWith('.tsx')) {
+                        else if (script?.src?.endsWith('.tsx')) {
                             newName = newName + '.tsx';
                         }
                         return newName;
@@ -182,15 +171,15 @@ function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htm
         codes.push(`export { default } from '${src}';\n`);
     }
     function generateScriptContentBeforeExportDefault() {
-        if (!sfc.script)
+        if (!script)
             return;
-        if (!!sfc.scriptSetup && scriptRanges?.exportDefault) {
+        if (!!scriptSetup && scriptRanges?.exportDefault) {
             addVirtualCode('script', 0, scriptRanges.exportDefault.expression.start);
         }
         else {
             let isExportRawObject = false;
             if (scriptRanges?.exportDefault) {
-                isExportRawObject = sfc.script.content.substring(scriptRanges.exportDefault.expression.start, scriptRanges.exportDefault.expression.end).startsWith('{');
+                isExportRawObject = script.content.substring(scriptRanges.exportDefault.expression.start, scriptRanges.exportDefault.expression.end).startsWith('{');
             }
             if (isExportRawObject && vueCompilerOptions.optionsWrapper.length === 2 && scriptRanges?.exportDefault) {
                 addVirtualCode('script', 0, scriptRanges.exportDefault.expression.start);
@@ -216,63 +205,63 @@ function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htm
                         }]);
                 }
                 codes.push(vueCompilerOptions.optionsWrapper[1]);
-                addVirtualCode('script', scriptRanges.exportDefault.expression.end, sfc.script.content.length);
+                addVirtualCode('script', scriptRanges.exportDefault.expression.end, script.content.length);
             }
             else {
-                addVirtualCode('script', 0, sfc.script.content.length);
+                addVirtualCode('script', 0, script.content.length);
             }
         }
     }
     function generateScriptContentAfterExportDefault() {
-        if (!sfc.script)
+        if (!script)
             return;
-        if (!!sfc.scriptSetup && scriptRanges?.exportDefault) {
-            addVirtualCode('script', scriptRanges.exportDefault.end, sfc.script.content.length);
+        if (!!scriptSetup && scriptRanges?.exportDefault) {
+            addVirtualCode('script', scriptRanges.exportDefault.end, script.content.length);
         }
     }
     function generateScriptSetupImports() {
-        if (!sfc.scriptSetup)
+        if (!scriptSetup)
             return;
         if (!scriptSetupRanges)
             return;
         codes.push([
-            sfc.scriptSetup.content.substring(0, Math.max(scriptSetupRanges.importSectionEndOffset, scriptSetupRanges.leadingCommentEndOffset)) + '\n',
+            scriptSetup.content.substring(0, Math.max(scriptSetupRanges.importSectionEndOffset, scriptSetupRanges.leadingCommentEndOffset)) + '\n',
             'scriptSetup',
             0,
             language_core_1.FileRangeCapabilities.full,
         ]);
     }
     function generateExportDefaultEndMapping() {
-        if (!sfc.scriptSetup) {
+        if (!scriptSetup) {
             return;
         }
         // fix https://github.com/vuejs/language-tools/issues/1127
         codes.push([
             '',
             'scriptSetup',
-            sfc.scriptSetup.content.length,
+            scriptSetup.content.length,
             { diagnostic: true },
         ]);
         codes.push(`\n`);
     }
     function generateScriptSetupAndTemplate() {
-        if (!sfc.scriptSetup || !scriptSetupRanges) {
+        if (!scriptSetup || !scriptSetupRanges) {
             return;
         }
         const definePropMirrors = {};
         let scriptSetupGeneratedOffset;
-        if (sfc.scriptSetup.generic) {
+        if (scriptSetup.generic) {
             if (!scriptRanges?.exportDefault) {
                 codes.push('export default ');
             }
             codes.push(`(<`);
             codes.push([
-                sfc.scriptSetup.generic,
-                sfc.scriptSetup.name,
-                sfc.scriptSetup.genericOffset,
+                scriptSetup.generic,
+                scriptSetup.name,
+                scriptSetup.genericOffset,
                 language_core_1.FileRangeCapabilities.full,
             ]);
-            if (!sfc.scriptSetup.generic.endsWith(',')) {
+            if (!scriptSetup.generic.endsWith(',')) {
                 codes.push(`,`);
             }
             codes.push(`>`);
@@ -285,13 +274,13 @@ function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htm
             //#region props
             codes.push(`const __VLS_fnComponent = `);
             codes.push(`(await import('${vueCompilerOptions.lib}')).defineComponent({\n`);
-            if (scriptSetupRanges.propsRuntimeArg) {
+            if (scriptSetupRanges.props.define?.arg) {
                 codes.push(`props: `);
-                addExtraReferenceVirtualCode('scriptSetup', scriptSetupRanges.propsRuntimeArg.start, scriptSetupRanges.propsRuntimeArg.end);
+                addExtraReferenceVirtualCode('scriptSetup', scriptSetupRanges.props.define.arg.start, scriptSetupRanges.props.define.arg.end);
                 codes.push(`,\n`);
             }
-            if (scriptSetupRanges.defineEmits) {
-                codes.push(`emits: ({} as __VLS_NormalizeEmits<typeof `, scriptSetupRanges.emitsAssignName ?? '__VLS_emit', `>),\n`);
+            if (scriptSetupRanges.emits.define) {
+                codes.push(`emits: ({} as __VLS_NormalizeEmits<typeof `, scriptSetupRanges.emits.name ?? '__VLS_emit', `>),\n`);
             }
             codes.push(`});\n`);
             if (scriptSetupRanges.defineProp.length) {
@@ -299,35 +288,35 @@ function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htm
                 for (const defineProp of scriptSetupRanges.defineProp) {
                     if (defineProp.defaultValue) {
                         if (defineProp.name) {
-                            codes.push(sfc.scriptSetup.content.substring(defineProp.name.start, defineProp.name.end));
+                            codes.push(scriptSetup.content.substring(defineProp.name.start, defineProp.name.end));
                         }
                         else {
                             codes.push('modelValue');
                         }
                         codes.push(`: `);
-                        codes.push(sfc.scriptSetup.content.substring(defineProp.defaultValue.start, defineProp.defaultValue.end));
+                        codes.push(scriptSetup.content.substring(defineProp.defaultValue.start, defineProp.defaultValue.end));
                         codes.push(`,\n`);
                     }
                 }
                 codes.push(`};\n`);
             }
             codes.push(`let __VLS_fnPropsTypeOnly!: {}`); // TODO: reuse __VLS_fnPropsTypeOnly even without generic, and remove __VLS_propsOption_defineProp
-            if (scriptSetupRanges.propsTypeArg) {
+            if (scriptSetupRanges.props.define?.typeArg) {
                 codes.push(` & `);
-                addVirtualCode('scriptSetup', scriptSetupRanges.propsTypeArg.start, scriptSetupRanges.propsTypeArg.end);
+                addVirtualCode('scriptSetup', scriptSetupRanges.props.define.typeArg.start, scriptSetupRanges.props.define.typeArg.end);
             }
             if (scriptSetupRanges.defineProp.length) {
                 codes.push(` & {\n`);
                 for (const defineProp of scriptSetupRanges.defineProp) {
                     let propName = 'modelValue';
                     if (defineProp.name) {
-                        propName = sfc.scriptSetup.content.substring(defineProp.name.start, defineProp.name.end);
+                        propName = scriptSetup.content.substring(defineProp.name.start, defineProp.name.end);
                         const propMirrorStart = muggle.getLength(codes);
                         definePropMirrors[propName] = [propMirrorStart, propMirrorStart + propName.length];
                     }
                     codes.push(`${propName}${defineProp.required ? '' : '?'}: `);
                     if (defineProp.type) {
-                        codes.push(sfc.scriptSetup.content.substring(defineProp.type.start, defineProp.type.end));
+                        codes.push(scriptSetup.content.substring(defineProp.type.start, defineProp.type.end));
                     }
                     else if (defineProp.defaultValue) {
                         codes.push(`typeof __VLS_defaults['`);
@@ -345,7 +334,7 @@ function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htm
             codes.push(`let __VLS_fnPropsDefineComponent!: InstanceType<typeof __VLS_fnComponent>['$props']`);
             codes.push(`;\n`);
             codes.push(`let __VLS_fnPropsSlots!: `);
-            if (scriptSetupRanges.defineSlots && vueCompilerOptions.jsxSlots) {
+            if (scriptSetupRanges.slots.define && vueCompilerOptions.jsxSlots) {
                 usedHelperTypes.PropsChildren = true;
                 codes.push(`__VLS_PropsChildren<typeof __VLS_slots>`);
             }
@@ -356,16 +345,16 @@ function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htm
             codes.push(`let __VLS_defaultProps!: `, `import('${vueCompilerOptions.lib}').VNodeProps`, `& import('${vueCompilerOptions.lib}').AllowedComponentProps`, `& import('${vueCompilerOptions.lib}').ComponentCustomProps`, `;\n`);
             //#endregion
             codes.push('return {} as {\n');
-            codes.push(`props: __VLS_Prettify<Omit<typeof __VLS_fnPropsDefineComponent & typeof __VLS_fnPropsTypeOnly, keyof typeof __VLS_defaultProps>> & typeof __VLS_fnPropsSlots & typeof __VLS_defaultProps,\n`);
-            codes.push(`expose(exposed: import('${vueCompilerOptions.lib}').ShallowUnwrapRef<${scriptSetupRanges.defineExpose ? 'typeof __VLS_exposed' : '{}'}>): void,\n`);
+            codes.push(`props: __VLS_Prettify<__VLS_OmitKeepDiscriminatedUnion<typeof __VLS_fnPropsDefineComponent & typeof __VLS_fnPropsTypeOnly, keyof typeof __VLS_defaultProps>> & typeof __VLS_fnPropsSlots & typeof __VLS_defaultProps,\n`);
+            codes.push(`expose(exposed: import('${vueCompilerOptions.lib}').ShallowUnwrapRef<${scriptSetupRanges.expose.define ? 'typeof __VLS_exposed' : '{}'}>): void,\n`);
             codes.push('attrs: any,\n');
             codes.push('slots: ReturnType<typeof __VLS_template>,\n');
-            codes.push(`emit: typeof ${scriptSetupRanges.emitsAssignName ?? '__VLS_emit'},\n`);
+            codes.push(`emit: typeof ${scriptSetupRanges.emits.name ?? '__VLS_emit'},\n`);
             codes.push('};\n');
             codes.push('})(),\n');
             codes.push(`) => ({} as import('${vueCompilerOptions.lib}').VNode & { __ctx?: Awaited<typeof __VLS_setup> }))`);
         }
-        else if (!sfc.script) {
+        else if (!script) {
             // no script block, generate script setup code at root
             scriptSetupGeneratedOffset = generateSetupFunction(false, 'export', definePropMirrors);
         }
@@ -383,7 +372,7 @@ function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htm
                 if (!defineProp.name) {
                     continue;
                 }
-                const propName = sfc.scriptSetup.content.substring(defineProp.name.start, defineProp.name.end);
+                const propName = scriptSetup.content.substring(defineProp.name.start, defineProp.name.end);
                 const propMirror = definePropMirrors[propName];
                 if (propMirror) {
                     mirrorBehaviorMappings.push({
@@ -399,11 +388,11 @@ function generate(ts, fileName, _sfc, lang, scriptRanges, scriptSetupRanges, htm
         }
     }
     function generateSetupFunction(functional, mode, definePropMirrors) {
-        if (!scriptSetupRanges || !sfc.scriptSetup) {
+        if (!scriptSetupRanges || !scriptSetup) {
             return;
         }
-        const definePropProposalA = sfc.scriptSetup.content.trimStart().startsWith('// @experimentalDefinePropProposal=kevinEdition') || vueCompilerOptions.experimentalDefinePropProposal === 'kevinEdition';
-        const definePropProposalB = sfc.scriptSetup.content.trimStart().startsWith('// @experimentalDefinePropProposal=johnsonEdition') || vueCompilerOptions.experimentalDefinePropProposal === 'johnsonEdition';
+        const definePropProposalA = scriptSetup.content.trimStart().startsWith('// @experimentalDefinePropProposal=kevinEdition') || vueCompilerOptions.experimentalDefinePropProposal === 'kevinEdition';
+        const definePropProposalB = scriptSetup.content.trimStart().startsWith('// @experimentalDefinePropProposal=johnsonEdition') || vueCompilerOptions.experimentalDefinePropProposal === 'johnsonEdition';
         if (vueCompilerOptions.target >= 3.3) {
             codes.push('const { ');
             for (const macro of Object.keys(vueCompilerOptions.macros)) {
@@ -429,35 +418,44 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
         }
         const scriptSetupGeneratedOffset = muggle.getLength(codes) - scriptSetupRanges.importSectionEndOffset;
         let setupCodeModifies = [];
-        if (scriptSetupRanges.defineProps && !scriptSetupRanges.propsAssignName) {
-            const ranges = scriptSetupRanges.withDefaults ?? scriptSetupRanges.defineProps;
-            codes.push(`const __VLS_props = `);
-            addVirtualCode('scriptSetup', ranges.start, ranges.end);
-            codes.push(`;\n`);
-            setupCodeModifies.push([() => codes.push(`__VLS_props`), ranges.start, ranges.end]);
+        if (scriptSetupRanges.props.define && !scriptSetupRanges.props.name) {
+            const range = scriptSetupRanges.props.withDefaults ?? scriptSetupRanges.props.define;
+            const statement = scriptSetupRanges.props.define.statement;
+            if (statement.start === range.start && statement.end === range.end) {
+                setupCodeModifies.push([() => codes.push(`const __VLS_props = `), range.start, range.start]);
+            }
+            else {
+                setupCodeModifies.push([() => {
+                        codes.push(`const __VLS_props = `);
+                        addVirtualCode('scriptSetup', range.start, range.end);
+                        codes.push(`;\n`);
+                        addVirtualCode('scriptSetup', statement.start, range.start);
+                        codes.push(`__VLS_props`);
+                    }, statement.start, range.end]);
+            }
         }
-        if (scriptSetupRanges.defineSlots && !scriptSetupRanges.slotsAssignName) {
-            setupCodeModifies.push([() => codes.push(`const __VLS_slots = `), scriptSetupRanges.defineSlots.start, scriptSetupRanges.defineSlots.start]);
+        if (scriptSetupRanges.slots.define && !scriptSetupRanges.slots.name) {
+            setupCodeModifies.push([() => codes.push(`const __VLS_slots = `), scriptSetupRanges.slots.define.start, scriptSetupRanges.slots.define.start]);
         }
-        if (scriptSetupRanges.defineEmits && !scriptSetupRanges.emitsAssignName) {
-            setupCodeModifies.push([() => codes.push(`const __VLS_emit = `), scriptSetupRanges.defineEmits.start, scriptSetupRanges.defineEmits.start]);
+        if (scriptSetupRanges.emits.define && !scriptSetupRanges.emits.name) {
+            setupCodeModifies.push([() => codes.push(`const __VLS_emit = `), scriptSetupRanges.emits.define.start, scriptSetupRanges.emits.define.start]);
         }
-        if (scriptSetupRanges.defineExpose) {
+        if (scriptSetupRanges.expose.define) {
             setupCodeModifies.push([() => {
-                    if (scriptSetupRanges?.exposeTypeArg) {
+                    if (scriptSetupRanges?.expose.define?.typeArg) {
                         codes.push(`let __VLS_exposed!: `);
-                        addExtraReferenceVirtualCode('scriptSetup', scriptSetupRanges.exposeTypeArg.start, scriptSetupRanges.exposeTypeArg.end);
+                        addExtraReferenceVirtualCode('scriptSetup', scriptSetupRanges.expose.define.typeArg.start, scriptSetupRanges.expose.define.typeArg.end);
                         codes.push(`;\n`);
                     }
-                    else if (scriptSetupRanges?.exposeRuntimeArg) {
+                    else if (scriptSetupRanges?.expose.define?.arg) {
                         codes.push(`const __VLS_exposed = `);
-                        addExtraReferenceVirtualCode('scriptSetup', scriptSetupRanges.exposeRuntimeArg.start, scriptSetupRanges.exposeRuntimeArg.end);
+                        addExtraReferenceVirtualCode('scriptSetup', scriptSetupRanges.expose.define.arg.start, scriptSetupRanges.expose.define.arg.end);
                         codes.push(`;\n`);
                     }
                     else {
                         codes.push(`const __VLS_exposed = {};\n`);
                     }
-                }, scriptSetupRanges.defineExpose.start, scriptSetupRanges.defineExpose.start]);
+                }, scriptSetupRanges.expose.define.start, scriptSetupRanges.expose.define.start]);
         }
         setupCodeModifies = setupCodeModifies.sort((a, b) => a[1] - b[1]);
         if (setupCodeModifies.length) {
@@ -477,10 +475,10 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
         else {
             addVirtualCode('scriptSetup', scriptSetupRanges.importSectionEndOffset);
         }
-        if (scriptSetupRanges.propsTypeArg && scriptSetupRanges.withDefaultsArg) {
+        if (scriptSetupRanges.props.define?.typeArg && scriptSetupRanges.props.withDefaults?.arg) {
             // fix https://github.com/vuejs/language-tools/issues/1187
             codes.push(`const __VLS_withDefaultsArg = (function <T>(t: T) { return t })(`);
-            addExtraReferenceVirtualCode('scriptSetup', scriptSetupRanges.withDefaultsArg.start, scriptSetupRanges.withDefaultsArg.end);
+            addExtraReferenceVirtualCode('scriptSetup', scriptSetupRanges.props.withDefaults.arg.start, scriptSetupRanges.props.withDefaults.arg.end);
             codes.push(`);\n`);
         }
         if (!functional && scriptSetupRanges.defineProp.length) {
@@ -492,7 +490,7 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
                     addExtraReferenceVirtualCode('scriptSetup', defineProp.name.start, defineProp.name.end);
                 }
                 else if (defineProp.name) {
-                    propName = sfc.scriptSetup.content.substring(defineProp.name.start, defineProp.name.end);
+                    propName = scriptSetup.content.substring(defineProp.name.start, defineProp.name.end);
                     const start = muggle.getLength(codes);
                     definePropMirrors[propName] = [start, start + propName.length];
                     codes.push(propName);
@@ -506,7 +504,7 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
                     type = `NonNullable<typeof ${propName}['value']>`;
                 }
                 else if (defineProp.type) {
-                    type = sfc.scriptSetup.content.substring(defineProp.type.start, defineProp.type.end);
+                    type = scriptSetup.content.substring(defineProp.type.start, defineProp.type.end);
                 }
                 if (defineProp.required) {
                     codes.push(`{ required: true, type: import('${vueCompilerOptions.lib}').PropType<${type}> },\n`);
@@ -519,7 +517,7 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
         }
         generateTemplate(functional);
         if (mode === 'return' || mode === 'export') {
-            if (!vueCompilerOptions.skipTemplateCodegen && (htmlGen?.hasSlot || scriptSetupRanges?.defineSlots)) {
+            if (!vueCompilerOptions.skipTemplateCodegen && (htmlGen?.hasSlot || scriptSetupRanges?.slots.define)) {
                 usedHelperTypes.WithTemplateSlots = true;
                 codes.push(`const __VLS_component = `);
                 generateComponent(functional);
@@ -553,7 +551,7 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
         codes.push(`setup() {\n`);
         codes.push(`return {\n`);
         generateSetupReturns();
-        if (scriptSetupRanges.defineExpose) {
+        if (scriptSetupRanges.expose.define) {
             codes.push(`...__VLS_exposed,\n`);
         }
         codes.push(`};\n`);
@@ -562,17 +560,20 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
     }
     function generateComponentOptions(functional) {
         if (scriptSetupRanges && !bypassDefineComponent) {
-            if (scriptSetupRanges.propsRuntimeArg || scriptSetupRanges.propsTypeArg || (!functional && scriptSetupRanges.defineProp.length)) {
-                codes.push(`props: {\n`);
-                if (scriptSetupRanges.propsRuntimeArg) {
-                    codes.push('...');
-                    addExtraReferenceVirtualCode('scriptSetup', scriptSetupRanges.propsRuntimeArg.start, scriptSetupRanges.propsRuntimeArg.end);
-                    codes.push(',\n');
-                }
-                if (scriptSetupRanges.propsTypeArg) {
+            const ranges = scriptSetupRanges;
+            const propsCodegens = [];
+            if (ranges.props.define?.arg) {
+                const arg = ranges.props.define.arg;
+                propsCodegens.push(() => {
+                    addExtraReferenceVirtualCode('scriptSetup', arg.start, arg.end);
+                });
+            }
+            if (ranges.props.define?.typeArg) {
+                const typeArg = ranges.props.define.typeArg;
+                propsCodegens.push(() => {
                     usedHelperTypes.DefinePropsToOptions = true;
-                    codes.push('...{} as ');
-                    if (scriptSetupRanges.withDefaultsArg) {
+                    codes.push(`{} as `);
+                    if (ranges.props.withDefaults?.arg) {
                         usedHelperTypes.MergePropDefaults = true;
                         codes.push(`__VLS_WithDefaults<`);
                     }
@@ -581,22 +582,38 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
                         codes.push(`typeof __VLS_fnPropsTypeOnly`);
                     }
                     else {
-                        addExtraReferenceVirtualCode('scriptSetup', scriptSetupRanges.propsTypeArg.start, scriptSetupRanges.propsTypeArg.end);
+                        addExtraReferenceVirtualCode('scriptSetup', typeArg.start, typeArg.end);
                     }
                     codes.push(`>`);
-                    if (scriptSetupRanges.withDefaultsArg) {
+                    if (ranges.props.withDefaults?.arg) {
                         codes.push(`, typeof __VLS_withDefaultsArg`);
                         codes.push(`>`);
                     }
-                    codes.push(',\n');
+                });
+            }
+            if (!functional && ranges.defineProp.length) {
+                propsCodegens.push(() => {
+                    codes.push(`__VLS_propsOption_defineProp`);
+                });
+            }
+            if (propsCodegens.length === 1) {
+                codes.push(`props: `);
+                for (const generate of propsCodegens) {
+                    generate();
                 }
-                if (!functional && scriptSetupRanges.defineProp.length) {
-                    codes.push(`...__VLS_propsOption_defineProp,\n`);
+                codes.push(`,\n`);
+            }
+            else if (propsCodegens.length >= 2) {
+                codes.push(`props: {\n`);
+                for (const generate of propsCodegens) {
+                    codes.push('...');
+                    generate();
+                    codes.push(',\n');
                 }
                 codes.push(`},\n`);
             }
-            if (scriptSetupRanges.defineEmits) {
-                codes.push(`emits: ({} as __VLS_NormalizeEmits<typeof `, scriptSetupRanges.emitsAssignName ?? '__VLS_emit', `>),\n`);
+            if (ranges.emits.define) {
+                codes.push(`emits: ({} as __VLS_NormalizeEmits<typeof `, ranges.emits.name ?? '__VLS_emit', `>),\n`);
             }
         }
         if (scriptRanges?.exportDefault?.args) {
@@ -606,14 +623,14 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
     function generateSetupReturns() {
         if (scriptSetupRanges && bypassDefineComponent) {
             // fill $props
-            if (scriptSetupRanges.defineProps) {
+            if (scriptSetupRanges.props.define) {
                 // NOTE: defineProps is inaccurate for $props
-                codes.push(`$props: __VLS_makeOptional(${scriptSetupRanges.propsAssignName ?? `__VLS_props`}),\n`);
-                codes.push(`...${scriptSetupRanges.propsAssignName ?? `__VLS_props`},\n`);
+                codes.push(`$props: __VLS_makeOptional(${scriptSetupRanges.props.name ?? `__VLS_props`}),\n`);
+                codes.push(`...${scriptSetupRanges.props.name ?? `__VLS_props`},\n`);
             }
             // fill $emit
-            if (scriptSetupRanges.defineEmits) {
-                codes.push(`$emit: ${scriptSetupRanges.emitsAssignName ?? '__VLS_emit'},\n`);
+            if (scriptSetupRanges.emits.define) {
+                codes.push(`$emit: ${scriptSetupRanges.emits.name ?? '__VLS_emit'},\n`);
             }
         }
     }
@@ -637,7 +654,7 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
         }
     }
     function generateComponentForTemplateUsage(functional, cssIds) {
-        if (sfc.scriptSetup && scriptSetupRanges) {
+        if (scriptSetup && scriptSetupRanges) {
             codes.push(`const __VLS_internalComponent = (await import('${vueCompilerOptions.lib}')).defineComponent({\n`);
             generateComponentOptions(functional);
             codes.push(`setup() {\n`);
@@ -646,9 +663,9 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
             // bindings
             const templateUsageVars = getTemplateUsageVars();
             for (const [content, bindings] of [
-                [sfc.scriptSetup.content, scriptSetupRanges.bindings],
-                scriptRanges && sfc.script
-                    ? [sfc.script.content, scriptRanges.bindings]
+                [scriptSetup.content, scriptSetupRanges.bindings],
+                scriptRanges && script
+                    ? [script.content, scriptRanges.bindings]
                     : ['', []],
             ]) {
                 for (const expose of bindings) {
@@ -678,7 +695,7 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
             codes.push(`},\n`); // setup() {
             codes.push(`});\n`); // defineComponent({
         }
-        else if (sfc.script) {
+        else if (script) {
             codes.push(`let __VLS_internalComponent!: typeof import('./${path_1.posix.basename(fileName)}')['default'];\n`);
         }
         else {
@@ -688,10 +705,10 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
     function generateExportOptions() {
         codes.push(`\n`);
         codes.push(`const __VLS_componentsOption = `);
-        if (sfc.script && scriptRanges?.exportDefault?.componentsOption) {
+        if (script && scriptRanges?.exportDefault?.componentsOption) {
             const componentsOption = scriptRanges.exportDefault.componentsOption;
             codes.push([
-                sfc.script.content.substring(componentsOption.start, componentsOption.end),
+                script.content.substring(componentsOption.start, componentsOption.end),
                 'script',
                 componentsOption.start,
                 {
@@ -707,13 +724,13 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
     }
     function generateConstNameOption() {
         codes.push(`\n`);
-        if (sfc.script && scriptRanges?.exportDefault?.nameOption) {
+        if (script && scriptRanges?.exportDefault?.nameOption) {
             const nameOption = scriptRanges.exportDefault.nameOption;
             codes.push(`const __VLS_name = `);
-            codes.push(`${sfc.script.content.substring(nameOption.start, nameOption.end)} as const`);
+            codes.push(`${script.content.substring(nameOption.start, nameOption.end)} as const`);
             codes.push(`;\n`);
         }
-        else if (sfc.scriptSetup) {
+        else if (scriptSetup) {
             codes.push(`let __VLS_name!: '${path_1.posix.basename(fileName.substring(0, fileName.lastIndexOf('.')))}';\n`);
         }
         else {
@@ -725,32 +742,33 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
         codes.push(`let __VLS_ctx!: ${useGlobalThisTypeInCtx ? 'typeof globalThis &' : ''}`);
         codes.push(`InstanceType<__VLS_PickNotAny<typeof __VLS_internalComponent, new () => {}>> & {\n`);
         /* CSS Module */
-        for (let i = 0; i < _sfc.styles.length; i++) {
-            const style = _sfc.styles[i];
-            if (!style.module)
-                continue;
-            codes.push(`${style.module}: Record<string, string> & __VLS_Prettify<{}`);
-            for (const className of style.classNames) {
-                generateCssClassProperty(i, className.text.substring(1), { start: className.offset, end: className.offset + className.text.length }, 'string', false, true);
+        for (let i = 0; i < styles.length; i++) {
+            const style = styles[i];
+            if (style.module) {
+                codes.push(`${style.module}: Record<string, string> & __VLS_Prettify<{}`);
+                for (const className of style.classNames) {
+                    generateCssClassProperty(i, className.text.substring(1), { start: className.offset, end: className.offset + className.text.length }, 'string', false, true);
+                }
+                codes.push('>;\n');
             }
-            codes.push('>;\n');
         }
         codes.push(`};\n`);
         /* Components */
         codes.push('/* Components */\n');
         codes.push(`let __VLS_otherComponents!: NonNullable<typeof __VLS_internalComponent extends { components: infer C } ? C : {}> & typeof __VLS_componentsOption;\n`);
-        codes.push(`let __VLS_own!: __VLS_SelfComponent<typeof __VLS_name, typeof __VLS_internalComponent & (new () => { ${(0, shared_1.getSlotsPropertyName)(vueCompilerOptions.target)}: typeof ${scriptSetupRanges?.slotsAssignName ?? '__VLS_slots'} })>;\n`);
+        codes.push(`let __VLS_own!: __VLS_SelfComponent<typeof __VLS_name, typeof __VLS_internalComponent & (new () => { ${(0, shared_1.getSlotsPropertyName)(vueCompilerOptions.target)}: typeof ${scriptSetupRanges?.slots?.name ?? '__VLS_slots'} })>;\n`);
         codes.push(`let __VLS_localComponents!: typeof __VLS_otherComponents & Omit<typeof __VLS_own, keyof typeof __VLS_otherComponents>;\n`);
         codes.push(`let __VLS_components!: typeof __VLS_localComponents & __VLS_GlobalComponents & typeof __VLS_ctx;\n`); // for html completion, TS references...
         /* Style Scoped */
         codes.push('/* Style Scoped */\n');
         codes.push('type __VLS_StyleScopedClasses = {}');
-        for (let i = 0; i < _sfc.styles.length; i++) {
-            const style = _sfc.styles[i];
-            if (!style.scoped && vueCompilerOptions.experimentalResolveStyleCssClasses !== 'always')
-                continue;
-            for (const className of style.classNames) {
-                generateCssClassProperty(i, className.text.substring(1), { start: className.offset, end: className.offset + className.text.length }, 'boolean', true, !style.module);
+        for (let i = 0; i < styles.length; i++) {
+            const style = styles[i];
+            const option = vueCompilerOptions.experimentalResolveStyleCssClasses;
+            if ((option === 'always' || option === 'scoped') && style.scoped) {
+                for (const className of style.classNames) {
+                    generateCssClassProperty(i, className.text.substring(1), { start: className.offset, end: className.offset + className.text.length }, 'boolean', true, !style.module);
+                }
             }
         }
         codes.push(';\n');
@@ -770,11 +788,11 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
         }
         if (!htmlGen) {
             codes.push(`// no template\n`);
-            if (!scriptSetupRanges?.defineSlots) {
+            if (!scriptSetupRanges?.slots.define) {
                 codes.push(`const __VLS_slots = {};\n`);
             }
         }
-        codes.push(`return ${scriptSetupRanges?.slotsAssignName ?? '__VLS_slots'};\n`);
+        codes.push(`return ${scriptSetupRanges?.slots.name ?? '__VLS_slots'};\n`);
         return { cssIds };
         function generateCssClassProperty(styleIndex, className, classRange, propertyType, optional, referencesCodeLens) {
             codes.push(`\n & { `);
@@ -811,9 +829,9 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
             codes.push(` }`);
         }
         function generateCssVars() {
-            const emptyLocalVars = {};
+            const emptyLocalVars = new Map();
             const identifiers = new Set();
-            for (const style of _sfc.styles) {
+            for (const style of styles) {
                 for (const cssBind of style.cssVars) {
                     (0, transform_1.walkInterpolationFragment)(ts, cssBind.text, ts.createSourceFile('/a.txt', cssBind.text, ts.ScriptTarget.ESNext), (frag, fragOffset, onlyForErrorMapping) => {
                         if (fragOffset === undefined) {
@@ -850,7 +868,7 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
                     usageVars.add(tag.split('.')[0]);
                 }
             }
-            for (const _id of htmlGen.identifiers) {
+            for (const _id of htmlGen.accessedGlobalVariables) {
                 usageVars.add(_id);
             }
         }
@@ -859,7 +877,7 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
     function addVirtualCode(vueTag, start, end) {
         muggle.offsetStack();
         codes.push([
-            sfc[vueTag].content.substring(start, end),
+            (vueTag === 'script' ? script : scriptSetup).content.substring(start, end),
             vueTag,
             start,
             language_core_1.FileRangeCapabilities.full, // diagnostic also working for setup() returns unused in template checking
@@ -869,7 +887,7 @@ declare function defineProp<T>(value?: T | (() => T), required?: boolean, rest?:
     function addExtraReferenceVirtualCode(vueTag, start, end) {
         muggle.offsetStack();
         codes.push([
-            sfc[vueTag].content.substring(start, end),
+            (vueTag === 'script' ? script : scriptSetup).content.substring(start, end),
             vueTag,
             start,
             {