@polka-codes/cli 0.4.5 → 0.4.7

package/dist/index.js

@@ -24629,7 +24629,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.4.5";
+var version = "0.4.7";
 
 // ../../node_modules/@anthropic-ai/sdk/version.mjs
 var VERSION = "0.36.2";
@@ -33240,12 +33240,8 @@ var capabilities = (toolNamePrefix) => `
 
 CAPABILITIES
 
-- You have access to a range of tools to aid you in your work. These tools help you effectively accomplish a wide range of tasks, such as writing code, making edits or improvements to existing files, understanding the current state of a project, performing system operations, and much more.
-- When the user initially gives you a task, a recursive list of all filepaths in the current working directory will be included in environment_details. This provides an overview of the project's file structure, offering key insights into the project from directory/file names (how developers conceptualize and organize their code) and file extensions (the language used). This can also guide decision-making on which files to explore further.
-- You can use ${toolNamePrefix}search_files to perform regex searches across files in a specified directory, outputting context-rich results that include surrounding lines. This is particularly useful for understanding code patterns, finding specific implementations, or identifying areas that need refactoring.
-- You can use the ${toolNamePrefix}list_code_definition_names tool to get an overview of source code definitions for all files at the top level of a specified directory. This can be particularly useful when you need to understand the broader context and relationships between certain parts of the code. You may need to call this tool multiple times to understand various parts of the codebase related to the task.
-\t- For example, when asked to make edits or improvements you might analyze the file structure in the initial environment_details to get an overview of the project, then use ${toolNamePrefix}list_code_definition_names to get further insight using source code definitions for files located in relevant directories, then ${toolNamePrefix}read_file to examine the contents of relevant files, analyze the code and suggest improvements or make necessary edits, then use the ${toolNamePrefix}replace_in_file tool to implement changes. If you refactored code that could affect other parts of the codebase, you could use ${toolNamePrefix}search_files to ensure you update other files as needed.
-- You can use the ${toolNamePrefix}execute_command tool to run commands on the user's computer whenever you feel it can help accomplish the user's task. When you need to execute a CLI command, you must provide a clear explanation of what the command does. Prefer to execute complex CLI commands over creating executable scripts, since they are more flexible and easier to run. Interactive and long-running commands are allowed, since the commands are run in the user's VSCode terminal. The user may keep commands running in the background and you will be kept updated on their status along the way. Each command you execute is run in a new terminal instance.`;
+- You have access to a range of tools to aid you in your work. These tools help you effectively accomplish a wide range of tasks.
+- When the user initially gives you a task, a recursive list of all filepaths in the current working directory will be included in context. This provides an overview of the project's file structure, offering key insights into the project from directory/file names (how developers conceptualize and organize their code) and file extensions (the language used). This can also guide decision-making on which files to explore further.`;
 var systemInformation = (info) => `
 ====
 
@@ -33310,7 +33306,7 @@ ${joined}`;
 };
 var responsePrompts = {
   errorInvokeTool: (tool, error) => `An error occurred while invoking the tool "${tool}": ${error}`,
-  requireUseTool: "Error: You must use a tool before proceeding",
+  requireUseTool: "Error: You must use a tool before proceeding. Making sure the tool is invoked using xml tags.",
   toolResults: (tool, result) => `<tool_response>
 <tool_name>${tool}</tool_name>
 <tool_result>
@@ -33364,6 +33360,7 @@ ${agents}`;
       text += `
 <context>${context}</context>`;
     }
+    callback({ kind: "StartTask" /* StartTask */, info: taskInfo, systemPrompt: this.config.systemPrompt });
     return await this.#processLoop(text, taskInfo, callback);
   }
   async#processLoop(userMessage, taskInfo, callback) {
