code-puppy 0.0.214__py3-none-any.whl → 0.0.366__py3-none-any.whl

code_puppy/tools/browser/__init__.py

@@ -0,0 +1,37 @@
+"""Browser tools for terminal automation.
+
+This module provides browser-based terminal automation tools.
+"""
+
+from code_puppy.config import get_banner_color
+
+from .browser_manager import (
+    cleanup_all_browsers,
+    get_browser_session,
+    get_session_browser_manager,
+    set_browser_session,
+)
+
+
+def format_terminal_banner(text: str) -> str:
+    """Format a terminal tool banner with the configured terminal_tool color.
+
+    Returns Rich markup string that can be used with Text.from_markup().
+
+    Args:
+        text: The banner text (e.g., "TERMINAL OPEN 🖥️ localhost:8765")
+
+    Returns:
+        Rich markup formatted string
+    """
+    color = get_banner_color("terminal_tool")
+    return f"[bold white on {color}] {text} [/bold white on {color}]"
+
+
+__all__ = [
+    "format_terminal_banner",
+    "cleanup_all_browsers",
+    "get_browser_session",
+    "get_session_browser_manager",
+    "set_browser_session",
+]

code_puppy/tools/browser/browser_control.py

@@ -4,10 +4,10 @@ from typing import Any, Dict, Optional
 
 from pydantic_ai import RunContext
 
-from code_puppy.messaging import emit_info
+from code_puppy.messaging import emit_error, emit_info, emit_success, emit_warning
 from code_puppy.tools.common import generate_group_id
 
