@polka-codes/cli 0.9.43 → 0.9.44

package/dist/index.js

@@ -57704,7 +57704,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.9.43";
+var version = "0.9.44";
 
 // src/commands/chat.ts
 import { readFile as readFile3 } from "node:fs/promises";
@@ -104051,8 +104051,9 @@ var runChat = async (opts, command) => {
 
 // src/runWorkflow.ts
 import { spawnSync as spawnSync2 } from "node:child_process";
-import fs2 from "node:fs/promises";
+import fs2, { mkdir as mkdir2 } from "node:fs/promises";
 import os4 from "node:os";
+import { dirname as dirname2 } from "node:path";
 
 // ../workflow/src/helpers.ts
 var makeStepFn = (fns) => {
@@ -107314,11 +107315,29 @@ Unstaged files:`);
         throw new UserCancelledError;
       }
     }
+    case "select": {
+      const { message, choices } = toolCall.input;
+      context.spinner.stop();
+      await new Promise((resolve4) => setTimeout(resolve4, 50));
+      try {
+        const result = await esm_default5({ message, choices });
+        context.spinner.start();
+        return result;
+      } catch (_e) {
+        throw new UserCancelledError;
+      }
+    }
     case "writeToFile": {
       const { path, content } = toolCall.input;
+      await mkdir2(dirname2(path), { recursive: true });
       await fs2.writeFile(path, content);
       return {};
     }
+    case "executeCommand": {
+      const { command } = toolCall.input;
+      const result = spawnSync2(command, { shell: true, stdio: "pipe", encoding: "utf-8" });
+      return { exitCode: result.status ?? -1, stdout: result.stdout, stderr: result.stderr };
+    }
     default:
       throw new Error(`Unknown tool: ${String(toolCall.tool)}`);
   }
@@ -107495,7 +107514,7 @@ ${commitMessage}`);
 };
 // src/workflows/init.workflow.ts
 import { existsSync as existsSync3, mkdirSync, readFileSync as readFileSync2, writeFileSync } from "node:fs";
-import { dirname as dirname2 } from "node:path";
+import { dirname as dirname3 } from "node:path";
 var import_lodash7 = __toESM(require_lodash(), 1);
 
 // ../../node_modules/yaml/dist/index.js
@@ -107665,7 +107684,7 @@ var initWorkflow = {
             const globalConfigPath = getGlobalConfigPath();
             const globalConfig2 = loadConfigAtPath(globalConfigPath) ?? {};
             import_lodash7.set(globalConfig2, ["providers", provider3, "apiKey"], providerConfig.apiKey);
-            mkdirSync(dirname2(globalConfigPath), { recursive: true });
+            mkdirSync(dirname3(globalConfigPath), { recursive: true });
             writeFileSync(globalConfigPath, $stringify2(globalConfig2));
             console.log(`API key saved to global config file: ${globalConfigPath}`);
             providerConfig.apiKey = undefined;
@@ -107718,7 +107737,7 @@ ${provider3.toUpperCase()}_API_KEY=${providerConfig.apiKey}`;
           import_lodash7.set(finalConfig, ["providers", providerConfig.provider, "apiKey"], providerConfig.apiKey);
         }
       }
-      mkdirSync(dirname2(configPath), { recursive: true });
+      mkdirSync(dirname3(configPath), { recursive: true });
       writeFileSync(configPath, $stringify2(finalConfig));
       console.log(`Configuration saved to ${configPath}`);
       return {};
@@ -107726,12 +107745,7 @@ ${provider3.toUpperCase()}_API_KEY=${providerConfig.apiKey}`;
     return { configPath };
   }
 };
-// src/workflows/plan.workflow.ts
-var PlanSchema = exports_external.object({
-  plan: exports_external.string().optional(),
-  ready: exports_external.boolean(),
-  question: exports_external.string().optional()
-});
+// src/workflows/plan.prompt.ts
 var PLAN_PROMPT = `
 You are an expert planner. Your goal is to create a detailed plan for a given task.
 
@@ -107757,61 +107771,100 @@ Respond with a JSON object that matches the following schema:
   "question": "The clarifying question to ask the user."
 }
 `;
