npm - @agentunion/kite - Versions diffs - 1.4.0 → 1.5.0 - Mend

@agentunion/kite 1.4.0 → 1.5.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (235) hide show

package/extensions/services/evol/static/test_relay.html ADDED Viewed

@@ -0,0 +1,462 @@
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>中转服务测试</title>
+    <style>
+        * {
+            margin: 0;
+            padding: 0;
+            box-sizing: border-box;
+        }
+        body {
+            font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif;
+            background: #f5f5f5;
+            padding: 20px;
+        }
+        .container {
+            max-width: 1200px;
+            margin: 0 auto;
+        }
+        h1 {
+            font-size: 24px;
+            color: #333;
+            margin-bottom: 20px;
+        }
+        .panel {
+            background: white;
+            border-radius: 8px;
+            padding: 20px;
+            margin-bottom: 20px;
+            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+        }
+        .panel h2 {
+            font-size: 18px;
+            color: #333;
+            margin-bottom: 15px;
+            border-bottom: 2px solid #667eea;
+            padding-bottom: 10px;
+        }
+        .status {
+            display: flex;
+            align-items: center;
+            gap: 10px;
+            margin-bottom: 15px;
+        }
+        .status-dot {
+            width: 12px;
+            height: 12px;
+            border-radius: 50%;
+            background: #ccc;
+        }
+        .status-dot.connected {
+            background: #27ae60;
+        }
+        .status-dot.disconnected {
+            background: #e74c3c;
+        }
+        .btn {
+            padding: 10px 20px;
+            border: none;
+            border-radius: 4px;
+            font-size: 14px;
+            cursor: pointer;
+            margin-right: 10px;
+            margin-bottom: 10px;
+        }
+        .btn-primary {
+            background: #667eea;
+            color: white;
+        }
+        .btn-success {
+            background: #27ae60;
+            color: white;
+        }
+        .btn-danger {
+            background: #e74c3c;
+            color: white;
+        }
+        .btn:disabled {
+            opacity: 0.5;
+            cursor: not-allowed;
+        }
+        .log {
+            background: #1e1e1e;
+            color: #d4d4d4;
+            padding: 15px;
+            border-radius: 4px;
+            font-family: monospace;
+            font-size: 12px;
+            max-height: 400px;
+            overflow-y: auto;
+            white-space: pre-wrap;
+            line-height: 1.4;
+        }
+        .log-entry {
+            margin-bottom: 5px;
+        }
+        .log-time {
+            color: #858585;
+        }
+        .log-info {
+            color: #4ec9b0;
+        }
+        .log-success {
+            color: #27ae60;
+        }
+        .log-error {
+            color: #f48771;
+        }
+        .log-event {
+            color: #dcdcaa;
+        }
+        input {
+            padding: 8px 12px;
+            border: 1px solid #ddd;
+            border-radius: 4px;
+            font-size: 14px;
+            margin-right: 10px;
+        }
+        .info-grid {
+            display: grid;
+            grid-template-columns: 150px 1fr;
+            gap: 10px;
+            margin-bottom: 15px;
+        }
+        .info-label {
+            color: #666;
+            font-weight: 500;
+        }
+        .info-value {
+            color: #333;
+            font-family: monospace;
+        }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <h1>🔌 中转服务测试</h1>
+        <!-- 连接状态 -->
+        <div class="panel">
+            <h2>连接状态</h2>
+            <div class="status">
+                <div class="status-dot" id="status-dot"></div>
+                <span id="status-text">未连接</span>
+            </div>
+            <div class="info-grid" id="connection-info" style="display: none;">
+                <div class="info-label">Module ID:</div>
+                <div class="info-value" id="module-id">-</div>
+                <div class="info-label">Role:</div>
+                <div class="info-value" id="role">-</div>
+                <div class="info-label">Session Token:</div>
+                <div class="info-value" id="session-token">-</div>
+            </div>
+            <div>
+                <button class="btn btn-primary" id="btn-connect">连接（使用已有 Token）</button>
+                <button class="btn btn-danger" id="btn-disconnect" disabled>断开</button>
+                <button class="btn btn-success" id="btn-request-code">首次配对（自动生成配对码）</button>
+            </div>
+        </div>
+        <!-- 配对码显示 -->
+        <div class="panel" id="pairing-code-panel" style="display: none;">
+            <h2>配对码</h2>
+            <div style="font-size: 32px; font-weight: bold; color: #667eea; text-align: center; padding: 20px; background: #f5f5f5; border-radius: 8px; font-family: monospace;">
+                <span id="pairing-code">------</span>
+            </div>
+            <div style="margin-top: 10px; text-align: center; color: #666;">
+                有效期：5 分钟
+            </div>
+        </div>
+        <!-- RPC 测试 -->
+        <div class="panel">
+            <h2>RPC 调用测试</h2>
+            <div>
+                <button class="btn btn-primary" id="btn-list-modules" disabled>launcher.list_modules</button>
+                <button class="btn btn-primary" id="btn-kernel-health" disabled>kernel.health</button>
+                <button class="btn btn-primary" id="btn-kernel-stats" disabled>kernel.stats</button>
+            </div>
+        </div>
+        <!-- 事件订阅测试 -->
+        <div class="panel">
+            <h2>事件订阅测试</h2>
+            <div style="margin-bottom: 15px;">
+                <button class="btn btn-success" id="btn-subscribe-all" disabled>订阅全部事件 (*)</button>
+                <button class="btn btn-success" id="btn-subscribe-module" disabled>订阅模块事件 (module.*)</button>
+                <button class="btn btn-danger" id="btn-unsubscribe" disabled>取消订阅</button>
+            </div>
+            <div>
+                <strong>当前订阅:</strong> <span id="current-subscription">无</span>
+            </div>
+        </div>
+        <!-- 日志 -->
+        <div class="panel">
+            <h2>日志</h2>
+            <button class="btn btn-danger" id="btn-clear-log" style="margin-bottom: 10px;">清空日志</button>
+            <div class="log" id="log"></div>
+        </div>
+    </div>
+    <script src="js/kernel-client.js"></script>
+    <script>
+        let client = null;
+        let currentSubscription = null;
+        // 日志函数
+        function log(message, type = 'info') {
+            const logEl = document.getElementById('log');
+            const time = new Date().toLocaleTimeString('zh-CN', { hour12: false, fractionalSecondDigits: 3 });
+            const entry = document.createElement('div');
+            entry.className = 'log-entry';
+            entry.innerHTML = `<span class="log-time">[${time}]</span> <span class="log-${type}">${message}</span>`;
+            logEl.insertBefore(entry, logEl.firstChild);
+        }
+        // 更新状态
+        function updateStatus(connected, info = null) {
+            const dot = document.getElementById('status-dot');
+            const text = document.getElementById('status-text');
+            const infoEl = document.getElementById('connection-info');
+            if (connected) {
+                dot.className = 'status-dot connected';
+                text.textContent = '已连接';
+                infoEl.style.display = 'grid';
+                document.getElementById('module-id').textContent = info.moduleId || '-';
+                document.getElementById('role').textContent = info.role || '-';
+                document.getElementById('session-token').textContent = info.sessionToken || '-';
+                // 启用按钮
+                document.getElementById('btn-disconnect').disabled = false;
+                document.getElementById('btn-list-modules').disabled = false;
+                document.getElementById('btn-kernel-health').disabled = false;
+                document.getElementById('btn-kernel-stats').disabled = false;
+                document.getElementById('btn-subscribe-all').disabled = false;
+                document.getElementById('btn-subscribe-module').disabled = false;
+                document.getElementById('btn-unsubscribe').disabled = false;
+            } else {
+                dot.className = 'status-dot disconnected';
+                text.textContent = '未连接';
+                infoEl.style.display = 'none';
+                // 禁用按钮
+                document.getElementById('btn-disconnect').disabled = true;
+                document.getElementById('btn-list-modules').disabled = true;
+                document.getElementById('btn-kernel-health').disabled = true;
+                document.getElementById('btn-kernel-stats').disabled = true;
+                document.getElementById('btn-subscribe-all').disabled = true;
+                document.getElementById('btn-subscribe-module').disabled = true;
+                document.getElementById('btn-unsubscribe').disabled = true;
+            }
+        }
+        // 连接
+        document.getElementById('btn-connect').addEventListener('click', async () => {
+            try {
+                log('正在连接...', 'info');
+                const proto = location.protocol === 'https:' ? 'wss:' : 'ws:';
+                client = new KernelClient(`${proto}//${location.host}/ws/relay`);
+                const result = await client.connect();
+                log(`连接成功！Module ID: ${result.moduleId}`, 'success');
+                updateStatus(true, {
+                    moduleId: result.moduleId,
+                    role: result.role,
+                    sessionToken: client.sessionToken
+                });
+                // 监听所有事件
+                client.eventListeners.set('*', [(data) => {
+                    log(`收到事件: ${JSON.stringify(data)}`, 'event');
+                }]);
+            } catch (err) {
+                log(`连接失败: ${err.message}`, 'error');
+                updateStatus(false);
+            }
+        });
+        // 断开
+        document.getElementById('btn-disconnect').addEventListener('click', () => {
+            if (client) {
+                client.disconnect();
+                log('已断开连接', 'info');
+                updateStatus(false);
+                client = null;
+            }
+        });
+        // 请求配对码并自动配对
+        document.getElementById('btn-request-code').addEventListener('click', async () => {
+            try {
+                log('正在请求配对码...', 'info');
+                const proto = location.protocol === 'https:' ? 'wss:' : 'ws:';
+                const ws = new WebSocket(`${proto}//${location.host}/ws/relay`);
+                ws.onopen = () => {
+                    log('WebSocket 已连接，发送请求...', 'info');
+                    ws.send(JSON.stringify({ type: 'request_code' }));
+                };
+                ws.onmessage = async (event) => {
+                    const msg = JSON.parse(event.data);
+                    log(`收到消息: ${JSON.stringify(msg)}`, 'info');
+                    if (msg.type === 'code_generated') {
+                        const code = msg.code;
+                        log(`配对码已生成: ${code}`, 'success');
+                        document.getElementById('pairing-code').textContent = code;
+                        document.getElementById('pairing-code-panel').style.display = 'block';
+                        // 同时在控制台输出
+                        console.log('='.repeat(50));
+                        console.log('配对码:', code);
+                        console.log('有效期: 5 分钟');
+                        console.log('='.repeat(50));
+                        // 关闭这个 WebSocket
+                        ws.close();
+                        // 自动配对
+                        log('自动配对中...', 'info');
+                        client = new KernelClient(`${proto}//${location.host}/ws/relay`);
+                        const result = await client.pair(code);
+                        log(`配对成功！Module ID: ${result.moduleId}`, 'success');
+                        updateStatus(true, {
+                            moduleId: result.moduleId,
+                            role: result.role,
+                            sessionToken: client.sessionToken
+                        });
+                        // 监听所有事件
+                        client.eventListeners.set('*', [(data) => {
+                            log(`收到事件: ${JSON.stringify(data)}`, 'event');
+                        }]);
+                    } else if (msg.type === 'error') {
+                        log(`生成配对码失败: ${msg.message}`, 'error');
+                    }
+                };
+                ws.onerror = (err) => {
+                    log('WebSocket 错误', 'error');
+                    console.error(err);
+                };
+            } catch (err) {
+                log(`配对失败: ${err.message}`, 'error');
+            }
+        });
+        // RPC 测试
+        document.getElementById('btn-list-modules').addEventListener('click', async () => {
+            try {
+                log('调用 launcher.list_modules...', 'info');
+                const result = await client.call('launcher.list_modules', {});
+                log(`成功！返回 ${result.modules.length} 个模块`, 'success');
+                log(JSON.stringify(result, null, 2), 'info');
+            } catch (err) {
+                log(`失败: ${err.message}`, 'error');
+            }
+        });
+        document.getElementById('btn-kernel-health').addEventListener('click', async () => {
+            try {
+                log('调用 kernel.health...', 'info');
+                const result = await client.call('kernel.health', {});
+                log(`成功！`, 'success');
+                log(JSON.stringify(result, null, 2), 'info');
+            } catch (err) {
+                log(`失败: ${err.message}`, 'error');
+            }
+        });
+        document.getElementById('btn-kernel-stats').addEventListener('click', async () => {
+            try {
+                log('调用 kernel.stats...', 'info');
+                const result = await client.call('kernel.stats', {});
+                log(`成功！`, 'success');
+                log(JSON.stringify(result, null, 2), 'info');
+            } catch (err) {
+                log(`失败: ${err.message}`, 'error');
+            }
+        });
+        // 事件订阅测试
+        document.getElementById('btn-subscribe-all').addEventListener('click', async () => {
+            try {
+                log('订阅全部事件 (*)...', 'info');
+                await client.call('event.subscribe', { events: ['*'] });
+                currentSubscription = '*';
+                document.getElementById('current-subscription').textContent = '*';
+                log('订阅成功！', 'success');
+            } catch (err) {
+                log(`订阅失败: ${err.message}`, 'error');
+            }
+        });
+        document.getElementById('btn-subscribe-module').addEventListener('click', async () => {
+            try {
+                log('订阅模块事件 (module.*)...', 'info');
+                await client.call('event.subscribe', { events: ['module.*'] });
+                currentSubscription = 'module.*';
+                document.getElementById('current-subscription').textContent = 'module.*';
+                log('订阅成功！', 'success');
+            } catch (err) {
+                log(`订阅失败: ${err.message}`, 'error');
+            }
+        });
+        document.getElementById('btn-unsubscribe').addEventListener('click', async () => {
+            try {
+                log('取消订阅...', 'info');
+                await client.call('event.subscribe', { events: [] });
+                currentSubscription = null;
+                document.getElementById('current-subscription').textContent = '无';
+                log('取消成功！', 'success');
+            } catch (err) {
+                log(`取消失败: ${err.message}`, 'error');
+            }
+        });
+        // 清空日志
+        document.getElementById('btn-clear-log').addEventListener('click', () => {
+            document.getElementById('log').innerHTML = '';
+        });
+        // 页面加载时的提示
+        log('中转服务测试页面已加载', 'info');
+        log('请先点击"连接"或"配对"按钮', 'info');
+    </script>
+</body>
+</html>

