115 lines
2.9 KiB
Go
115 lines
2.9 KiB
Go
package tools
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
toolutil "git.mlow.ca/mlow/lmcli/pkg/lmcli/tools/util"
|
|
"git.mlow.ca/mlow/lmcli/pkg/lmcli/model"
|
|
)
|
|
|
|
const FILE_INSERT_LINES_DESCRIPTION = `Insert lines into a file, must specify path.
|
|
|
|
Make sure your inserts match the flow and indentation of surrounding content.`
|
|
|
|
var FileInsertLinesTool = model.Tool{
|
|
Name: "file_insert_lines",
|
|
Description: FILE_INSERT_LINES_DESCRIPTION,
|
|
Parameters: []model.ToolParameter{
|
|
{
|
|
Name: "path",
|
|
Type: "string",
|
|
Description: "Path of the file to be modified, relative to the current working directory.",
|
|
Required: true,
|
|
},
|
|
{
|
|
Name: "position",
|
|
Type: "integer",
|
|
Description: `Which line to insert content *before*.`,
|
|
Required: true,
|
|
},
|
|
{
|
|
Name: "content",
|
|
Type: "string",
|
|
Description: `The content to insert.`,
|
|
Required: true,
|
|
},
|
|
},
|
|
Impl: func(tool *model.Tool, 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 position int
|
|
tmp, ok = args["position"]
|
|
if ok {
|
|
tmp, ok := tmp.(float64)
|
|
if !ok {
|
|
return "", fmt.Errorf("Invalid position in function arguments: %v", tmp)
|
|
}
|
|
position = 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 := fileInsertLines(path, position, content)
|
|
ret, err := result.ToJson()
|
|
if err != nil {
|
|
return "", fmt.Errorf("Could not serialize result: %v", err)
|
|
}
|
|
return ret, nil
|
|
},
|
|
}
|
|
|
|
func fileInsertLines(path string, position int, content string) model.CallResult {
|
|
ok, reason := toolutil.IsPathWithinCWD(path)
|
|
if !ok {
|
|
return model.CallResult{Message: reason}
|
|
}
|
|
|
|
// Read the existing file's content
|
|
data, err := os.ReadFile(path)
|
|
if err != nil {
|
|
if !os.IsNotExist(err) {
|
|
return model.CallResult{Message: fmt.Sprintf("Could not read path: %s", err.Error())}
|
|
}
|
|
_, err = os.Create(path)
|
|
if err != nil {
|
|
return model.CallResult{Message: fmt.Sprintf("Could not create new file: %s", err.Error())}
|
|
}
|
|
data = []byte{}
|
|
}
|
|
|
|
if position < 1 {
|
|
return model.CallResult{Message: "start_line cannot be less than 1"}
|
|
}
|
|
|
|
lines := strings.Split(string(data), "\n")
|
|
contentLines := strings.Split(strings.Trim(content, "\n"), "\n")
|
|
|
|
before := lines[:position-1]
|
|
after := lines[position-1:]
|
|
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 model.CallResult{Message: fmt.Sprintf("Could not write to path: %s", err.Error())}
|
|
}
|
|
|
|
return model.CallResult{Result: newContent}
|
|
}
|