Function initWs

ext/src/extension.ts:46–110

(context: vscode.ExtensionContext)

  

function initWs(context: vscode.ExtensionContext) {
	// change to something else
	const ws = new WebSocket(process.env.WS_RELAYER_URL || "ws://localhost:9093");
	
	ws.onerror = (e: any) => {
		console.log("error", );
		console.log(e);
		console.log(JSON.stringify(e));
	}

	ws.onopen = () => {
		ws.send(JSON.stringify({
			event: "subscribe",
			data: null
		}));
	}

	ws.onmessage = async (e: any) => {
		const data: AdminMessage = JSON.parse(e.data);
		console.log('data');
		console.log(data);
		if (data.type === "command") {
			vscode.commands.executeCommand('extension.sendToAiTerminal', data.content);
		}

		if (data.type === "update-file") {
			const fileUri = await ensureFileExists(data.path!, data.content);

			const document = await vscode.workspace.openTextDocument(fileUri);
			await vscode.window.showTextDocument(document);

			const edit = new vscode.WorkspaceEdit();
			const range = new vscode.Range(
				new vscode.Position(0, 0),
				new vscode.Position(document.lineCount, 0)
			);

			edit.replace(document.uri, range, data.content);
			await vscode.workspace.applyEdit(edit);
		}

		if (data.type === "prompt-start") {
			const terminals = vscode.window.terminals;
			if (terminals.length > 0) {
				const activeTerminal = vscode.window.activeTerminal;
				activeTerminal?.sendText('\x03');
			}
			// get response from git diff command silently and forward it via ws to the server
			const diff = await vscode.commands.executeCommand('git diff');
			ws.send(JSON.stringify({
				event: "vscode_diff",
				diff: diff,
				callbackId: data.callbackId
			}));
		}

		if (data.type === "prompt-end") {
			vscode.commands.executeCommand('git add .');