MCP Browser Use Server

src
mcp_browser_use
controller
# -*- coding: utf-8 -*-

import logging
import pyperclip
from browser_use.agent.views import ActionResult
from browser_use.browser.context import BrowserContext
from browser_use.controller.service import Controller
from browser_use.controller.registry import ActionRegistryError

logger = logging.getLogger(__name__)


class CustomController(Controller):
    """
    A custom controller registering two clipboard actions: copy and paste.
    """

    def __init__(self):
        super().__init__()
        self._register_custom_actions()

    def _register_custom_actions(self) -> None:
        """Register all custom browser actions for this controller."""

        @self.registry.action("Copy text to clipboard")
        def copy_to_clipboard(text: str) -> ActionResult:
            """
            Copy the given text to the system's clipboard.
            Returns an ActionResult with the same text as extracted_content.
            """
            try:
                pyperclip.copy(text)
                # Be cautious about logging the actual text, if sensitive
                logger.debug("Copied text to clipboard.")
                return ActionResult(extracted_content=text)
            except Exception as e:
                logger.error(f"Error copying text to clipboard: {e}")
                return ActionResult(error=str(e), extracted_content=None)

        @self.registry.action("Paste text from clipboard", requires_browser=True)
        async def paste_from_clipboard(browser: BrowserContext) -> ActionResult:
            """
            Paste whatever is currently in the system's clipboard
            into the active browser page by simulating keyboard typing.
            """
            try:
                text = pyperclip.paste()
            except Exception as e:
                logger.error(f"Error reading text from clipboard: {e}")
                return ActionResult(error=str(e), extracted_content=None)

            try:
                # Send text to browser
                page = await browser.get_current_page()
                if page is None:
                    # It's possible there's no current page
                    raise ActionRegistryError("No active browser page found.")
                await page.keyboard.type(text)
                logger.debug("Pasted text from clipboard into the browser.")
                return ActionResult(extracted_content=text)
            except Exception as e:
                logger.error(f"Error pasting text into the browser: {e}")
                return ActionResult(error=str(e), extracted_content=None)