agent
76 lines
2.3 KiB
Go
76 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/mark3labs/mcp-go/mcp"
|
|
"github.com/mark3labs/mcp-go/server"
|
|
|
|
"fn-registry/functions/browser"
|
|
)
|
|
|
|
// registerStorageTools wires storage_save (read) + storage_load (MUTA).
|
|
func registerStorageTools(s *server.MCPServer, d *deps) {
|
|
s.AddTool(storageSaveTool(), mcp.NewTypedToolHandler(d.handleStorageSave))
|
|
|
|
if !d.readOnly {
|
|
s.AddTool(storageLoadTool(), mcp.NewTypedToolHandler(d.handleStorageLoad))
|
|
}
|
|
}
|
|
|
|
// ---- storage_save ----
|
|
|
|
type storageSaveArgs struct {
|
|
Port int `json:"port"`
|
|
Path string `json:"path"`
|
|
}
|
|
|
|
func storageSaveTool() mcp.Tool {
|
|
return mcp.NewTool("storage_save",
|
|
mcp.WithDescription("Save the current session storage state (cookies + localStorage) to a JSON file for later reuse."),
|
|
mcp.WithNumber("port", mcp.Description("CDP port. Default 9222.")),
|
|
mcp.WithString("path", mcp.Required(), mcp.Description("Output JSON file path.")),
|
|
)
|
|
}
|
|
|
|
func (d *deps) handleStorageSave(_ context.Context, _ mcp.CallToolRequest, a storageSaveArgs) (*mcp.CallToolResult, error) {
|
|
if a.Path == "" {
|
|
return mcp.NewToolResultError("path is required"), nil
|
|
}
|
|
err := d.withConn(portOr(a.Port), func(c *browser.CDPConn) error {
|
|
return browser.CdpSaveStorageState(c, a.Path)
|
|
})
|
|
if err != nil {
|
|
return mcp.NewToolResultError(err.Error()), nil
|
|
}
|
|
return mcp.NewToolResultText("storage state saved to " + a.Path), nil
|
|
}
|
|
|
|
// ---- storage_load (MUTA) ----
|
|
|
|
type storageLoadArgs struct {
|
|
Port int `json:"port"`
|
|
Path string `json:"path"`
|
|
}
|
|
|
|
func storageLoadTool() mcp.Tool {
|
|
return mcp.NewTool("storage_load",
|
|
mcp.WithDescription("Load a previously saved session storage state (cookies + localStorage) from a JSON file into the live browser."),
|
|
mcp.WithNumber("port", mcp.Description("CDP port. Default 9222.")),
|
|
mcp.WithString("path", mcp.Required(), mcp.Description("Input JSON file path.")),
|
|
)
|
|
}
|
|
|
|
func (d *deps) handleStorageLoad(_ context.Context, _ mcp.CallToolRequest, a storageLoadArgs) (*mcp.CallToolResult, error) {
|
|
if a.Path == "" {
|
|
return mcp.NewToolResultError("path is required"), nil
|
|
}
|
|
err := d.withConn(portOr(a.Port), func(c *browser.CDPConn) error {
|
|
return browser.CdpLoadStorageState(c, a.Path)
|
|
})
|
|
if err != nil {
|
|
return mcp.NewToolResultError(err.Error()), nil
|
|
}
|
|
return mcp.NewToolResultText("storage state loaded from " + a.Path), nil
|
|
}
|