From 51f58dd8712c84cba73e12e69a9affa4516612a8 Mon Sep 17 00:00:00 2001
From: Johnson Chu <johnsoncodehk@gmail.com>
Date: Tue, 21 Nov 2023 07:44:50 +0800
Subject: [PATCH] sync https://github.com/volarjs/volar.js/pull/91

---
 .../language-server/src/core/astro2tsx.ts     |  59 ++++-----
 packages/language-server/src/core/index.ts    | 118 +++++++++---------
 packages/language-server/src/core/parseCSS.ts |  51 +++++---
 .../language-server/src/core/parseHTML.ts     |  16 +--
 packages/language-server/src/core/parseJS.ts  |  38 ++----
 packages/language-server/src/core/svelte.ts   |  24 ++--
 packages/language-server/src/core/utils.ts    |  12 +-
 packages/language-server/src/core/vue.ts      |  24 ++--
 .../src/languageServerPlugin.ts               |  18 +--
 packages/language-server/src/plugins/astro.ts |   4 +-
 packages/language-server/src/plugins/html.ts  |   2 +-
 .../src/plugins/typescript-addons/index.ts    |   2 +-
 .../src/plugins/typescript/index.ts           |  30 +++--
 packages/ts-plugin/src/astro2tsx.ts           |  31 ++---
 packages/ts-plugin/src/index.ts               |   2 +-
 packages/ts-plugin/src/language.ts            |  17 ++-
 16 files changed, 213 insertions(+), 235 deletions(-)

diff --git a/packages/language-server/src/core/astro2tsx.ts b/packages/language-server/src/core/astro2tsx.ts
index 2658f603..94224f9d 100644
--- a/packages/language-server/src/core/astro2tsx.ts
+++ b/packages/language-server/src/core/astro2tsx.ts
@@ -1,7 +1,7 @@
 import { convertToTSX } from '@astrojs/compiler/sync';
 import type { ConvertToTSXOptions, TSXResult } from '@astrojs/compiler/types';
 import { decode } from '@jridgewell/sourcemap-codec';
-import { FileKind, FileRangeCapabilities, VirtualFile } from '@volar/language-core';
+import { FileKind, CodeInformations, VirtualFile } from '@volar/language-core';
 import { HTMLDocument, TextDocument } from 'vscode-html-languageservice';
 import { patchTSX } from './utils.js';
 
