refactor: remove ~780 lines of dead code

Author: NikkeTryHard

api_utils/auth_manager.py

@@ -78,11 +78,6 @@ def mark_profile_failed(self, profile_path: Optional[str] = None) -> None:
                 "Attempted to mark profile failed but no profile provided or active."
             )
 
-    def reset_failures(self) -> None:
-        """Reset the failure tracking."""
-        self.failed_profiles.clear()
-        logger.info("Auth profile failure tracking reset.")
-
 
 # Global instance
 auth_manager = AuthManager()

api_utils/client_connection.py

@@ -1,22 +1,14 @@
 import asyncio
 from asyncio import Event, Task
 from logging import Logger
-from typing import Any, Callable, Coroutine, Dict, Protocol, Tuple
+from typing import Any, Callable, Coroutine, Dict, Tuple
 
 from fastapi import HTTPException, Request
 
 from logging_utils import set_request_id
 from models import ClientDisconnectedError
 
 
-class SupportsReceive(Protocol):
-    """Protocol for request objects that support _receive method."""
-
-    def _receive(self) -> Coroutine[Any, Any, Dict[str, Any]]:
-        """Internal method to receive messages from ASGI."""
-        ...
-
-
 async def check_client_connection(req_id: str, http_request: Request) -> bool:
     """
     Checks if the client is still connected.

api_utils/mcp_adapter.py

@@ -55,39 +55,3 @@ async def execute_mcp_tool_with_endpoint(
         except Exception:
             data = {"raw": resp.text}
     return json.dumps(data, ensure_ascii=False)
-
-
-# Synchronous helpers for use inside threads
-def execute_mcp_tool_sync(name: str, params: Dict[str, Any]) -> str:
-    ep = os.environ.get("MCP_HTTP_ENDPOINT")
-    if not ep:
-        raise RuntimeError("MCP_HTTP_ENDPOINT not configured")
-    url = f"{_normalize_endpoint(ep)}/tools/execute"
-    payload = {"name": name, "arguments": params}
-    headers = {"Content-Type": "application/json"}
-    timeout = float(os.environ.get("MCP_HTTP_TIMEOUT", "15"))
-    with httpx.Client(timeout=timeout) as client:
-        resp = client.post(url, json=payload, headers=headers)
-        resp.raise_for_status()
-        try:
-            data = resp.json()
-        except Exception:
-            data = {"raw": resp.text}
-    return json.dumps(data, ensure_ascii=False)
-
-
-def execute_mcp_tool_with_endpoint_sync(
-    endpoint: str, name: str, params: Dict[str, Any]
-) -> str:
-    url = f"{_normalize_endpoint(endpoint)}/tools/execute"
-    payload = {"name": name, "arguments": params}
-    headers = {"Content-Type": "application/json"}
-    timeout = float(os.environ.get("MCP_HTTP_TIMEOUT", "15"))
-    with httpx.Client(timeout=timeout) as client:
-        resp = client.post(url, json=payload, headers=headers)
-        resp.raise_for_status()
-        try:
-            data = resp.json()
-        except Exception:
-            data = {"raw": resp.text}
-    return json.dumps(data, ensure_ascii=False)

api_utils/response_generators.py

@@ -50,13 +50,8 @@ async def gen_sse_from_aux_stream(
     full_body_content = ""
     data_receiving = False
 
-    loop_count = 0
-
     try:
         async for raw_data in use_stream_response(req_id):
-            loop_count += 1
-            # logger.debug(f"[{req_id}] gen_sse_from_aux_stream loop iteration #{loop_count}")
-
             data_receiving = True
 
             try:

api_utils/server_state.py

@@ -99,18 +99,6 @@ def clear_debug_logs(self) -> None:
         self.console_logs = []
         self.network_log = {"requests": [], "responses": []}
 
-    def get_server_status(self) -> Dict[str, Any]:
-        """Get current server status as a dictionary."""
-        return {
-            "is_initializing": self.is_initializing,
-            "is_playwright_ready": self.is_playwright_ready,
-            "is_browser_connected": self.is_browser_connected,
-            "is_page_ready": self.is_page_ready,
-            "current_model": self.current_ai_studio_model_id,
-            "queue_size": self.request_queue.qsize() if self.request_queue else 0,
-            "worker_running": bool(self.worker_task and not self.worker_task.done()),
-        }
-
 
 # Global singleton instance
 state = ServerState()

browser_utils/__init__.py

@@ -34,7 +34,6 @@
     save_error_snapshot,
 )
 from .page_controller import PageController
-from .script_manager import ScriptManager, script_manager
 
 __all__ = [
     # 初始化相关
@@ -60,9 +59,6 @@
     "_force_ui_state_settings",
     "_force_ui_state_with_retry",
     "_verify_and_apply_ui_state",
-    # 脚本管理相关
-    "ScriptManager",
-    "script_manager",
     # Page Controller
     "PageController",
     # Debug utilities (comprehensive error snapshots)

browser_utils/operations_modules/interactions.py

@@ -349,7 +349,6 @@ async def _wait_for_response_completion(
     edit_button_locator: Locator,
     req_id: str,
     check_client_disconnected_func: Callable,
-    current_chat_id: Optional[str],
     timeout_ms=RESPONSE_COMPLETION_TIMEOUT,
     initial_wait_ms=INITIAL_WAIT_MS_BEFORE_POLLING,
 ) -> bool:

browser_utils/page_controller_modules/chat.py

@@ -290,11 +290,6 @@ async def _execute_chat_clear(
                 else:
                     raise
 
-            await self._check_disconnect(
-                check_client_disconnected,
-                f"清空聊天 - 消失检查尝试 {attempt_disappear + 1} 后",
-            )
-
     async def _dismiss_backdrops(self):
         """尝试关闭可能残留的 cdk 透明遮罩层以避免点击被拦截。"""
         try:

browser_utils/page_controller_modules/input.py

@@ -1,6 +1,4 @@
 import asyncio
-import base64
-import mimetypes
 from typing import Callable, List
 
 from playwright.async_api import TimeoutError
@@ -16,7 +14,6 @@
 )
 from config.selector_utils import (
     AUTOSIZE_WRAPPER_SELECTORS,
-    DRAG_DROP_TARGET_SELECTORS,
     build_combined_selector,
 )
 from logging_utils import set_request_id
@@ -328,152 +325,6 @@ async def _handle_post_upload_dialog(self):
         except Exception:
             pass
 
-    async def _ensure_files_attached(
-        self, wrapper_locator, expected_min: int = 1, timeout_ms: int = 5000
-    ) -> bool:
-        """轮询检查输入区域内 file input 的 files 是否 >= 期望数量。"""
-        end = asyncio.get_event_loop().time() + (timeout_ms / 1000)
-        while asyncio.get_event_loop().time() < end:
-            try:
-                # NOTE: normalize JS eval string to avoid parser confusion
-                counts = await wrapper_locator.evaluate(
-                    """
-                    (el) => {
-                      const result = {inputs:0, chips:0, blobs:0};
-                      try { el.querySelectorAll('input[type="file"]').forEach(i => { result.inputs += (i.files ? i.files.length : 0); }); } catch(e){}
-                      try { result.chips = el.querySelectorAll('button[aria-label*="Remove" i], button[aria-label*="asset" i]').length; } catch(e){}
-                      try { result.blobs = el.querySelectorAll('img[src^="blob:"], video[src^="blob:"]').length; } catch(e){}
-                      return result;
-                    }
-                    """
-                )
-
-                total = 0
-                if isinstance(counts, dict):
-                    total = max(
-                        int(counts.get("inputs") or 0),
-                        int(counts.get("chips") or 0),
-                        int(counts.get("blobs") or 0),
-                    )
-                if total >= expected_min:
-                    self.logger.info(
-                        f" 已检测到已附加文件: inputs={counts.get('inputs')}, chips={counts.get('chips')}, blobs={counts.get('blobs')} (>= {expected_min})"
-                    )
-                    return True
-            except asyncio.CancelledError:
-                raise
-            except Exception:
-                pass
-            await asyncio.sleep(0.2)
-        self.logger.warning(f" 未能在超时内检测到已附加文件 (期望 >= {expected_min})")
-        return False
-
-    async def _simulate_drag_drop_files(
-        self, target_locator, files_list: List[str]
-    ) -> None:
-        """将本地文件以拖放事件的方式注入到目标元素。
-        仅负责触发 dragenter/dragover/drop，不在此处做附加验证以节省时间。
-        """
-        payloads = []
-        for path in files_list:
-            try:
-                with open(path, "rb") as f:
-                    raw = f.read()
-                b64 = base64.b64encode(raw).decode("ascii")
-                mime, _ = mimetypes.guess_type(path)
-                payloads.append(
-                    {
-                        "name": path.split("/")[-1],
-                        "mime": mime or "application/octet-stream",
-                        "b64": b64,
-                    }
-                )
-            except Exception as e:
-                self.logger.warning(f" 读取文件失败，跳过拖放: {path} - {e}")
-
-        if not payloads:
-            raise Exception("无可用文件用于拖放")
-
-        # 使用集中管理的选择器列表作为拖放候选目标
-        candidates = [target_locator, self.page.locator(PROMPT_TEXTAREA_SELECTOR)]
-        candidates.extend(
-            [self.page.locator(sel) for sel in DRAG_DROP_TARGET_SELECTORS]
-        )
-
-        last_err = None
-        for idx, cand in enumerate(candidates):
-            try:
-                await expect_async(cand).to_be_visible(timeout=3000)
-                await cand.evaluate(
-                    """
-                    (el, files) => {
-                      const dt = new DataTransfer();
-                      for (const p of files) {
-                        const bstr = atob(p.b64);
-                        const len = bstr.length;
-                        const u8 = new Uint8Array(len);
-                        for (let i = 0; i < len; i++) u8[i] = bstr.charCodeAt(i);
-                        const blob = new Blob([u8], { type: p.mime || 'application/octet-stream' });
-                        const file = new File([blob], p.name, { type: p.mime || 'application/octet-stream' });
-                        dt.items.add(file);
-                      }
-                      const evEnter = new DragEvent('dragenter', { bubbles: true, cancelable: true, dataTransfer: dt });
-                      el.dispatchEvent(evEnter);
-                      const evOver = new DragEvent('dragover', { bubbles: true, cancelable: true, dataTransfer: dt });
-                      el.dispatchEvent(evOver);
-                      const evDrop = new DragEvent('drop', { bubbles: true, cancelable: true, dataTransfer: dt });
-                      el.dispatchEvent(evDrop);
-                    }
-                    """,
-                    payloads,
-                )
-                await asyncio.sleep(0.5)
-                self.logger.info(
-                    f" 拖放事件已在候选目标 {idx + 1}/{len(candidates)} 上触发。"
-                )
-                return
-            except asyncio.CancelledError:
-                raise
-            except Exception as e_try:
-                last_err = e_try
-                continue
-
-        # 兜底：在 document.body 上尝试一次
-        try:
-            await self.page.evaluate(
-                """
-                (files) => {
-                  const dt = new DataTransfer();
-                  for (const p of files) {
-                    const bstr = atob(p.b64);
-                    const len = bstr.length;
-                    const u8 = new Uint8Array(len);
-                    for (let i = 0; i < len; i++) u8[i] = bstr.charCodeAt(i);
-                    const blob = new Blob([u8], { type: p.mime || 'application/octet-stream' });
-                    const file = new File([blob], p.name, { type: p.mime || 'application/octet-stream' });
-                    dt.items.add(file);
-                  }
-                  const el = document.body;
-                  const evEnter = new DragEvent('dragenter', { bubbles: true, cancelable: true, dataTransfer: dt });
-                  el.dispatchEvent(evEnter);
-                  const evOver = new DragEvent('dragover', { bubbles: true, cancelable: true, dataTransfer: dt });
-                  el.dispatchEvent(evOver);
-                  const evDrop = new DragEvent('drop', { bubbles: true, cancelable: true, dataTransfer: dt });
-                  el.dispatchEvent(evDrop);
-                }
-                """,
-                payloads,
-            )
-            await asyncio.sleep(0.5)
-            self.logger.info(" 拖放事件已在 document.body 上触发（兜底）。")
-            return
-        except asyncio.CancelledError:
-            raise
-        except Exception:
-            pass
-
-        raise last_err or Exception("拖放未能在任何候选目标上触发")
-
     async def _try_enter_submit(
         self, prompt_textarea_locator, check_client_disconnected: Callable
     ) -> bool:

browser_utils/page_controller_modules/response.py

@@ -59,7 +59,6 @@ async def get_response(
                 edit_button_locator,
                 self.req_id,
                 check_client_disconnected,
-                None,
             )
 
             if not completion_detected: