summaryrefslogtreecommitdiffhomepage
path: root/lib/dispatch/tool/files/write_file.rb
blob: 218c310ea2b31a5026f9bebe2a135c03bd442b88 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
# frozen_string_literal: true

module Dispatch
  module Tool
    module Files
      WRITE_FILE = Dispatch::Tools::Definition.new(
        name: "write_file",
        description: "Write or overwrite a file with the given content. Creates parent directories if needed.",
        parameters: {
          type: "object",
          properties: {
            path: {
              type: "string",
              description: "Path to the file to write, relative to the worktree root."
            },
            content: {
              type: "string",
              description: "The content to write to the file."
            }
          },
          required: %w[path content],
          additionalProperties: false
        }
      ) do |params, context|
        worktree_path = context[:worktree_path]
        path = params[:path]
        content = params[:content]

        resolved = begin
          Sandbox.resolve_path(path, worktree_path:)
        rescue SandboxError => e
          next Dispatch::Tools::Result.failure(error: e.message)
        end

        FileUtils.mkdir_p(File.dirname(resolved))
        File.write(resolved, content)

        byte_count = content.bytesize
        Dispatch::Tools::Result.success(output: "Wrote #{byte_count} bytes to #{path}")
      end
    end
  end
end