@agentunion/kite 1.5.0 → 1.6.0

package/extensions/services/testUi/entry.py

@@ -0,0 +1,446 @@
+"""
+testUi 模块入口
+演示 Evol 页面插件化功能
+"""
+import asyncio
+import json
+import os
+import sys
+import threading
+import traceback
+import uuid
+from datetime import datetime, timezone
+
+import websockets
+
+MODULE_NAME = "testUi"
+
+# ── 弹性多连接 ──
+_extra_ws: dict = {}           # slot → WebSocket
+_extra_ws_tasks: dict = {}     # slot → recv loop Task
+_kernel_port = ""              # 缓存 kernel_port
+
+# ── 日志 & 崩溃报告基础设施 ──
+
+_log_lock = threading.Lock()
+_log_latest_path = None
+_log_daily_path = None
+_log_daily_date = ""
+_log_dir = None
+_crash_log_path = None
+
+def _resolve_daily_log_path():
+    global _log_daily_path, _log_daily_date
+    if not _log_dir: return
+    today = datetime.now().strftime("%Y-%m-%d")
+    if today == _log_daily_date and _log_daily_path: return
+    month_dir = os.path.join(_log_dir, today[:7])
+    os.makedirs(month_dir, exist_ok=True)
+    _log_daily_path = os.path.join(month_dir, f"{today}.log")
+    _log_daily_date = today
+
+def _write_log(line: str):
+    with _log_lock:
+        if _log_latest_path:
+            try:
+                with open(_log_latest_path, "a", encoding="utf-8") as f: f.write(line)
+            except Exception: pass
+        _resolve_daily_log_path()
+        if _log_daily_path:
+            try:
+                with open(_log_daily_path, "a", encoding="utf-8") as f: f.write(line)
+            except Exception: pass
+
+def _write_crash(exc_type, exc_value, exc_tb, thread_name=None, severity="critical", handled=False):
+    record = {
+        "timestamp": datetime.now(timezone.utc).isoformat(),
+        "module": MODULE_NAME,
+        "thread": thread_name or threading.current_thread().name,
+        "exception_type": exc_type.__name__ if exc_type else "Unknown",
+        "exception_message": str(exc_value),
+        "traceback": "".join(traceback.format_exception(exc_type, exc_value, exc_tb)),
+        "severity": severity, "handled": handled,
+        "process_id": os.getpid(), "platform": sys.platform,
+    }
+    if exc_tb:
+        entries = traceback.extract_tb(exc_tb)
+        if entries:
+            last = entries[-1]
+            record["context"] = {"function": last.name, "file": os.path.basename(last.filename), "line": last.lineno}
+    line = json.dumps(record, ensure_ascii=False) + "\n"
+    if _crash_log_path:
+        try:
+            with open(_crash_log_path, "a", encoding="utf-8") as f: f.write(line)
+        except Exception: pass
+    if _log_dir:
+        try:
+            today = datetime.now().strftime("%Y-%m-%d")
+            archive_dir = os.path.join(_log_dir, "crashes", today[:7])
+            os.makedirs(archive_dir, exist_ok=True)
+            with open(os.path.join(archive_dir, f"{today}.jsonl"), "a", encoding="utf-8") as f: f.write(line)
+        except Exception: pass
+
+def _setup_exception_hooks():
+    _orig = sys.excepthook
+    def _excepthook(et, ev, tb):
+        _write_crash(et, ev, tb, severity="critical", handled=False)
+        print(f"[{MODULE_NAME}] 崩溃: {et.__name__}")
+        if _crash_log_path: print(f"[{MODULE_NAME}] 崩溃日志: {_crash_log_path}")
+        _orig(et, ev, tb)
+    sys.excepthook = _excepthook
+    if hasattr(threading, "excepthook"):
+        def _thread_excepthook(args):
+            _write_crash(args.exc_type, args.exc_value, args.exc_traceback,
+                        thread_name=args.thread.name if args.thread else "unknown",
+                        severity="error", handled=False)
+        threading.excepthook = _thread_excepthook
+
+async def _run_local_mode(kernel_port: int):
+    global _kernel_port
+    _kernel_port = str(kernel_port)
+
+    # 初始化日志
+    global _log_dir, _log_latest_path, _crash_log_path
+    module_data = os.environ.get("KITE_MODULE_DATA")
+    if module_data:
+        _log_dir = os.path.join(module_data, "log")
+        os.makedirs(_log_dir, exist_ok=True)
+        suffix = os.environ.get("KITE_INSTANCE_SUFFIX", "")
+        _log_latest_path = os.path.join(_log_dir, f"latest{suffix}.log")
+        try:
+            with open(_log_latest_path, "w", encoding="utf-8") as f: pass
+        except Exception: _log_latest_path = None
+        _crash_log_path = os.path.join(_log_dir, f"crashes{suffix}.jsonl")
+        try:
+            with open(_crash_log_path, "w", encoding="utf-8") as f: pass
+        except Exception: _crash_log_path = None
+        _resolve_daily_log_path()
+    _setup_exception_hooks()
+
+    # 从 stdin 读取 token
+    line = sys.stdin.readline().strip()
+    if not line:
+        print(f"[{MODULE_NAME}] ERROR: stdin closed")
+        sys.exit(1)
+
+    msg = json.loads(line)
+    token = msg.get("token", "")
+    if not token:
+        print(f"[{MODULE_NAME}] ERROR: No token")
+        sys.exit(1)
+
+    ws_url = f"ws://127.0.0.1:{kernel_port}/ws?id={MODULE_NAME}"
+
+    print(f"[{MODULE_NAME}] Connecting to Kernel at {ws_url}")
+
+    _has_registered = False
+
+    async with websockets.connect(ws_url) as ws:
+        # 认证
+        await ws.send(json.dumps({
+            "jsonrpc": "2.0",
+            "id": "auth",
+            "method": "auth",
+            "params": {"token": token}
+        }))
+
+        auth_resp = await ws.recv()
+        print(f"[{MODULE_NAME}] Auth response: {auth_resp}")
+
+        # subscribe/register/module.ready 由 _do_init 在收到 system.require_init 后执行
+
+        # 接收消息循环
+        print(f"[{MODULE_NAME}] Running, waiting for messages...")
+        try:
+            async for raw in ws:
+                try:
+                    msg = json.loads(raw)
+                except Exception:
+                    continue
+                # 检测事件
+                params = msg.get("params", {})
+                event = params.get("event", "")
+                data = params.get("data", {})
+
+                # 检测 system.require_init，触发注册流程
+                if event == "system.require_init":
+                    asyncio.create_task(_do_init(ws))
+                    continue
+
+                if event == "system.connection.offer":
+                    asyncio.create_task(_handle_connection_offer(data))
+                    continue
+                if event == "system.connection.release":
+                    asyncio.create_task(_handle_connection_release(data))
+                    continue
+                if event == "module.shutdown" and data.get("module_id") == MODULE_NAME:
+                    print(f"[{MODULE_NAME}] Received module.shutdown, graceful exit")
+                    # 关闭所有附加连接
+                    for slot, ews in list(_extra_ws.items()):
+                        try: await ews.close(code=1000, reason="shutdown")
+                        except Exception: pass
+                    for task in _extra_ws_tasks.values():
+                        task.cancel()
+                    _extra_ws.clear()
+                    _extra_ws_tasks.clear()
+                    # ack
+                    await ws.send(json.dumps({
+                        "jsonrpc": "2.0", "id": str(uuid.uuid4()),
+                        "method": "event.publish",
+                        "params": {"event_id": str(uuid.uuid4()), "event": "module.shutdown.ack",
+                                   "data": {"module_id": MODULE_NAME}}
+                    }))
+                    # exiting
+                    await ws.send(json.dumps({
+                        "jsonrpc": "2.0", "id": str(uuid.uuid4()),
+                        "method": "event.publish",
+                        "params": {"event_id": str(uuid.uuid4()), "event": "module.exiting",
+                                   "data": {"module_id": MODULE_NAME, "action": "none"}}
+                    }))
+                    # shutdown.ready
+                    await ws.send(json.dumps({
+                        "jsonrpc": "2.0", "id": str(uuid.uuid4()),
+                        "method": "event.publish",
+                        "params": {"event_id": str(uuid.uuid4()), "event": "module.shutdown.ready",
+                                   "data": {"module_id": MODULE_NAME}}
+                    }))
+                    # 等待 Kernel 处理完 shutdown 事件后再关闭连接
+                    await asyncio.sleep(0.2)
+                    await ws.close()
+                    break
+        except websockets.ConnectionClosed:
+            print(f"[{MODULE_NAME}] Connection closed")
+        except KeyboardInterrupt:
+            print(f"[{MODULE_NAME}] Shutting down")
+
+
+async def _do_init(ws):
+    """收到 system.require_init 后执行：注册 + 订阅 + module.ready + 页面注册"""
+    global _has_registered
+    try:
+        reason = "startup" if not _has_registered else "recovery"
+
+        # 注册模块
+        await ws.send(json.dumps({
+            "jsonrpc": "2.0",
+            "id": str(uuid.uuid4()),
+            "method": "registry.register",
+            "params": {
+                "module_id": MODULE_NAME,
+                "module_type": "service",
+                "max_connections": 3,
+            }
+        }))
+        reg_resp = await ws.recv()
+        print(f"[{MODULE_NAME}] Registry response: {reg_resp}")
+
+        # 订阅事件
+        await ws.send(json.dumps({
+            "jsonrpc": "2.0",
+            "id": str(uuid.uuid4()),
+            "method": "event.subscribe",
+            "params": {
+                "events": ["module.shutdown"]
+            }
+        }))
+        await ws.recv()
+
+        # 发布 module.ready
+        await ws.send(json.dumps({
+            "jsonrpc": "2.0",
+            "id": str(uuid.uuid4()),
+            "method": "event.publish",
+            "params": {
+                "event_id": str(uuid.uuid4()),
+                "event": "module.ready",
+                "data": {"module_id": MODULE_NAME, "graceful_shutdown": True, "reason": reason}
+            }
+        }))
+        print(f"[{MODULE_NAME}] Module ready (reason={reason})")
+
+        # 注册测试页面
+        pages = [
+            {
+                "page_id": "testUi.particles",
+                "path": "/test-particles",
+                "label": {"zh": "粒子动画", "en": "Particles"},
+                "icon": "Star",
+                "order": 200,
+                "type": "hosted",
+                "requires_auth": True,
+            },
+            {
+                "page_id": "testUi.cards",
+                "path": "/test-cards",
+                "label": {"zh": "3D卡片", "en": "3D Cards"},
+                "icon": "Grid",
+                "order": 201,
+                "type": "hosted",
+                "requires_auth": True,
+            }
+        ]
+
+        for page in pages:
+            await ws.send(json.dumps({
+                "jsonrpc": "2.0",
+                "id": str(uuid.uuid4()),
+                "method": "event.publish",
+                "params": {
+                    "event_id": str(uuid.uuid4()),
+                    "event": "ui.page.register",
+                    "data": page
+                }
+            }))
+            print(f"[{MODULE_NAME}] Registered page: {page['page_id']}")
+        _has_registered = True
+    except Exception as e:
+        print(f"[{MODULE_NAME}] _do_init failed: {e}")
+        _has_registered = False
+
+
+async def _run_remote_mode(gateway_url: str):
+    # 远程日志初始化
+    global _log_dir, _log_latest_path, _crash_log_path
+    home = os.environ.get("HOME") or os.environ.get("USERPROFILE") or os.path.expanduser("~")
+    data_dir = os.path.join(home, ".kite", "remote", MODULE_NAME)
+    os.makedirs(data_dir, exist_ok=True)
+    _log_dir = os.path.join(data_dir, "log")
+    os.makedirs(_log_dir, exist_ok=True)
+    _log_latest_path = os.path.join(_log_dir, "latest.log")
+    try:
+        with open(_log_latest_path, "w", encoding="utf-8") as f: pass
+    except Exception: _log_latest_path = None
+    _crash_log_path = os.path.join(_log_dir, "crashes.jsonl")
+    try:
+        with open(_crash_log_path, "w", encoding="utf-8") as f: pass
+    except Exception: _crash_log_path = None
+    _resolve_daily_log_path()
+    _setup_exception_hooks()
+
+    kite_token = _get_kite_token(MODULE_NAME, gateway_url)
+    print(f"[{MODULE_NAME}] Remote mode: gateway={gateway_url}")
+
+    # TODO: 实现远程模式的 WebSocket 连接到 gateway
+    print(f"[{MODULE_NAME}] Remote mode not yet fully implemented for testUi")
+    print(f"[{MODULE_NAME}] token={kite_token[:8]}...")
+
+    try:
+        while True:
+            await asyncio.sleep(10)
+    except KeyboardInterrupt:
+        print(f"[{MODULE_NAME}] Shutting down")
+    except websockets.ConnectionClosed as e:
+        if e.code in (4001, 4010, 4011):
+            print(f"[{MODULE_NAME}] 认证失败 (code={e.code}), 清除 token 缓存")
+            _clear_token_cache(MODULE_NAME, gateway_url)
+        raise
+
+
+# ── Token management ──
+
+def _gateway_to_filename(gateway_url: str) -> str:
+    try:
+        from urllib.parse import urlparse
+        parsed = urlparse(gateway_url)
+        host = parsed.hostname or "unknown"
+        port = parsed.port or (443 if parsed.scheme == "wss" else 80)
+        return f"{host}-{port}.json".replace(":", "-").replace("/", "-")
+    except Exception: return "default.json"
+
+def _load_token_cache(module_name: str, gateway_url: str) -> dict | None:
+    home = os.environ.get("HOME") or os.environ.get("USERPROFILE") or os.path.expanduser("~")
+    token_file = os.path.join(home, ".kite", "remote", module_name, "tokens", _gateway_to_filename(gateway_url))
+    if not os.path.exists(token_file): return None
+    try:
+        with open(token_file, "r") as f: return json.load(f)
+    except Exception: return None
+
+def _clear_token_cache(module_name: str, gateway_url: str):
+    home = os.environ.get("HOME") or os.environ.get("USERPROFILE") or os.path.expanduser("~")
+    token_file = os.path.join(home, ".kite", "remote", module_name, "tokens", _gateway_to_filename(gateway_url))
+    try: os.remove(token_file)
+    except Exception: pass
+
+def _get_kite_token(module_name: str, gateway_url: str) -> str:
+    token = os.environ.get("KITE_TOKEN")
+    if token: return token
+    cache = _load_token_cache(module_name, gateway_url)
+    if cache: return cache["token"]
+    print(f"[{module_name}] No token for {gateway_url}")
+    print(f"  export KITE_TOKEN=<token>")
+    sys.exit(1)
+
+
+# ── 弹性多连接 handler ──
+
+async def _handle_connection_offer(data):
+    """处理 Kernel 下发的 slot token，建立附加连接。"""
+    slots = data.get("slots", {})
+    for slot_str, info in slots.items():
+        slot = int(slot_str)
+        token = info.get("token", "")
+        if not token or slot in _extra_ws:
+            continue
+        asyncio.create_task(_connect_slot(slot, token))
+
+async def _connect_slot(slot, token):
+    """建立单个 slot 附加连接。"""
+    ws_url = f"ws://127.0.0.1:{_kernel_port}/ws"
+    try:
+        ws = await websockets.connect(ws_url, open_timeout=5, ping_interval=None, close_timeout=5)
+        auth_req = {"jsonrpc": "2.0", "id": f"auth-slot-{slot}", "method": "auth", "params": {"token": token}}
+        await ws.send(json.dumps(auth_req))
+        resp = json.loads(await asyncio.wait_for(ws.recv(), timeout=5))
+        if "error" in resp:
+            await ws.close()
+            return
+        _extra_ws[slot] = ws
+        _extra_ws_tasks[slot] = asyncio.create_task(_slot_recv_loop(slot, ws))
+        print(f"[{MODULE_NAME}] Slot {slot} connected")
+    except Exception as e:
+        print(f"[{MODULE_NAME}] Slot {slot} connect failed: {e}")
+
+async def _slot_recv_loop(slot, ws):
+    """附加连接的接收循环：与主连接平等处理所有消息。"""
+    try:
+        async for raw in ws:
+            try:
+                msg = json.loads(raw)
+                params = msg.get("params", {})
+                event = params.get("event", "")
+                data = params.get("data", {})
+                if event == "module.shutdown" and data.get("module_id") == MODULE_NAME:
+                    print(f"[{MODULE_NAME}] Slot {slot} received shutdown")
+                    # 主连接会处理 shutdown，这里只需要关闭自己
+                    break
+            except Exception as e:
+                print(f"[{MODULE_NAME}] Slot {slot} 消息处理异常: {e}")
+    except Exception:
+        pass
+    finally:
+        _extra_ws.pop(slot, None)
+        _extra_ws_tasks.pop(slot, None)
+
+async def _handle_connection_release(data):
+    """Kernel 请求释放 slot。"""
+    for slot in data.get("slots", []):
+        ws = _extra_ws.pop(slot, None)
+        task = _extra_ws_tasks.pop(slot, None)
+        if ws:
+            try: await ws.close(code=1000, reason="release")
+            except Exception: pass
+        if task:
+            task.cancel()
+
+
+if __name__ == "__main__":
+    kernel_port = os.environ.get("KITE_KERNEL_PORT")
+    if kernel_port:
+        asyncio.run(_run_local_mode(int(kernel_port)))
+    else:
+        gateway_url = os.environ.get("KITE_GATEWAY_URL")
+        if gateway_url:
+            asyncio.run(_run_remote_mode(gateway_url))
+        else:
+            print(f"[{MODULE_NAME}] ERROR: No KITE_KERNEL_PORT and no KITE_GATEWAY_URL")
+            sys.exit(1)

