diff --git a/.vscode/launch.json b/.vscode/launch.json index 531210a572..b575d4b706 100644 --- a/.vscode/launch.json +++ b/.vscode/launch.json @@ -7,8 +7,8 @@ "type": "extensionHost", "request": "launch", "runtimeExecutable": "${execPath}", - "args": ["--extensionDevelopmentPath=${workspaceRoot}/tools/astro-vscode"], - "outFiles": ["${workspaceRoot}/tools/astro-vscode/dist/**/*.js"] + "args": ["--extensionDevelopmentPath=${workspaceRoot}/tools/vscode"], + "outFiles": ["${workspaceRoot}/tools/vscode/dist/**/*.js"] }, { "type": "node", @@ -16,7 +16,7 @@ "name": "Attach to Server", "port": 6040, "restart": true, - "outFiles": ["${workspaceRoot}/tools/astro-languageserver/dist/**/*.js"], + "outFiles": ["${workspaceRoot}/tools/languageserver/dist/**/*.js"], "skipFiles": ["/**"] } ], diff --git a/tools/language-server/src/plugins/typescript/DocumentSnapshot.ts b/tools/language-server/src/plugins/typescript/DocumentSnapshot.ts index 9e2e778c67..606db8d58b 100644 --- a/tools/language-server/src/plugins/typescript/DocumentSnapshot.ts +++ b/tools/language-server/src/plugins/typescript/DocumentSnapshot.ts @@ -5,6 +5,8 @@ import { isInTag, positionAt, offsetAt } from '../../core/documents/utils'; import { pathToUrl } from '../../utils'; import { getScriptKindFromFileName, isAstroFilePath, toVirtualAstroFilePath } from './utils'; +const FILLER_DEFAULT_EXPORT = `\nexport default function() { return ''; };`; + /** * The mapper to get from original snapshot positions to generated and vice versa. */ @@ -66,7 +68,15 @@ class AstroDocumentSnapshot implements DocumentSnapshot { } get text() { - return this.doc.getText(); + let raw = this.doc.getText(); + return this.transformContent(raw); + } + + /** @internal */ + private transformContent(content: string) { + return content.replace(/---/g, '///') + + // TypeScript needs this to know there's a default export. + FILLER_DEFAULT_EXPORT; } get filePath() { @@ -128,7 +138,9 @@ export class DocumentFragmentSnapshot implements Omit { + defs.definitions!.map(async (def) => { const { fragment, snapshot } = await docs.retrieve(def.fileName); if (isNoTextSpanInGeneratedCode(snapshot.getFullText(), def.textSpan)) { const fileName = isVirtualFilePath(def.fileName) ? def.fileName.substr(0, def.fileName.length - 3) : def.fileName; + const textSpan = isVirtualAstroFilePath(tsFilePath) ? { start: 0, length: 0 } : def.textSpan; return LocationLink.create( pathToUrl(fileName), - convertToLocationRange(fragment, def.textSpan), - convertToLocationRange(fragment, def.textSpan), - convertToLocationRange(mainFragment, defs.textSpan) + convertToLocationRange(fragment, textSpan), + convertToLocationRange(fragment, textSpan), + convertToLocationRange(mainFragment, defs!.textSpan) ); } }) @@ -110,4 +128,46 @@ export class TypeScriptPlugin implements CompletionsProvider { private isInsideFrontmatter(document: Document, position: Position) { return isInsideFrontmatter(document.getText(), document.offsetAt(position)); } + + private goToDefinitionFoundOnlyAlias(tsFileName: string, defs: readonly ts.DefinitionInfo[]) { + return !!(defs.length === 1 && + defs[0].kind === 'alias' && + defs[0].fileName === tsFileName); + } + + private getGoToDefinitionRefsForImportSpecifier(tsFilePath: string, offset: number, lang: ts.LanguageService): ts.DefinitionInfoAndBoundSpan | undefined { + const program = lang.getProgram(); + const sourceFile = program?.getSourceFile(tsFilePath); + if (sourceFile) { + let node = (ts as BetterTS).getTouchingPropertyName(sourceFile, offset); + if(node && node.kind === SyntaxKind.Identifier) { + if(node.parent.kind === SyntaxKind.ImportClause) { + let decl = node.parent.parent as ImportDeclaration; + let spec = ts.isStringLiteral(decl.moduleSpecifier) && decl.moduleSpecifier.text; + if(spec) { + let fileName = pathJoin(pathDirname(tsFilePath), spec); + let start = node.pos + 1; + let def: ts.DefinitionInfoAndBoundSpan = { + definitions: [{ + kind: 'alias', + fileName, + name: '', + containerKind: '', + containerName: '', + textSpan: { + start: 0, + length: 0 + } + } as ts.DefinitionInfo], + textSpan: { + start, + length: node.end - start + } + }; + return def; + } + } + } + } + } } diff --git a/tools/language-server/src/plugins/typescript/utils.ts b/tools/language-server/src/plugins/typescript/utils.ts index 9acbe2ed82..d84f35da1a 100644 --- a/tools/language-server/src/plugins/typescript/utils.ts +++ b/tools/language-server/src/plugins/typescript/utils.ts @@ -190,7 +190,7 @@ export function ensureRealAstroFilePath(filePath: string) { } export function ensureRealFilePath(filePath: string) { - return isVirtualFilePath(filePath) ? filePath.slice(0, 3) : filePath; + return isVirtualFilePath(filePath) ? filePath.slice(0, filePath.length - 3) : filePath; } export function findTsConfigPath(fileName: string, rootUris: string[]) {