pybao-cli 1.3.52 → 1.3.54

package/dist/{chunk-LR5UHV7J.js → chunk-EX4CINNF.js}

@@ -21,6 +21,7 @@ import {
   WebSearchTool,
   applyMarkdown,
   countTokens,
+  fileExistsBun,
   getAbsolutePath,
   getAgentPrompt,
   getMaxThinkingTokens,
@@ -29,45 +30,45 @@ import {
   hasWritePermission,
   loadLanguage,
   query
-} from "./chunk-74D2ZU54.js";
+} from "./chunk-QZOP7ZRC.js";
 import {
   FallbackToolUseRejectedMessage,
   MCPTool,
   getClients,
   getMCPTools
-} from "./chunk-D2ACSVOB.js";
+} from "./chunk-JYGKQP6W.js";
 import {
   queryLLM
-} from "./chunk-WADIOYER.js";
+} from "./chunk-QUFE2RKU.js";
 import {
   tree
-} from "./chunk-KCGULRGJ.js";
+} from "./chunk-RX7KHSRR.js";
 import {
   generateAgentId
-} from "./chunk-GQVZF5YT.js";
+} from "./chunk-CXAAFWX6.js";
 import {
   getActiveAgents,
   getAgentByType,
   getAvailableAgentTypes
-} from "./chunk-2T5RFGJ2.js";
+} from "./chunk-MMT6FJI2.js";
 import {
   INTERRUPT_MESSAGE,
   createAssistantMessage,
   createUserMessage,
   getLastAssistantMessageId
-} from "./chunk-ZFTWJ7YB.js";
+} from "./chunk-ZNTAAH36.js";
 import {
   getModelManager
-} from "./chunk-ATJCQ3LJ.js";
+} from "./chunk-6FELR65K.js";
 import {
   getContext
-} from "./chunk-HZY4TS6I.js";
+} from "./chunk-AGRH2PYY.js";
 import {
   getTheme
-} from "./chunk-VNKJNO7F.js";
+} from "./chunk-QFFNOAQC.js";
 import {
   debug
-} from "./chunk-45CWC3JW.js";
+} from "./chunk-AD5SXLOX.js";
 import {
   BunShell,
   getCwd,
@@ -76,7 +77,7 @@ import {
   logError,
   overwriteLog,
   readTaskOutput
-} from "./chunk-UH2HA3JE.js";
+} from "./chunk-VC2SEKFU.js";
 import {
   formatDuration,
   formatNumber
@@ -3104,15 +3105,255 @@ var LspTool = {
   }
 };
 
