@polka-codes/runner 0.7.24 → 0.8.1

package/dist/index.js

@@ -25790,11 +25790,11 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.7.24";
+var version = "0.8.1";
 
 // src/runner.ts
 import { execSync } from "node:child_process";
-import { existsSync as existsSync2, readFileSync as readFileSync2 } from "node:fs";
+import { promises as fs5 } from "node:fs";
 
 // ../cli-shared/src/config.ts
 import { existsSync, readFileSync } from "node:fs";
@@ -54535,6 +54535,11 @@ var wsOutgoingMessageSchema = z3.discriminatedUnion("type", [
       response: z3.string()
     }))
   }),
+  z3.object({
+    type: z3.literal("pending_tools_response_completed"),
+    step: z3.number(),
+    index: z3.number()
+  }),
   z3.object({
     type: z3.literal("file"),
     path: z3.string(),
@@ -54742,29 +54747,71 @@ class Runner {
     console.log(`Received tool requests for step ${message.step}:`, message.requests.map((r2) => r2.tool));
     const responses = [];
     for (const request of message.requests) {
-      let respMsg;
-      try {
-        console.log(`Executing tool: ${request.tool} with params:`, request.params);
-        const tool2 = this.availableTools[request.tool];
-        if (tool2) {
-          const resp = await tool2.handler(this.provider, request.params);
-          if (resp.type === "Reply" /* Reply */) {
-            respMsg = responsePrompts.toolResults(request.tool, resp.message);
-          } else {
-            respMsg = responsePrompts.errorInvokeTool(request.tool, `Unexpected tool response: ${JSON.stringify(resp)}`);
+      const fn = async () => {
+        try {
+          console.log(`Executing tool: ${request.tool} with params:`, request.params);
+          if (request.params.overridenAgent === "coder" && this.provider.executeCommand) {
+            const { format, check, test } = request.params;
+            if (format) {
+              try {
+                await this.provider.executeCommand(format, false);
+              } catch (error) {
+                console.warn(`Failed to format code using command: ${format}`, error);
+              }
+            }
+            if (check) {
+              try {
+                const { exitCode, stdout, stderr } = await this.provider.executeCommand(check, false);
+                if (exitCode !== 0) {
+                  return responsePrompts.commandResult(check, exitCode, stdout, stderr);
+                }
+              } catch (error) {
+                console.warn(`Failed to check code using command: ${check}`, error);
+              }
+            }
+            if (test) {
+              try {
+                const { exitCode, stdout, stderr } = await this.provider.executeCommand(test, false);
+                if (exitCode !== 0) {
+                  return responsePrompts.commandResult(test, exitCode, stdout, stderr);
+                }
+              } catch (error) {
+                console.warn(`Failed to test code using command: ${test}`, error);
+              }
+            }
+            return {
+              type: "exit"
+            };
           }
-        } else {
-          respMsg = responsePrompts.errorInvokeTool(request.tool, "Tool not available");
+          const tool2 = this.availableTools[request.tool];
+          if (tool2) {
+            const resp = await tool2.handler(this.provider, request.params);
+            if (resp.type === "Reply" /* Reply */) {
+              return responsePrompts.toolResults(request.tool, resp.message);
+            }
+            return responsePrompts.errorInvokeTool(request.tool, `Unexpected tool response: ${JSON.stringify(resp)}`);
+          }
+          return responsePrompts.errorInvokeTool(request.tool, "Tool not available");
+        } catch (toolError) {
+          console.error(`Error executing tool ${request.tool}:`, toolError);
+          return responsePrompts.errorInvokeTool(request.tool, toolError);
         }
-      } catch (toolError) {
-        console.error(`Error executing tool ${request.tool}:`, toolError);
-        respMsg = responsePrompts.errorInvokeTool(request.tool, toolError);
+      };
+      const respMsg = await fn();
+      if (typeof respMsg === "string") {
+        responses.push({
+          index: request.index,
+          tool: request.tool,
+          response: respMsg
+        });
+      } else if (respMsg.type === "exit") {
+        this.wsManager.sendMessage({
+          type: "pending_tools_response_completed",
+          step: message.step,
+          index: request.index
+        });
+        return;
       }
-      responses.push({
-        index: request.index,
-        tool: request.tool,
-        response: respMsg
-      });
     }
     this.wsManager.sendMessage({
       type: "pending_tools_response",
@@ -54781,7 +54828,7 @@ class Runner {
         switch (change.status) {
           case "added":
           case "modified":
-            this.sendFileContent(change.path);
+            await this.sendFileContent(change.path);
             break;
           case "deleted":
             this.sendFileDeleted(change.path);
@@ -54789,7 +54836,7 @@ class Runner {
           case "renamed":
             if (change.oldPath) {
               this.sendFileDeleted(change.oldPath);
-              this.sendFileContent(change.path);
+              await this.sendFileContent(change.path);
             }
             break;
         }
@@ -54841,10 +54888,20 @@ class Runner {
     }
     return changes;
   }
-  sendFileContent(path4) {
+  async sendFileContent(path4) {
     try {
-      if (existsSync2(path4)) {
-        const content = readFileSync2(path4, "utf8");
+      const stat = await fs5.stat(path4).catch(() => null);
+      if (!stat) {
+        console.error(`File or directory not found: ${path4}`);
+        return;
+      }
+      if (stat.isDirectory()) {
+        const [files] = await listFiles(path4, true, 1000, process.cwd());
+        for (const file of files) {
+          await this.sendFileContent(file);
+        }
+      } else if (stat.isFile()) {
+        const content = await fs5.readFile(path4, "utf8");
         this.wsManager.sendMessage({
           type: "file",
           path: path4,
@@ -54852,15 +54909,10 @@ class Runner {
         });
         console.log(`Sent content for file: ${path4}`);
       } else {
-        console.error(`File not found: ${path4}`);
-        this.wsManager.sendMessage({
-          type: "error",
-          message: "File not found",
-          details: path4
-        });
+        console.error(`Path is not a file or directory: ${path4}`);
       }
     } catch (error) {
-      console.error(`Error reading file ${path4}:`, error);
+      console.error(`Error processing path ${path4}:`, error);
     }
   }
   sendFileDeleted(path4) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/runner",
-  "version": "0.7.24",
+  "version": "0.8.1",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",