summaryrefslogtreecommitdiffhomepage
path: root/lib/dispatch/tool/files/read_file.rb
blob: c0c5edc728d81a67723cb6bac9610b3365a3356b (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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
# frozen_string_literal: true

module Dispatch
  module Tool
    module Files
      READ_FILE = Dispatch::Tools::Definition.new(
        name: "read_file",
        description: "Read the contents of a file. Returns file contents with line numbers.",
        parameters: {
          type: "object",
          properties: {
            path: {
              type: "string",
              description: "Path to the file to read, relative to the worktree root."
            },
            start_line: {
              type: "integer",
              description: "The line number to start reading from (0-based). Defaults to 0."
            },
            end_line: {
              type: "integer",
              description: "The inclusive line number to stop reading at (0-based). Use -1 for end of file."
            }
          },
          required: %w[path],
          additionalProperties: false
        }
      ) do |params, context|
        worktree_path = context[:worktree_path]
        path = params[:path]

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

        unless File.exist?(resolved)
          next Dispatch::Tools::Result.failure(error: "File not found: #{path}")
        end

        # Binary file detection: check first 8192 bytes for null bytes
        sample = File.binread(resolved, 8192)
        if sample.include?("\x00")
          next Dispatch::Tools::Result.failure(error: "Cannot read binary file: #{path}")
        end

        lines = File.readlines(resolved)

        start_line = params.fetch(:start_line, 0)
        end_line = params.fetch(:end_line, -1)
        end_line = lines.length - 1 if end_line == -1

        selected = lines[start_line..end_line] || []

        # Calculate padding width based on the highest line number
        width = end_line.to_s.length

        output = selected.each_with_index.map do |line, idx|
          line_num = start_line + idx
          "#{line_num.to_s.rjust(width)}: #{line}"
        end.join

        Dispatch::Tools::Result.success(output:)
      end
    end
  end
end