@@ -33507,6 +33504,8 @@ __export(exports_allTools, {
   writeToFile: () => writeToFile_default,
   searchFiles: () => searchFiles_default,
   replaceInFile: () => replaceInFile_default,
+  renameFile: () => renameFile_default,
+  removeFile: () => removeFile_default,
   readFile: () => readFile_default,
   listFiles: () => listFiles_default,
   listCodeDefinitionNames: () => listCodeDefinitionNames_default,
@@ -33590,7 +33589,7 @@ var getStringArray = (args, name, defaultValue) => {
   if (ret === "") {
     return [];
   }
-  return ret.split(",");
+  return ret.split(",").map((s2) => s2.trim());
 };
 var getBoolean = (args, name, defaultValue) => {
   const ret = args[name];
@@ -34222,7 +34221,7 @@ var handler9 = async (provider, args) => {
   await provider.writeFile(path, content);
   return {
     type: "Reply" /* Reply */,
-    message: `<write_file_path>${path}</write_file_path><status>Success</status>`
+    message: `<write_to_file_path>${path}</write_to_file_path><status>Success</status>`
   };
 };
 var isAvailable9 = (provider) => {
@@ -34259,7 +34258,7 @@ var toolInfo10 = {
     {
       name: "files",
       description: "The files relevant to the task",
-      required: true,
+      required: false,
       usageValue: "Relevant files"
     }
   ],
@@ -34308,6 +34307,109 @@ var handOver_default = {
   handler: handler10,
   isAvailable: isAvailable10
 };
+// ../core/src/tools/removeFile.ts
+var toolInfo11 = {
+  name: "remove_file",
+  description: "Request to remove a file at the specified path.",
+  parameters: [
+    {
+      name: "path",
+      description: "The path of the file to remove",
+      required: true,
+      usageValue: "File path here"
+    }
+  ],
+  examples: [
+    {
+      description: "Request to remove a file",
+      parameters: [
+        {
+          name: "path",
+          value: "src/main.js"
+        }
+      ]
+    }
+  ]
+};
+var handler11 = async (provider, args) => {
+  if (!provider.removeFile) {
+    return {
+      type: "Error" /* Error */,
+      message: "Not possible to remove file. Abort."
+    };
+  }
+  const path = getString(args, "path");
+  await provider.removeFile(path);
+  return {
+    type: "Reply" /* Reply */,
+    message: `<remove_file_path>${path}</remove_file_path><status>Success</status>`
+  };
+};
+var isAvailable11 = (provider) => {
+  return !!provider.removeFile;
+};
+var removeFile_default = {
+  ...toolInfo11,
+  handler: handler11,
+  isAvailable: isAvailable11
+};
+// ../core/src/tools/renameFile.ts
+var toolInfo12 = {
+  name: "rename_file",
+  description: "Request to rename a file from source path to target path.",
+  parameters: [
+    {
+      name: "sourcePath",
+      description: "The current path of the file",
+      required: true,
+      usageValue: "Source file path here"
+    },
+    {
+      name: "targetPath",
+      description: "The new path for the file",
+      required: true,
+      usageValue: "Target file path here"
+    }
+  ],
+  examples: [
+    {
+      description: "Request to rename a file",
+      parameters: [
+        {
+          name: "sourcePath",
+          value: "src/old-name.js"
+        },
+        {
+          name: "targetPath",
+          value: "src/new-name.js"
+        }
+      ]
+    }
+  ]
+};
+var handler12 = async (provider, args) => {
+  if (!provider.renameFile) {
+    return {
+      type: "Error" /* Error */,
+      message: "Not possible to rename file. Abort."
+    };
+  }
+  const sourcePath = getString(args, "sourcePath");
+  const targetPath = getString(args, "targetPath");
+  await provider.renameFile(sourcePath, targetPath);
+  return {
+    type: "Reply" /* Reply */,
+    message: `<rename_file_path>${targetPath}</rename_file_path><status>Success</status>`
+  };
+};
+var isAvailable12 = (provider) => {
+  return !!provider.renameFile;
+};
+var renameFile_default = {
+  ...toolInfo12,
+  handler: handler12,
+  isAvailable: isAvailable12
+};
 // ../core/src/Agent/CoderAgent/prompts.ts
 var basePrompt = "You are a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.";
 var editingFilesPrompt = (toolNamePrefix) => `
@@ -34917,7 +35019,7 @@ import os from "node:os";
 // src/provider.ts
 var import_ignore2 = __toESM(require_ignore(), 1);
 import { spawn as spawn2 } from "node:child_process";
-import { mkdir, readFile, writeFile } from "node:fs/promises";
+import { mkdir, readFile, rename, unlink, writeFile } from "node:fs/promises";
 import { dirname } from "node:path";
 
 // src/utils/listFiles.ts
@@ -35030,7 +35132,7 @@ async function searchFiles(path2, regex, filePattern, cwd, excludeFiles) {
     rg.stderr.on("data", (data) => {
       const err = data.toString();
       if (!err.startsWith("WARNING:")) {
-        console.error(err);
+        console.warn(err);
       }
     });
     rg.on("error", (error) => {
@@ -35047,9 +35149,9 @@ async function searchFiles(path2, regex, filePattern, cwd, excludeFiles) {
 }
 
 // src/provider.ts
-var getProvider = (options) => {
+var getProvider = (agentName, config, options) => {
   const ig = import_ignore2.default().add(options.excludeFiles ?? []);
-  return {
+  const provider2 = {
     readFile: async (path2) => {
       if (ig.ignores(path2)) {
         throw new Error(`Not allow to access file ${path2}`);
@@ -35063,6 +35165,18 @@ var getProvider = (options) => {
       await mkdir(dirname(path2), { recursive: true });
       return await writeFile(path2, content, "utf8");
     },
+    removeFile: async (path2) => {
+      if (ig.ignores(path2)) {
+        throw new Error(`Not allow to access file ${path2}`);
+      }
+      return await unlink(path2);
+    },
+    renameFile: async (sourcePath, targetPath) => {
+      if (ig.ignores(sourcePath) || ig.ignores(targetPath)) {
+        throw new Error(`Not allow to access file ${sourcePath} or ${targetPath}`);
+      }
+      return await rename(sourcePath, targetPath);
+    },
     listFiles: async (path2, recursive, maxCount) => {
       return await listFiles(path2, recursive, maxCount, dirname(path2), options.excludeFiles);
     },
@@ -35101,8 +35215,31 @@ var getProvider = (options) => {
           reject(err);
         });
       });
+    },
+    attemptCompletion: async (result) => {
+      const cmd = (config.hooks?.agents?.[agentName]?.beforeCompletion ?? config.hooks?.agents?.default?.beforeCompletion)?.trim();
+      if (cmd) {
+        try {
+          const { exitCode, stdout, stderr } = await provider2.executeCommand(cmd, false);
+          if (exitCode !== 0) {
+            return `
+<command>${cmd}</command>
+<command_exit_code>${exitCode}</command_exit_code>
+<command_stdout>
+${stdout}
+</command_stdout>
+<command_stderr>
+${stderr}
+</command_stderr>`;
+          }
+        } catch (error) {
+          console.warn(`Failed to execute hook: ${error}`);
+        }
+      }
+      return;
     }
   };
+  return provider2;
 };
 
 // src/Runner.ts
@@ -35126,7 +35263,7 @@ class Runner {
     if (typeof rules2 === "string") {
       rules2 = [rules2];
     }
-    const provider2 = getProvider({
+    const providerOptions = {
       command: {
         onStarted(command) {
           console.log(`$ >>>> $ ${command}`);
@@ -35145,19 +35282,20 @@ class Runner {
         }
       },
       excludeFiles: options.config.excludeFiles
-    });
+    };
     const platform = os.platform();
     const agents = [coderAgentInfo, architectAgentInfo];
     this.#multiAgent = new MultiAgent({
       createAgent: async (name) => {
-        switch (name.trim().toLowerCase()) {
+        const agentName = name.trim().toLowerCase();
+        switch (agentName) {
           case coderAgentInfo.name.toLowerCase():
             return new CoderAgent({
               ai: service,
               os: platform,
               customInstructions: rules2,
               scripts: options.config.scripts,
-              provider: provider2,
+              provider: getProvider("coder", options.config, providerOptions),
               interactive: options.interactive,
               agents
             });
@@ -35167,7 +35305,7 @@ class Runner {
               os: platform,
               customInstructions: rules2,
               scripts: options.config.scripts,
-              provider: provider2,
+              provider: getProvider("architect", options.config, providerOptions),
               interactive: options.interactive,
               agents
             });
@@ -39306,6 +39444,8 @@ var z2 = /* @__PURE__ */ Object.freeze({
 });
 
 // src/config.ts
+var agentNames = z2.enum(["coder", "architect"]);
+var agentNameOrDefault = z2.union([agentNames, z2.literal("default")]);
 var providerModelSchema = z2.object({
   provider: z2.string().optional(),
   model: z2.string().optional()
@@ -39324,13 +39464,14 @@ var configSchema = z2.object({
   defaultProvider: z2.string().optional(),
   defaultModel: z2.string().optional(),
   maxIterations: z2.number().int().positive().optional(),
+  hooks: z2.object({
+    agents: z2.record(agentNameOrDefault, z2.object({ beforeCompletion: z2.string().optional() })).optional()
+  }).optional(),
   scripts: z2.record(z2.string(), z2.string().or(z2.object({
     command: z2.string(),
     description: z2.string()
   }))).optional(),
-  agents: z2.object({
-    default: agentSchema.optional()
-  }).catchall(agentSchema).optional(),
+  agents: z2.record(agentNameOrDefault, agentSchema).optional(),
   commands: z2.object({
     default: providerModelSchema.optional()
   }).catchall(providerModelSchema).optional(),
@@ -39416,7 +39557,7 @@ var readConfig = (path2) => {
 
 // src/options.ts
 function addSharedOptions(command) {
-  return command.option("-c --config <path>", "Path to config file").option("--api-provider <provider>", "API provider").option("--model <model>", "Model ID").option("--api-key <key>", "API key").option("--max-iterations <iterations>", "Maximum number of iterations to run. Default to 30", Number.parseInt);
+  return command.option("-c --config <path>", "Path to config file").option("--api-provider <provider>", "API provider").option("--model <model>", "Model ID").option("--api-key <key>", "API key").option("--max-iterations <iterations>", "Maximum number of iterations to run. Default to 30", Number.parseInt).option("-v --verbose", "Enable verbose output. Use -v for level 1, -vv for level 2", (value, prev) => prev + 1, 0);
 }
 
 class ApiProviderConfig {
@@ -39485,6 +39626,7 @@ function parseOptions(options, cwd = process.cwd(), home = os2.homedir()) {
   });
   return {
     maxIterations: options.maxIterations ?? 30,
+    verbose: options.verbose ?? 0,
     config,
     providerConfig
   };
@@ -39975,13 +40117,32 @@ var chalkStderr = createChalk({ level: stderrColor ? stderrColor.level : 0 });
 var source_default = chalk;
 
 // src/utils/eventHandler.ts
-var printEvent = (event) => {
+var printEvent = (verbose) => (event) => {
+  let hadReasoning = false;
   switch (event.kind) {
+    case "StartTask" /* StartTask */:
+      if (verbose > 1) {
+        console.log(`
+====== System Prompt ======
+${event.systemPrompt}`);
+        console.log(`
+
+================
+`);
+      }
+      break;
     case "StartRequest" /* StartRequest */:
       console.log(`
 
 ======== New Request ========
 `);
+      if (verbose) {
+        console.log(event.userMessage);
+        console.log(`
+
+======== Request Message Ended ========
+`);
+      }
       break;
     case "EndRequest" /* EndRequest */:
       console.log(`
@@ -39992,10 +40153,17 @@ var printEvent = (event) => {
     case "Usage" /* Usage */:
       break;
     case "Text" /* Text */:
+      if (hadReasoning) {
+        process.stdout.write(`
+
+`);
+        hadReasoning = false;
+      }
       process.stdout.write(event.newText);
       break;
     case "Reasoning" /* Reasoning */:
       process.stdout.write(source_default.dim(event.newText));
+      hadReasoning = true;
       break;
     case "ToolUse" /* ToolUse */:
       break;
@@ -40591,15 +40759,15 @@ function useKeypress(userHandler) {
   signal.current = userHandler;
   useEffect((rl) => {
     let ignore3 = false;
-    const handler11 = withUpdates((_input, event) => {
+    const handler13 = withUpdates((_input, event) => {
       if (ignore3)
         return;
       signal.current(event, rl);
     });
-    rl.input.on("keypress", handler11);
+    rl.input.on("keypress", handler13);
     return () => {
       ignore3 = true;
-      rl.input.removeListener("keypress", handler11);
+      rl.input.removeListener("keypress", handler13);
     };
   }, []);
 }
@@ -40781,16 +40949,16 @@ class Emitter {
 
 class SignalExitBase {
 }
-var signalExitWrap = (handler11) => {
+var signalExitWrap = (handler13) => {
   return {
     onExit(cb, opts) {
-      return handler11.onExit(cb, opts);
+      return handler13.onExit(cb, opts);
     },
     load() {
-      return handler11.load();
+      return handler13.load();
     },
     unload() {
-      return handler11.unload();
+      return handler13.unload();
     }
   };
 };
@@ -41470,7 +41638,7 @@ async function printConfig(configPath) {
   }
   console.log($stringify(config));
 }
-async function configCommand(options) {
+var configCommand = new Command("config").description("Configure global or local settings").option("-g, --global", "Use global config").option("-p, --print", "Print config").action(async (options) => {
   const globalConfigPath = getGlobalConfigPath();
   let configPath = options.global ? globalConfigPath : localConfigFileName;
   if (options.print) {
@@ -41557,12 +41725,11 @@ ${provider2.toUpperCase()}_API_KEY=${apiKey}`;
   mkdirSync(dirname2(configPath), { recursive: true });
   writeFileSync(configPath, $stringify(newConfig));
   console.log(`Config file saved at: ${configPath}`);
-  return;
-}
+});
 
 // src/commands/chat.ts
 var runChat = async (options) => {
-  const { config, providerConfig, maxIterations } = parseOptions(options);
+  const { config, providerConfig, maxIterations, verbose } = parseOptions(options);
   let { provider: provider2, model, apiKey } = providerConfig.getConfigForAgent("coder") ?? {};
   if (!provider2) {
     const newConfig = await configPrompt({ provider: provider2, model, apiKey });
@@ -41582,7 +41749,7 @@ var runChat = async (options) => {
     config: config ?? {},
     maxIterations,
     interactive: true,
-    eventCallback: printEvent
+    eventCallback: printEvent(verbose)
   });
   let taskInfo;
   const chat2 = new Chat3({
@@ -42351,12 +42518,57 @@ var prCommand = new Command("pr").description("Create a GitHub pull request").ar
 });
 
 // src/commands/task.ts
+var readStdin = async (timeoutMs = 30000) => {
+  if (process.stdin.isTTY) {
+    return "";
+  }
+  return new Promise((resolve2, reject) => {
+    let input = "";
+    let timeoutId = undefined;
+    const cleanup = () => {
+      if (timeoutId)
+        clearTimeout(timeoutId);
+      process.stdin.removeAllListeners();
+      process.stdin.resume();
+    };
+    timeoutId = setTimeout(() => {
+      cleanup();
+      reject(new Error("Stdin read timeout"));
+    }, timeoutMs);
+    process.stdin.on("data", (chunk) => {
+      input += chunk.toString();
+    });
+    process.stdin.on("end", () => {
+      cleanup();
+      if (!input) {
+        reject(new Error("Empty stdin input"));
+        return;
+      }
+      resolve2(input);
+    });
+    process.stdin.on("error", (err) => {
+      cleanup();
+      reject(err);
+    });
+  });
+};
 var runTask = async (taskArg, options) => {
-  if (!taskArg) {
-    runChat(options);
-    return;
+  let task = taskArg;
+  if (!task) {
+    try {
+      const stdinInput = await readStdin();
+      if (stdinInput) {
+        task = stdinInput;
+      } else {
+        runChat(options);
+        return;
+      }
+    } catch (error) {
+      console.error("Error reading stdin:", error);
+      process.exit(1);
+    }
   }
-  const { providerConfig, config, maxIterations } = parseOptions(options);
+  const { providerConfig, config, maxIterations, verbose } = parseOptions(options);
   let { provider: provider2, model, apiKey } = providerConfig.getConfigForAgent("coder") ?? {};
   if (!provider2) {
     const newConfig = await configPrompt({ provider: provider2, model, apiKey });
@@ -42373,18 +42585,18 @@ var runTask = async (taskArg, options) => {
     config: config ?? {},
     maxIterations,
     interactive: false,
-    eventCallback: printEvent
+    eventCallback: printEvent(verbose)
   });
-  await runner.startTask(taskArg);
+  await runner.startTask(task);
   runner.printUsage();
 };
 
 // src/index.ts
 var program2 = new Command;
-program2.name("polka").description("Polka Codes CLI").version(version);
-program2.command("chat").description("Start an interactive chat session").action(runChat);
-program2.command("config").description("Configure global or local settings").option("-g, --global", "Use global config").option("-p, --print", "Print config").action(configCommand);
+program2.name("polka-codes").description("Polka Codes CLI").version(version);
 program2.argument("[task]", "The task to execute").action(runTask);
+program2.command("chat").description("Start an interactive chat session").action(runChat);
+program2.addCommand(configCommand);
 program2.addCommand(commitCommand);
 program2.addCommand(prCommand);
 addSharedOptions(program2);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/cli",
-  "version": "0.4.5",
+  "version": "0.4.7",
   "license": "AGPL-3.0",
   "type": "module",
   "bin": {
@@ -17,7 +17,7 @@
   },
   "dependencies": {
     "@inquirer/prompts": "^7.2.3",
-    "@polka-codes/core": "0.4.3",
+    "@polka-codes/core": "0.4.5",
     "@vscode/ripgrep": "^1.15.10",
     "chalk": "^5.4.1",
     "commander": "^13.0.0",