+// src/tools/filesystem/PatchTool/PatchTool.tsx
+import { z as z7 } from "zod";
+
+// src/utils/diff/patch-manager.ts
+var PatchManager = class {
+  static parseMultiFileDiff(diff) {
+    const lines = diff.split("\n");
+    const patches = [];
+    let currentPatch = null;
+    let currentHunk = null;
+    for (const line of lines) {
+      if (line.startsWith("---")) {
+        if (currentHunk && currentPatch) {
+          currentPatch.hunks.push(currentHunk);
+          currentHunk = null;
+        }
+        const pathPart = line.substring(3).trim();
+        continue;
+      }
+      if (line.startsWith("+++")) {
+        const pathPart = line.substring(3).trim();
+        let filePath = pathPart;
+        if (filePath.startsWith("b/")) {
+          filePath = filePath.substring(2);
+        } else if (filePath.startsWith("a/")) {
+          filePath = filePath.substring(2);
+        }
+        currentPatch = {
+          filePath,
+          hunks: []
+        };
+        patches.push(currentPatch);
+        continue;
+      }
+      if (line.startsWith("@@")) {
+        if (!currentPatch) {
+          continue;
+        }
+        if (currentHunk) {
+          currentPatch.hunks.push(currentHunk);
+        }
+        currentHunk = {
+          header: line,
+          oldLines: [],
+          newLines: []
+        };
+        continue;
+      }
+      if (currentHunk) {
+        if (line.startsWith(" ")) {
+          currentHunk.oldLines.push(line.slice(1));
+          currentHunk.newLines.push(line.slice(1));
+        } else if (line.startsWith("-")) {
+          currentHunk.oldLines.push(line.slice(1));
+        } else if (line.startsWith("+")) {
+          currentHunk.newLines.push(line.slice(1));
+        }
+      }
+    }
+    if (currentHunk && currentPatch) {
+      currentPatch.hunks.push(currentHunk);
+    }
+    return patches;
+  }
+  static parse(diff) {
+    const patches = this.parseMultiFileDiff(diff);
+    if (patches.length === 0) {
+      return this.parseRawHunks(diff);
+    }
+    return patches.flatMap((p) => p.hunks);
+  }
+  static parseRawHunks(diff) {
+    const lines = diff.split("\n");
+    const hunks = [];
+    let currentHunk = null;
+    for (const line of lines) {
+      if (line.startsWith("---") || line.startsWith("+++")) continue;
+      if (line.startsWith("@@")) {
+        if (currentHunk) hunks.push(currentHunk);
+        currentHunk = { header: line, oldLines: [], newLines: [] };
+        continue;
+      }
+      if (currentHunk) {
+        if (line.startsWith(" ")) {
+          currentHunk.oldLines.push(line.slice(1));
+          currentHunk.newLines.push(line.slice(1));
+        } else if (line.startsWith("-")) {
+          currentHunk.oldLines.push(line.slice(1));
+        } else if (line.startsWith("+")) {
+          currentHunk.newLines.push(line.slice(1));
+        }
+      }
+    }
+    if (currentHunk) hunks.push(currentHunk);
+    return hunks;
+  }
+  static apply(content, hunks) {
+    const lines = content.split("\n");
+    const replacements = [];
+    let searchStartIndex = 0;
+    for (const hunk of hunks) {
+      const matchIndex = this.findContext(lines, hunk.oldLines, searchStartIndex);
+      if (matchIndex === -1) {
+        throw new Error(`Could not find context for hunk: ${hunk.header}`);
+      }
+      replacements.push({
+        start: matchIndex,
+        length: hunk.oldLines.length,
+        newLines: hunk.newLines
+      });
+      searchStartIndex = matchIndex + hunk.oldLines.length;
+    }
+    replacements.sort((a, b) => b.start - a.start);
+    const resultLines = [...lines];
+    for (const rep of replacements) {
+      resultLines.splice(rep.start, rep.length, ...rep.newLines);
+    }
+    return resultLines.join("\n");
+  }
+  static findContext(lines, context, startIndex) {
+    if (context.length === 0) return startIndex;
+    for (let i = startIndex; i <= lines.length - context.length; i++) {
+      let match = true;
+      for (let j = 0; j < context.length; j++) {
+        if (lines[i + j] !== context[j]) {
+          match = false;
+          break;
+        }
+      }
+      if (match) {
+        return i;
+      }
+    }
+    return -1;
+  }
+};
+
+// src/tools/filesystem/PatchTool/PatchTool.tsx
+import { readFileSync as readFileSync4, writeFileSync as writeFileSync2 } from "fs";
+
+// src/tools/filesystem/PatchTool/prompt.ts
+var DESCRIPTION5 = `Apply a unified diff patch to one or more files.
+
+Usage:
+- Provide a standard unified diff string (starting with \`--- a/file\` and \`+++ b/file\`).
+- The tool uses "Smart Patch" logic: it ignores strict line numbers in the diff header (e.g., \`@@ -10,5 +10,5 @@\`) and instead uses the context lines (lines starting with space) to locate the code block.
+- This allows you to apply patches even if the file has changed slightly (e.g., lines shifted), as long as the context matches.
+- Supports applying changes to MULTIPLE files in a single atomic operation. If any file fails to patch, NO files will be changed.
+
+When to use:
+- Complex refactoring involving multiple files.
+- Applying changes where you are confident about the context but unsure about exact line numbers.
+- When you want to ensure atomicity across multiple file edits.
+
+Example:
+\`\`\`diff
+--- src/utils.ts
++++ src/utils.ts
+@@ -10,3 +10,3 @@
+ function add(a, b) {
+-  return a + b;
++  return a + b + 0;
+ }
+\`\`\`
+`;
+
+// src/tools/filesystem/PatchTool/PatchTool.tsx
+var inputSchema7 = z7.strictObject({
+  patch: z7.string().describe("The unified diff to apply. Must include file headers (---/+++)")
+});
+var PatchTool = {
+  name: "Patch",
+  async description() {
+    return "Apply a unified diff patch to one or more files";
+  },
+  async prompt() {
+    return DESCRIPTION5;
+  },
+  inputSchema: inputSchema7,
+  userFacingName() {
+    return "Patch";
+  },
+  async isEnabled() {
+    return true;
+  },
+  isReadOnly() {
+    return false;
+  },
+  isConcurrencySafe() {
+    return false;
+  },
+  needsPermissions() {
+    return true;
+  },
+  renderToolUseMessage(input, { verbose }) {
+    return `Applying patch...`;
+  },
+  renderToolResultMessage(result) {
+    return `Patch applied successfully to ${result.metadata?.files_changed?.length ?? 0} files.`;
+  },
+  async validateInput({ patch }) {
+    if (!patch || patch.trim().length === 0) {
+      return { result: false, message: "Patch content is required" };
+    }
+    return { result: true };
+  },
+  async execute({ patch }, context) {
+    const hunks = PatchManager.parse(patch);
+    if (hunks.length === 0) {
+      throw new Error("No valid hunks found in patch");
+    }
+    const patches = PatchManager.parseMultiFileDiff(patch);
+    const plan = [];
+    for (const p of patches) {
+      const filePath = p.filePath.trim();
+      if (!fileExistsBun(filePath)) {
+        throw new Error(`File not found: ${filePath}`);
+      }
+      const content = readFileSync4(filePath, "utf-8");
+      try {
+        const newContent = PatchManager.apply(content, p.hunks);
+        plan.push({ filePath, newContent });
+      } catch (e) {
+        throw new Error(`Failed to apply patch to ${filePath}: ${e.message}`);
+      }
+    }
+    const changedFiles = [];
+    for (const item of plan) {
+      writeFileSync2(item.filePath, item.newContent);
+      changedFiles.push(item.filePath);
+    }
+    return {
+      output: `Successfully applied patch to ${changedFiles.length} files.`,
+      metadata: {
+        files_changed: changedFiles
+      }
+    };
+  }
+};
+
 // src/tools/mcp/ReadMcpResourceTool/ReadMcpResourceTool.tsx
 import { Box as Box7, Text as Text7 } from "ink";
 import React7 from "react";
