06a782a2e3
This commit adds "find references" functionality to the Ivy integrated language service. The basic approach is as follows: 1. Generate shims for all files to ensure we find references in shims throughout the entire program 2. Determine if the position for the reference request is within a template. * Yes, it is in a template: Find which node in the template AST the position refers to. Then find the position in the shim file for that template node. Pass the shim file and position in the shim file along to step 3. * No, the request for references was made outside a template: Forward the file and position to step 3. 3. (`getReferencesAtTypescriptPosition`): Call the native TypeScript LS `getReferencesAtPosition`. For each reference that is in a shim file, map those back to a template location, otherwise return it as-is. PR Close #39768
73 lines
2.5 KiB
TypeScript
73 lines
2.5 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright Google LLC All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by an MIT-style license that can be
|
|
* found in the LICENSE file at https://angular.io/license
|
|
*/
|
|
|
|
import * as ts from 'typescript/lib/tsserverlibrary';
|
|
import {LanguageService} from './language_service';
|
|
|
|
export function create(info: ts.server.PluginCreateInfo): ts.LanguageService {
|
|
const {project, languageService: tsLS, config} = info;
|
|
const angularOnly = config?.angularOnly === true;
|
|
|
|
const ngLS = new LanguageService(project, tsLS);
|
|
|
|
function getSemanticDiagnostics(fileName: string): ts.Diagnostic[] {
|
|
const diagnostics: ts.Diagnostic[] = [];
|
|
if (!angularOnly) {
|
|
diagnostics.push(...tsLS.getSemanticDiagnostics(fileName));
|
|
}
|
|
diagnostics.push(...ngLS.getSemanticDiagnostics(fileName));
|
|
return diagnostics;
|
|
}
|
|
|
|
function getQuickInfoAtPosition(fileName: string, position: number): ts.QuickInfo|undefined {
|
|
if (angularOnly) {
|
|
return ngLS.getQuickInfoAtPosition(fileName, position);
|
|
} else {
|
|
// If TS could answer the query, then return that result. Otherwise, return from Angular LS.
|
|
return tsLS.getQuickInfoAtPosition(fileName, position) ??
|
|
ngLS.getQuickInfoAtPosition(fileName, position);
|
|
}
|
|
}
|
|
|
|
function getTypeDefinitionAtPosition(
|
|
fileName: string, position: number): readonly ts.DefinitionInfo[]|undefined {
|
|
if (angularOnly) {
|
|
return ngLS.getTypeDefinitionAtPosition(fileName, position);
|
|
} else {
|
|
// If TS could answer the query, then return that result. Otherwise, return from Angular LS.
|
|
return tsLS.getTypeDefinitionAtPosition(fileName, position) ??
|
|
ngLS.getTypeDefinitionAtPosition(fileName, position);
|
|
}
|
|
}
|
|
|
|
function getDefinitionAndBoundSpan(
|
|
fileName: string, position: number): ts.DefinitionInfoAndBoundSpan|undefined {
|
|
if (angularOnly) {
|
|
return ngLS.getDefinitionAndBoundSpan(fileName, position);
|
|
} else {
|
|
// If TS could answer the query, then return that result. Otherwise, return from Angular LS.
|
|
return tsLS.getDefinitionAndBoundSpan(fileName, position) ??
|
|
ngLS.getDefinitionAndBoundSpan(fileName, position);
|
|
}
|
|
}
|
|
|
|
function getReferencesAtPosition(fileName: string, position: number): ts.ReferenceEntry[]|
|
|
undefined {
|
|
return ngLS.getReferencesAtPosition(fileName, position);
|
|
}
|
|
|
|
return {
|
|
...tsLS,
|
|
getSemanticDiagnostics,
|
|
getTypeDefinitionAtPosition,
|
|
getQuickInfoAtPosition,
|
|
getDefinitionAndBoundSpan,
|
|
getReferencesAtPosition,
|
|
};
|
|
}
|