""" DeepSeek Web Adapter 控制面板 提供 Web UI 管理：配置、Worker 注册、浏览器控制、日志查看 """ import asyncio import json import logging import threading from pathlib import Path from typing import Optional from flask import Flask, render_template, request, jsonify from flask_socketio import SocketIO from .config import config, DeepSeekWebConfig, CONFIG_FILE logger = logging.getLogger(__name__) TEMPLATE_DIR = str(Path(__file__).parent.parent / "templates") class AdminServer: """控制面板 HTTP 服务""" def __init__(self, adapter=None): self.adapter = adapter self.app = Flask(__name__, template_folder=TEMPLATE_DIR, static_folder=None) self.socketio = SocketIO(self.app, cors_allowed_origins="*", async_mode="threading") self._adapter_loop = None self._setup_routes() def _setup_routes(self): @self.app.route("/") def index(): return render_template("admin.html") @self.app.route("/api/config", methods=["GET"]) def get_config(): from dataclasses import asdict return jsonify(asdict(config)) @self.app.route("/api/config", methods=["POST"]) def save_config(): try: data = request.json for key, value in data.items(): if hasattr(config, key): setattr(config, key, value) config.save() return jsonify({"success": True}) except Exception as e: return jsonify({"success": False, "error": str(e)}) @self.app.route("/api/register", methods=["POST"]) def register_worker(): try: data = request.json import urllib.request import ssl base_url = config.array_base_url.rstrip("/") url = f"{base_url}/api/v1/workers/register" reg_data = { "name": data.get("name", "DeepSeek-Web"), "type": data.get("type", "ai"), "description": data.get("description", ""), "invite_code": data.get("invite_code") or None, "capabilities": config.profile.get("capabilities", {}), "organization": "Array", "timezone": "Asia/Shanghai", "tags": ["AI", "DeepSeek", "Agent", "Web"], "max_concurrent_tasks": 1, } req = urllib.request.Request( url, data=json.dumps(reg_data).encode("utf-8"), headers={"Content-Type": "application/json"}, method="POST", ) ctx = ssl.create_default_context() ctx.check_hostname = False ctx.verify_mode = ssl.CERT_NONE try: with urllib.request.urlopen(req, context=ctx, timeout=10) as resp: result = json.loads(resp.read().decode("utf-8")) except urllib.error.HTTPError as e: error_body = e.read().decode("utf-8", errors="replace") return jsonify({"error": f"HTTP {e.code}: {error_body}"}) worker_id = result.get("worker", {}).get("id", "") api_key = result.get("api_key", "") if api_key: config.api_key = api_key config.worker_id = worker_id config.worker_name = data.get("name", config.worker_name) config.save() return jsonify({"worker_id": worker_id, "api_key": api_key}) except Exception as e: return jsonify({"error": str(e)}) @self.app.route("/api/status") def get_status(): if self.adapter: status = self.adapter.get_status() status["worker_name"] = config.worker_name return jsonify(status) return jsonify({ "connected": False, "deepseek_ready": False, "processing": False, "current_task": None, "worker_id": None, "worker_name": config.worker_name, }) @self.app.route("/api/browser/start", methods=["POST"]) def start_browser(): if not self.adapter or not self._adapter_loop: return jsonify({"success": False, "error": "Not ready"}) future = asyncio.run_coroutine_threadsafe( self.adapter.start_deepseek_session(), self._adapter_loop ) try: result = future.result(timeout=30) return jsonify({"success": result}) except Exception as e: return jsonify({"success": False, "error": str(e)}) @self.app.route("/api/browser/stop", methods=["POST"]) def stop_browser(): if not self.adapter or not self._adapter_loop: return jsonify({"success": False, "error": "Not ready"}) future = asyncio.run_coroutine_threadsafe( self.adapter.stop_deepseek_session(), self._adapter_loop ) try: future.result(timeout=10) return jsonify({"success": True}) except Exception as e: return jsonify({"success": False, "error": str(e)}) @self.app.route("/api/browser/reinit", methods=["POST"]) def reinit_prompt(): if not self.adapter or not self.adapter.session or not self._adapter_loop: return jsonify({"success": False, "error": "DeepSeek session not started"}) from .deepseek_agent_adapter import ADAPTER_SYSTEM_PROMPT async def _reinit(): await self.adapter.session.controller.send( f"[系统重新初始化]\n\n{ADAPTER_SYSTEM_PROMPT}" ) return True future = asyncio.run_coroutine_threadsafe(_reinit(), self._adapter_loop) try: future.result(timeout=15) return jsonify({"success": True}) except Exception as e: return jsonify({"success": False, "error": str(e)}) @self.app.route("/api/logs/deepseek") def get_deepseek_logs(): if self.adapter and self.adapter.session: return jsonify({"logs": self.adapter.session.get_logs(100)}) return jsonify({"logs": None}) @self.app.route("/api/logs/deepseek", methods=["DELETE"]) def clear_deepseek_logs(): if self.adapter and self.adapter.session: self.adapter.session._log_buffer.clear() return jsonify({"success": True}) @self.app.route("/api/logs/system") def get_system_logs(): log_file = Path("adapter_debug.log") if log_file.exists(): try: content = log_file.read_text(encoding="utf-8", errors="replace") lines = content.split("\n") return jsonify({"logs": "\n".join(lines[-200:])}) except: pass return jsonify({"logs": None}) def set_adapter(self, adapter, loop): self.adapter = adapter self._adapter_loop = loop def run(self, host="0.0.0.0", port=None): if port is None: port = config.admin_port logger.info(f"控制面板启动: http://{host}:{port}") self.socketio.run(self.app, host=host, port=port, debug=False, allow_unsafe_werkzeug=True)