-import { z as z7 } from "zod";
+import { z as z8 } from "zod";
 import { ReadResourceResultSchema } from "@modelcontextprotocol/sdk/types.js";
 
 // src/tools/mcp/ReadMcpResourceTool/prompt.ts
 var TOOL_NAME2 = "ReadMcpResourceTool";
-var DESCRIPTION5 = `Reads a specific resource from an MCP server.
+var DESCRIPTION6 = `Reads a specific resource from an MCP server.
 - server: The name of the MCP server to read from
 - uri: The URI of the resource to read
 
@@ -3125,19 +3366,19 @@ Parameters:
 - uri (required): The URI of the resource to read`;
 
 // src/tools/mcp/ReadMcpResourceTool/ReadMcpResourceTool.tsx
-var inputSchema7 = z7.strictObject({
-  server: z7.string().describe("The MCP server name"),
-  uri: z7.string().describe("The resource URI to read")
+var inputSchema8 = z8.strictObject({
+  server: z8.string().describe("The MCP server name"),
+  uri: z8.string().describe("The resource URI to read")
 });
 var ReadMcpResourceTool = {
   name: TOOL_NAME2,
   async description() {
-    return DESCRIPTION5;
+    return DESCRIPTION6;
   },
   async prompt() {
     return PROMPT5;
   },
-  inputSchema: inputSchema7,
+  inputSchema: inputSchema8,
   userFacingName() {
     return "readMcpResource";
   },
@@ -3239,9 +3480,9 @@ var ReadMcpResourceTool = {
 import { last, memoize } from "lodash-es";
 import React8 from "react";
 import { Box as Box8, Text as Text8 } from "ink";
-import { z as z8 } from "zod";
+import { z as z9 } from "zod";
 import { randomUUID as randomUUID2 } from "crypto";
-import { existsSync as existsSync6, readFileSync as readFileSync4 } from "fs";
+import { existsSync as existsSync6, readFileSync as readFileSync5 } from "fs";
 
 // src/utils/agent/transcripts.ts
 var transcripts = /* @__PURE__ */ new Map();
@@ -3337,17 +3578,17 @@ assistant: "I'm going to use the Task tool to launch the with the greeting-respo
 var TOOL_NAME3 = "Task";
 
 // src/tools/agent/TaskTool/TaskTool.tsx
-var inputSchema8 = z8.object({
-  description: z8.string().describe("A short (3-5 word) description of the task"),
-  prompt: z8.string().describe("The task for the agent to perform"),
-  subagent_type: z8.string().describe("The type of specialized agent to use for this task"),
-  model: z8.enum(["sonnet", "opus", "haiku"]).optional().describe(
+var inputSchema9 = z9.object({
+  description: z9.string().describe("A short (3-5 word) description of the task"),
+  prompt: z9.string().describe("The task for the agent to perform"),
+  subagent_type: z9.string().describe("The type of specialized agent to use for this task"),
+  model: z9.enum(["sonnet", "opus", "haiku"]).optional().describe(
     "Optional model to use for this agent. If not specified, inherits from parent. Prefer haiku for quick, straightforward tasks to minimize cost and latency."
   ),
-  resume: z8.string().optional().describe(
+  resume: z9.string().optional().describe(
     "Optional agent ID to resume from. If provided, the agent will continue from the previous execution transcript."
   ),
-  run_in_background: z8.boolean().optional().describe(
+  run_in_background: z9.boolean().optional().describe(
     "Set to true to run this agent in the background. Use TaskOutput to read the output later."
   )
 });
@@ -3417,7 +3658,7 @@ function applyAgentPermissionMode(base, options) {
 function readJsonArrayFile(path) {
   if (!existsSync6(path)) return null;
   try {
-    const raw = readFileSync4(path, "utf8");
+    const raw = readFileSync5(path, "utf8");
     const parsed = JSON.parse(raw);
     return Array.isArray(parsed) ? parsed : null;
   } catch {
@@ -3498,7 +3739,7 @@ function buildForkContextForAgent(options) {
 }
 var TaskTool = {
   name: TOOL_NAME3,
-  inputSchema: inputSchema8,
+  inputSchema: inputSchema9,
   async description() {
     return "Launch a new task";
   },
@@ -3948,6 +4189,7 @@ var getAllTools = () => [
   FileWriteTool,
   DeleteTool,
   NotebookEditTool,
+  PatchTool,
   TodoWriteTool,
   WebSearchTool,
   WebFetchTool,