Merge pull request #631 from codestoryai/features/grab-the-references

[ide] grab the references

Author: theskcd

extensions/codestory/src/server/goToReferences.ts

@@ -0,0 +1,40 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as vscode from 'vscode';
+import { SidecarGoToReferencesRequest, SidecarGoToRefernecesResponse } from './types';
+import { shouldTrackFile } from '../utilities/openTabs';
+
+export async function goToReferences(request: SidecarGoToReferencesRequest): Promise<SidecarGoToRefernecesResponse> {
+	const locations: vscode.LocationLink[] = await vscode.commands.executeCommand(
+		'vscode.executeReferenceProvider',
+		request.fs_file_path,
+		request.position,
+	);
+	const implementations = await Promise.all(locations.map(async (location) => {
+		const uri = location.targetUri;
+		const range = location.targetRange;
+		if (shouldTrackFile(uri)) {
+			console.log('we are trakcing this uri');
+			console.log(uri);
+		}
+		return {
+			fs_file_path: uri.fsPath,
+			range: {
+				startPosition: {
+					line: range.start.line,
+					character: range.start.character,
+				},
+				endPosition: {
+					line: range.end.line,
+					character: range.end.character,
+				},
+			}
+		};
+	}));
+	return {
+		reference_locations: implementations,
+	};
+}

extensions/codestory/src/server/requestHandler.ts

@@ -3,7 +3,7 @@
  *  Licensed under the MIT License. See License.txt in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
 import * as http from 'http';
-import { SidecarApplyEditsRequest, SidecarDiagnosticsRequest, SidecarGoToDefinitionRequest, SidecarGoToImplementationRequest, SidecarOpenFileToolRequest, SidecarQuickFixInvocationRequest, SidecarQuickFixRequest } from './types';
+import { SidecarApplyEditsRequest, SidecarDiagnosticsRequest, SidecarGoToDefinitionRequest, SidecarGoToImplementationRequest, SidecarGoToReferencesRequest, SidecarOpenFileToolRequest, SidecarQuickFixInvocationRequest, SidecarQuickFixRequest } from './types';
 import { Position, Range } from 'vscode';
 import { getDiagnosticsFromEditor } from './diagnostics';
 import { openFileEditor } from './openFile';
@@ -12,6 +12,7 @@ import { SIDECAR_CLIENT } from '../extension';
 import { goToImplementation } from './goToImplementation';
 import { quickFixInvocation, quickFixList } from './quickFix';
 import { applyEdits } from './applyEdits';
+import { goToReferences } from './goToReferences';
 
 // Helper function to read the request body
 function readRequestBody(req: http.IncomingMessage): Promise<string> {
@@ -95,6 +96,13 @@ export async function handleRequest(req: http.IncomingMessage, res: http.ServerR
 			const response = await applyEdits(request);
 			res.writeHead(200, { 'Content-Type': 'application/json' });
 			res.end(JSON.stringify(response));
+		} else if (req.method === 'POST' && req.url === '/go_to_references') {
+			console.log('go-to-references');
+			const body = await readRequestBody(req);
+			const request: SidecarGoToReferencesRequest = JSON.parse(body);
+			const response = await goToReferences(request);
+			res.writeHead(200, { 'Content-Type': 'application/json' });
+			res.end(JSON.stringify(response));
 		} else {
 			console.log('HC request');
 			res.writeHead(200, { 'Content-Type': 'application/json' });

extensions/codestory/src/server/types.ts

@@ -68,6 +68,18 @@ export type SidecarGoToImplementationResponse = {
 	implementation_locations: FileAndRange[];
 };
 
+export type SidecarGoToReferencesRequest = {
+	fs_file_path: string;
+	position: {
+		line: number;
+		character: number;
+	};
+};
+
+export type SidecarGoToRefernecesResponse = {
+	reference_locations: FileAndRange[];
+};
+
 export type SidecarQuickFixRequest = {
 	fs_file_path: string;
 	request_id: string;