write.go RawHistoryFolder

1package tools
2
3import (
4	"context"
5	"encoding/json"
6	"fmt"
7	"os"
8	"path/filepath"
9
10	"github.com/patriceckhart/zot/internal/core"
11	"github.com/patriceckhart/zot/internal/provider"
12)
13
14// WriteTool writes content to a file, creating parent directories.
15type WriteTool struct {
16	CWD     string
17	Sandbox *Sandbox
18}
19
20type writeArgs struct {
21	Path    string `json:"path"`
22	Content string `json:"content"`
23}
24
25const writeSchema = `{
26  "type":"object",
27  "properties":{
28    "path":{"type":"string","description":"Path to the file to write (relative or absolute)"},
29    "content":{"type":"string","description":"File contents. Overwrites any existing file."}
30  },
31  "required":["path","content"],
32  "additionalProperties":false
33}`
34
35func (t *WriteTool) Name() string { return "write" }
36func (t *WriteTool) Description() string {
37	return "Write content to a file. Creates parent directories. Overwrites existing files."
38}
39func (t *WriteTool) Schema() json.RawMessage { return json.RawMessage(writeSchema) }
40
41func (t *WriteTool) Execute(ctx context.Context, raw json.RawMessage, progress func(string)) (core.ToolResult, error) {
42	var a writeArgs
43	if err := json.Unmarshal(raw, &a); err != nil {
44		return core.ToolResult{}, fmt.Errorf("invalid args: %w", err)
45	}
46	if a.Path == "" {
47		return core.ToolResult{}, fmt.Errorf("path is required")
48	}
49	path := resolvePath(t.CWD, a.Path)
50	if err := t.Sandbox.CheckPath(path); err != nil {
51		return core.ToolResult{}, err
52	}
53
54	if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
55		return core.ToolResult{}, err
56	}
57	if err := os.WriteFile(path, []byte(a.Content), 0o644); err != nil {
58		return core.ToolResult{}, err
59	}
60
61	msg := fmt.Sprintf("wrote %d bytes to %s", len(a.Content), a.Path)
62	return core.ToolResult{
63		Content: []provider.Content{provider.TextBlock{Text: msg}},
64		Details: map[string]any{"path": path, "bytes": len(a.Content)},
65	}, nil
66}