lmcli/pkg/agent/toolbox/file_replace_lines.go

package toolbox

import (
	"fmt"
	"os"
	"strings"

	toolutil "git.mlow.ca/mlow/lmcli/pkg/agent/toolbox/util"
	"git.mlow.ca/mlow/lmcli/pkg/api"
)

const FILE_REPLACE_LINES_DESCRIPTION = `Replace or remove a range of lines within a file, must specify path.

Useful for re-writing snippets/blocks of code or entire functions.

Plan your edits carefully and ensure any new content matches the flow and indentation of surrounding text.`

var FileReplaceLinesTool = api.ToolSpec{
	Name:        "file_replace_lines",
	Description: FILE_REPLACE_LINES_DESCRIPTION,
	Parameters: []api.ToolParameter{
		{
			Name:        "path",
			Type:        "string",
			Description: "Path of the file to be modified, relative to the current working directory.",
			Required:    true,
		},
		{
			Name:        "start_line",
			Type:        "integer",
			Description: `Line number which specifies the start of the replacement range (inclusive).`,
			Required:    true,
		},
		{
			Name:        "end_line",
			Type:        "integer",
			Description: `Line number which specifies the end of the replacement range (inclusive). If unset, range extends to end of file.`,
		},
		{
			Name:        "content",
			Type:        "string",
			Description: `Content to replace specified range. Omit to remove the specified range.`,
		},
	},
	Impl: func(tool *api.ToolSpec, args map[string]interface{}) (string, error) {
		tmp, ok := args["path"]
		if !ok {
			return "", fmt.Errorf("path parameter to write_file was not included.")
		}
		path, ok := tmp.(string)
		if !ok {
			return "", fmt.Errorf("Invalid path in function arguments: %v", tmp)
		}
		var start_line int
		tmp, ok = args["start_line"]
		if ok {
			tmp, ok := tmp.(float64)
			if !ok {
				return "", fmt.Errorf("Invalid start_line in function arguments: %v", tmp)
			}
			start_line = int(tmp)
		}
		var end_line int
		tmp, ok = args["end_line"]
		if ok {
			tmp, ok := tmp.(float64)
			if !ok {
				return "", fmt.Errorf("Invalid end_line in function arguments: %v", tmp)
			}
			end_line = int(tmp)
		}
		var content string
		tmp, ok = args["content"]
		if ok {
			content, ok = tmp.(string)
			if !ok {
				return "", fmt.Errorf("Invalid content in function arguments: %v", tmp)
			}
		}

		result := fileReplaceLines(path, start_line, end_line, content)
		ret, err := result.ToJson()
		if err != nil {
			return "", fmt.Errorf("Could not serialize result: %v", err)
		}
		return ret, nil
	},
}

func fileReplaceLines(path string, startLine int, endLine int, content string) api.CallResult {
	ok, reason := toolutil.IsPathWithinCWD(path)
	if !ok {
		return api.CallResult{Message: reason}
	}

	// Read the existing file's content
	data, err := os.ReadFile(path)
	if err != nil {
		if !os.IsNotExist(err) {
			return api.CallResult{Message: fmt.Sprintf("Could not read path: %s", err.Error())}
		}
		_, err = os.Create(path)
		if err != nil {
			return api.CallResult{Message: fmt.Sprintf("Could not create new file: %s", err.Error())}
		}
		data = []byte{}
	}

	if startLine < 1 {
		return api.CallResult{Message: "start_line cannot be less than 1"}
	}

	lines := strings.Split(string(data), "\n")
	contentLines := strings.Split(strings.Trim(content, "\n"), "\n")

	if endLine == 0 || endLine > len(lines) {
		endLine = len(lines)
	}

	before := lines[:startLine-1]
	after := lines[endLine:]

	lines = append(before, append(contentLines, after...)...)
	newContent := strings.Join(lines, "\n")

	// Join the lines and write back to the file
	err = os.WriteFile(path, []byte(newContent), 0644)
	if err != nil {
		return api.CallResult{Message: fmt.Sprintf("Could not write to path: %s", err.Error())}
	}

	return api.CallResult{Result: newContent}
}