+
+// src/workflows/plan.workflow.ts
+var PlanSchema = exports_external.object({
+  plan: exports_external.string().optional(),
+  ready: exports_external.boolean(),
+  question: exports_external.string().optional()
+});
 var planWorkflow = {
   name: "Plan Task",
   description: "Create or update a plan for a given task.",
   async* fn(input, _step, tools3) {
     const { task, fileContent, filePath } = input;
     let plan = fileContent || "";
-    let isPlanReady = false;
     let userFeedback = "";
-    while (true) {
-      const currentTask = userFeedback ? `${task}
+    let state = "Generating";
+    while (state !== "Done") {
+      switch (state) {
+        case "Generating": {
+          const currentTask = userFeedback ? `${task}
 
 User feedback: ${userFeedback}` : task;
-      const prompt3 = PLAN_PROMPT.replace("{task}", currentTask).replace("{fileContent}", plan);
-      const {
-        plan: newPlan,
-        ready,
-        question
-      } = yield* tools3.invokeAgent({
-        agent: "architect",
-        messages: [{ type: "user", content: prompt3 }],
-        outputSchema: PlanSchema
-      });
-      if (newPlan !== undefined) {
-        plan = newPlan;
-      }
-      isPlanReady = ready;
-      if (!isPlanReady && question) {
-        const answer = yield* tools3.input({ message: question });
-        userFeedback = `Question: ${question}
+          const prompt3 = PLAN_PROMPT.replace("{task}", currentTask).replace("{fileContent}", plan);
+          const {
+            plan: newPlan,
+            ready,
+            question
+          } = yield* tools3.invokeAgent({
+            agent: "architect",
+            messages: [{ type: "user", content: prompt3 }],
+            outputSchema: PlanSchema
+          });
+          if (newPlan !== undefined) {
+            plan = newPlan;
+          }
+          if (!ready && question) {
+            const answer = yield* tools3.input({ message: question });
+            userFeedback = `Question: ${question}
 Answer: ${answer}`;
-        continue;
-      }
-      console.log(`
+            state = "Generating";
+          } else {
+            userFeedback = "";
+            state = "Reviewing";
+          }
+          break;
+        }
+        case "Reviewing": {
+          console.log(`
 Generated Plan:
 `);
-      console.log(plan);
-      try {
-        userFeedback = yield* tools3.input({
-          message: "What changes do you want to make? (leave empty or press Ctrl+C to exit)"
-        });
-      } catch (_error) {
-        userFeedback = "";
-      }
-      if (!userFeedback) {
-        break;
-      }
-    }
-    const savePlan = yield* tools3.confirm({ message: "Do you want to save the plan?", default: false });
-    if (savePlan) {
-      let savePath = filePath;
-      if (!savePath) {
-        savePath = yield* tools3.input({ message: "Where do you want to save the plan?", default: "docs/plan.md" });
+          console.log(plan);
+          const choice = yield* tools3.select({
+            message: "What do you want to do?",
+            choices: [
+              { name: "Accept and Save", value: "1" },
+              { name: "Provide Feedback", value: "2" },
+              { name: "Regenerate Plan", value: "3" },
+              { name: "Exit", value: "4" }
+            ]
+          });
+          switch (choice) {
+            case "1": {
+              const defaultPath = `.plans/plan-${new Date().toISOString().replace(/:/g, "-")}.md`;
+              const savePath = filePath || (yield* tools3.input({ message: "Where do you want to save the plan?", default: defaultPath }));
+              yield* tools3.writeToFile({ path: savePath, content: plan });
+              console.log(`Plan saved to ${savePath}`);
+              state = "Done";
+              break;
+            }
+            case "2": {
+              try {
+                userFeedback = yield* tools3.input({
+                  message: "What changes do you want to make?"
+                });
+                state = "Generating";
+              } catch (_error) {
+                userFeedback = "";
+                state = "Reviewing";
+              }
+              break;
+            }
+            case "3": {
+              plan = "";
+              userFeedback = "";
+              state = "Generating";
+              break;
+            }
+            case "4": {
+              state = "Done";
+              break;
+            }
+          }
+          break;
+        }
       }
-      yield* tools3.writeToFile({ path: savePath, content: plan });
-      console.log(`Plan saved to ${savePath}`);
     }
     return {};
   }
@@ -108285,7 +108338,7 @@ var commitCommand = new Command("commit").description("Create a commit with AI-g
 
 // src/commands/create.ts
 import { existsSync as existsSync4 } from "node:fs";
-import { mkdir as mkdir2, stat } from "node:fs/promises";
+import { mkdir as mkdir3, stat } from "node:fs/promises";
 import { join as join3 } from "node:path";
 var askForPath = async (projectName) => {
   let targetPath = join3(process.cwd(), projectName);
@@ -108334,7 +108387,7 @@ var createCommand2 = new Command("create").description("Create a new project").a
   }
   const targetPath = await askForPath(projectName);
   try {
-    await mkdir2(targetPath, { recursive: true });
+    await mkdir3(targetPath, { recursive: true });
   } catch (error47) {
     console.error(`Failed to create directory: ${targetPath}`, error47);
     process.exit(1);
@@ -108352,6 +108405,84 @@ var createCommand2 = new Command("create").description("Create a new project").a
   await createNewProject(runner.multiAgent, projectName);
 });
 
+// src/workflows/fix.workflow.ts
+var fixWorkflow = {
+  name: "Fix",
+  description: "Fix an issue by running a command repeatedly.",
+  async* fn(input, _step, tools3) {
+    let command = input.command;
+    if (!command) {
+      const config4 = loadConfig();
+      const check2 = config4?.scripts?.check;
+      const test = config4?.scripts?.test;
+      let checkCommand;
+      if (typeof check2 === "string") {
+        checkCommand = check2;
+      } else if (check2) {
+        checkCommand = check2.command;
+      }
+      let testCommand;
+      if (typeof test === "string") {
+        testCommand = test;
+      } else if (test) {
+        testCommand = test.command;
+      }
+      let defaultCommand;
+      if (checkCommand && testCommand) {
+        defaultCommand = `${checkCommand} && ${testCommand}`;
+      } else if (checkCommand) {
+        defaultCommand = checkCommand;
+      } else if (testCommand) {
+        defaultCommand = testCommand;
+      }
+      command = yield* tools3.input({
+        message: "Please enter the command to run to identify issues:",
+        default: defaultCommand
+      });
+      if (!command) {
+        throw new Error("No command provided. Aborting.");
+      }
+    }
+    for (let i = 0;i < 10; i++) {
+      console.log(`Running command: ${command}`);
+      const { exitCode, stdout, stderr } = yield* tools3.executeCommand({ command });
+      if (stdout) {
+        console.log(`Command stdout:
+`, stdout);
+      }
+      if (stderr) {
+        console.error(`Command stderr:
+`, stderr);
+      }
+      if (exitCode === 0) {
+        console.log("Command succeeded!");
+        return { success: true };
+      }
+      console.log(`Command failed with exit code ${exitCode}. Asking agent to fix it...`);
+      const prompt3 = `The command "${command}" failed with exit code ${exitCode}.
+Please fix the code.
+
+stdout:
+${stdout}
+
+stderr:
+${stderr}
+`;
+      yield* tools3.invokeAgent({
+        agent: "coder",
+        messages: [{ type: "user", content: prompt3 }]
+      });
+    }
+    throw new Error("Failed to fix the issue after 10 attempts.");
+  }
+};
+
+// src/commands/fix.ts
+var fixCommand = new Command("fix").description("Fix issues by running a command and letting an agent fix it.").argument("[command]", "The command to run").action(async (command, _options, cmd) => {
+  const workflowInput = { command };
+  await runWorkflow("fix", fixWorkflow, cmd, workflowInput);
+});
+
 // src/commands/init.ts
 var initCommand = new Command("init").description("Initialize polkacodes configuration").option("-g, --global", "Use global config").action(async (options, command) => {
   await runWorkflow("init", initWorkflow, command, {
@@ -108587,6 +108718,7 @@ program2.addCommand(prCommand);
 program2.addCommand(reviewCommand);
 program2.addCommand(createCommand2);
 program2.addCommand(planCommand);
+program2.addCommand(fixCommand);
 addSharedOptions(program2);
 program2.parse();
 process.on("uncaughtException", (error47) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/cli",
-  "version": "0.9.43",
+  "version": "0.9.44",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",