more tools

packages/opencode/src/tool/write.ts

@@ -0,0 +1,67 @@
+import { z } from "zod"
+import * as path from "path"
+import { Tool } from "./tool"
+import { FileTimes } from "./util/file-times"
+import { LSP } from "../lsp"
+import { Permission } from "../permission"
+import DESCRIPTION from "./write.txt"
+
+export const WriteTool = Tool.define({
+  id: "opencode.write",
+  description: DESCRIPTION,
+  parameters: z.object({
+    filePath: z
+      .string()
+      .describe(
+        "The absolute path to the file to write (must be absolute, not relative)",
+      ),
+    content: z.string().describe("The content to write to the file"),
+  }),
+  async execute(params, ctx) {
+    const filepath = path.isAbsolute(params.filePath)
+      ? params.filePath
+      : path.join(process.cwd(), params.filePath)
+
+    const file = Bun.file(filepath)
+    const exists = await file.exists()
+    if (exists) await FileTimes.assert(ctx.sessionID, filepath)
+
+    await Permission.ask({
+      id: "opencode.write",
+      sessionID: ctx.sessionID,
+      title: exists
+        ? "Overwrite this file: " + filepath
+        : "Create new file: " + filepath,
+      metadata: {
+        filePath: filepath,
+        content: params.content,
+        exists,
+      },
+    })
+
+    await Bun.write(filepath, params.content)
+    FileTimes.read(ctx.sessionID, filepath)
+
+    let output = ""
+    await LSP.file(filepath)
+    const diagnostics = await LSP.diagnostics()
+    for (const [file, issues] of Object.entries(diagnostics)) {
+      if (issues.length === 0) continue
+      if (file === filepath) {
+        output += `\nThis file has errors, please fix\n<file_diagnostics>\n${issues.map(LSP.Diagnostic.pretty).join("\n")}\n</file_diagnostics>\n`
+        continue
+      }
+      output += `\n<project_diagnostics>\n${file}\n${issues.map(LSP.Diagnostic.pretty).join("\n")}\n</project_diagnostics>\n`
+    }
+
+    return {
+      metadata: {
+        diagnostics,
+        filepath,
+        exists: exists,
+      },
+      output,
+    }
+  },
+})
+