Add commands for dealing with cache and site data, cookies, etc

2025-09-30 08:00:30 -05:00
parent 74d99db629
commit 72d4a5653d
10 changed files with 1553 additions and 1 deletions
--- a/mcp/main.go
+++ b/mcp/main.go
@@ -2168,6 +2168,278 @@ func main() {
 		}, nil
 	})

+	// Cache management tools
+	mcpServer.AddTool(mcp.Tool{
+		Name:        "web_disable_cache_cremotemcp",
+		Description: "Disable browser cache for a tab",
+		InputSchema: mcp.ToolInputSchema{
+			Type: "object",
+			Properties: map[string]any{
+				"tab": map[string]any{
+					"type":        "string",
+					"description": "Tab ID (optional, uses current tab)",
+				},
+				"timeout": map[string]any{
+					"type":        "integer",
+					"description": "Timeout in seconds (default: 5)",
+					"default":     5,
+				},
+			},
+		},
+	}, func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+		// Convert arguments to map
+		params, ok := request.Params.Arguments.(map[string]any)
+		if !ok {
+			return nil, fmt.Errorf("invalid arguments format")
+		}
+
+		tab := getStringParam(params, "tab", "")
+		timeout := getIntParam(params, "timeout", 5)
+
+		err := cremoteServer.client.DisableCache(tab, timeout)
+		if err != nil {
+			return &mcp.CallToolResult{
+				Content: []mcp.Content{
+					mcp.NewTextContent(fmt.Sprintf("Error disabling cache: %v", err)),
+				},
+				IsError: true,
+			}, nil
+		}
+
+		return &mcp.CallToolResult{
+			Content: []mcp.Content{
+				mcp.NewTextContent("Cache disabled successfully"),
+			},
+			IsError: false,
+		}, nil
+	})
+
+	mcpServer.AddTool(mcp.Tool{
+		Name:        "web_enable_cache_cremotemcp",
+		Description: "Enable browser cache for a tab",
+		InputSchema: mcp.ToolInputSchema{
+			Type: "object",
+			Properties: map[string]any{
+				"tab": map[string]any{
+					"type":        "string",
+					"description": "Tab ID (optional, uses current tab)",
+				},
+				"timeout": map[string]any{
+					"type":        "integer",
+					"description": "Timeout in seconds (default: 5)",
+					"default":     5,
+				},
+			},
+		},
+	}, func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+		// Convert arguments to map
+		params, ok := request.Params.Arguments.(map[string]any)
+		if !ok {
+			return nil, fmt.Errorf("invalid arguments format")
+		}
+
+		tab := getStringParam(params, "tab", "")
+		timeout := getIntParam(params, "timeout", 5)
+
+		err := cremoteServer.client.EnableCache(tab, timeout)
+		if err != nil {
+			return &mcp.CallToolResult{
+				Content: []mcp.Content{
+					mcp.NewTextContent(fmt.Sprintf("Error enabling cache: %v", err)),
+				},
+				IsError: true,
+			}, nil
+		}
+
+		return &mcp.CallToolResult{
+			Content: []mcp.Content{
+				mcp.NewTextContent("Cache enabled successfully"),
+			},
+			IsError: false,
+		}, nil
+	})
+
+	mcpServer.AddTool(mcp.Tool{
+		Name:        "web_clear_cache_cremotemcp",
+		Description: "Clear browser cache for a tab",
+		InputSchema: mcp.ToolInputSchema{
+			Type: "object",
+			Properties: map[string]any{
+				"tab": map[string]any{
+					"type":        "string",
+					"description": "Tab ID (optional, uses current tab)",
+				},
+				"timeout": map[string]any{
+					"type":        "integer",
+					"description": "Timeout in seconds (default: 5)",
+					"default":     5,
+				},
+			},
+		},
+	}, func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+		// Convert arguments to map
+		params, ok := request.Params.Arguments.(map[string]any)
+		if !ok {
+			return nil, fmt.Errorf("invalid arguments format")
+		}
+
+		tab := getStringParam(params, "tab", "")
+		timeout := getIntParam(params, "timeout", 5)
+
+		err := cremoteServer.client.ClearCache(tab, timeout)
+		if err != nil {
+			return &mcp.CallToolResult{
+				Content: []mcp.Content{
+					mcp.NewTextContent(fmt.Sprintf("Error clearing cache: %v", err)),
+				},
+				IsError: true,
+			}, nil
+		}
+
+		return &mcp.CallToolResult{
+			Content: []mcp.Content{
+				mcp.NewTextContent("Cache cleared successfully"),
+			},
+			IsError: false,
+		}, nil
+	})
+
+	// Site data management tools
+	mcpServer.AddTool(mcp.Tool{
+		Name:        "web_clear_all_site_data_cremotemcp",
+		Description: "Clear all site data including cookies, storage, cache, etc. for a tab",
+		InputSchema: mcp.ToolInputSchema{
+			Type: "object",
+			Properties: map[string]any{
+				"tab": map[string]any{
+					"type":        "string",
+					"description": "Tab ID (optional, uses current tab)",
+				},
+				"timeout": map[string]any{
+					"type":        "integer",
+					"description": "Timeout in seconds (default: 5)",
+					"default":     5,
+				},
+			},
+		},
+	}, func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+		// Convert arguments to map
+		params, ok := request.Params.Arguments.(map[string]any)
+		if !ok {
+			return nil, fmt.Errorf("invalid arguments format")
+		}
+
+		tab := getStringParam(params, "tab", "")
+		timeout := getIntParam(params, "timeout", 5)
+
+		err := cremoteServer.client.ClearAllSiteData(tab, timeout)
+		if err != nil {
+			return &mcp.CallToolResult{
+				Content: []mcp.Content{
+					mcp.NewTextContent(fmt.Sprintf("Error clearing all site data: %v", err)),
+				},
+				IsError: true,
+			}, nil
+		}
+
+		return &mcp.CallToolResult{
+			Content: []mcp.Content{
+				mcp.NewTextContent("All site data cleared successfully"),
+			},
+			IsError: false,
+		}, nil
+	})
+
+	mcpServer.AddTool(mcp.Tool{
+		Name:        "web_clear_cookies_cremotemcp",
+		Description: "Clear cookies for a tab",
+		InputSchema: mcp.ToolInputSchema{
+			Type: "object",
+			Properties: map[string]any{
+				"tab": map[string]any{
+					"type":        "string",
+					"description": "Tab ID (optional, uses current tab)",
+				},
+				"timeout": map[string]any{
+					"type":        "integer",
+					"description": "Timeout in seconds (default: 5)",
+					"default":     5,
+				},
+			},
+		},
+	}, func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+		// Convert arguments to map
+		params, ok := request.Params.Arguments.(map[string]any)
+		if !ok {
+			return nil, fmt.Errorf("invalid arguments format")
+		}
+
+		tab := getStringParam(params, "tab", "")
+		timeout := getIntParam(params, "timeout", 5)
+
+		err := cremoteServer.client.ClearCookies(tab, timeout)
+		if err != nil {
+			return &mcp.CallToolResult{
+				Content: []mcp.Content{
+					mcp.NewTextContent(fmt.Sprintf("Error clearing cookies: %v", err)),
+				},
+				IsError: true,
+			}, nil
+		}
+
+		return &mcp.CallToolResult{
+			Content: []mcp.Content{
+				mcp.NewTextContent("Cookies cleared successfully"),
+			},
+			IsError: false,
+		}, nil
+	})
+
+	mcpServer.AddTool(mcp.Tool{
+		Name:        "web_clear_storage_cremotemcp",
+		Description: "Clear web storage (localStorage, sessionStorage, IndexedDB, etc.) for a tab",
+		InputSchema: mcp.ToolInputSchema{
+			Type: "object",
+			Properties: map[string]any{
+				"tab": map[string]any{
+					"type":        "string",
+					"description": "Tab ID (optional, uses current tab)",
+				},
+				"timeout": map[string]any{
+					"type":        "integer",
+					"description": "Timeout in seconds (default: 5)",
+					"default":     5,
+				},
+			},
+		},
+	}, func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+		// Convert arguments to map
+		params, ok := request.Params.Arguments.(map[string]any)
+		if !ok {
+			return nil, fmt.Errorf("invalid arguments format")
+		}
+
+		tab := getStringParam(params, "tab", "")
+		timeout := getIntParam(params, "timeout", 5)
+
+		err := cremoteServer.client.ClearStorage(tab, timeout)
+		if err != nil {
+			return &mcp.CallToolResult{
+				Content: []mcp.Content{
+					mcp.NewTextContent(fmt.Sprintf("Error clearing storage: %v", err)),
+				},
+				IsError: true,
+			}, nil
+		}
+
+		return &mcp.CallToolResult{
+			Content: []mcp.Content{
+				mcp.NewTextContent("Storage cleared successfully"),
+			},
+			IsError: false,
+		}, nil
+	})
+
 	// Start the server
 	log.Printf("Cremote MCP server ready")
 	if err := server.ServeStdio(mcpServer); err != nil {