Skip to content

Commit 52c59db

Browse files
sandersnsandersn
authored
Add CopilotRelated command (#59963)
1 parent da1fb07 commit 52c59db

17 files changed

+298
-1
lines changed

‎src/harness/client.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -800,6 +800,7 @@ export class SessionClient implements LanguageService {
800800
}
801801

802802
mapCode: typeof notImplemented = notImplemented;
803+
getImports: typeof notImplemented = notImplemented;
803804

804805
private createFileLocationOrRangeRequestArgs(positionOrRange: number | TextRange, fileName: string): protocol.FileLocationOrRangeRequestArgs {
805806
return typeof positionOrRange === "number"

‎src/harness/fourslashImpl.ts

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4638,6 +4638,28 @@ ${changes.join("\n// ---\n")}
46384638
${after}`;
46394639
this.baseline("mapCode", baseline, ".mapCode.ts");
46404640
}
4641+
4642+
public verifyGetImports(fileName: string, expectedImports: string[]): void {
4643+
const actualImports = this.languageService.getImports(fileName);
4644+
if (actualImports.length !== expectedImports.length) {
4645+
throw new Error(`Expected ${expectedImports.length} imports for ${fileName}, got ${actualImports.length}
4646+
Expected:
4647+
${expectedImports}
4648+
Actual:
4649+
${actualImports}
4650+
`);
4651+
}
4652+
for (let i = 0; i < expectedImports.length; i++) {
4653+
if (actualImports[i] !== expectedImports[i]) {
4654+
throw new Error(`Expected at ${fileName} index ${i}: ${expectedImports[i]}, got ${actualImports[i]}
4655+
Expected:
4656+
${expectedImports}
4657+
Actual:
4658+
${actualImports}
4659+
`);
4660+
}
4661+
}
4662+
}
46414663
}
46424664

46434665
function updateTextRangeForTextChanges({ pos, end }: ts.TextRange, textChanges: readonly ts.TextChange[]): ts.TextRange {

‎src/harness/fourslashInterfaceImpl.ts

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -257,6 +257,10 @@ export class VerifyNegatable {
257257
public baselineMapCode(ranges: FourSlash.Range[][], changes: string[] = []): void {
258258
this.state.baselineMapCode(ranges, changes);
259259
}
260+
261+
public getImports(fileName: string, imports: string[]): void {
262+
return this.state.verifyGetImports(fileName, imports);
263+
}
260264
}
261265

262266
export interface CompletionsResult {
@@ -2047,3 +2051,8 @@ export interface RenameOptions {
20472051
readonly providePrefixAndSuffixTextForRename?: boolean;
20482052
readonly quotePreference?: "auto" | "double" | "single";
20492053
}
2054+
2055+
export interface VerifyGetImportsOptions {
2056+
fileName: string;
2057+
imports: string[];
2058+
}

‎src/server/protocol.ts

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -202,6 +202,7 @@ export const enum CommandTypes {
202202
ProvideInlayHints = "provideInlayHints",
203203
WatchChange = "watchChange",
204204
MapCode = "mapCode",
205+
CopilotRelated = "copilotRelated",
205206
}
206207

207208
/**
@@ -2406,6 +2407,18 @@ export interface MapCodeResponse extends Response {
24062407
body: readonly FileCodeEdits[];
24072408
}
24082409

2410+
export interface CopilotRelatedRequest extends FileRequest {
2411+
command: CommandTypes.CopilotRelated;
2412+
arguments: FileRequestArgs;
2413+
}
2414+
2415+
export interface CopilotRelatedItems {
2416+
relatedFiles: readonly string[];
2417+
}
2418+
2419+
export interface CopilotRelatedResponse extends Response {
2420+
body: CopilotRelatedItems;
2421+
}
24092422
/**
24102423
* Synchronous request for semantic diagnostics of one file.
24112424
*/

‎src/server/session.ts

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -937,6 +937,7 @@ const invalidPartialSemanticModeCommands: readonly protocol.CommandTypes[] = [
937937
protocol.CommandTypes.ProvideCallHierarchyIncomingCalls,
938938
protocol.CommandTypes.ProvideCallHierarchyOutgoingCalls,
939939
protocol.CommandTypes.GetPasteEdits,
940+
protocol.CommandTypes.CopilotRelated,
940941
];
941942

942943
const invalidSyntacticModeCommands: readonly protocol.CommandTypes[] = [
@@ -2030,7 +2031,6 @@ export class Session<TMessage = string> implements EventSender {
20302031
};
20312032
});
20322033
}
2033-
20342034
private mapCode(args: protocol.MapCodeRequestArgs): protocol.FileCodeEdits[] {
20352035
const formatOptions = this.getHostFormatOptions();
20362036
const preferences = this.getHostPreferences();
@@ -2051,6 +2051,14 @@ export class Session<TMessage = string> implements EventSender {
20512051
return this.mapTextChangesToCodeEdits(changes);
20522052
}
20532053

2054+
private getCopilotRelatedInfo(args: protocol.FileRequestArgs): protocol.CopilotRelatedItems {
2055+
const { file, project } = this.getFileAndProject(args);
2056+
2057+
return {
2058+
relatedFiles: project.getLanguageService().getImports(file),
2059+
};
2060+
}
2061+
20542062
private setCompilerOptionsForInferredProjects(args: protocol.SetCompilerOptionsForInferredProjectsArgs): void {
20552063
this.projectService.setCompilerOptionsForInferredProjects(args.options, args.projectRootPath);
20562064
}
@@ -3791,6 +3799,9 @@ export class Session<TMessage = string> implements EventSender {
37913799
[protocol.CommandTypes.MapCode]: (request: protocol.MapCodeRequest) => {
37923800
return this.requiredResponse(this.mapCode(request.arguments));
37933801
},
3802+
[protocol.CommandTypes.CopilotRelated]: (request: protocol.CopilotRelatedRequest) => {
3803+
return this.requiredResponse(this.getCopilotRelatedInfo(request.arguments));
3804+
},
37943805
}));
37953806

37963807
public addProtocolHandler(command: string, handler: (request: protocol.Request) => HandlerResponse): void {

‎src/services/services.ts

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ import {
22
__String,
33
ApplicableRefactorInfo,
44
ApplyCodeActionCommandResult,
5+
arrayFrom,
56
AssignmentDeclarationKind,
67
BaseType,
78
BinaryExpression,
@@ -233,6 +234,7 @@ import {
233234
Node,
234235
NodeArray,
235236
NodeFlags,
237+
nodeIsSynthesized,
236238
noop,
237239
normalizePath,
238240
normalizeSpans,
@@ -1602,6 +1604,7 @@ const invalidOperationsInPartialSemanticMode: readonly (keyof LanguageService)[]
16021604
"provideInlayHints",
16031605
"getSupportedCodeFixes",
16041606
"getPasteEdits",
1607+
"getImports",
16051608
];
16061609

16071610
const invalidOperationsInSyntacticMode: readonly (keyof LanguageService)[] = [
@@ -3364,6 +3367,18 @@ export function createLanguageService(
33643367
);
33653368
}
33663369

3370+
function getImports(fileName: string): readonly string[] {
3371+
synchronizeHostData();
3372+
const file = getValidSourceFile(fileName);
3373+
let imports: Set<string> | undefined;
3374+
for (const specifier of file.imports) {
3375+
if (nodeIsSynthesized(specifier)) continue;
3376+
const name = program.getResolvedModuleFromModuleSpecifier(specifier, file)?.resolvedModule?.resolvedFileName;
3377+
if (name) (imports ??= new Set()).add(name);
3378+
}
3379+
return imports ? arrayFrom(imports) : emptyArray;
3380+
}
3381+
33673382
const ls: LanguageService = {
33683383
dispose,
33693384
cleanupSemanticCache,
@@ -3438,6 +3453,7 @@ export function createLanguageService(
34383453
preparePasteEditsForFile,
34393454
getPasteEdits,
34403455
mapCode,
3456+
getImports,
34413457
};
34423458

34433459
switch (languageServiceMode) {

‎src/services/types.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -697,6 +697,7 @@ export interface LanguageService {
697697
getSupportedCodeFixes(fileName?: string): readonly string[];
698698

699699
/** @internal */ mapCode(fileName: string, contents: string[], focusLocations: TextSpan[][] | undefined, formatOptions: FormatCodeSettings, preferences: UserPreferences): readonly FileTextChanges[];
700+
/** @internal */ getImports(fileName: string): readonly string[];
700701

701702
dispose(): void;
702703
preparePasteEditsForFile(fileName: string, copiedTextRanges: TextRange[]): boolean;

‎tests/baselines/reference/api/typescript.d.ts

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,7 @@ declare namespace ts {
123123
ProvideInlayHints = "provideInlayHints",
124124
WatchChange = "watchChange",
125125
MapCode = "mapCode",
126+
CopilotRelated = "copilotRelated",
126127
}
127128
/**
128129
* A TypeScript Server message
@@ -1830,6 +1831,16 @@ declare namespace ts {
18301831
export interface MapCodeResponse extends Response {
18311832
body: readonly FileCodeEdits[];
18321833
}
1834+
export interface CopilotRelatedRequest extends FileRequest {
1835+
command: CommandTypes.CopilotRelated;
1836+
arguments: FileRequestArgs;
1837+
}
1838+
export interface CopilotRelatedItems {
1839+
relatedFiles: readonly string[];
1840+
}
1841+
export interface CopilotRelatedResponse extends Response {
1842+
body: CopilotRelatedItems;
1843+
}
18331844
/**
18341845
* Synchronous request for semantic diagnostics of one file.
18351846
*/
@@ -3514,6 +3525,7 @@ declare namespace ts {
35143525
private getDocumentHighlights;
35153526
private provideInlayHints;
35163527
private mapCode;
3528+
private getCopilotRelatedInfo;
35173529
private setCompilerOptionsForInferredProjects;
35183530
private getProjectInfo;
35193531
private getProjectInfoWorker;

‎tests/cases/fourslash/fourslash.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -472,6 +472,7 @@ declare namespace FourSlashInterface {
472472
}
473473
}): void;
474474
baselineMapCode(ranges: Range[][], changes: string[]): void;
475+
getImports(fileName: string, imports: string[]): void;
475476
}
476477
class edit {
477478
caretPosition(): Marker;

‎tests/cases/fourslash/getImportsDuplicate.ts

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
///<reference path="fourslash.ts"/>
2+
3+
// @Filename: /first.ts
4+
//// export function foo() {
5+
//// return 1;
6+
//// }
7+
//// export function bar() {
8+
//// return 2;
9+
//// }
10+
11+
// @Filename: /index.ts
12+
//// import { foo } from "./first";
13+
//// import { bar } from './first';
14+
//// console.log(foo() + bar())
15+
16+
verify.getImports('/index.ts', ['/first.ts'])

0 commit comments

Comments
 (0)