d1c6f6e44c
Integrate `PermissionCache` into `WriteFileTool`, `EditFileTool`, `AppendFileTool`, and `ListDirTool` to allow user approval for actions outside the workspace. - Add `PermissionPrompter` interface and `TerminalPrompter` implementation. - Update `AgentInstance` to initialize and register the permission cache and `RequestPermissionTool`. - Refactor filesystem tools to support permission checks via a `PermissionChecker` interface. - Add `plan.md` documenting the multi-wave permission implementation strategy.
128 lines
3.2 KiB
Go
128 lines
3.2 KiB
Go
package tools
|
|
|
|
import (
|
|
"regexp"
|
|
|
|
"github.com/sipeed/picoclaw/pkg/media"
|
|
fstools "github.com/sipeed/picoclaw/pkg/tools/fs"
|
|
)
|
|
|
|
type (
|
|
ReadFileTool = fstools.ReadFileTool
|
|
ReadFileLinesTool = fstools.ReadFileLinesTool
|
|
WriteFileTool = fstools.WriteFileTool
|
|
ListDirTool = fstools.ListDirTool
|
|
EditFileTool = fstools.EditFileTool
|
|
AppendFileTool = fstools.AppendFileTool
|
|
LoadImageTool = fstools.LoadImageTool
|
|
SendFileTool = fstools.SendFileTool
|
|
)
|
|
|
|
const MaxReadFileSize = fstools.MaxReadFileSize
|
|
|
|
func NewReadFileTool(
|
|
workspace string,
|
|
restrict bool,
|
|
maxReadFileSize int,
|
|
allowPaths ...[]*regexp.Regexp,
|
|
) *ReadFileTool {
|
|
return fstools.NewReadFileTool(workspace, restrict, maxReadFileSize, allowPaths...)
|
|
}
|
|
|
|
func NewReadFileBytesTool(
|
|
workspace string,
|
|
restrict bool,
|
|
maxReadFileSize int,
|
|
allowPaths ...[]*regexp.Regexp,
|
|
) *ReadFileTool {
|
|
return fstools.NewReadFileBytesTool(workspace, restrict, maxReadFileSize, allowPaths...)
|
|
}
|
|
|
|
func NewReadFileLinesTool(
|
|
workspace string,
|
|
restrict bool,
|
|
maxReadFileSize int,
|
|
allowPaths ...[]*regexp.Regexp,
|
|
) *ReadFileLinesTool {
|
|
return fstools.NewReadFileLinesTool(workspace, restrict, maxReadFileSize, allowPaths...)
|
|
}
|
|
|
|
func NewWriteFileTool(
|
|
workspace string,
|
|
restrict bool,
|
|
permCache any,
|
|
allowPaths ...[]*regexp.Regexp,
|
|
) *WriteFileTool {
|
|
var pc fstools.PermissionChecker
|
|
if permCache != nil {
|
|
pc = permCache.(fstools.PermissionChecker)
|
|
}
|
|
return fstools.NewWriteFileTool(workspace, restrict, pc, allowPaths...)
|
|
}
|
|
|
|
func NewListDirTool(
|
|
workspace string,
|
|
restrict bool,
|
|
permCache any,
|
|
allowPaths ...[]*regexp.Regexp,
|
|
) *ListDirTool {
|
|
var checker interface{ Check(path string) string }
|
|
if permCache != nil {
|
|
checker = permCache.(interface{ Check(path string) string })
|
|
}
|
|
return fstools.NewListDirTool(workspace, restrict, checker, allowPaths...)
|
|
}
|
|
|
|
func NewEditFileTool(
|
|
workspace string,
|
|
restrict bool,
|
|
allowPaths ...[]*regexp.Regexp,
|
|
) *EditFileTool {
|
|
return fstools.NewEditFileTool(workspace, restrict, allowPaths...)
|
|
}
|
|
|
|
func NewEditFileToolWithPermission(
|
|
workspace string,
|
|
restrict bool,
|
|
permCache any,
|
|
allowPaths ...[]*regexp.Regexp,
|
|
) *EditFileTool {
|
|
var checker interface{ Check(path string) string }
|
|
if permCache != nil {
|
|
checker = permCache.(interface{ Check(path string) string })
|
|
}
|
|
return fstools.NewEditFileToolWithPermission(workspace, restrict, checker, allowPaths...)
|
|
}
|
|
|
|
func NewAppendFileTool(
|
|
workspace string,
|
|
restrict bool,
|
|
permCache any,
|
|
allowPaths ...[]*regexp.Regexp,
|
|
) *AppendFileTool {
|
|
var checker interface{ Check(path string) string }
|
|
if permCache != nil {
|
|
checker = permCache.(interface{ Check(path string) string })
|
|
}
|
|
return fstools.NewAppendFileTool(workspace, restrict, checker, allowPaths...)
|
|
}
|
|
|
|
func NewLoadImageTool(
|
|
workspace string,
|
|
restrict bool,
|
|
maxFileSize int,
|
|
store media.MediaStore,
|
|
allowPaths ...[]*regexp.Regexp,
|
|
) *LoadImageTool {
|
|
return fstools.NewLoadImageTool(workspace, restrict, maxFileSize, store, allowPaths...)
|
|
}
|
|
|
|
func NewSendFileTool(
|
|
workspace string,
|
|
restrict bool,
|
|
maxFileSize int,
|
|
store media.MediaStore,
|
|
allowPaths ...[]*regexp.Regexp,
|
|
) *SendFileTool {
|
|
return fstools.NewSendFileTool(workspace, restrict, maxFileSize, store, allowPaths...)
|
|
}
|