|
@@ -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,
|
|
|
|
|
+ }
|
|
|
|
|
+ },
|
|
|
|
|
+})
|
|
|
|
|
+
|