-from .camoufox_manager import get_camoufox_manager
+from .browser_manager import get_session_browser_manager
 
 
 async def initialize_browser(
@@ -18,11 +18,11 @@ async def initialize_browser(
     """Initialize the browser with specified settings."""
     group_id = generate_group_id("browser_initialize", f"{browser_type}_{homepage}")
     emit_info(
-        f"[bold white on blue] BROWSER INITIALIZE [/bold white on blue] 🌐 {browser_type} → {homepage}",
+        f"BROWSER INITIALIZE 🌐 {browser_type} → {homepage}",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
 
         # Configure browser settings
         browser_manager.headless = headless
@@ -41,9 +41,9 @@ async def initialize_browser(
             url = "Unknown"
             title = "Unknown"
 
-        emit_info(
-            "[green]Browser initialized successfully[/green]", message_group=group_id
-        )
+        # emit_info(
+        #     "[green]Browser initialized successfully[/green]", message_group=group_id
+        # )  # Removed to reduce console spam
 
         return {
             "success": True,
@@ -55,8 +55,8 @@ async def initialize_browser(
         }
 
     except Exception as e:
-        emit_info(
-            f"[red]Browser initialization failed: {str(e)}[/red]",
+        emit_error(
+            f"Browser initialization failed: {str(e)}",
             message_group=group_id,
         )
         return {
@@ -71,16 +71,14 @@ async def close_browser() -> Dict[str, Any]:
     """Close the browser and clean up resources."""
     group_id = generate_group_id("browser_close")
     emit_info(
-        "[bold white on blue] BROWSER CLOSE [/bold white on blue] 🔒",
+        "BROWSER CLOSE 🔒",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         await browser_manager.close()
 
-        emit_info(
-            "[yellow]Browser closed successfully[/yellow]", message_group=group_id
-        )
+        emit_warning("Browser closed successfully", message_group=group_id)
 
         return {"success": True, "message": "Browser closed"}
 
@@ -92,11 +90,11 @@ async def get_browser_status() -> Dict[str, Any]:
     """Get current browser status and information."""
     group_id = generate_group_id("browser_status")
     emit_info(
-        "[bold white on blue] BROWSER STATUS [/bold white on blue] 📊",
+        "BROWSER STATUS 📊",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
 
         if not browser_manager._initialized:
             return {
@@ -137,11 +135,11 @@ async def create_new_page(url: Optional[str] = None) -> Dict[str, Any]:
     """Create a new browser page/tab."""
     group_id = generate_group_id("browser_new_page", url or "blank")
     emit_info(
-        f"[bold white on blue] BROWSER NEW PAGE [/bold white on blue] 📄 {url or 'blank page'}",
+        f"BROWSER NEW PAGE 📄 {url or 'blank page'}",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
 
         if not browser_manager._initialized:
             return {
@@ -154,9 +152,7 @@ async def create_new_page(url: Optional[str] = None) -> Dict[str, Any]:
         final_url = page.url
         title = await page.title()
 
-        emit_info(
-            f"[green]Created new page: {final_url}[/green]", message_group=group_id
-        )
+        emit_success(f"Created new page: {final_url}", message_group=group_id)
 
         return {"success": True, "url": final_url, "title": title, "requested_url": url}
 
@@ -168,11 +164,11 @@ async def list_pages() -> Dict[str, Any]:
     """List all open browser pages/tabs."""
     group_id = generate_group_id("browser_list_pages")
     emit_info(
-        "[bold white on blue] BROWSER LIST PAGES [/bold white on blue] 📋",
+        "BROWSER LIST PAGES 📋",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
 
         if not browser_manager._initialized:
             return {"success": False, "error": "Browser not initialized"}

code_puppy/tools/browser/browser_interactions.py

@@ -4,10 +4,10 @@ from typing import Any, Dict, List, Optional
 
 from pydantic_ai import RunContext
 
-from code_puppy.messaging import emit_info
+from code_puppy.messaging import emit_error, emit_info, emit_success
 from code_puppy.tools.common import generate_group_id
 
-from .camoufox_manager import get_camoufox_manager
+from .browser_manager import get_session_browser_manager
 
 
 async def click_element(
@@ -20,18 +20,19 @@ async def click_element(
     """Click on an element."""
     group_id = generate_group_id("browser_click", selector[:100])
     emit_info(
-        f"[bold white on blue] BROWSER CLICK [/bold white on blue] 🖱️ selector='{selector}' button={button}",
+        f"BROWSER CLICK 🖱️ selector='{selector}' button={button}",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
             return {"success": False, "error": "No active browser page available"}
 
-        # Find element
-        element = page.locator(selector)
+        # Find element - use .first to handle cases where selector matches multiple elements
+        # This avoids Playwright's strict mode violation errors
+        element = page.locator(selector).first
 
         # Wait for element to be visible and enabled
         await element.wait_for(state="visible", timeout=timeout)
@@ -48,12 +49,12 @@ async def click_element(
 
         await element.click(**click_options)
 
-        emit_info(f"[green]Clicked element: {selector}[/green]", message_group=group_id)
+        emit_success(f"Clicked element: {selector}", message_group=group_id)
 
         return {"success": True, "selector": selector, "action": f"{button}_click"}
 
     except Exception as e:
-        emit_info(f"[red]Click failed: {str(e)}[/red]", message_group=group_id)
+        emit_error(f"Click failed: {str(e)}", message_group=group_id)
         return {"success": False, "error": str(e), "selector": selector}
 
 
@@ -65,23 +66,21 @@ async def double_click_element(
     """Double-click on an element."""
     group_id = generate_group_id("browser_double_click", selector[:100])
     emit_info(
-        f"[bold white on blue] BROWSER DOUBLE CLICK [/bold white on blue] 🖱️🖱️ selector='{selector}'",
+        f"BROWSER DOUBLE CLICK 🖱️🖱️ selector='{selector}'",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
             return {"success": False, "error": "No active browser page available"}
 
-        element = page.locator(selector)
+        element = page.locator(selector).first
         await element.wait_for(state="visible", timeout=timeout)
         await element.dblclick(force=force, timeout=timeout)
 
-        emit_info(
-            f"[green]Double-clicked element: {selector}[/green]", message_group=group_id
-        )
+        emit_success(f"Double-clicked element: {selector}", message_group=group_id)
 
         return {"success": True, "selector": selector, "action": "double_click"}
 
@@ -97,23 +96,21 @@ async def hover_element(
     """Hover over an element."""
     group_id = generate_group_id("browser_hover", selector[:100])
     emit_info(
-        f"[bold white on blue] BROWSER HOVER [/bold white on blue] 👆 selector='{selector}'",
+        f"BROWSER HOVER 👆 selector='{selector}'",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
             return {"success": False, "error": "No active browser page available"}
 
-        element = page.locator(selector)
+        element = page.locator(selector).first
         await element.wait_for(state="visible", timeout=timeout)
         await element.hover(force=force, timeout=timeout)
 
-        emit_info(
-            f"[green]Hovered over element: {selector}[/green]", message_group=group_id
-        )
+        emit_success(f"Hovered over element: {selector}", message_group=group_id)
 
         return {"success": True, "selector": selector, "action": "hover"}
 
@@ -130,17 +127,17 @@ async def set_element_text(
     """Set text in an input element."""
     group_id = generate_group_id("browser_set_text", f"{selector[:50]}_{text[:30]}")
     emit_info(
-        f"[bold white on blue] BROWSER SET TEXT [/bold white on blue] ✏️ selector='{selector}' text='{text[:50]}{'...' if len(text) > 50 else ''}'",
+        f"BROWSER SET TEXT ✏️ selector='{selector}' text='{text[:50]}{'...' if len(text) > 50 else ''}'",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
             return {"success": False, "error": "No active browser page available"}
 
-        element = page.locator(selector)
+        element = page.locator(selector).first
         await element.wait_for(state="visible", timeout=timeout)
 
         if clear_first:
@@ -148,9 +145,7 @@ async def set_element_text(
 
         await element.fill(text, timeout=timeout)
 
-        emit_info(
-            f"[green]Set text in element: {selector}[/green]", message_group=group_id
-        )
+        emit_success(f"Set text in element: {selector}", message_group=group_id)
 
         return {
             "success": True,
@@ -160,7 +155,7 @@ async def set_element_text(
         }
 
     except Exception as e:
-        emit_info(f"[red]Set text failed: {str(e)}[/red]", message_group=group_id)
+        emit_error(f"Set text failed: {str(e)}", message_group=group_id)
         return {"success": False, "error": str(e), "selector": selector, "text": text}
 
 
@@ -171,17 +166,17 @@ async def get_element_text(
     """Get text content from an element."""
     group_id = generate_group_id("browser_get_text", selector[:100])
     emit_info(
-        f"[bold white on blue] BROWSER GET TEXT [/bold white on blue] 📝 selector='{selector}'",
+        f"BROWSER GET TEXT 📝 selector='{selector}'",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
             return {"success": False, "error": "No active browser page available"}
 
-        element = page.locator(selector)
+        element = page.locator(selector).first
         await element.wait_for(state="visible", timeout=timeout)
 
         text = await element.text_content()
@@ -199,17 +194,17 @@ async def get_element_value(
     """Get value from an input element."""
     group_id = generate_group_id("browser_get_value", selector[:100])
     emit_info(
-        f"[bold white on blue] BROWSER GET VALUE [/bold white on blue] 📎 selector='{selector}'",
+        f"BROWSER GET VALUE 📎 selector='{selector}'",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
             return {"success": False, "error": "No active browser page available"}
 
-        element = page.locator(selector)
+        element = page.locator(selector).first
         await element.wait_for(state="visible", timeout=timeout)
 
         value = await element.input_value()
@@ -233,17 +228,17 @@ async def select_option(
         "browser_select_option", f"{selector[:50]}_{option_desc}"
     )
     emit_info(
-        f"[bold white on blue] BROWSER SELECT OPTION [/bold white on blue] 📄 selector='{selector}' option='{option_desc}'",
+        f"BROWSER SELECT OPTION 📄 selector='{selector}' option='{option_desc}'",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
             return {"success": False, "error": "No active browser page available"}
 
-        element = page.locator(selector)
+        element = page.locator(selector).first
         await element.wait_for(state="visible", timeout=timeout)
 
         if value is not None:
@@ -262,8 +257,8 @@ async def select_option(
                 "selector": selector,
             }
 
-        emit_info(
-            f"[green]Selected option in {selector}: {selection}[/green]",
+        emit_success(
+            f"Selected option in {selector}: {selection}",
             message_group=group_id,
         )
 
@@ -280,21 +275,21 @@ async def check_element(
     """Check a checkbox or radio button."""
     group_id = generate_group_id("browser_check", selector[:100])
     emit_info(
-        f"[bold white on blue] BROWSER CHECK [/bold white on blue] ☑️ selector='{selector}'",
+        f"BROWSER CHECK ☑️ selector='{selector}'",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
             return {"success": False, "error": "No active browser page available"}
 
-        element = page.locator(selector)
+        element = page.locator(selector).first
         await element.wait_for(state="visible", timeout=timeout)
         await element.check(timeout=timeout)
 
-        emit_info(f"[green]Checked element: {selector}[/green]", message_group=group_id)
+        emit_success(f"Checked element: {selector}", message_group=group_id)
 
         return {"success": True, "selector": selector, "action": "check"}
 
@@ -309,23 +304,21 @@ async def uncheck_element(
     """Uncheck a checkbox."""
     group_id = generate_group_id("browser_uncheck", selector[:100])
     emit_info(
-        f"[bold white on blue] BROWSER UNCHECK [/bold white on blue] ☐️ selector='{selector}'",
+        f"BROWSER UNCHECK ☐️ selector='{selector}'",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
             return {"success": False, "error": "No active browser page available"}
 
-        element = page.locator(selector)
+        element = page.locator(selector).first
         await element.wait_for(state="visible", timeout=timeout)
         await element.uncheck(timeout=timeout)
 
-        emit_info(
-            f"[green]Unchecked element: {selector}[/green]", message_group=group_id
-        )
+        emit_success(f"Unchecked element: {selector}", message_group=group_id)
 
         return {"success": True, "selector": selector, "action": "uncheck"}
 

code_puppy/tools/browser/browser_locators.py

@@ -4,10 +4,10 @@ from typing import Any, Dict, Optional
 
 from pydantic_ai import RunContext
 
-from code_puppy.messaging import emit_info
+from code_puppy.messaging import emit_info, emit_success
 from code_puppy.tools.common import generate_group_id
 
-from .camoufox_manager import get_camoufox_manager
+from .browser_manager import get_session_browser_manager
 
 
 async def find_by_role(
@@ -19,11 +19,11 @@ async def find_by_role(
     """Find elements by ARIA role."""
     group_id = generate_group_id("browser_find_by_role", f"{role}_{name or 'any'}")
     emit_info(
-        f"[bold white on blue] BROWSER FIND BY ROLE [/bold white on blue] 🎨 role={role} name={name}",
+        f"BROWSER FIND BY ROLE 🎨 role={role} name={name}",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
@@ -46,8 +46,8 @@ async def find_by_role(
                 text = await element.text_content()
                 elements.append({"index": i, "text": text, "visible": True})
 
-        emit_info(
-            f"[green]Found {count} elements with role '{role}'[/green]",
+        emit_success(
+            f"Found {count} elements with role '{role}'",
             message_group=group_id,
         )
 
@@ -71,11 +71,11 @@ async def find_by_text(
     """Find elements containing specific text."""
     group_id = generate_group_id("browser_find_by_text", text[:50])
     emit_info(
-        f"[bold white on blue] BROWSER FIND BY TEXT [/bold white on blue] 🔍 text='{text}' exact={exact}",
+        f"BROWSER FIND BY TEXT 🔍 text='{text}' exact={exact}",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
@@ -98,8 +98,8 @@ async def find_by_text(
                     {"index": i, "tag": tag_name, "text": full_text, "visible": True}
                 )
 
-        emit_info(
-            f"[green]Found {count} elements containing text '{text}'[/green]",
+        emit_success(
+            f"Found {count} elements containing text '{text}'",
             message_group=group_id,
         )
 
@@ -123,11 +123,11 @@ async def find_by_label(
     """Find form elements by their associated label text."""
     group_id = generate_group_id("browser_find_by_label", text[:50])
     emit_info(
-        f"[bold white on blue] BROWSER FIND BY LABEL [/bold white on blue] 🏷️ label='{text}' exact={exact}",
+        f"BROWSER FIND BY LABEL 🏷️ label='{text}' exact={exact}",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
@@ -161,8 +161,8 @@ async def find_by_label(
                     }
                 )
 
-        emit_info(
-            f"[green]Found {count} elements with label '{text}'[/green]",
+        emit_success(
+            f"Found {count} elements with label '{text}'",
             message_group=group_id,
         )
 
@@ -186,11 +186,11 @@ async def find_by_placeholder(
     """Find elements by placeholder text."""
     group_id = generate_group_id("browser_find_by_placeholder", text[:50])
     emit_info(
-        f"[bold white on blue] BROWSER FIND BY PLACEHOLDER [/bold white on blue] 📝 placeholder='{text}' exact={exact}",
+        f"BROWSER FIND BY PLACEHOLDER 📝 placeholder='{text}' exact={exact}",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
@@ -220,8 +220,8 @@ async def find_by_placeholder(
                     }
                 )
 
-        emit_info(
-            f"[green]Found {count} elements with placeholder '{text}'[/green]",
+        emit_success(
+            f"Found {count} elements with placeholder '{text}'",
             message_group=group_id,
         )
 
@@ -244,11 +244,11 @@ async def find_by_test_id(
     """Find elements by test ID attribute."""
     group_id = generate_group_id("browser_find_by_test_id", test_id)
     emit_info(
-        f"[bold white on blue] BROWSER FIND BY TEST ID [/bold white on blue] 🧪 test_id='{test_id}'",
+        f"BROWSER FIND BY TEST ID 🧪 test_id='{test_id}'",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
@@ -277,8 +277,8 @@ async def find_by_test_id(
                     }
                 )
 
-        emit_info(
-            f"[green]Found {count} elements with test-id '{test_id}'[/green]",
+        emit_success(
+            f"Found {count} elements with test-id '{test_id}'",
             message_group=group_id,
         )
 
@@ -300,11 +300,11 @@ async def run_xpath_query(
     """Find elements using XPath selector."""
     group_id = generate_group_id("browser_xpath_query", xpath[:100])
     emit_info(
-        f"[bold white on blue] BROWSER XPATH QUERY [/bold white on blue] 🔍 xpath='{xpath}'",
+        f"BROWSER XPATH QUERY 🔍 xpath='{xpath}'",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
@@ -338,8 +338,8 @@ async def run_xpath_query(
                     }
                 )
 
-        emit_info(
-            f"[green]Found {count} elements with XPath '{xpath}'[/green]",
+        emit_success(
+            f"Found {count} elements with XPath '{xpath}'",
             message_group=group_id,
         )
 
@@ -355,11 +355,11 @@ async def find_buttons(
     """Find all button elements on the page."""
     group_id = generate_group_id("browser_find_buttons", text_filter or "all")
     emit_info(
-        f"[bold white on blue] BROWSER FIND BUTTONS [/bold white on blue] 🔘 filter='{text_filter or 'none'}'",
+        f"BROWSER FIND BUTTONS 🔘 filter='{text_filter or 'none'}'",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
@@ -382,10 +382,9 @@ async def find_buttons(
 
         filtered_count = len(buttons)
 
-        emit_info(
-            f"[green]Found {filtered_count} buttons"
-            + (f" containing '{text_filter}'" if text_filter else "")
-            + "[/green]",
+        emit_success(
+            f"Found {filtered_count} buttons"
+            + (f" containing '{text_filter}'" if text_filter else ""),
             message_group=group_id,
         )
 
@@ -407,11 +406,11 @@ async def find_links(
     """Find all link elements on the page."""
     group_id = generate_group_id("browser_find_links", text_filter or "all")
     emit_info(
-        f"[bold white on blue] BROWSER FIND LINKS [/bold white on blue] 🔗 filter='{text_filter or 'none'}'",
+        f"BROWSER FIND LINKS 🔗 filter='{text_filter or 'none'}'",
         message_group=group_id,
     )
     try:
-        browser_manager = get_camoufox_manager()
+        browser_manager = get_session_browser_manager()
         page = await browser_manager.get_current_page()
 
         if not page:
@@ -436,10 +435,9 @@ async def find_links(
 
         filtered_count = len(links)
 
-        emit_info(
-            f"[green]Found {filtered_count} links"
-            + (f" containing '{text_filter}'" if text_filter else "")
-            + "[/green]",
+        emit_success(
+            f"Found {filtered_count} links"
+            + (f" containing '{text_filter}'" if text_filter else ""),
             message_group=group_id,
         )