code-agent: (#1440)

Implement code editor actions.

- user requests like `Create a TypeScript function called toSnakeCase
that takes a string and returns a string` will be handled by the
createFunction action
- Leverage copilot inline suggestion to create the function body and
automatically accept the suggestion. This will able to us to experiment
with rapid coding using natural language.
- Detect if coda extension is connected.
- Errors out instead of hanging if the coda extension if not running
- Added code to handle creation of file, and save actions

Author: pcdeadeasy

ts/packages/agents/code/src/codeActionHandler.ts

@@ -16,6 +16,7 @@ import { fileURLToPath } from "url";
 import os from "os";
 import registerDebug from "debug";
 import chalk from "chalk";
+import { createActionResultFromError } from "@typeagent/agent-sdk/helpers/action";
 
 const debug = registerDebug("typeagent:code");
 
@@ -82,7 +83,9 @@ async function updateCodeContext(
                     if (pendingCall) {
                         agentContext.pendingCall.delete(Number(data.id));
                         const { resolve, context } = pendingCall;
-                        context.actionIO.setDisplay(data.result);
+                        if (context?.actionIO) {
+                            context.actionIO.setDisplay(data.result);
+                        }
                         resolve();
                     }
                 }
@@ -174,6 +177,88 @@ async function ensureVSCodeProcess(): Promise<void> {
     });
 }
 
+async function sendPingToCodaExtension(
+    agentContext: CodeActionContext,
+): Promise<boolean> {
+    const ws = agentContext.webSocket;
+    if (!ws || ws.readyState !== 1) return false;
+
+    const callId = agentContext.nextCallId++;
+    return new Promise((resolve) => {
+        const timeout = setTimeout(() => {
+            agentContext.pendingCall.delete(callId);
+            resolve(false);
+        }, 1000);
+
+        agentContext.pendingCall.set(callId, {
+            resolve: () => {
+                clearTimeout(timeout);
+                resolve(true);
+            },
+            context: undefined as any,
+        });
+
+        ws.send(
+            JSON.stringify({
+                id: callId,
+                method: "code/ping",
+                params: {},
+            }),
+        );
+    });
+}
+
+type ActiveFile = {
+    filePath: string;
+    languageId: string;
+    isUntitled: boolean;
+    isDirty: boolean;
+};
+
+export async function getActiveFileFromVSCode(
+    agentContext: CodeActionContext,
+    timeoutMs = 2000,
+): Promise<ActiveFile | undefined> {
+    const ws = agentContext.webSocket;
+
+    if (!ws || ws.readyState !== 1 /* OPEN */) {
+        return undefined;
+    }
+
+    const callId = agentContext.nextCallId++;
+
+    return new Promise<ActiveFile | undefined>((resolve) => {
+        // Hard timeout so we never hang
+        const t = setTimeout(() => {
+            agentContext.pendingCall.delete(callId);
+            resolve(undefined);
+        }, timeoutMs);
+
+        // NOTE: pendingCall entry has no ActionContext because this isn’t a UI action
+        agentContext.pendingCall.set(callId, {
+            resolve: (value?: any) => {
+                clearTimeout(t);
+                resolve(value as ActiveFile | undefined);
+            },
+            context: undefined as any,
+        });
+
+        try {
+            ws.send(
+                JSON.stringify({
+                    id: callId,
+                    method: "code/getActiveFile",
+                    params: {},
+                }),
+            );
+        } catch {
+            clearTimeout(t);
+            agentContext.pendingCall.delete(callId);
+            resolve(undefined);
+        }
+    });
+}
+
 async function executeCodeAction(
     action: AppAction,
     context: ActionContext<CodeActionContext>,
@@ -185,8 +270,17 @@ async function executeCodeAction(
 
     const agentContext = context.sessionContext.agentContext;
     const webSocketEndpoint = agentContext.webSocket;
+
     if (webSocketEndpoint) {
         try {
+            const isExtensionAlive =
+                await sendPingToCodaExtension(agentContext);
+            if (!isExtensionAlive) {
+                return createActionResultFromError(
+                    "❌ Coda VSCode extension is not connected.",
+                );
+            }
+
             const callId = agentContext.nextCallId++;
             return new Promise<undefined>((resolve) => {
                 agentContext.pendingCall.set(callId, {

ts/packages/agents/code/src/codeManifest.json

@@ -34,9 +34,9 @@
     },
     "code-editor": {
       "schema": {
-        "description": "Code agent that helps you perform vscode editor actions like go to a reference, reveal a declaration, clipboard copy or paste, add a comment line etc.",
-        "schemaFile": "./vscode/editorCommandActionsSchema.ts",
-        "schemaType": "CodeEditorActions"
+        "description": "Code agent that helps you perform vscode editor actions like create a new file, go to a reference, reveal a declaration, clipboard copy or paste, add a comment line etc.",
+        "schemaFile": "./vscode/editorCodeActionsSchema.ts",
+        "schemaType": "EditorCodeActions"
       }
     },
     "code-workbench": {