1"""
2Browser Utilities for NotebookLM Skill
3Handles browser launching, stealth features, and common interactions
4"""
5 
6import json
7import time
8import random
9from typing import Optional, List
10 
11from patchright.sync_api import Playwright, BrowserContext, Page
12from config import BROWSER_PROFILE_DIR, STATE_FILE, BROWSER_ARGS, USER_AGENT
13 
14 
15class BrowserFactory:
16    """Factory for creating configured browser contexts"""
17 
18    @staticmethod
19    def launch_persistent_context(
20        playwright: Playwright,
21        headless: bool = True,
22        user_data_dir: str = str(BROWSER_PROFILE_DIR)
23    ) -> BrowserContext:
24        """
25        Launch a persistent browser context with anti-detection features
26        and cookie workaround.
27        """
28        # Launch persistent context
29        context = playwright.chromium.launch_persistent_context(
30            user_data_dir=user_data_dir,
31            channel="chrome",  # Use real Chrome
32            headless=headless,
33            no_viewport=True,
34            ignore_default_args=["--enable-automation"],
35            user_agent=USER_AGENT,
36            args=BROWSER_ARGS
37        )
38 
39        # Cookie Workaround for Playwright bug #36139
40        # Session cookies (expires=-1) don't persist in user_data_dir automatically
41        BrowserFactory._inject_cookies(context)
42 
43        return context
44 
45    @staticmethod
46    def _inject_cookies(context: BrowserContext):
47        """Inject cookies from state.json if available"""
48        if STATE_FILE.exists():
49            try:
50                with open(STATE_FILE, 'r') as f:
51                    state = json.load(f)
52                    if 'cookies' in state and len(state['cookies']) > 0:
53                        context.add_cookies(state['cookies'])
54                        # print(f"  🔧 Injected {len(state['cookies'])} cookies from state.json")
55            except Exception as e:
56                print(f"  ⚠️  Could not load state.json: {e}")
57 
58 
59class StealthUtils:
60    """Human-like interaction utilities"""
61 
62    @staticmethod
63    def random_delay(min_ms: int = 100, max_ms: int = 500):
64        """Add random delay"""
65        time.sleep(random.uniform(min_ms / 1000, max_ms / 1000))
66 
67    @staticmethod
68    def human_type(page: Page, selector: str, text: str, wpm_min: int = 320, wpm_max: int = 480):
69        """Type with human-like speed"""
70        element = page.query_selector(selector)
71        if not element:
72            # Try waiting if not immediately found
73            try:
74                element = page.wait_for_selector(selector, timeout=2000)
75            except:
76                pass
77        
78        if not element:
79            print(f"⚠️ Element not found for typing: {selector}")
80            return
81 
82        # Click to focus
83        element.click()
84        
85        # Type
86        for char in text:
87            element.type(char, delay=random.uniform(25, 75))
88            if random.random() < 0.05:
89                time.sleep(random.uniform(0.15, 0.4))
90 
91    @staticmethod
92    def realistic_click(page: Page, selector: str):
93        """Click with realistic movement"""
94        element = page.query_selector(selector)
95        if not element:
96            return
97 
98        # Optional: Move mouse to element (simplified)
99        box = element.bounding_box()
100        if box:
101            x = box['x'] + box['width'] / 2
102            y = box['y'] + box['height'] / 2
103            page.mouse.move(x, y, steps=5)
104 
105        StealthUtils.random_delay(100, 300)
106        element.click()
107        StealthUtils.random_delay(100, 300)
108