Merge pull request #620 from codestoryai/features/apply-edits-endpoint

[ide] apply edits endpoint

Author: theskcd

extensions/codestory/src/server/applyEdits.ts

@@ -0,0 +1,54 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { SidecarApplyEditsRequest, SidecarApplyEditsResponse } from './types';
+import * as vscode from 'vscode';
+
+/**
+ * We want to apply edits to the codebase over here and try to get this ti work
+ */
+
+export async function applyEdits(request: SidecarApplyEditsRequest): Promise<SidecarApplyEditsResponse> {
+	const filePath = request.fs_file_path;
+	const startPosition = request.selected_range.startPosition;
+	const endPosition = request.selected_range.endPosition;
+	const replacedText = request.edited_content;
+	const range = new vscode.Range(new vscode.Position(startPosition.line, startPosition.character), new vscode.Position(endPosition.line, endPosition.character));
+	const workspaceEdit = new vscode.WorkspaceEdit();
+	workspaceEdit.replace(
+		vscode.Uri.file(filePath),
+		range,
+		replacedText,
+	);
+	const success = await vscode.workspace.applyEdit(workspaceEdit);
+	// we calculate how many lines we get after replacing the text
+	// once we make the edit on the range, the new range is presented to us
+	// we have to calculate the new range and use that instead
+	// simple algo here would be the following:
+	const lines = replacedText.split(/\r\n|\r|\n/);
+	let lastLineColumn = 0;
+	if (lines.length > 0) {
+		lastLineColumn = lines[lines.length - 1].length
+	} else {
+		lastLineColumn = replacedText.length + startPosition.character;
+	}
+	const newRange = {
+		startPosition: {
+			line: startPosition.line,
+			character: startPosition.character,
+			byte_offset: 0,
+		},
+		endPosition: {
+			line: startPosition.line + replacedText.split(/\r\n|\r|\n/).length,
+			character: lastLineColumn,
+			byte_offset: 0,
+		}
+	};
+	return {
+		fs_file_path: filePath,
+		success,
+		new_range: newRange,
+	};
+}

extensions/codestory/src/server/requestHandler.ts

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

@@ -101,3 +101,35 @@ export type SidecarQuickFixInvocationResponse = {
 	request_id: string;
 	invocation_success: boolean;
 };
+
+export type SidecarApplyEditsRequest = {
+	fs_file_path: string;
+	edited_content: string;
+	selected_range: {
+		startPosition: {
+			line: number;
+			character: number;
+		};
+		endPosition: {
+			line: number;
+			character: number;
+		};
+	};
+}
+
+export type SidecarApplyEditsResponse = {
+	fs_file_path: string;
+	success: boolean;
+	new_range: {
+		startPosition: {
+			line: number;
+			character: number;
+			byte_offset: number;
+		};
+		endPosition: {
+			line: number;
+			character: number;
+			byte_offset: number;
+		};
+	};
+};