zhuyongxin
cefabd68dc
- 新增 BashTool（bash 命令执行，30s 超时 + 自愈 + 截断） - 新增 WriteFileTool（文件写入） - Windows GBK 编码自动转 UTF-8（golang.org/x/text） - OpenAIProvider 新增 API 交互日志 - dumpMessages 移除截断，完整展示上下文 - cmd/claw 三段式任务演示
73 lines
1.9 KiB
Go
73 lines
1.9 KiB
Go
package tools
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"go-tiny-claw/internal/schema"
|
|
)
|
|
|
|
type WriteFileTool struct {
|
|
workDir string // 工作区约束
|
|
}
|
|
|
|
func NewWriteFileTool(workDir string) *WriteFileTool {
|
|
return &WriteFileTool{workDir: workDir}
|
|
}
|
|
|
|
func (t *WriteFileTool) Name() string {
|
|
return "write_file"
|
|
}
|
|
|
|
func (t *WriteFileTool) Definition() schema.ToolDefinition {
|
|
return schema.ToolDefinition{
|
|
Name: t.Name(),
|
|
Description: "创建或覆盖写入一个文件。如果目录不存在会自动创建。请提供相对于工作区的相对路径。",
|
|
InputSchema: map[string]interface{}{
|
|
"type": "object",
|
|
"properties": map[string]interface{}{
|
|
"path": map[string]interface{}{
|
|
"type": "string",
|
|
"description": "要写入的文件路径，如 src/main.go",
|
|
},
|
|
"content": map[string]interface{}{
|
|
"type": "string",
|
|
"description": "要写入的完整文件内容",
|
|
},
|
|
},
|
|
"required": []string{"path", "content"},
|
|
},
|
|
}
|
|
}
|
|
|
|
type writeFileArgs struct {
|
|
Path string `json:"path"`
|
|
Content string `json:"content"`
|
|
}
|
|
|
|
func (t *WriteFileTool) Execute(ctx context.Context, args json.RawMessage) (string, error) {
|
|
var input writeFileArgs
|
|
if err := json.Unmarshal(args, &input); err != nil {
|
|
return "", fmt.Errorf("参数解析失败: %w", err)
|
|
}
|
|
|
|
// 【安全防线】：限制在 WorkDir 下执行，防止大模型修改系统级文件
|
|
fullPath := filepath.Join(t.workDir, input.Path)
|
|
|
|
// 自动创建缺失的父级目录
|
|
if err := os.MkdirAll(filepath.Dir(fullPath), 0755); err != nil {
|
|
return "", fmt.Errorf("创建父目录失败: %w", err)
|
|
}
|
|
|
|
// 写入文件内容，权限设为 0644
|
|
err := os.WriteFile(fullPath, []byte(input.Content), 0644)
|
|
if err != nil {
|
|
return "", fmt.Errorf("写入文件失败: %w", err)
|
|
}
|
|
|
|
return fmt.Sprintf("成功将内容写入到文件: %s", input.Path), nil
|
|
}
|