[ide] add new endpoints for tools

Author: theskcd

extensions/codestory/src/extension.ts

@@ -51,6 +51,7 @@ export async function activate(context: ExtensionContext) {
 						const newAddress: AddressInfo = serverAddress as AddressInfo;
 						port = newAddress.port;
 					}
+					console.log(serverAddress);
 					console.log(`Server for talking to sidecar is running at http://localhost:${server.address()}/`);
 				}); // Use 0 to let the OS pick an available port
 			} else {

extensions/codestory/src/server/diagnostics.ts

@@ -0,0 +1,15 @@
+/*---------------------------------------------------------------------------------------------
+ *  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';
+
+export function getDiagnosticsFromEditor(filePath: string, interestedRange: vscode.Range): vscode.Diagnostic[] {
+	const fileUri = vscode.Uri.file(filePath);
+	const diagnostics = vscode.languages.getDiagnostics(fileUri);
+	diagnostics.filter((diagnostic) => {
+		return interestedRange.contains(diagnostic.range);
+	});
+	return diagnostics;
+}

extensions/codestory/src/server/goToDefinition.ts

@@ -0,0 +1,48 @@
+/*---------------------------------------------------------------------------------------------
+ *  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 { SidecarGoToDefinitionRequest, SidecarGoToDefinitionResponse } from './types';
+import { shouldTrackFile } from '../utilities/openTabs';
+
+export async function goToDefinition(request: SidecarGoToDefinitionRequest): Promise<SidecarGoToDefinitionResponse> {
+	const locations: vscode.LocationLink[] = await vscode.commands.executeCommand(
+		'vscode.executeTypeDefinitionProvider',
+		request.fs_file_path,
+		request.position
+	);
+	const definitons = await Promise.all(locations.map(async (location) => {
+		const uri = location.targetUri;
+		const range = location.targetRange;
+		// we have to always open the text document first, this ends up sending
+		// it over to the sidecar as a side-effect but that is fine
+
+		// No need to await on this
+		if (shouldTrackFile(uri)) {
+			console.log('we are tracking this uri');
+			console.log(uri);
+			// 	sidecarClient.documentOpen(textDocument.uri.fsPath, textDocument.getText(), textDocument.languageId);
+		}
+
+		// return the value as we would normally
+		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,
+				}
+			},
+		};
+	}));
+	// lets return all of them over here
+	return {
+		symbols: definitons,
+	};
+}

extensions/codestory/src/server/openFile.ts

@@ -0,0 +1,18 @@
+/*---------------------------------------------------------------------------------------------
+ *  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 { SidecarOpenFileToolRequest, SidecarOpenFileToolResponse } from './types';
+
+export async function openFileEditor(request: SidecarOpenFileToolRequest): Promise<SidecarOpenFileToolResponse> {
+	const filePath = request.fs_file_path;
+	const textDocument = await vscode.workspace.openTextDocument(filePath);
+	// we get back the text document over here
+	const contents = textDocument.getText();
+	return {
+		fs_file_path: filePath,
+		file_contents: contents,
+	};
+}

extensions/codestory/src/server/requestHandler.ts

@@ -3,6 +3,11 @@
  *  Licensed under the MIT License. See License.txt in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
 import * as http from 'http';
+import { SidecarDiagnosticsRequest, SidecarGoToDefinitionRequest, SidecarOpenFileToolRequest } from './types';
+import { Position, Range } from 'vscode';
+import { getDiagnosticsFromEditor } from './diagnostics';
+import { openFileEditor } from './openFile';
+import { goToDefinition } from './goToDefinition';
 
 // Helper function to read the request body
 function readRequestBody(req: http.IncomingMessage): Promise<string> {
@@ -23,18 +28,37 @@ function readRequestBody(req: http.IncomingMessage): Promise<string> {
 // Async handler function to handle incoming requests
 export async function handleRequest(req: http.IncomingMessage, res: http.ServerResponse) {
 	try {
-		const body = await readRequestBody(req);
-		console.log(body);
-		// const request = JSON.parse(body);
-		// console.log(request);
+		if (req.method === 'POST' && req.url === '/diagnostics') {
+			const body = await readRequestBody(req);
+			console.log('body from post request for diagnostics');
+			console.log(body);
+			console.log('log after post for diagnostics');
+			const diagnosticsBody: SidecarDiagnosticsRequest = JSON.parse(body);
+			const selectionRange = new Range(new Position(diagnosticsBody.range.startPosition.line, diagnosticsBody.range.startPosition.character), new Position(diagnosticsBody.range.endPosition.line, diagnosticsBody.range.endPosition.character));
+			const diagnosticsFromEditor = getDiagnosticsFromEditor(diagnosticsBody.fs_file_path, selectionRange);
+			// Process the diagnostics request asynchronously
+			const response = {
+				'diagnostics': diagnosticsFromEditor,
+			};
 
-		// Process the request asynchronously
-		const response = {
-			'reply': 'gg from vscode',
-		};
-
-		res.writeHead(200, { 'Content-Type': 'application/json' });
-		res.end(JSON.stringify(response));
+			res.writeHead(200, { 'Content-Type': 'application/json' });
+			res.end(JSON.stringify(response));
+		} else if (req.method === 'POST' && req.url === '/file_open') {
+			const body = await readRequestBody(req);
+			const openFileRequest: SidecarOpenFileToolRequest = JSON.parse(body);
+			const response = await openFileEditor(openFileRequest);
+			res.writeHead(200, { 'Content-Type': 'application/json' });
+			res.end(JSON.stringify(response));
+		} else if (req.method === 'POST' && req.url === 'go_to_definition') {
+			const body = await readRequestBody(req);
+			const request: SidecarGoToDefinitionRequest = JSON.parse(body);
+			const response = await goToDefinition(request);
+			res.writeHead(200, { 'Content-Type': 'application/json' });
+			res.end(JSON.stringify(response));
+		} else {
+			res.writeHead(200, { 'Content-Type': 'application/json' });
+			res.end(JSON.stringify({ reply: 'gg' }));
+		}
 	} catch (err) {
 		console.error(err);
 		res.writeHead(500, { 'Content-Type': 'application/json' });

extensions/codestory/src/server/types.ts

@@ -0,0 +1,55 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+// We store the various types of requests we are getting from the sidecar over here
+
+export type SidecarDiagnosticsRequest = {
+	fs_file_path: string;
+	range: {
+		startPosition: {
+			line: number;
+			character: number;
+		};
+		endPosition: {
+			line: number;
+			character: number;
+		};
+	};
+};
+
+export type SidecarGoToDefinitionRequest = {
+	fs_file_path: string;
+	position: {
+		line: number;
+		character: number;
+	};
+};
+
+export type SidecarGoToDefinitionResponse = {
+	symbols: FileAndRange[];
+};
+
+export type FileAndRange = {
+	fs_file_path: string;
+	range: {
+		startPosition: {
+			line: number;
+			character: number;
+		};
+		endPosition: {
+			line: number;
+			character: number;
+		};
+	};
+};
+
+export type SidecarOpenFileToolRequest = {
+	fs_file_path: string;
+};
+
+export type SidecarOpenFileToolResponse = {
+	fs_file_path: string;
+	file_contents: string;
+};