gnoma/internal/tool/fs/edit.go

package fs

import (
	"context"
	"encoding/json"
	"fmt"
	"os"
	"strings"

	"somegit.dev/Owlibou/gnoma/internal/tool"
)

const editToolName = "fs.edit"

var editParams = json.RawMessage(`{
	"type": "object",
	"properties": {
		"path": {
			"type": "string",
			"description": "Absolute path to the file to edit"
		},
		"old_string": {
			"type": "string",
			"description": "The exact text to find and replace"
		},
		"new_string": {
			"type": "string",
			"description": "The replacement text"
		},
		"replace_all": {
			"type": "boolean",
			"description": "Replace all occurrences (default false)"
		}
	},
	"required": ["path", "old_string", "new_string"]
}`)

type EditTool struct {
	guard *Guard
}

func NewEditTool() *EditTool { return &EditTool{} }

func (t *EditTool) SetGuard(g *Guard) { t.guard = g }

func (t *EditTool) Name() string               { return editToolName }
func (t *EditTool) Description() string         { return "Perform exact string replacement in a file" }
func (t *EditTool) Parameters() json.RawMessage { return editParams }
func (t *EditTool) IsReadOnly() bool            { return false }
func (t *EditTool) IsDestructive() bool         { return false }
func (t *EditTool) Category() tool.Category     { return tool.CategoryWrite }

func (t *EditTool) ExtractPaths(args json.RawMessage) []string {
	var a editArgs
	if err := json.Unmarshal(args, &a); err != nil {
		return nil
	}
	return []string{a.Path}
}

type editArgs struct {
	Path       string `json:"path"`
	OldString  string `json:"old_string"`
	NewString  string `json:"new_string"`
	ReplaceAll bool   `json:"replace_all,omitempty"`
}

func (t *EditTool) Execute(_ context.Context, args json.RawMessage) (tool.Result, error) {
	var a editArgs
	if err := json.Unmarshal(args, &a); err != nil {
		return tool.Result{}, fmt.Errorf("fs.edit: invalid args: %w", err)
	}
	if a.Path == "" {
		return tool.Result{}, fmt.Errorf("fs.edit: path required")
	}
	if a.OldString == a.NewString {
		return tool.Result{}, fmt.Errorf("fs.edit: old_string and new_string must differ")
	}

	path := a.Path
	if t.guard != nil {
		resolved, err := t.guard.ResolveRead(path)
		if err != nil {
			return tool.Result{Output: fmt.Sprintf("Error: %v", err)}, nil
		}
		path = resolved
	}

	data, err := os.ReadFile(path)
	if err != nil {
		return tool.Result{Output: fmt.Sprintf("Error: %v", err)}, nil
	}

	content := string(data)

	count := strings.Count(content, a.OldString)
	if count == 0 {
		return tool.Result{
			Output:   "Error: old_string not found in file",
			Metadata: map[string]any{"matches": 0},
		}, nil
	}

	if !a.ReplaceAll && count > 1 {
		return tool.Result{
			Output:   fmt.Sprintf("Error: old_string has %d matches (must be unique, or use replace_all)", count),
			Metadata: map[string]any{"matches": count},
		}, nil
	}

	var newContent string
	if a.ReplaceAll {
		newContent = strings.ReplaceAll(content, a.OldString, a.NewString)
	} else {
		newContent = strings.Replace(content, a.OldString, a.NewString, 1)
	}

	if err := os.WriteFile(path, []byte(newContent), 0o644); err != nil {
		return tool.Result{Output: fmt.Sprintf("Error writing file: %v", err)}, nil
	}

	replacements := 1
	if a.ReplaceAll {
		replacements = count
	}

	// Generate diff-style output with context
	diff := buildEditDiff(content, a.OldString, a.NewString, path, replacements)

	return tool.Result{
		Output:   diff,
		Metadata: map[string]any{"replacements": replacements, "path": path},
	}, nil
}

// buildEditDiff generates a diff display with context lines around the edit.
func buildEditDiff(original, oldStr, newStr, path string, replacements int) string {
	contextLines := 3
	lines := strings.Split(original, "\n")

	// Find the line where the old string starts
	editStart := -1
	for i, line := range lines {
		if strings.Contains(line, strings.Split(oldStr, "\n")[0]) {
			editStart = i
			break
		}
	}

	if editStart == -1 {
		return fmt.Sprintf("Replaced %d occurrence(s) in %s", replacements, path)
	}

	oldLines := strings.Split(oldStr, "\n")
	newLines := strings.Split(newStr, "\n")

	var b strings.Builder
	fmt.Fprintf(&b, "Edit(%s)\n", path)
	fmt.Fprintf(&b, "  Added %d lines, removed %d lines\n", len(newLines), len(oldLines))

	// Context before
	start := editStart - contextLines
	if start < 0 {
		start = 0
	}
	for i := start; i < editStart; i++ {
		fmt.Fprintf(&b, "  %4d   %s\n", i+1, lines[i])
	}

	// Removed lines (old)
	for i, line := range oldLines {
		fmt.Fprintf(&b, "  %4d - %s\n", editStart+i+1, line)
	}

	// Added lines (new)
	for i, line := range newLines {
		fmt.Fprintf(&b, "  %4d + %s\n", editStart+i+1, line)
	}

	// Context after
	afterStart := editStart + len(oldLines)
	afterEnd := afterStart + contextLines
	if afterEnd > len(lines) {
		afterEnd = len(lines)
	}
	for i := afterStart; i < afterEnd; i++ {
		fmt.Fprintf(&b, "  %4d   %s\n", i+1, lines[i])
	}

	return b.String()
}