@@ -38,14 +38,14 @@ function safeConvertToTSX(content: string, options: ConvertToTSXOptions) {
 
 export function astro2tsx(
 	input: string,
-	fileName: string,
+	fileId: string,
 	ts: typeof import('typescript/lib/tsserverlibrary.js'),
 	htmlDocument: HTMLDocument
 ) {
-	const tsx = safeConvertToTSX(input, { filename: fileName });
+	const tsx = safeConvertToTSX(input, { filename: fileId });
 
 	return {
-		virtualFile: getVirtualFileTSX(input, tsx, fileName, ts, htmlDocument),
+		virtualFile: getVirtualFileTSX(input, tsx, fileId, ts, htmlDocument),
 		diagnostics: tsx.diagnostics,
 	};
 }
@@ -53,14 +53,14 @@ export function astro2tsx(
 function getVirtualFileTSX(
 	input: string,
 	tsx: TSXResult,
-	fileName: string,
+	fileId: string,
 	ts: typeof import('typescript/lib/tsserverlibrary.js'),
 	htmlDocument: HTMLDocument
 ): VirtualFile {
-	tsx.code = patchTSX(tsx.code, fileName);
+	tsx.code = patchTSX(tsx.code, fileId);
 	const v3Mappings = decode(tsx.map.mappings);
-	const sourcedDoc = TextDocument.create(fileName, 'astro', 0, input);
-	const genDoc = TextDocument.create(fileName + '.tsx', 'typescriptreact', 0, tsx.code);
+	const sourcedDoc = TextDocument.create(fileId, 'astro', 0, input);
+	const genDoc = TextDocument.create(fileId + '.tsx', 'typescriptreact', 0, tsx.code);
 
 	const mappings: VirtualFile['mappings'] = [];
 
@@ -102,19 +102,29 @@ function getVirtualFileTSX(
 						// Disable features inside script tags. This is a bit annoying to do, I wonder if maybe leaving script tags
 						// unmapped would be better.
 						const node = htmlDocument.findNodeAt(current.sourceOffset);
-						const rangeCapabilities: FileRangeCapabilities =
+						const rangeCapabilities: CodeInformations =
 							node.tag !== 'script'
-								? FileRangeCapabilities.full
+								? {}
 								: {
-										completion: false,
-										definition: false,
-										diagnostic: false,
-										displayWithLink: false,
-										hover: false,
-										references: false,
-										referencesCodeLens: false,
-										rename: false,
-										semanticTokens: false,
+									diagnostics: false,
+									renameEdits: false,
+									formattingEdits: false,
+									definitions: false,
+									references: false,
+									foldingRanges: false,
+									inlayHints: false,
+									codeActions: false,
+									symbols: false,
+									selectionRanges: false,
+									linkedEditingRanges: false,
+									colors: false,
+									autoInserts: false,
+									codeLenses: false,
+									highlights: false,
+									links: false,
+									semanticTokens: false,
+									hover: false,
+									signatureHelps: false,
 								  };
 
 						mappings.push({
@@ -146,18 +156,9 @@ function getVirtualFileTSX(
 	}
 
 	return {
-		fileName: fileName + '.tsx',
+		id: fileId + '.tsx',
 		languageId: 'typescriptreact',
 		kind: FileKind.TypeScriptHostFile,
-		capabilities: {
-			codeAction: true,
-			documentFormatting: false,
-			diagnostic: true,
-			documentSymbol: true,
-			inlayHint: true,
-			foldingRange: true,
-		},
-		codegenStacks: [],
 		snapshot: {
 			getText: (start, end) => tsx.code.substring(start, end),
 			getLength: () => tsx.code.length,
diff --git a/packages/language-server/src/core/index.ts b/packages/language-server/src/core/index.ts
index f9dd37a2..aa913568 100644
--- a/packages/language-server/src/core/index.ts
+++ b/packages/language-server/src/core/index.ts
@@ -1,8 +1,6 @@
 import type { DiagnosticMessage, ParseResult } from '@astrojs/compiler/types';
 import {
-	FileCapabilities,
 	FileKind,
-	FileRangeCapabilities,
 	type Language,
 	type VirtualFile,
 } from '@volar/language-core';
@@ -21,83 +19,83 @@ export function getLanguageModule(
 	ts: typeof import('typescript/lib/tsserverlibrary.js')
 ): Language<AstroFile> {
 	return {
-		createVirtualFile(fileName, snapshot) {
-			if (fileName.endsWith('.astro')) {
-				return new AstroFile(fileName, snapshot, ts);
+		createVirtualFile(id, languageId, snapshot) {
+			if (languageId === 'astro') {
+				return new AstroFile(id, snapshot, ts);
 			}
 		},
 		updateVirtualFile(astroFile, snapshot) {
 			astroFile.update(snapshot);
 		},
-		resolveTypeScriptProjectHost(host) {
-			return {
-				...host,
-				resolveModuleName(moduleName, impliedNodeFormat) {
-					if (
-						impliedNodeFormat === ts.ModuleKind.ESNext &&
-						(moduleName.endsWith('.astro') ||
-							moduleName.endsWith('.vue') ||
-							moduleName.endsWith('.svelte'))
-					) {
-						return `${moduleName}.js`;
-					}
-					return host.resolveModuleName?.(moduleName, impliedNodeFormat) ?? moduleName;
-				},
-				getScriptFileNames() {
-					const fileNames = host.getScriptFileNames();
-					return [
-						...fileNames,
-						...(astroInstall
-							? ['./env.d.ts', './astro-jsx.d.ts'].map((filePath) =>
+		typescript: {
+			resolveModuleName(moduleName, impliedNodeFormat) {
+				if (
+					impliedNodeFormat === ts.ModuleKind.ESNext &&
+					(moduleName.endsWith('.astro') ||
+						moduleName.endsWith('.vue') ||
+						moduleName.endsWith('.svelte'))
+				) {
+					return `${moduleName}.js`;
+				}
+			},
+			resolveLanguageServiceHost(host) {
+				return {
+					...host,
+					getScriptFileNames() {
+						const fileNames = host.getScriptFileNames();
+						return [
+							...fileNames,
+							...(astroInstall
+								? ['./env.d.ts', './astro-jsx.d.ts'].map((filePath) =>
 									ts.sys.resolvePath(path.resolve(astroInstall.path, filePath))
-							  )
-							: []),
-					];
-				},
-				getCompilationSettings() {
-					const baseCompilationSettings = host.getCompilationSettings();
-					return {
-						...baseCompilationSettings,
-						module: ts.ModuleKind.ESNext ?? 99,
-						target: ts.ScriptTarget.ESNext ?? 99,
-						jsx: ts.JsxEmit.Preserve ?? 1,
-						jsxImportSource: undefined,
-						jsxFactory: 'astroHTML',
-						resolveJsonModule: true,
-						allowJs: true,
-						isolatedModules: true,
-						moduleResolution:
-							baseCompilationSettings.moduleResolution === ts.ModuleResolutionKind.Classic ||
-							!baseCompilationSettings.moduleResolution
-								? ts.ModuleResolutionKind.Node10
-								: baseCompilationSettings.moduleResolution,
-					};
-				},
-			};
+								)
+								: []),
+						];
+					},
+					getCompilationSettings() {
+						const baseCompilationSettings = host.getCompilationSettings();
+						return {
+							...baseCompilationSettings,
+							module: ts.ModuleKind.ESNext ?? 99,
+							target: ts.ScriptTarget.ESNext ?? 99,
+							jsx: ts.JsxEmit.Preserve ?? 1,
+							jsxImportSource: undefined,
+							jsxFactory: 'astroHTML',
+							resolveJsonModule: true,
+							allowJs: true,
+							isolatedModules: true,
+							moduleResolution:
+								baseCompilationSettings.moduleResolution === ts.ModuleResolutionKind.Classic ||
+									!baseCompilationSettings.moduleResolution
+									? ts.ModuleResolutionKind.Node10
+									: baseCompilationSettings.moduleResolution,
+						};
+					},
+				};
+			},
 		},
 	};
 }
 
 export class AstroFile implements VirtualFile {
 	kind = FileKind.TextFile;
-	capabilities = FileCapabilities.full;
 
-	fileName: string;
+	id: string;
 	languageId = 'astro';
 	mappings!: VirtualFile['mappings'];
 	embeddedFiles!: VirtualFile['embeddedFiles'];
 	astroMeta!: ParseResult & { frontmatter: FrontmatterStatus };
 	compilerDiagnostics!: DiagnosticMessage[];
 	htmlDocument!: HTMLDocument;
-	scriptFiles!: string[];
+	scriptFileIds!: string[];
 	codegenStacks = [];
 
 	constructor(
-		public sourceFileName: string,
+		public sourceFileId: string,
 		public snapshot: ts.IScriptSnapshot,
 		private readonly ts: typeof import('typescript/lib/tsserverlibrary.js')
 	) {
-		this.fileName = sourceFileName;
+		this.id = sourceFileId;
 		this.onSnapshotUpdated();
 	}
 
@@ -115,14 +113,14 @@ export class AstroFile implements VirtualFile {
 			{
 				sourceRange: [0, this.snapshot.getLength()],
 				generatedRange: [0, this.snapshot.getLength()],
-				data: FileRangeCapabilities.full,
+				data: {},
 			},
 		];
 
 		this.astroMeta = getAstroMetadata(this.snapshot.getText(0, this.snapshot.getLength()));
 
 		const { htmlDocument, virtualFile: htmlVirtualFile } = parseHTML(
-			this.fileName,
+			this.id,
 			this.snapshot,
 			this.astroMeta.frontmatter.status === 'closed'
 				? this.astroMeta.frontmatter.position.end.offset
@@ -131,16 +129,16 @@ export class AstroFile implements VirtualFile {
 		this.htmlDocument = htmlDocument;
 
 		const scriptTags = extractScriptTags(
-			this.fileName,
+			this.id,
 			this.snapshot,
 			htmlDocument,
 			this.astroMeta.ast
 		);
 
-		this.scriptFiles = scriptTags.map((scriptTag) => scriptTag.fileName);
+		this.scriptFileIds = scriptTags.map((scriptTag) => scriptTag.id);
 
 		htmlVirtualFile.embeddedFiles.push(
-			...extractStylesheets(this.fileName, this.snapshot, htmlDocument, this.astroMeta.ast),
+			...extractStylesheets(this.id, this.snapshot, htmlDocument, this.astroMeta.ast),
 			...scriptTags
 		);
 
@@ -149,7 +147,7 @@ export class AstroFile implements VirtualFile {
 
 		const tsx = astro2tsx(
 			this.snapshot.getText(0, this.snapshot.getLength()),
-			this.fileName,
+			this.id,
 			this.ts,
 			htmlDocument
 		);
diff --git a/packages/language-server/src/core/parseCSS.ts b/packages/language-server/src/core/parseCSS.ts
index 8d9a79af..f0ead279 100644
--- a/packages/language-server/src/core/parseCSS.ts
+++ b/packages/language-server/src/core/parseCSS.ts
@@ -1,6 +1,6 @@
 import type { ParentNode, ParseResult } from '@astrojs/compiler/types';
 import { is } from '@astrojs/compiler/utils';
-import { FileKind, FileRangeCapabilities, VirtualFile } from '@volar/language-core';
+import { FileKind, CodeInformations, VirtualFile } from '@volar/language-core';
 import * as SourceMap from '@volar/source-map';
 import * as muggle from 'muggle-string';
 import type ts from 'typescript/lib/tsserverlibrary';
@@ -8,27 +8,48 @@ import type { HTMLDocument, Node } from 'vscode-html-languageservice';
 import type { AttributeNodeWithPosition } from './compilerUtils.js';
 
 export function extractStylesheets(
-	fileName: string,
+	fileId: string,
 	snapshot: ts.IScriptSnapshot,
 	htmlDocument: HTMLDocument,
 	ast: ParseResult['ast']
 ): VirtualFile[] {
 	const embeddedCSSFiles: VirtualFile[] = findEmbeddedStyles(
-		fileName,
+		fileId,
 		snapshot,
 		htmlDocument.roots
 	);
 
 	const inlineStyles = findInlineStyles(ast);
 	if (inlineStyles.length > 0) {
-		const codes: muggle.Segment<FileRangeCapabilities>[] = [];
+		const codes: muggle.Segment<CodeInformations>[] = [];
 		for (const inlineStyle of inlineStyles) {
 			codes.push('x { ');
 			codes.push([
 				inlineStyle.value,
 				undefined,
 				inlineStyle.position.start.offset + 'style="'.length,
-				FileRangeCapabilities.full,
+				// disable all but only keep document colors
+				{
+					diagnostics: false,
+					renameEdits: false,
+					formattingEdits: false,
+					definitions: false,
+					references: false,
+					foldingRanges: false,
+					inlayHints: false,
+					codeActions: false,
+					symbols: false,
+					selectionRanges: false,
+					linkedEditingRanges: false,
+					// colors: false,
+					autoInserts: false,
+					codeLenses: false,
+					highlights: false,
+					links: false,
+					semanticTokens: false,
+					hover: false,
+					signatureHelps: false,
+				},
 			]);
 			codes.push(' }\n');
 		}
@@ -37,15 +58,13 @@ export function extractStylesheets(
 		const text = muggle.toString(codes);
 
 		embeddedCSSFiles.push({
-			fileName: fileName + '.inline.css',
+			id: fileId + '.inline.css',
 			languageId: 'css',
-			codegenStacks: [],
 			snapshot: {
 				getText: (start, end) => text.substring(start, end),
 				getLength: () => text.length,
 				getChangeRange: () => undefined,
 			},
-			capabilities: { documentSymbol: true },
 			embeddedFiles: [],
 			kind: FileKind.TextFile,
 			mappings,
@@ -60,7 +79,7 @@ export function extractStylesheets(
  * Embedded styles are styles that are defined in `<style>` tags.
  */
 function findEmbeddedStyles(
-	fileName: string,
+	fileId: string,
 	snapshot: ts.IScriptSnapshot,
 	roots: Node[]
 ): VirtualFile[] {
@@ -78,7 +97,7 @@ function findEmbeddedStyles(
 			) {
 				const styleText = snapshot.getText(node.startTagEnd, node.endTagStart);
 				embeddedCSSFiles.push({
-					fileName: fileName + `.${cssIndex}.css`,
+					id: fileId + `.${cssIndex}.css`,
 					languageId: 'css',
 					kind: FileKind.TextFile,
 					snapshot: {
@@ -86,20 +105,16 @@ function findEmbeddedStyles(
 						getLength: () => styleText.length,
 						getChangeRange: () => undefined,
 					},
-					codegenStacks: [],
 					mappings: [
 						{
 							sourceRange: [node.startTagEnd, node.endTagStart],
 							generatedRange: [0, styleText.length],
-							data: FileRangeCapabilities.full,
+							data: {
+								diagnostics: false,
+								formattingEdits: false,
+							},
 						},
 					],
-					capabilities: {
-						diagnostic: false,
-						documentSymbol: true,
-						foldingRange: true,
-						documentFormatting: false,
-					},
 					embeddedFiles: [],
 				});
 				cssIndex++;
diff --git a/packages/language-server/src/core/parseHTML.ts b/packages/language-server/src/core/parseHTML.ts
index c25f9e4c..8d98eb44 100644
--- a/packages/language-server/src/core/parseHTML.ts
+++ b/packages/language-server/src/core/parseHTML.ts
@@ -1,4 +1,4 @@
-import { FileKind, FileRangeCapabilities, VirtualFile } from '@volar/language-core';
+import { FileKind, VirtualFile } from '@volar/language-core';
 import type ts from 'typescript/lib/tsserverlibrary';
 import * as html from 'vscode-html-languageservice';
 import { isInsideExpression } from '../plugins/utils';
@@ -84,9 +84,9 @@ export function preprocessHTML(text: string, frontmatterEnd?: number) {
 	}
 }
 
-function getHTMLVirtualFile(fileName: string, preprocessedHTML: string): VirtualFile {
+function getHTMLVirtualFile(fileId: string, preprocessedHTML: string): VirtualFile {
 	return {
-		fileName: fileName + `.html`,
+		id: fileId + `.html`,
 		languageId: 'html',
 		kind: FileKind.TextFile,
 		snapshot: {
@@ -94,19 +94,15 @@ function getHTMLVirtualFile(fileName: string, preprocessedHTML: string): Virtual
 			getLength: () => preprocessedHTML.length,
 			getChangeRange: () => undefined,
 		},
-		codegenStacks: [],
 		mappings: [
 			{
 				sourceRange: [0, preprocessedHTML.length],
 				generatedRange: [0, preprocessedHTML.length],
-				data: FileRangeCapabilities.full,
+				data: {
+					formattingEdits: false,
+				},
 			},
 		],
-		capabilities: {
-			documentSymbol: true,
-			foldingRange: true,
-			documentFormatting: false,
-		},
 		embeddedFiles: [],
 	};
 }
diff --git a/packages/language-server/src/core/parseJS.ts b/packages/language-server/src/core/parseJS.ts
index 9a0e9e06..0b0ff490 100644
--- a/packages/language-server/src/core/parseJS.ts
+++ b/packages/language-server/src/core/parseJS.ts
@@ -1,6 +1,6 @@
 import type { ParentNode, ParseResult } from '@astrojs/compiler/types';
 import { is } from '@astrojs/compiler/utils';
-import { FileKind, FileRangeCapabilities, VirtualFile } from '@volar/language-core';
+import { FileKind, CodeInformations, VirtualFile } from '@volar/language-core';
 import * as SourceMap from '@volar/source-map';
 import * as muggle from 'muggle-string';
 import type ts from 'typescript/lib/tsserverlibrary';
@@ -49,7 +49,7 @@ function isIsolatedScriptTag(scriptTag: Node): boolean {
  * All the isolated scripts are passed to the TypeScript language server as separate `.mts` files.
  */
 function findIsolatedScripts(
-	fileName: string,
+	fileId: string,
 	snapshot: ts.IScriptSnapshot,
 	roots: Node[]
 ): VirtualFile[] {
@@ -68,7 +68,7 @@ function findIsolatedScripts(
 			) {
 				const scriptText = snapshot.getText(node.startTagEnd, node.endTagStart);
 				embeddedScripts.push({
-					fileName: fileName + `.${scriptIndex}.mts`,
+					id: fileId + `.${scriptIndex}.mts`,
 					languageId: 'typescript',
 					kind: FileKind.TypeScriptHostFile,
 					snapshot: {
@@ -76,22 +76,15 @@ function findIsolatedScripts(
 						getLength: () => scriptText.length,
 						getChangeRange: () => undefined,
 					},
-					codegenStacks: [],
 					mappings: [
 						{
 							sourceRange: [node.startTagEnd, node.endTagStart],
 							generatedRange: [0, scriptText.length],
-							data: FileRangeCapabilities.full,
+							data: {
+								formattingEdits: false,
+							},
 						},
 					],
-					capabilities: {
-						diagnostic: true,
-						codeAction: true,
-						inlayHint: true,
-						documentSymbol: true,
-						foldingRange: true,
-						documentFormatting: false,
-					},
 					embeddedFiles: [],
 				});
 				scriptIndex++;
@@ -198,15 +191,17 @@ function findEventAttributes(ast: ParseResult['ast']): JavaScriptContext[] {
 /**
  * Merge all the inline and non-hoisted scripts into a single `.mjs` file
  */
-function mergeJSContexts(fileName: string, javascriptContexts: JavaScriptContext[]): VirtualFile {
-	const codes: muggle.Segment<FileRangeCapabilities>[] = [];
+function mergeJSContexts(fileId: string, javascriptContexts: JavaScriptContext[]): VirtualFile {
+	const codes: muggle.Segment<CodeInformations>[] = [];
 
 	for (const javascriptContext of javascriptContexts) {
 		codes.push([
 			javascriptContext.content,
 			undefined,
 			javascriptContext.startOffset,
-			FileRangeCapabilities.full,
+			{
+				formattingEdits: false,
+			},
 		]);
 	}
 
@@ -214,22 +209,13 @@ function mergeJSContexts(fileName: string, javascriptContexts: JavaScriptContext
 	const text = muggle.toString(codes);
 
 	return {
-		fileName: fileName + '.inline.mjs',
+		id: fileId + '.inline.mjs',
 		languageId: 'javascript',
-		codegenStacks: [],
 		snapshot: {
 			getText: (start, end) => text.substring(start, end),
 			getLength: () => text.length,
 			getChangeRange: () => undefined,
 		},
-		capabilities: {
-			codeAction: true,
-			diagnostic: true,
-			documentFormatting: false,
-			documentSymbol: true,
-			foldingRange: true,
-			inlayHint: true,
-		},
 		embeddedFiles: [],
 		kind: FileKind.TypeScriptHostFile,
 		mappings,
diff --git a/packages/language-server/src/core/svelte.ts b/packages/language-server/src/core/svelte.ts
index 1f049ddc..81a80ebd 100644
--- a/packages/language-server/src/core/svelte.ts
+++ b/packages/language-server/src/core/svelte.ts
@@ -1,7 +1,6 @@
 import {
-	FileCapabilities,
 	FileKind,
-	FileRangeCapabilities,
+	CodeInformations,
 	Language,
 	VirtualFile,
 } from '@volar/language-core';
@@ -11,9 +10,9 @@ import { framework2tsx } from './utils.js';
 
 export function getSvelteLanguageModule(): Language<SvelteFile> {
 	return {
-		createVirtualFile(fileName, snapshot) {
-			if (fileName.endsWith('.svelte')) {
-				return new SvelteFile(fileName, snapshot);
+		createVirtualFile(id, languageId, snapshot) {
+			if (languageId === 'svelte') {
+				return new SvelteFile(id, snapshot);
 			}
 		},
 		updateVirtualFile(svelteFile, snapshot) {
@@ -24,19 +23,18 @@ export function getSvelteLanguageModule(): Language<SvelteFile> {
 
 class SvelteFile implements VirtualFile {
 	kind = FileKind.TextFile;
-	capabilities = FileCapabilities.full;
 
-	fileName: string;
+	id: string;
 	languageId = 'svelte';
-	mappings!: Mapping<FileRangeCapabilities>[];
+	mappings!: Mapping<CodeInformations>[];
 	embeddedFiles!: VirtualFile[];
 	codegenStacks = [];
 
 	constructor(
-		public sourceFileName: string,
+		public sourceFileId: string,
 		public snapshot: ts.IScriptSnapshot
 	) {
-		this.fileName = sourceFileName;
+		this.id = sourceFileId;
 		this.onSnapshotUpdated();
 	}
 
@@ -50,15 +48,15 @@ class SvelteFile implements VirtualFile {
 			{
 				sourceRange: [0, this.snapshot.getLength()],
 				generatedRange: [0, this.snapshot.getLength()],
-				data: FileRangeCapabilities.full,
+				data: {},
 			},
 		];
 
 		this.embeddedFiles = [];
 		this.embeddedFiles.push(
 			framework2tsx(
-				this.fileName,
-				this.fileName,
+				this.id,
+				this.id,
 				this.snapshot.getText(0, this.snapshot.getLength()),
 				'svelte'
 			)
diff --git a/packages/language-server/src/core/utils.ts b/packages/language-server/src/core/utils.ts
index c0b45216..946aae46 100644
--- a/packages/language-server/src/core/utils.ts
+++ b/packages/language-server/src/core/utils.ts
@@ -1,7 +1,5 @@
 import {
-	FileCapabilities,
 	FileKind,
-	FileRangeCapabilities,
 	type VirtualFile,
 } from '@volar/language-core';
 import * as path from 'node:path';
@@ -9,7 +7,7 @@ import { URI, Utils } from 'vscode-uri';
 import { importSvelteIntegration, importVueIntegration } from '../importPackage';
 
 export function framework2tsx(
-	fileName: string,
+	fileId: string,
 	filePath: string,
 	sourceCode: string,
 	framework: 'vue' | 'svelte'
@@ -23,27 +21,25 @@ export function framework2tsx(
 	}
 
 	const className = classNameFromFilename(filePath);
-	const tsx = patchTSX(integrationEditorEntrypoint.toTSX(sourceCode, className), fileName);
+	const tsx = patchTSX(integrationEditorEntrypoint.toTSX(sourceCode, className), fileId);
 
 	return getVirtualFile(tsx);
 
 	function getVirtualFile(content: string): VirtualFile {
 		return {
-			fileName: fileName + '.tsx',
+			id: fileId + '.tsx',
 			languageId: 'typescript',
-			capabilities: FileCapabilities.full,
 			kind: FileKind.TypeScriptHostFile,
 			snapshot: {
 				getText: (start, end) => content.substring(start, end),
 				getLength: () => content.length,
 				getChangeRange: () => undefined,
 			},
-			codegenStacks: [],
 			mappings: [
 				{
 					sourceRange: [0, content.length],
 					generatedRange: [0, 0],
-					data: FileRangeCapabilities.full,
+					data: {},
 				},
 			],
 			embeddedFiles: [],
diff --git a/packages/language-server/src/core/vue.ts b/packages/language-server/src/core/vue.ts
index 7016f149..2588d84a 100644
--- a/packages/language-server/src/core/vue.ts
+++ b/packages/language-server/src/core/vue.ts
@@ -1,7 +1,6 @@
 import {
-	FileCapabilities,
 	FileKind,
-	FileRangeCapabilities,
+	CodeInformations,
 	Language,
 	VirtualFile,
 } from '@volar/language-core';
@@ -11,9 +10,9 @@ import { framework2tsx } from './utils.js';
 
 export function getVueLanguageModule(): Language<VueFile> {
 	return {
-		createVirtualFile(fileName, snapshot) {
-			if (fileName.endsWith('.vue')) {
-				return new VueFile(fileName, snapshot);
+		createVirtualFile(id, languageId, snapshot) {
+			if (languageId === 'vue') {
+				return new VueFile(id, snapshot);
 			}
 		},
 		updateVirtualFile(vueFile, snapshot) {
@@ -24,19 +23,18 @@ export function getVueLanguageModule(): Language<VueFile> {
 
 class VueFile implements VirtualFile {
 	kind = FileKind.TextFile;
-	capabilities = FileCapabilities.full;
 
-	fileName: string;
+	id: string;
 	languageId = 'vue';
-	mappings!: Mapping<FileRangeCapabilities>[];
+	mappings!: Mapping<CodeInformations>[];
 	embeddedFiles!: VirtualFile[];
 	codegenStacks = [];
 
 	constructor(
-		public sourceFileName: string,
+		public sourceFileId: string,
 		public snapshot: ts.IScriptSnapshot
 	) {
-		this.fileName = sourceFileName;
+		this.id = sourceFileId;
 		this.onSnapshotUpdated();
 	}
 
@@ -50,15 +48,15 @@ class VueFile implements VirtualFile {
 			{
 				sourceRange: [0, this.snapshot.getLength()],
 				generatedRange: [0, this.snapshot.getLength()],
-				data: FileRangeCapabilities.full,
+				data: {},
 			},
 		];
 
 		this.embeddedFiles = [];
 		this.embeddedFiles.push(
 			framework2tsx(
-				this.fileName,
-				this.fileName,
+				this.id,
+				this.id,
 				this.snapshot.getText(0, this.snapshot.getLength()),
 				'vue'
 			)
diff --git a/packages/language-server/src/languageServerPlugin.ts b/packages/language-server/src/languageServerPlugin.ts
index 55d4f3b0..2c3d48f1 100644
--- a/packages/language-server/src/languageServerPlugin.ts
+++ b/packages/language-server/src/languageServerPlugin.ts
@@ -42,14 +42,16 @@ export function createPlugin(connection: Connection): TypeScriptServerPlugin {
 			'vue',
 			'svelte',
 		],
-		resolveConfig(config, env, projectHost) {
+		resolveConfig(config, env, projectContext) {
 			config.languages ??= {};
-			if (projectHost) {
-				const astroInstall = getAstroInstall([projectHost.getCurrentDirectory()]);
+			if (env && projectContext) {
+				const workspacePath = env.uriToFileName(env.workspaceFolder.uri.toString());
+				// const tsconfigPath = projectContext.configFileName ? env.uriToFileName(projectContext.configFileName) : undefined;
+				const astroInstall = getAstroInstall([workspacePath]);
 
 				if (!astroInstall) {
 					connection.sendNotification(ShowMessageNotification.type, {
-						message: `Couldn't find Astro in workspace "${projectHost.getCurrentDirectory()}". Experience might be degraded. For the best experience, please make sure Astro is installed into your project and restart the language server.`,
+						message: `Couldn't find Astro in workspace "${workspacePath}". Experience might be degraded. For the best experience, please make sure Astro is installed into your project and restart the language server.`,
 						type: MessageType.Warning,
 					});
 				}
@@ -68,10 +70,10 @@ export function createPlugin(connection: Connection): TypeScriptServerPlugin {
 			config.services.typescriptaddons ??= createTypescriptAddonsService();
 			config.services.astro ??= createAstroService();
 
-			if (env && projectHost) {
-				const rootDir = projectHost.getCurrentDirectory();
-				const prettier = importPrettier(rootDir);
-				const prettierPluginPath = getPrettierPluginPath(rootDir);
+			if (env && projectContext) {
+				const workspacePath = env.uriToFileName(env.workspaceFolder.uri.toString());
+				const prettier = importPrettier(workspacePath);
+				const prettierPluginPath = getPrettierPluginPath(workspacePath);
 
 				if (prettier && prettierPluginPath) {
 					config.services.prettier ??= createPrettierService({
diff --git a/packages/language-server/src/plugins/astro.ts b/packages/language-server/src/plugins/astro.ts
index d0d6a1fd..0aef7120 100644
--- a/packages/language-server/src/plugins/astro.ts
+++ b/packages/language-server/src/plugins/astro.ts
@@ -26,7 +26,7 @@ export const create =
 				if (token.isCancellationRequested) return null;
 				let items: CompletionItem[] = [];
 
-				const [file] = context!.documents.getVirtualFileByUri(document.uri);
+				const [file] = context!.project.fileProvider.getVirtualFile(document.uri);
 				if (!(file instanceof AstroFile)) return;
 
 				if (completionContext.triggerCharacter === '-') {
@@ -42,7 +42,7 @@ export const create =
 			provideSemanticDiagnostics(document, token) {
 				if (token.isCancellationRequested) return [];
 
-				const [file] = context!.documents.getVirtualFileByUri(document.uri);
+				const [file] = context!.project.fileProvider.getVirtualFile(document.uri);
 				if (!(file instanceof AstroFile)) return;
 
 				return file.compilerDiagnostics.map(compilerMessageToDiagnostic);
diff --git a/packages/language-server/src/plugins/html.ts b/packages/language-server/src/plugins/html.ts
index c795bc31..9eef84dc 100644
--- a/packages/language-server/src/plugins/html.ts
+++ b/packages/language-server/src/plugins/html.ts
@@ -24,7 +24,7 @@ export const create =
 			async provideCompletionItems(document, position, completionContext, token) {
 				if (document.languageId !== 'html') return;
 
-				const [_, source] = context.documents.getVirtualFileByUri(document.uri);
+				const [_, source] = context.project.fileProvider.getVirtualFile(document.uri);
 				const rootVirtualFile = source?.root;
 				if (!(rootVirtualFile instanceof AstroFile)) return;
 
diff --git a/packages/language-server/src/plugins/typescript-addons/index.ts b/packages/language-server/src/plugins/typescript-addons/index.ts
index e0d35b25..47ab9f4a 100644
--- a/packages/language-server/src/plugins/typescript-addons/index.ts
+++ b/packages/language-server/src/plugins/typescript-addons/index.ts
@@ -23,7 +23,7 @@ export const create =
 				)
 					return null;
 
-				const [_, source] = context.documents.getVirtualFileByUri(document.uri);
+				const [_, source] = context.project.fileProvider.getVirtualFile(document.uri);
 				const file = source?.root;
 				if (!(file instanceof AstroFile)) return undefined;
 
diff --git a/packages/language-server/src/plugins/typescript/index.ts b/packages/language-server/src/plugins/typescript/index.ts
index 4e222531..1328599d 100644
--- a/packages/language-server/src/plugins/typescript/index.ts
+++ b/packages/language-server/src/plugins/typescript/index.ts
@@ -25,12 +25,12 @@ export const create =
 		return {
 			...typeScriptPlugin,
 			transformCompletionItem(item) {
-				const [_, source] = context.documents.getVirtualFileByUri(item.data.uri);
+				const [_, source] = context.project.fileProvider.getVirtualFile(item.data.uri);
 				const file = source?.root;
 				if (!(file instanceof AstroFile) || !context.project.typescript) return undefined;
-				if (file.scriptFiles.includes(item.data.fileName)) return undefined;
+				if (file.scriptFileIds.includes(context.env.fileNameToUri(item.data.fileName))) return undefined;
 
-				const newLine = context.project.typescript.projectHost.getCompilationSettings().newLine?.toString() ?? '\n';
+				const newLine = context.project.typescript.languageServiceHost.getCompilationSettings().newLine?.toString() ?? '\n';
 				if (item.additionalTextEdits) {
 					item.additionalTextEdits = item.additionalTextEdits.map((edit) => {
 						// HACK: There's a weird situation sometimes where some components (especially Svelte) will get imported as type imports
@@ -51,24 +51,22 @@ export const create =
 			},
 			transformCodeAction(item) {
 				if (item.kind !== 'quickfix') return undefined;
-				const originalFileName = item.data.uri.replace('.tsx', '');
+				const originalUri = item.data.uri.replace('.tsx', '');
 
-				const [_, source] = context.documents.getVirtualFileByUri(originalFileName);
+				const [_, source] = context.project.fileProvider.getVirtualFile(originalUri);
 				const file = source?.root;
 				if (!(file instanceof AstroFile) || !context.project.typescript) return undefined;
 				if (
-					file.scriptFiles.includes(item.diagnostics?.[0].data.documentUri.replace('file://', ''))
+					file.scriptFileIds.includes(item.diagnostics?.[0].data.documentUri)
 				)
 					return undefined;
 
-				const document = context.getTextDocument(originalFileName);
-				if (!document) return undefined;
-
-				const newLine = context.project.typescript.projectHost.getCompilationSettings().newLine?.toString() ?? '\n';
+				const document = context.documents.get(file.id, file.languageId, file.snapshot);
+				const newLine = context.project.typescript.languageServiceHost.getCompilationSettings().newLine?.toString() ?? '\n';
 				if (!item.edit?.documentChanges) return undefined;
 				item.edit.documentChanges = item.edit.documentChanges.map((change) => {
 					if (TextDocumentEdit.is(change)) {
-						change.textDocument.uri = originalFileName;
+						change.textDocument.uri = originalUri;
 						if (change.edits.length === 1) {
 							change.edits = change.edits.map((edit) => {
 								const editInFrontmatter = editShouldBeInFrontmatter(edit.range, document);
@@ -134,7 +132,7 @@ export const create =
 				return enhancedResolveCompletionItem(resolvedCompletionItem);
 			},
 			async provideSemanticDiagnostics(document, token) {
-				const [_, source] = context.documents.getVirtualFileByUri(document.uri);
+				const [_, source] = context.project.fileProvider.getVirtualFile(document.uri);
 				const file = source?.root;
 				if (!(file instanceof AstroFile)) return null;
 
@@ -144,10 +142,10 @@ export const create =
 				const diagnostics = await typeScriptPlugin.provideSemanticDiagnostics!(document, token);
 				if (!diagnostics) return null;
 
-				const astroDocument = context.documents.getDocumentByFileName(
-					file.snapshot,
-					file.sourceFileName,
-					file.languageId
+				const astroDocument = context.documents.get(
+					file.sourceFileId,
+					file.languageId,
+					file.snapshot
 				);
 
 				return enhancedProvideSemanticDiagnostics(diagnostics, astroDocument.lineCount);
diff --git a/packages/ts-plugin/src/astro2tsx.ts b/packages/ts-plugin/src/astro2tsx.ts
index 7bf5aa6c..61d586cd 100644
--- a/packages/ts-plugin/src/astro2tsx.ts
+++ b/packages/ts-plugin/src/astro2tsx.ts
@@ -1,7 +1,7 @@
 import { convertToTSX } from '@astrojs/compiler/sync';
 import type { ConvertToTSXOptions, TSXResult } from '@astrojs/compiler/types';
 import { decode } from '@jridgewell/sourcemap-codec';
-import { FileKind, FileRangeCapabilities, VirtualFile } from '@volar/language-core';
+import { FileKind, VirtualFile } from '@volar/language-core';
 import path from 'node:path';
 import { TextDocument } from 'vscode-languageserver-textdocument';
 
@@ -52,21 +52,21 @@ export function astro2tsx(
 function getVirtualFileTSX(
 	input: string,
 	tsx: TSXResult,
-	fileName: string,
+	fileId: string,
 	ts: typeof import('typescript/lib/tsserverlibrary.js')
 ): VirtualFile {
-	tsx.code = patchTSX(tsx.code, fileName);
+	tsx.code = patchTSX(tsx.code, fileId);
 	const v3Mappings = decode(tsx.map.mappings);
-	const sourcedDoc = TextDocument.create(fileName, 'astro', 0, input);
-	const genDoc = TextDocument.create(fileName + '.tsx', 'typescriptreact', 0, tsx.code);
+	const sourcedDoc = TextDocument.create(fileId, 'astro', 0, input);
+	const genDoc = TextDocument.create(fileId + '.tsx', 'typescriptreact', 0, tsx.code);
 
 	const mappings: VirtualFile['mappings'] = [];
 
 	let current:
 		| {
-				genOffset: number;
-				sourceOffset: number;
-		  }
+			genOffset: number;
+			sourceOffset: number;
+		}
 		| undefined;
 
 	for (let genLine = 0; genLine < v3Mappings.length; genLine++) {
@@ -100,7 +100,7 @@ function getVirtualFileTSX(
 						mappings.push({
 							sourceRange: [current.sourceOffset, current.sourceOffset + length],
 							generatedRange: [current.genOffset, current.genOffset + length],
-							data: FileRangeCapabilities.full,
+							data: {},
 						});
 					}
 				}
@@ -125,19 +125,12 @@ function getVirtualFileTSX(
 		});
 	}
 
+	mappings.forEach(mapping => mapping.data.formattingEdits = false);
+
 	return {
-		fileName: fileName + '.tsx',
+		id: fileId + '.tsx',
 		languageId: 'typescriptreact',
 		kind: FileKind.TypeScriptHostFile,
-		capabilities: {
-			codeAction: true,
-			documentFormatting: false,
-			diagnostic: true,
-			documentSymbol: true,
-			inlayHint: true,
-			foldingRange: true,
-		},
-		codegenStacks: [],
 		snapshot: {
 			getText: (start, end) => tsx.code.substring(start, end),
 			getLength: () => tsx.code.length,
diff --git a/packages/ts-plugin/src/index.ts b/packages/ts-plugin/src/index.ts
index 491728fb..39232d7e 100644
--- a/packages/ts-plugin/src/index.ts
+++ b/packages/ts-plugin/src/index.ts
@@ -14,7 +14,7 @@ const init: ts.server.PluginModuleFactory = (modules) => {
 	const { typescript: ts } = modules;
 	const pluginModule: ts.server.PluginModule = {
 		create(info) {
-			const virtualFiles = createFileProvider([getLanguageModule(ts)], () => {});
+			const virtualFiles = createFileProvider([getLanguageModule(ts)], ts.sys.useCaseSensitiveFileNames, () => {});
 
 			decorateLanguageService(virtualFiles, info.languageService, true);
 			decorateLanguageServiceHost(virtualFiles, info.languageServiceHost, ts, ['.astro']);
diff --git a/packages/ts-plugin/src/language.ts b/packages/ts-plugin/src/language.ts
index 416eba35..29925138 100644
--- a/packages/ts-plugin/src/language.ts
+++ b/packages/ts-plugin/src/language.ts
@@ -1,7 +1,5 @@
 import {
-	FileCapabilities,
 	FileKind,
-	FileRangeCapabilities,
 	type Language,
 	type VirtualFile,
 } from '@volar/language-core';
@@ -12,8 +10,8 @@ export function getLanguageModule(
 	ts: typeof import('typescript/lib/tsserverlibrary.js')
 ): Language<AstroFile> {
 	return {
-		createVirtualFile(fileName, snapshot) {
-			if (fileName.endsWith('.astro')) {
+		createVirtualFile(fileName, languageId, snapshot) {
+			if (languageId === 'astro') {
 				return new AstroFile(fileName, snapshot, ts);
 			}
 		},
@@ -25,20 +23,19 @@ export function getLanguageModule(
 
 export class AstroFile implements VirtualFile {
 	kind = FileKind.TextFile;
-	capabilities = FileCapabilities.full;
 
-	fileName: string;
+	id: string;
 	languageId = 'astro';
 	mappings!: VirtualFile['mappings'];
 	embeddedFiles!: VirtualFile['embeddedFiles'];
 	codegenStacks = [];
 
 	constructor(
-		public sourceFileName: string,
+		public sourceFileId: string,
 		public snapshot: ts.IScriptSnapshot,
 		private readonly ts: typeof import('typescript/lib/tsserverlibrary.js')
 	) {
-		this.fileName = sourceFileName;
+		this.id = sourceFileId;
 		this.onSnapshotUpdated();
 	}
 
@@ -52,7 +49,7 @@ export class AstroFile implements VirtualFile {
 			{
 				sourceRange: [0, this.snapshot.getLength()],
 				generatedRange: [0, this.snapshot.getLength()],
-				data: FileRangeCapabilities.full,
+				data: {},
 			},
 		];
 
@@ -60,7 +57,7 @@ export class AstroFile implements VirtualFile {
 
 		const tsx = astro2tsx(
 			this.snapshot.getText(0, this.snapshot.getLength()),
-			this.fileName,
+			this.id,
 			this.ts
 		);