fix: always call getProgram() to get latest snapshots

Author: so1ve

packages/complex-types/src/core/transform.ts

@@ -9,8 +9,8 @@ import type { TransformOptions } from "./types";
 export function transform(code: string, id: string, options: TransformOptions) {
 	const s = new MagicString(code);
 	const language = getLanguage();
-	const typeChecker = language.__internal__.typeChecker;
-	const printer = new Printer(typeChecker);
+	const checker = language.tsLs.getProgram()!.getTypeChecker();
+	const printer = new Printer(checker);
 	const transformers = getTransformers(options);
 
 	for (const [, transform] of transformers) {

packages/language/src/helpers.ts

@@ -3,7 +3,7 @@ import type ts from "typescript/lib/tsserverlibrary";
 
 export function createHelpers(
 	language: vue.Language<string>,
-	program: ts.Program,
+	tsLs: ts.LanguageService,
 	vueCompilerOptions: vue.VueCompilerOptions,
 	ts: typeof import("typescript/lib/tsserverlibrary"),
 ) {
@@ -37,6 +37,7 @@ export function createHelpers(
 	}
 
 	function getVirtualFileOrTsAst(normalizedFilepath: string) {
+		const program = tsLs.getProgram()!;
 		let virtualFileOrTsAst = program.getSourceFile(normalizedFilepath);
 		if (virtualFileOrTsAst) {
 			return virtualFileOrTsAst;

packages/language/src/index.ts

@@ -135,13 +135,11 @@ function createLanguageWorker(
 		return getScriptKind!(fileName);
 	};
 
-	const program = tsLs.getProgram()!;
-	const typeChecker = program.getTypeChecker();
-
-	const helpers = createHelpers(language, program, vueOptions, ts);
+	const helpers = createHelpers(language, tsLs, vueOptions, ts);
 
 	return {
 		...helpers,
+		tsLs,
 		updateFile(fileName: string, text: string) {
 			fileName = normalizePath(fileName);
 			scriptSnapshots.set(fileName, ts.ScriptSnapshot.fromString(text));
@@ -167,11 +165,6 @@ function createLanguageWorker(
 			scriptSnapshots.clear();
 			projectVersion++;
 		},
-		__internal__: {
-			tsLs,
-			program,
-			typeChecker,
-		},
 	};
 }
 

packages/tsx-auto-props/src/core/transform.ts

@@ -17,7 +17,7 @@ export function transform(code: string, id: string): TransformResult {
 	const s = new MagicString(code);
 	const normalizedFilepath = normalizePath(id);
 	const language = getLanguage();
-	const typeChecker = language.__internal__.typeChecker;
+	const checker = language.tsLs.getProgram()!.getTypeChecker();
 	// This transform is supposed to be used in `.tsx` files not Vue SFCs.
 	const ast = language.getVirtualFileOrTsAst(normalizedFilepath);
 	if (!ast) {
@@ -29,7 +29,7 @@ export function transform(code: string, id: string): TransformResult {
 			return;
 		}
 		const identifier = node.expression;
-		const symbol = typeChecker.getSymbolAtLocation(identifier);
+		const symbol = checker.getSymbolAtLocation(identifier);
 		// This is a `defineComponent` call...
 		if (
 			symbol?.declarations?.some(
@@ -73,7 +73,7 @@ export function transform(code: string, id: string): TransformResult {
 			}
 			const props = setupFn?.parameters[0];
 			if (props) {
-				const propsList = typeChecker
+				const propsList = checker
 					.getTypeAtLocation(props)
 					.getProperties()
 					.map((p) => p.name);