package/extensions/services/evol/stats_manager.py ADDED Viewed

@@ -0,0 +1,240 @@
+"""
+积分统计管理模块 - 定时采集、数据分析
+数据存储结构：
+- stats/YYYY-MM/YYYY-MM-DD.jsonl - 每日账户信息记录
+- 每 15 分钟自动采集一次
+- 手动访问账户页面时也会触发采集
+"""
+import asyncio
+import json
+import os
+import time
+from datetime import datetime, timezone
+from typing import Optional
+class StatsManager:
+    """积分统计管理器"""
+    def __init__(self, data_dir: str, evol_api, auth_manager):
+        """
+        初始化统计管理器
+        Args:
+            data_dir: 数据目录（$KITE_DATA/evol）
+            evol_api: Evol API 实例
+            auth_manager: 认证管理器实例
+        """
+        self.data_dir = data_dir
+        self.evol_api = evol_api
+        self.auth_manager = auth_manager
+        self.stats_dir = os.path.join(data_dir, "stats")
+        os.makedirs(self.stats_dir, exist_ok=True)
+        self.interval = 15 * 60  # 15 分钟
+        self.running = False
+        self.task = None
+    def _get_daily_file(self, dt: datetime = None) -> str:
+        """获取每日记录文件路径"""
+        if dt is None:
+            dt = datetime.now(timezone.utc)
+        month_dir = os.path.join(self.stats_dir, dt.strftime("%Y-%m"))
+        os.makedirs(month_dir, exist_ok=True)
+        return os.path.join(month_dir, dt.strftime("%Y-%m-%d.jsonl"))
+    async def start(self):
+        """启动定时采集任务"""
+        if self.running:
+            return
+        self.running = True
+        self.task = asyncio.create_task(self._collect_loop())
+        print("[stats] 积分监控已启动，采集间隔: 15分钟")
+    async def stop(self):
+        """停止定时采集任务"""
+        self.running = False
+        if self.task:
+            self.task.cancel()
+            try:
+                await self.task
+            except asyncio.CancelledError:
+                pass
+    async def _collect_loop(self):
+        """定时采集循环"""
+        while self.running:
+            try:
+                await self._collect_once()
+            except Exception as e:
+                print(f"[stats] 采集失败: {e}")
+            # Wait for next interval
+            await asyncio.sleep(self.interval)
+    async def _collect_once(self):
+        """执行一次采集（采集所有已登录用户）"""
+        # 获取所有有效的 Evol Token
+        evol_records = self.auth_manager.list_all_evol_tokens()
+        if not evol_records:
+            return
+        # 为每个用户采集积分信息
+        for evol_record in evol_records:
+            try:
+                # Fetch user info
+                result = await self.evol_api.get_user_info(evol_record["token"])
+                if not result.get("success"):
+                    continue
+                data = result["data"]
+                self._save_account_info(data)
+            except Exception as e:
+                phone = evol_record.get("phone", "unknown")
+                print(f"[stats] 采集用户 {phone} 失败: {e}")
+    def _save_account_info(self, data: dict):
+        """保存账户信息到每日文件"""
+        now = datetime.now(timezone.utc)
+        account_info = data.get("accountInfo", {})
+        user_info = data.get("userInfo", {})
+        credits_package = account_info.get("creditsPackageSummary", {})
+        record = {
+            "timestamp": time.time(),
+            "timestamp_human": now.isoformat(),
+            "userInfo": {
+                "userId": user_info.get("userId"),
+                "phone": user_info.get("phone"),
+                "nickName": user_info.get("nickName")
+            },
+            "accountInfo": {
+                "credits": account_info.get("credits", 0),
+                "creditsLimit": account_info.get("creditsLimit", 0),
+                "vipType": account_info.get("vipType", 0),
+                "vipTypeName": account_info.get("vipTypeName", "Unknown"),
+                "vipExpireTime": account_info.get("vipExpireTime"),
+                "vipRemainingDays": account_info.get("vipRemainingDays", 0)
+            },
+            "creditsPackage": {
+                "chatbotCredits": credits_package.get("chatbotCredits", 0),
+                "codexCredits": credits_package.get("codexCredits", 0),
+                "claudeCodeCredits": credits_package.get("claudeCodeCredits", 0),
+                "geminiCredits": credits_package.get("geminiCredits", 0),
+                "openclawCredits": credits_package.get("openclawCredits", 0),
+                "generalCredits": credits_package.get("generalCredits", 0)
+            },
+            "apiKey": data.get("apiKey", "")
+        }
+        # 追加到每日文件
+        daily_file = self._get_daily_file(now)
+        with open(daily_file, "a", encoding="utf-8") as f:
+            f.write(json.dumps(record, ensure_ascii=False) + "\n")
+        print(f"[stats] 账户信息已保存: credits={record['accountInfo']['credits']}")
+    async def collect_manual(self):
+        """手动触发采集（用户访问账户页面时）"""
+        try:
+            await self._collect_once()
+        except Exception as e:
+            print(f"[stats] 手动采集失败: {e}")
+    def get_stats(self, period: str = "day", date: Optional[str] = None) -> dict:
+        """
+        获取统计数据
+        Args:
+            period: 统计周期 (day/week/month)
+            date: 日期 (YYYY-MM-DD)
+        Returns:
+            统计数据
+        """
+        if date is None:
+            date = datetime.now(timezone.utc).strftime("%Y-%m-%d")
+        try:
+            dt = datetime.strptime(date, "%Y-%m-%d")
+        except ValueError:
+            return {"success": False, "msg": "Invalid date format"}
+        # 读取对应日期的文件
+        daily_file = self._get_daily_file(dt)
+        if not os.path.exists(daily_file):
+            return {
+                "success": True,
+                "data": {
+                    "period": period,
+                    "date": date,
+                    "records": [],
+                    "summary": {
+                        "startCredits": 0,
+                        "endCredits": 0,
+                        "consumed": 0,
+                        "recovered": 0
+                    }
+                }
+            }
+        # 读取所有记录
+        records = []
+        try:
+            with open(daily_file, "r", encoding="utf-8") as f:
+                for line in f:
+                    line = line.strip()
+                    if not line:
+                        continue
+                    try:
+                        record = json.loads(line)
+                        records.append(record)
+                    except json.JSONDecodeError:
+                        continue
+        except Exception as e:
+            return {"success": False, "msg": f"读取文件失败: {str(e)}"}
+        if not records:
+            return {
+                "success": True,
+                "data": {
+                    "period": period,
+                    "date": date,
+                    "records": [],
+                    "summary": {
+                        "startCredits": 0,
+                        "endCredits": 0,
+                        "consumed": 0,
+                        "recovered": 0
+                    }
+                }
+            }
+        # 计算汇总
+        start_credits = records[0]["accountInfo"]["credits"]
+        end_credits = records[-1]["accountInfo"]["credits"]
+        consumed = max(0, start_credits - end_credits)
+        recovered = max(0, end_credits - start_credits)
+        return {
+            "success": True,
+            "data": {
+                "period": period,
+                "date": date,
+                "records": records,
+                "summary": {
+                    "startCredits": start_credits,
+                    "endCredits": end_credits,
+                    "consumed": consumed,
+                    "recovered": recovered,
+                    "dataPoints": len(records)
+                }
+            }
+        }