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

@agentunion/kite 1.5.0 → 1.6.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 (573) hide show

package/extensions/services/kite_console/static/test_kernel_client_token.html ADDED Viewed

@@ -0,0 +1,352 @@
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Kernel Client Token 测试</title>
+    <style>
+        body {
+            font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+            max-width: 900px;
+            margin: 50px auto;
+            padding: 20px;
+            background: #f5f5f5;
+        }
+        .container {
+            background: white;
+            padding: 30px;
+            border-radius: 8px;
+            box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+        }
+        h1 {
+            color: #333;
+            margin-bottom: 10px;
+        }
+        .subtitle {
+            color: #666;
+            margin-bottom: 30px;
+        }
+        .section {
+            margin-bottom: 30px;
+            padding: 20px;
+            background: #f9f9f9;
+            border-radius: 6px;
+        }
+        .section h2 {
+            margin-top: 0;
+            color: #444;
+            font-size: 18px;
+        }
+        .info {
+            margin: 10px 0;
+            padding: 10px;
+            background: #e8f4f8;
+            border-left: 4px solid #2196F3;
+            border-radius: 4px;
+        }
+        .success {
+            background: #e8f5e9;
+            border-left-color: #4CAF50;
+        }
+        .error {
+            background: #ffebee;
+            border-left-color: #f44336;
+        }
+        button {
+            padding: 10px 20px;
+            margin: 5px;
+            border: none;
+            border-radius: 4px;
+            cursor: pointer;
+            font-size: 14px;
+            transition: background 0.3s;
+        }
+        .btn-primary {
+            background: #2196F3;
+            color: white;
+        }
+        .btn-primary:hover {
+            background: #1976D2;
+        }
+        .btn-success {
+            background: #4CAF50;
+            color: white;
+        }
+        .btn-success:hover {
+            background: #388E3C;
+        }
+        .btn-danger {
+            background: #f44336;
+            color: white;
+        }
+        .btn-danger:hover {
+            background: #d32f2f;
+        }
+        .token-display {
+            font-family: monospace;
+            font-size: 12px;
+            word-break: break-all;
+            padding: 10px;
+            background: #fff;
+            border: 1px solid #ddd;
+            border-radius: 4px;
+            margin: 10px 0;
+        }
+        .log {
+            background: #263238;
+            color: #aed581;
+            padding: 15px;
+            border-radius: 4px;
+            font-family: 'Courier New', monospace;
+            font-size: 13px;
+            max-height: 400px;
+            overflow-y: auto;
+            margin-top: 10px;
+        }
+        .log-entry {
+            margin: 5px 0;
+        }
+        .log-time {
+            color: #78909c;
+        }
+        .log-success {
+            color: #81c784;
+        }
+        .log-error {
+            color: #e57373;
+        }
+        .log-info {
+            color: #64b5f6;
+        }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <h1>🧪 Kernel Client Token 测试</h1>
+        <p class="subtitle">测试 kernel-client.js 的 Kite Token 支持</p>
+        <!-- Token 状态 -->
+        <div class="section">
+            <h2>📋 Token 状态</h2>
+            <div id="token-status">
+                <p>正在检查...</p>
+            </div>
+        </div>
+        <!-- 操作按钮 -->
+        <div class="section">
+            <h2>🎮 操作</h2>
+            <button class="btn-primary" onclick="testConnect()">测试连接</button>
+            <button class="btn-success" onclick="testRpc()">测试 RPC</button>
+            <button class="btn-danger" onclick="clearTokens()">清除所有 Token</button>
+            <button class="btn-primary" onclick="clearLog()">清空日志</button>
+        </div>
+        <!-- 连接状态 -->
+        <div class="section">
+            <h2>🔌 连接状态</h2>
+            <div id="connection-status" class="info">
+                未连接
+            </div>
+        </div>
+        <!-- 日志 -->
+        <div class="section">
+            <h2>📝 日志</h2>
+            <div id="log" class="log"></div>
+        </div>
+    </div>
+    <script src="/static/js/kernel-client.js"></script>
+    <script>
+        let kernelClient = null;
+        // 日志函数
+        function log(message, type = 'info') {
+            const logDiv = document.getElementById('log');
+            const time = new Date().toLocaleTimeString();
+            const className = `log-${type}`;
+            const entry = document.createElement('div');
+            entry.className = 'log-entry';
+            entry.innerHTML = `<span class="log-time">[${time}]</span> <span class="${className}">${message}</span>`;
+            logDiv.appendChild(entry);
+            logDiv.scrollTop = logDiv.scrollHeight;
+        }
+        function clearLog() {
+            document.getElementById('log').innerHTML = '';
+        }
+        // 检查 Token 状态
+        function checkTokenStatus() {
+            const kiteToken = localStorage.getItem('kiteToken');
+            const frontendToken = localStorage.getItem('frontend_token');
+            const sessionToken = sessionStorage.getItem('session_token');
+            let html = '';
+            if (kiteToken) {
+                html += `<div class="info success">
+                    <strong>✓ Kite Token 存在</strong>
+                    <div class="token-display">${kiteToken}</div>
+                    <small>将优先使用此 Token 进行认证</small>
+                </div>`;
+            } else {
+                html += `<div class="info">
+                    <strong>○ Kite Token 不存在</strong>
+                    <small>需要通过手机号登录获取</small>
+                </div>`;
+            }
+            if (frontendToken) {
+                html += `<div class="info ${kiteToken ? '' : 'success'}">
+                    <strong>${kiteToken ? '○' : '✓'} Frontend Token 存在</strong>
+                    <div class="token-display">${frontendToken}</div>
+                    <small>${kiteToken ? '将作为备用 Token' : '将使用此 Token 进行认证'}</small>
+                </div>`;
+            } else {
+                html += `<div class="info">
+                    <strong>○ Frontend Token 不存在</strong>
+                    <small>需要通过配对码获取</small>
+                </div>`;
+            }
+            if (sessionToken) {
+                html += `<div class="info">
+                    <strong>Session Token: ${sessionToken}</strong>
+                </div>`;
+            }
+            if (!kiteToken && !frontendToken) {
+                html += `<div class="info error">
+                    <strong>⚠️ 没有可用的 Token</strong>
+                    <p>请先通过以下方式之一获取 Token：</p>
+                    <ul>
+                        <li>手机号登录（获取 Kite Token）</li>
+                        <li>配对码配对（获取 Frontend Token）</li>
+                    </ul>
+                </div>`;
+            }
+            document.getElementById('token-status').innerHTML = html;
+        }
+        // 测试连接
+        async function testConnect() {
+            log('开始测试连接...', 'info');
+            const kiteToken = localStorage.getItem('kiteToken');
+            const frontendToken = localStorage.getItem('frontend_token');
+            if (!kiteToken && !frontendToken) {
+                log('错误: 没有可用的 Token', 'error');
+                updateConnectionStatus('error', '没有可用的 Token');
+                return;
+            }
+            const tokenType = kiteToken ? 'Kite Token' : 'Frontend Token';
+            log(`使用 ${tokenType} 进行认证`, 'info');
+            try {
+                updateConnectionStatus('connecting', '连接中...');
+                const proto = location.protocol === 'https:' ? 'wss:' : 'ws:';
+                kernelClient = new KernelClient(`${proto}//${location.host}/ws/relay`);
+                log(`创建 KernelClient 实例`, 'success');
+                log(`Token: ${kernelClient.token ? kernelClient.token.substring(0, 20) + '...' : 'null'}`, 'info');
+                log(`Session Token: ${kernelClient.sessionToken}`, 'info');
+                const result = await kernelClient.connect();
+                log('连接成功！', 'success');
+                log(`Module ID: ${result.moduleId}`, 'success');
+                log(`Role: ${result.role}`, 'success');
+                log(`Reconnected: ${result.reconnected}`, 'info');
+                updateConnectionStatus('success', `已连接 (${result.moduleId})`);
+                // 订阅事件
+                await kernelClient.subscribe(['module.*', 'registry.*']);
+                log('已订阅事件: module.*, registry.*', 'success');
+            } catch (err) {
+                log(`连接失败: ${err.message}`, 'error');
+                updateConnectionStatus('error', `连接失败: ${err.message}`);
+            }
+        }
+        // 测试 RPC
+        async function testRpc() {
+            if (!kernelClient || !kernelClient.authenticated) {
+                log('错误: 未连接或未认证', 'error');
+                return;
+            }
+            log('测试 RPC 调用...', 'info');
+            try {
+                // 测试 launcher.list_modules
+                log('调用 launcher.list_modules', 'info');
+                const result = await kernelClient.call('launcher.list_modules');
+                log(`RPC 调用成功`, 'success');
+                log(`模块数量: ${result.modules ? result.modules.length : 0}`, 'success');
+                if (result.modules && result.modules.length > 0) {
+                    log(`模块列表: ${result.modules.map(m => m.name).join(', ')}`, 'info');
+                }
+                // 测试 kernel.health
+                log('调用 kernel.health', 'info');
+                const health = await kernelClient.call('kernel.health');
+                log(`Kernel 状态: ${health.status}`, 'success');
+            } catch (err) {
+                log(`RPC 调用失败: ${err.message}`, 'error');
+            }
+        }
+        // 清除所有 Token
+        function clearTokens() {
+            if (confirm('确定要清除所有 Token 吗？')) {
+                localStorage.removeItem('kiteToken');
+                localStorage.removeItem('frontend_token');
+                sessionStorage.removeItem('session_token');
+                log('已清除所有 Token', 'info');
+                checkTokenStatus();
+                if (kernelClient) {
+                    kernelClient.disconnect();
+                    kernelClient = null;
+                }
+                updateConnectionStatus('disconnected', '未连接');
+            }
+        }
+        // 更新连接状态
+        function updateConnectionStatus(status, message) {
+            const statusDiv = document.getElementById('connection-status');
+            statusDiv.className = 'info';
+            if (status === 'success') {
+                statusDiv.className = 'info success';
+                statusDiv.innerHTML = `<strong>✓ ${message}</strong>`;
+            } else if (status === 'error') {
+                statusDiv.className = 'info error';
+                statusDiv.innerHTML = `<strong>✗ ${message}</strong>`;
+            } else if (status === 'connecting') {
+                statusDiv.className = 'info';
+                statusDiv.innerHTML = `<strong>◐ ${message}</strong>`;
+            } else {
+                statusDiv.innerHTML = `<strong>${message}</strong>`;
+            }
+        }
+        // 页面加载时检查 Token 状态
+        window.addEventListener('DOMContentLoaded', () => {
+            checkTokenStatus();
+            log('页面加载完成', 'info');
+            log('准备测试 Kite Token 支持', 'info');
+        });
+    </script>
+</body>
+</html>

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

@@ -0,0 +1,247 @@
+"""
+积分统计管理模块 - 定时采集、数据分析
+数据存储结构：
+- 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
+        self._last_collect_time = 0  # 上次采集时间戳，用于防抖
+        self._collect_cooldown = 30  # 30 秒内不重复采集
+    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):
+        """执行一次采集（采集所有已登录用户）"""
+        now = time.time()
+        if now - self._last_collect_time < self._collect_cooldown:
+            return
+        self._last_collect_time = now
+        # 获取所有有效的 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)
+                }
+            }
+        }