package/extensions/services/testUi/module.md

@@ -0,0 +1,18 @@
+---
+name: testUi
+display_name: 测试UI模块
+type: service
+state: manual
+max_connections: 3
+runtime: python
+entry: entry.py
+preferred_port: 0
+monitor: true
+---
+# 测试UI模块
+
+用于测试 Evol 页面插件化功能的演示模块。
+
+注册两个花里胡哨的测试页面：
+- 粒子动画页面
+- 3D 卡片展示页面

package/extensions/services/testUi/ui/cards.html

@@ -0,0 +1,131 @@
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <title>3D卡片</title>
+  <style>
+    * { margin: 0; padding: 0; box-sizing: border-box; }
+    body {
+      background: linear-gradient(135deg, #1e3c72 0%, #2a5298 100%);
+      min-height: 100vh;
+      display: flex;
+      flex-direction: column;
+      align-items: center;
+      padding: 40px 20px;
+      font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif;
+    }
+    h1 {
+      color: white;
+      font-size: 42px;
+      margin-bottom: 40px;
+      text-shadow: 0 2px 20px rgba(0,0,0,0.3);
+    }
+    .cards-container {
+      display: grid;
+      grid-template-columns: repeat(auto-fit, minmax(280px, 1fr));
+      gap: 30px;
+      max-width: 1200px;
+      width: 100%;
+    }
+    .card {
+      background: rgba(255, 255, 255, 0.1);
+      backdrop-filter: blur(10px);
+      border-radius: 20px;
+      padding: 30px;
+      color: white;
+      cursor: pointer;
+      transition: all 0.3s ease;
+      border: 1px solid rgba(255, 255, 255, 0.2);
+      transform-style: preserve-3d;
+      perspective: 1000px;
+    }
+    .card:hover {
+      transform: translateY(-10px) rotateX(5deg);
+      box-shadow: 0 20px 40px rgba(0,0,0,0.3);
+      background: rgba(255, 255, 255, 0.15);
+    }
+    .card-icon {
+      font-size: 48px;
+      margin-bottom: 20px;
+      display: block;
+    }
+    .card-title {
+      font-size: 24px;
+      font-weight: 600;
+      margin-bottom: 10px;
+    }
+    .card-desc {
+      font-size: 14px;
+      opacity: 0.8;
+      line-height: 1.6;
+    }
+    .card-badge {
+      display: inline-block;
+      background: rgba(255, 255, 255, 0.2);
+      padding: 4px 12px;
+      border-radius: 12px;
+      font-size: 12px;
+      margin-top: 15px;
+    }
+  </style>
+</head>
+<body>
+  <h1>🎴 3D 卡片展示</h1>
+  <div class="cards-container">
+    <div class="card" style="--hue: 200">
+      <span class="card-icon">🚀</span>
+      <div class="card-title">快速启动</div>
+      <div class="card-desc">一键启动所有服务，快速进入工作状态</div>
+      <span class="card-badge">推荐</span>
+    </div>
+    <div class="card" style="--hue: 280">
+      <span class="card-icon">📊</span>
+      <div class="card-title">数据分析</div>
+      <div class="card-desc">实时监控系统性能，可视化数据展示</div>
+      <span class="card-badge">热门</span>
+    </div>
+    <div class="card" style="--hue: 340">
+      <span class="card-icon">🔧</span>
+      <div class="card-title">系统配置</div>
+      <div class="card-desc">灵活配置各项参数，满足个性化需求</div>
+      <span class="card-badge">新功能</span>
+    </div>
+    <div class="card" style="--hue: 120">
+      <span class="card-icon">🎨</span>
+      <div class="card-title">主题定制</div>
+      <div class="card-desc">多种主题风格，打造专属界面</div>
+      <span class="card-badge">Beta</span>
+    </div>
+    <div class="card" style="--hue: 40">
+      <span class="card-icon">📦</span>
+      <div class="card-title">插件市场</div>
+      <div class="card-desc">丰富的插件生态，扩展无限可能</div>
+      <span class="card-badge">即将推出</span>
+    </div>
+    <div class="card" style="--hue: 180">
+      <span class="card-icon">🛡️</span>
+      <div class="card-title">安全中心</div>
+      <div class="card-desc">全方位保护数据安全，权限精细管理</div>
+      <span class="card-badge">企业版</span>
+    </div>
+  </div>
+  <script>
+    document.querySelectorAll('.card').forEach(card => {
+      card.addEventListener('mousemove', (e) => {
+        const rect = card.getBoundingClientRect();
+        const x = e.clientX - rect.left;
+        const y = e.clientY - rect.top;
+        const centerX = rect.width / 2;
+        const centerY = rect.height / 2;
+        const rotateX = (y - centerY) / 10;
+        const rotateY = (centerX - x) / 10;
+        card.style.transform = `translateY(-10px) rotateX(${rotateX}deg) rotateY(${rotateY}deg)`;
+      });
+      card.addEventListener('mouseleave', () => {
+        card.style.transform = 'translateY(0) rotateX(0) rotateY(0)';
+      });
+    });
+  </script>
+</body>
+</html>

package/extensions/services/testUi/ui/index.html

@@ -0,0 +1,22 @@
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <title>TestUI</title>
+</head>
+<body>
+  <script>
+    // 根据 hash 路由到不同页面
+    const hash = window.location.hash.slice(1);
+    if (hash === '/particles' || window.location.pathname.includes('particles')) {
+      window.location.href = 'particles.html';
+    } else if (hash === '/cards' || window.location.pathname.includes('cards')) {
+      window.location.href = 'cards.html';
+    } else {
+      // 默认显示粒子页面
+      window.location.href = 'particles.html';
+    }
+  </script>
+</body>
+</html>