code-puppy 0.0.287__py3-none-any.whl → 0.0.323__py3-none-any.whl

code_puppy/tools/browser/browser_scripts.py

@@ -4,7 +4,7 @@ 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
 from code_puppy.tools.common import generate_group_id
 
 from .camoufox_manager import get_camoufox_manager
@@ -17,7 +17,7 @@ async def execute_javascript(
     """Execute JavaScript code in the browser context."""
     group_id = generate_group_id("browser_execute_js", script[:100])
     emit_info(
-        f"[bold white on blue] BROWSER EXECUTE JS [/bold white on blue] 📜 script='{script[:100]}{'...' if len(script) > 100 else ''}'",
+        f"BROWSER EXECUTE JS 📜 script='{script[:100]}{'...' if len(script) > 100 else ''}'",
         message_group=group_id,
     )
     try:
@@ -30,16 +30,12 @@ async def execute_javascript(
         # Execute JavaScript
         result = await page.evaluate(script, timeout=timeout)
 
-        emit_info(
-            "[green]JavaScript executed successfully[/green]", message_group=group_id
-        )
+        emit_success("JavaScript executed successfully", message_group=group_id)
 
         return {"success": True, "script": script, "result": result}
 
     except Exception as e:
-        emit_info(
-            f"[red]JavaScript execution failed: {str(e)}[/red]", message_group=group_id
-        )
+        emit_error(f"JavaScript execution failed: {str(e)}", message_group=group_id)
         return {"success": False, "error": str(e), "script": script}
 
 
@@ -52,7 +48,7 @@ async def scroll_page(
     target = element_selector or "page"
     group_id = generate_group_id("browser_scroll", f"{direction}_{amount}_{target}")
     emit_info(
-        f"[bold white on blue] BROWSER SCROLL [/bold white on blue] 📋 direction={direction} amount={amount} target='{target}'",
+        f"BROWSER SCROLL 📋 direction={direction} amount={amount} target='{target}'",
         message_group=group_id,
     )
     try:
@@ -120,9 +116,7 @@ async def scroll_page(
             })
         """)
 
-        emit_info(
-            f"[green]Scrolled {target} {direction}[/green]", message_group=group_id
-        )
+        emit_success(f"Scrolled {target} {direction}", message_group=group_id)
 
         return {
             "success": True,
@@ -148,7 +142,7 @@ async def scroll_to_element(
     """Scroll to bring an element into view."""
     group_id = generate_group_id("browser_scroll_to_element", selector[:100])
     emit_info(
-        f"[bold white on blue] BROWSER SCROLL TO ELEMENT [/bold white on blue] 🎯 selector='{selector}'",
+        f"BROWSER SCROLL TO ELEMENT 🎯 selector='{selector}'",
         message_group=group_id,
     )
     try:
@@ -165,9 +159,7 @@ async def scroll_to_element(
         # Check if element is now visible
         is_visible = await element.is_visible()
 
-        emit_info(
-            f"[green]Scrolled to element: {selector}[/green]", message_group=group_id
-        )
+        emit_success(f"Scrolled to element: {selector}", message_group=group_id)
 
         return {"success": True, "selector": selector, "visible": is_visible}
 
@@ -182,7 +174,7 @@ async def set_viewport_size(
     """Set the viewport size."""
     group_id = generate_group_id("browser_set_viewport", f"{width}x{height}")
     emit_info(
-        f"[bold white on blue] BROWSER SET VIEWPORT [/bold white on blue] 🖥️ size={width}x{height}",
+        f"BROWSER SET VIEWPORT 🖥️ size={width}x{height}",
         message_group=group_id,
     )
     try:
@@ -194,8 +186,8 @@ async def set_viewport_size(
 
         await page.set_viewport_size({"width": width, "height": height})
 
-        emit_info(
-            f"[green]Set viewport size to {width}x{height}[/green]",
+        emit_success(
+            f"Set viewport size to {width}x{height}",
             message_group=group_id,
         )
 
@@ -213,7 +205,7 @@ async def wait_for_element(
     """Wait for an element to reach a specific state."""
     group_id = generate_group_id("browser_wait_for_element", f"{selector[:50]}_{state}")
     emit_info(
-        f"[bold white on blue] BROWSER WAIT FOR ELEMENT [/bold white on blue] ⏱️ selector='{selector}' state={state} timeout={timeout}ms",
+        f"BROWSER WAIT FOR ELEMENT ⏱️ selector='{selector}' state={state} timeout={timeout}ms",
         message_group=group_id,
     )
     try:
@@ -226,9 +218,7 @@ async def wait_for_element(
         element = page.locator(selector)
         await element.wait_for(state=state, timeout=timeout)
 
-        emit_info(
-            f"[green]Element {selector} is now {state}[/green]", message_group=group_id
-        )
+        emit_success(f"Element {selector} is now {state}", message_group=group_id)
 
         return {"success": True, "selector": selector, "state": state}
 
@@ -246,7 +236,7 @@ async def highlight_element(
         "browser_highlight_element", f"{selector[:50]}_{color}"
     )
     emit_info(
-        f"[bold white on blue] BROWSER HIGHLIGHT ELEMENT [/bold white on blue] 🔦 selector='{selector}' color={color}",
+        f"BROWSER HIGHLIGHT ELEMENT 🔦 selector='{selector}' color={color}",
         message_group=group_id,
     )
     try:
@@ -271,9 +261,7 @@ async def highlight_element(
 
         await element.evaluate(highlight_script)
 
-        emit_info(
-            f"[green]Highlighted element: {selector}[/green]", message_group=group_id
-        )
+        emit_success(f"Highlighted element: {selector}", message_group=group_id)
 
         return {"success": True, "selector": selector, "color": color}
 
@@ -285,7 +273,7 @@ async def clear_highlights() -> Dict[str, Any]:
     """Clear all element highlights."""
     group_id = generate_group_id("browser_clear_highlights")
     emit_info(
-        "[bold white on blue] BROWSER CLEAR HIGHLIGHTS [/bold white on blue] 🧹",
+        "BROWSER CLEAR HIGHLIGHTS 🧹",
         message_group=group_id,
     )
     try:
@@ -311,7 +299,7 @@ async def clear_highlights() -> Dict[str, Any]:
 
         count = await page.evaluate(clear_script)
 
-        emit_info(f"[green]Cleared {count} highlights[/green]", message_group=group_id)
+        emit_success(f"Cleared {count} highlights", message_group=group_id)
 
         return {"success": True, "cleared_count": count}
 

code_puppy/tools/browser/browser_workflows.py

@@ -5,15 +5,16 @@ from typing import Any, Dict
 
 from pydantic_ai import RunContext
 
-from code_puppy.messaging import emit_info
+from code_puppy import config
+from code_puppy.messaging import emit_error, emit_info, emit_success, emit_warning
 from code_puppy.tools.common import generate_group_id
 
 
 def get_workflows_directory() -> Path:
-    """Get the browser workflows directory, creating it if it doesn't exist."""
-    home_dir = Path.home()
-    workflows_dir = home_dir / ".code_puppy" / "browser_workflows"
-    workflows_dir.mkdir(parents=True, exist_ok=True)
+    """Get the browser workflows directory, creating it if it doesn't exist (uses XDG_DATA_HOME)."""
+    data_dir = Path(config.DATA_DIR)
+    workflows_dir = data_dir / "browser_workflows"
+    workflows_dir.mkdir(parents=True, exist_ok=True, mode=0o700)
     return workflows_dir
 
 
@@ -21,7 +22,7 @@ async def save_workflow(name: str, content: str) -> Dict[str, Any]:
     """Save a browser workflow as a markdown file."""
     group_id = generate_group_id("save_workflow", name)
     emit_info(
-        f"[bold white on blue] SAVE WORKFLOW [/bold white on blue] 💾 name='{name}'",
+        f"SAVE WORKFLOW 💾 name='{name}'",
         message_group=group_id,
     )
 
@@ -61,8 +62,8 @@ async def save_workflow(name: str, content: str) -> Dict[str, Any]:
         with open(workflow_path, "w", encoding="utf-8") as f:
             f.write(content)
 
-        emit_info(
-            f"[green]✅ Workflow saved successfully: {workflow_path}[/green]",
+        emit_success(
+            f"Workflow saved successfully: {workflow_path}",
             message_group=group_id,
         )
 
@@ -74,8 +75,8 @@ async def save_workflow(name: str, content: str) -> Dict[str, Any]:
         }
 
     except Exception as e:
-        emit_info(
-            f"[red]❌ Failed to save workflow: {e}[/red]",
+        emit_error(
+            f"Failed to save workflow: {e}",
             message_group=group_id,
         )
         return {"success": False, "error": str(e), "name": name}
@@ -85,7 +86,7 @@ async def list_workflows() -> Dict[str, Any]:
     """List all available browser workflows."""
     group_id = generate_group_id("list_workflows")
     emit_info(
-        "[bold white on blue] LIST WORKFLOWS [/bold white on blue] 📋",
+        "LIST WORKFLOWS 📋",
         message_group=group_id,
     )
 
@@ -108,15 +109,13 @@ async def list_workflows() -> Dict[str, Any]:
                     }
                 )
             except Exception as e:
-                emit_info(
-                    f"[yellow]Warning: Could not read {workflow_file}: {e}[/yellow]"
-                )
+                emit_warning(f"Could not read {workflow_file}: {e}")
 
         # Sort by modification time (newest first)
         workflows.sort(key=lambda x: x["modified"], reverse=True)
 
-        emit_info(
-            f"[green]✅ Found {len(workflows)} workflow(s)[/green]",
+        emit_success(
+            f"Found {len(workflows)} workflow(s)",
             message_group=group_id,
         )
 
@@ -128,8 +127,8 @@ async def list_workflows() -> Dict[str, Any]:
         }
 
     except Exception as e:
-        emit_info(
-            f"[red]❌ Failed to list workflows: {e}[/red]",
+        emit_error(
+            f"Failed to list workflows: {e}",
             message_group=group_id,
         )
         return {"success": False, "error": str(e)}
@@ -139,7 +138,7 @@ async def read_workflow(name: str) -> Dict[str, Any]:
     """Read a saved browser workflow."""
     group_id = generate_group_id("read_workflow", name)
     emit_info(
-        f"[bold white on blue] READ WORKFLOW [/bold white on blue] 📖 name='{name}'",
+        f"READ WORKFLOW 📖 name='{name}'",
         message_group=group_id,
     )
 
@@ -153,8 +152,8 @@ async def read_workflow(name: str) -> Dict[str, Any]:
         workflow_path = workflows_dir / name
 
         if not workflow_path.exists():
-            emit_info(
-                f"[red]❌ Workflow not found: {name}[/red]",
+            emit_error(
+                f"Workflow not found: {name}",
                 message_group=group_id,
             )
             return {
@@ -167,8 +166,8 @@ async def read_workflow(name: str) -> Dict[str, Any]:
         with open(workflow_path, "r", encoding="utf-8") as f:
             content = f.read()
 
-        emit_info(
-            f"[green]✅ Workflow read successfully: {len(content)} characters[/green]",
+        emit_success(
+            f"Workflow read successfully: {len(content)} characters",
             message_group=group_id,
         )
 
@@ -181,8 +180,8 @@ async def read_workflow(name: str) -> Dict[str, Any]:
         }
 
     except Exception as e:
-        emit_info(
-            f"[red]❌ Failed to read workflow: {e}[/red]",
+        emit_error(
+            f"Failed to read workflow: {e}",
             message_group=group_id,
         )
         return {"success": False, "error": str(e), "name": name}

code_puppy/tools/browser/camoufox_manager.py

@@ -6,7 +6,8 @@ from typing import Optional
 
 from playwright.async_api import Browser, BrowserContext, Page
 
-from code_puppy.messaging import emit_info
+from code_puppy import config
+from code_puppy.messaging import emit_info, emit_success, emit_warning
 
 
 class CamoufoxManager:
@@ -48,13 +49,14 @@ class CamoufoxManager:
         return cls._instance
 
     def _get_profile_directory(self) -> Path:
-        """Get or create the persistent profile directory.
+        """Get or create the persistent profile directory (uses XDG_CACHE_HOME).
 
-        Returns a Path object pointing to ~/.code_puppy/camoufox_profile
+        Returns a Path object pointing to XDG_CACHE_HOME/code_puppy/camoufox_profile
         where browser data (cookies, history, bookmarks, etc.) will be stored.
         """
-        profile_path = Path.home() / ".code_puppy" / "camoufox_profile"
-        profile_path.mkdir(parents=True, exist_ok=True)
+        cache_dir = Path(config.CACHE_DIR)
+        profile_path = cache_dir / "camoufox_profile"
+        profile_path.mkdir(parents=True, exist_ok=True, mode=0o700)
         return profile_path
 
     async def async_initialize(self) -> None:
@@ -63,7 +65,7 @@ class CamoufoxManager:
             return
 
         try:
-            emit_info("[yellow]Initializing Camoufox (privacy Firefox)...[/yellow]")
+            emit_info("Initializing Camoufox (privacy Firefox)...")
 
             # Ensure Camoufox binary and dependencies are fetched before launching
             await self._prefetch_camoufox()
@@ -80,7 +82,7 @@ class CamoufoxManager:
 
     async def _initialize_camoufox(self) -> None:
         """Try to start Camoufox with the configured privacy settings."""
-        emit_info(f"[cyan]📁 Using persistent profile: {self.profile_dir}[/cyan]")
+        emit_info(f"Using persistent profile: {self.profile_dir}")
         # Lazy import camoufox to avoid triggering heavy optional deps at import time
         try:
             import camoufox
@@ -103,8 +105,8 @@ class CamoufoxManager:
         except Exception:
             from playwright.async_api import async_playwright
 
-            emit_info(
-                "[yellow]Camoufox no disponible. Usando Playwright (Chromium) como alternativa.[/yellow]"
+            emit_warning(
+                "Camoufox no disponible. Usando Playwright (Chromium) como alternativa."
             )
             pw = await async_playwright().start()
             # Use persistent context directory for Chromium to emulate previous behavior
@@ -142,9 +144,7 @@ class CamoufoxManager:
 
     async def _prefetch_camoufox(self) -> None:
         """Prefetch Camoufox binary and dependencies."""
-        emit_info(
-            "[cyan]🔍 Ensuring Camoufox binary and dependencies are up-to-date...[/cyan]"
-        )
+        emit_info("Ensuring Camoufox binary and dependencies are up-to-date...")
 
         # Lazy import camoufox utilities to avoid side effects during module import
         try:
@@ -152,20 +152,20 @@ class CamoufoxManager:
             from camoufox.locale import ALLOW_GEOIP, download_mmdb
             from camoufox.pkgman import CamoufoxFetcher, camoufox_path
         except Exception:
-            emit_info(
-                "[yellow]Camoufox no disponible. Omitiendo prefetch y preparándose para usar Playwright.[/yellow]"
+            emit_warning(
+                "Camoufox no disponible. Omitiendo prefetch y preparándose para usar Playwright."
             )
             return
 
         needs_install = False
         try:
             camoufox_path(download_if_missing=False)
-            emit_info("[cyan]🗃️ Using cached Camoufox installation[/cyan]")
+            emit_info("Using cached Camoufox installation")
         except (CamoufoxNotInstalled, FileNotFoundError):
-            emit_info("[cyan]📥 Camoufox not found, installing fresh copy[/cyan]")
+            emit_info("Camoufox not found, installing fresh copy")
             needs_install = True
         except UnsupportedVersion:
-            emit_info("[cyan]♻️ Camoufox update required, reinstalling[/cyan]")
+            emit_info("Camoufox update required, reinstalling")
             needs_install = True
 
         if needs_install:
@@ -175,7 +175,7 @@ class CamoufoxManager:
         if ALLOW_GEOIP:
             download_mmdb()
 
-        emit_info("[cyan]📦 Camoufox dependencies ready[/cyan]")
+        emit_info("Camoufox dependencies ready")
 
     async def close_page(self, page: Page) -> None:
         """Close a specific page."""
@@ -195,13 +195,9 @@ class CamoufoxManager:
                 try:
                     storage_state_path = self.profile_dir / "storage_state.json"
                     await self._context.storage_state(path=str(storage_state_path))
-                    emit_info(
-                        f"[green]💾 Browser state saved to {storage_state_path}[/green]"
-                    )
+                    emit_success(f"Browser state saved to {storage_state_path}")
                 except Exception as e:
-                    emit_info(
-                        f"[yellow]Warning: Could not save storage state: {e}[/yellow]"
-                    )
+                    emit_warning(f"Could not save storage state: {e}")
 
                 await self._context.close()
                 self._context = None
@@ -210,12 +206,12 @@ class CamoufoxManager:
                 self._browser = None
             self._initialized = False
         except Exception as e:
-            emit_info(f"[yellow]Warning during cleanup: {e}[/yellow]")
+            emit_warning(f"Warning during cleanup: {e}")
 
     async def close(self) -> None:
         """Close the browser and clean up resources."""
         await self._cleanup()
-        emit_info("[yellow]Camoufox browser closed[/yellow]")
+        emit_info("Camoufox browser closed")
 
     def __del__(self):
         """Ensure cleanup on object destruction."""