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...)
}