@minded-ai/mindedjs 2.0.22-beta.3 → 2.0.23

package/src/browserTask/cookieStore.py

@@ -1,135 +0,0 @@
-"""
-Cookie management for browser sessions.
-
-This module provides cookie persistence across browser task executions,
-enabling features like staying logged in and maintaining session state.
-"""
-
-import json
-import os
-import time
-from pathlib import Path
-from typing import List, Optional, TypedDict
-
-from browser_use.browser import BrowserSession
-
-
-class StoredCookie(TypedDict, total=False):
-  name: str
-  value: str
-  domain: str
-  path: str
-  expires: float  # seconds since epoch
-  secure: bool
-  httpOnly: bool
-  sameSite: str  # "Strict" | "Lax" | "None"
-
-
-class CookieStore:
-  """Manages persistent cookie storage for browser sessions"""
-  
-  def __init__(self, base_dir: str):
-    self.base_dir = Path(base_dir)
-    
-  def _file_for(self) -> Path:
-    return self.base_dir / "last-session.json"
-  
-  async def save(self, cookies: List[StoredCookie]) -> None:
-    """Save cookies to disk, filtering expired ones"""
-    os.makedirs(self.base_dir, exist_ok=True)
-    file_path = self._file_for()
-    tmp_path = Path(f"{file_path}.tmp.{os.getpid()}")
-    
-    # Filter out expired cookies
-    now = time.time()
-    to_persist = [
-      c for c in cookies 
-      if 'expires' not in c or c.get('expires', 0) > now
-    ]
-    
-    # Write to temp file then rename for atomic write
-    try:
-      with open(tmp_path, 'w') as f:
-        f.write(json.dumps({
-          'version': 1,
-          'cookies': to_persist
-        }, indent=2))
-      
-      # Atomic rename
-      os.rename(tmp_path, file_path)
-    finally:
-      # Clean up temp file if it still exists
-      if tmp_path.exists():
-        tmp_path.unlink()
-  
-  async def load(self) -> Optional[List[StoredCookie]]:
-    """Load cookies from disk"""
-    file_path = self._file_for()
-    
-    if not file_path.exists():
-      return None
-    
-    try:
-      with open(file_path, 'r') as f:
-        content = f.read()
-        parsed = json.loads(content)
-        
-        if not parsed or not isinstance(parsed.get('cookies'), list):
-          return None
-        
-        return parsed['cookies']
-    except Exception as e:
-      print(f"Failed to load cookies: {e}")
-      return None
-
-
-async def load_cookies_for_session(browser_session: BrowserSession, cookie_store: CookieStore) -> None:
-  """Load cookies from store and set them in the browser via Playwright"""
-  stored_cookies = await cookie_store.load()
-  if not stored_cookies or len(stored_cookies) == 0:
-    print("No stored cookies to load")
-    return
-  
-  try:
-    # Ensure browser is started by getting the current page
-    # This initializes the browser_context if needed
-    await browser_session.get_current_page()
-    
-    # Access browser_context after ensuring it's initialized
-    if not hasattr(browser_session, 'browser_context') or not browser_session.browser_context:
-      print("No browser context available to set cookies")
-      return
-    
-    context = browser_session.browser_context
-    
-    # Add cookies to the browser context
-    # Playwright's add_cookies expects a list of cookies
-    await context.add_cookies(stored_cookies)
-    
-    print(f"Loaded {len(stored_cookies)} cookies into browser session")
-  except Exception as e:
-    print(f"Failed to load cookies: {e}")
-
-
-async def save_cookies_for_session(browser_session: BrowserSession, cookie_store: CookieStore) -> None:
-  """Get cookies from browser and save them to the store"""
-  try:
-    # Ensure browser_context is available
-    if not hasattr(browser_session, 'browser_context') or not browser_session.browser_context:
-      print("No browser context available to get cookies")
-      return
-    
-    # Get the browser context from the session
-    context = browser_session.browser_context
-    
-    # Get all cookies from the browser context
-    cookies = await context.cookies()
-    
-    if cookies:
-      await cookie_store.save(cookies)
-      print(f"Saved {len(cookies)} cookies from browser session")
-    else:
-      print("No cookies found in browser session")
-  except Exception as e:
-    print(f"Failed to save cookies: {e}")
-