botrun-horse 2.29.2 → 2.30.0

package/botrun-c/admin/users-cli.py

@@ -0,0 +1,298 @@
+#!/usr/bin/env python3
+"""
+波特人 Users CLI - 統一使用者管理工具 (Python 版本)
+
+避免 Node.js google-gax 問題，使用 Python Firebase Admin SDK
+"""
+
+import sys
+import argparse
+from pathlib import Path
+from datetime import datetime
+import firebase_admin
+from firebase_admin import credentials, firestore, auth
+
+# 初始化 Firebase Admin
+project_root = Path(__file__).parent.parent
+key_path = project_root / 'keys' / 'compute-sa-key.json'
+
+if not key_path.exists():
+    print(f"❌ 找不到 service account key: {key_path}")
+    sys.exit(1)
+
+cred = credentials.Certificate(str(key_path))
+if not firebase_admin._apps:
+    firebase_admin.initialize_app(cred)
+
+db = firestore.client()
+
+def get_user_by_email_or_uid(email_or_uid):
+    """透過 email 或 UID 取得使用者"""
+    try:
+        if '@' in email_or_uid:
+            user = auth.get_user_by_email(email_or_uid)
+        else:
+            user = auth.get_user(email_or_uid)
+        return user
+    except auth.UserNotFoundError:
+        print(f'❌ 找不到使用者: {email_or_uid}')
+        return None
+
+def query_user(email_or_uid):
+    """查詢使用者的完整資訊"""
+    print('\n' + '=' * 80)
+    print('  使用者資訊查詢')
+    print('=' * 80 + '\n')
+
+    # 取得使用者
+    user = get_user_by_email_or_uid(email_or_uid)
+    if not user:
+        return
+
+    uid = user.uid
+    email = user.email or '(無 email)'
+
+    print(f'📧 Email: {email}')
+    print(f'👤 UID: {uid}\n')
+
+    # 查詢 Firestore users collection
+    user_doc = db.collection('users').document(uid).get()
+
+    if user_doc.exists:
+        user_data = user_doc.to_dict()
+        custom_limits = user_data.get('customLimits', {})
+
+        print('=' * 80)
+        print('📊 Rate Limit 方案與自訂限制')
+        print('=' * 80 + '\n')
+
+        plan = user_data.get('plan', 'Lite')
+        print(f'當前方案: {plan}')
+
+        # 預設限制
+        plan_limits = {
+            'Lite': {'rpm': 6, 'tpm': 200000, 'rpd': 32},
+            'Max-1': {'rpm': 60, 'tpm': 2000000, 'rpd': 3200},
+            'Max-2': {'rpm': 600, 'tpm': 20000000, 'rpd': 32000},
+        }
+        default_limits = plan_limits.get(plan, plan_limits['Lite'])
+
+        if custom_limits:
+            print('\n⚙️  自訂限制:')
+            if 'rpm' in custom_limits:
+                print(f'  - RPM: {custom_limits["rpm"]} (覆寫預設值 {default_limits["rpm"]})')
+            if 'tpm' in custom_limits:
+                print(f'  - TPM: {custom_limits["tpm"]:,} (覆寫預設值 {default_limits["tpm"]:,})')
+            if 'rpd' in custom_limits:
+                print(f'  - RPD: {custom_limits["rpd"]} (覆寫預設值 {default_limits["rpd"]})')
+
+        # 實際生效的限制
+        effective_rpm = custom_limits.get('rpm', default_limits['rpm'])
+        effective_tpm = custom_limits.get('tpm', default_limits['tpm'])
+        effective_rpd = custom_limits.get('rpd', default_limits['rpd'])
+
+        print(f'\n{"實際生效限制" if custom_limits else "方案限制"}:')
+        print(f'  - RPM (每分鐘請求數): {effective_rpm}')
+        print(f'  - TPM (每分鐘 Token 數): {effective_tpm:,}')
+        print(f'  - RPD (每日請求數): {effective_rpd}')
+
+        # 查詢今日使用量
+        today = datetime.now().strftime('%Y-%m-%d')
+        rpd_key = f'rpd:{uid}:{today}'
+
+        counter_doc = db.collection('rate_limit_counters').document(rpd_key).get()
+        if counter_doc.exists:
+            current_rpd = counter_doc.to_dict().get('count', 0)
+        else:
+            current_rpd = 0
+
+        print(f'\n當前使用狀況 (今日):')
+        print(f'  - RPD: {current_rpd} / {effective_rpd} ({int((current_rpd / effective_rpd) * 100)}%)')
+
+        if current_rpd >= effective_rpd:
+            print('\n⛔ 狀態: 已達限制 (BLOCKED)')
+        else:
+            print('\n✅ 狀態: 正常 (OK)')
+
+    print('\n' + '=' * 80 + '\n')
+
+def set_custom_limits(email_or_uid, rpm=None, tpm=None, rpd=None):
+    """設定使用者的自訂限制"""
+    user = get_user_by_email_or_uid(email_or_uid)
+    if not user:
+        return
+
+    uid = user.uid
+    email = user.email or '(無 email)'
+
+    print(f'\n⚙️  設定自訂 Rate Limit...')
+    print(f'📧 使用者: {email}')
+    print(f'👤 UID: {uid}\n')
+
+    custom_limits = {}
+    if rpm is not None:
+        custom_limits['rpm'] = int(rpm)
+    if tpm is not None:
+        custom_limits['tpm'] = int(tpm)
+    if rpd is not None:
+        custom_limits['rpd'] = int(rpd)
+
+    if not custom_limits:
+        print('❌ 請至少指定一個限制值（--rpm, --tpm, 或 --rpd）')
+        return
+
+    print('📝 設定的自訂限制:')
+    for key, value in custom_limits.items():
+        print(f'   {key.upper()}: {value}')
+
+    # 寫入 Firestore
+    db.collection('users').document(uid).set(
+        {'customLimits': custom_limits},
+        merge=True
+    )
+
+    print(f'\n✅ 成功設定自訂限制！')
+    print('ℹ️  設定將在下次 API 請求時生效\n')
+
+def reset_custom_limits(email_or_uid):
+    """移除使用者的自訂限制"""
+    user = get_user_by_email_or_uid(email_or_uid)
+    if not user:
+        return
+
+    uid = user.uid
+    email = user.email or '(無 email)'
+
+    print(f'\n🔄 重置自訂限制...')
+    print(f'📧 使用者: {email}')
+    print(f'👤 UID: {uid}\n')
+
+    # 從 Firestore 移除 customLimits
+    db.collection('users').document(uid).update({
+        'customLimits': firestore.DELETE_FIELD
+    })
+
+    print(f'✅ 已移除自訂限制，恢復使用方案預設值\n')
+
+def reset_rate_limit(email_or_uid):
+    """清除使用者的所有 Rate Limit 計數器"""
+    user = get_user_by_email_or_uid(email_or_uid)
+    if not user:
+        return
+
+    uid = user.uid
+    email = user.email or '(無 email)'
+
+    print(f'\n🔄 清除 Rate Limit 計數器...')
+    print(f'📧 使用者: {email}')
+    print(f'👤 UID: {uid}\n')
+
+    # 查詢所有計數器
+    counters_ref = db.collection('rate_limit_counters')
+
+    # 查詢 RPM, TPM, RPD 計數器
+    counter_types = ['rpm', 'tpm', 'rpd']
+    deleted_count = 0
+
+    for counter_type in counter_types:
+        query = counters_ref.where('__name__', '>=', f'{counter_type}:{uid}') \
+                            .where('__name__', '<=', f'{counter_type}:{uid}\uf8ff')
+
+        docs = query.stream()
+        for doc in docs:
+            doc.reference.delete()
+            deleted_count += 1
+            print(f'  🗑️  已刪除: {doc.id}')
+
+    print(f'\n✅ 已清除 {deleted_count} 個計數器')
+    print('🎉 使用者現在可以重新開始使用服務\n')
+
+def list_users_with_custom_limits():
+    """列出所有有自訂限制的使用者"""
+    print('\n' + '=' * 80)
+    print('  有自訂限制的使用者列表')
+    print('=' * 80 + '\n')
+
+    users_ref = db.collection('users')
+    users_with_limits = []
+
+    # 查詢所有使用者文件
+    for doc in users_ref.stream():
+        data = doc.to_dict()
+        if 'customLimits' in data:
+            users_with_limits.append({
+                'uid': doc.id,
+                'customLimits': data['customLimits']
+            })
+
+    if not users_with_limits:
+        print('📭 目前沒有任何使用者設定自訂限制\n')
+        return
+
+    print(f'找到 {len(users_with_limits)} 位使用者:\n')
+
+    for user_info in users_with_limits:
+        uid = user_info['uid']
+        limits = user_info['customLimits']
+
+        # 取得 email
+        try:
+            user = auth.get_user(uid)
+            email = user.email or '(無 email)'
+        except:
+            email = '(無法取得)'
+
+        print(f'📧 {email}')
+        print(f'   UID: {uid}')
+        print(f'   限制: {limits}')
+        print()
+
+    print('=' * 80 + '\n')
+
+def main():
+    parser = argparse.ArgumentParser(
+        description='波特人使用者管理 CLI (Python 版本 - 避免 google-gax 問題)'
+    )
+    subparsers = parser.add_subparsers(dest='command', help='可用命令')
+
+    # query 命令
+    query_parser = subparsers.add_parser('query', help='查詢使用者資訊')
+    query_parser.add_argument('user', help='使用者 email 或 UID')
+
+    # set-custom 命令
+    set_parser = subparsers.add_parser('set-custom', help='設定自訂 Rate Limit')
+    set_parser.add_argument('user', help='使用者 email 或 UID')
+    set_parser.add_argument('--rpm', type=int, help='每分鐘請求數')
+    set_parser.add_argument('--tpm', type=int, help='每分鐘 Token 數')
+    set_parser.add_argument('--rpd', type=int, help='每日請求數')
+
+    # reset-custom 命令
+    reset_custom_parser = subparsers.add_parser('reset-custom', help='移除自訂限制')
+    reset_custom_parser.add_argument('user', help='使用者 email 或 UID')
+
+    # reset 命令
+    reset_parser = subparsers.add_parser('reset', help='清除 Rate Limit 計數器')
+    reset_parser.add_argument('user', help='使用者 email 或 UID')
+
+    # list 命令
+    subparsers.add_parser('list-custom', help='列出所有有自訂限制的使用者')
+
+    args = parser.parse_args()
+
+    if not args.command:
+        parser.print_help()
+        return
+
+    if args.command == 'query':
+        query_user(args.user)
+    elif args.command == 'set-custom':
+        set_custom_limits(args.user, args.rpm, args.tpm, args.rpd)
+    elif args.command == 'reset-custom':
+        reset_custom_limits(args.user)
+    elif args.command == 'reset':
+        reset_rate_limit(args.user)
+    elif args.command == 'list-custom':
+        list_users_with_custom_limits()
+
+if __name__ == '__main__':
+    main()

package/botrun-c/admin/users.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+# 波特人 Users CLI Wrapper
+# 確保從正確的目錄執行，並載入所需的依賴
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"
+
+# 設定 NODE_PATH 讓 tsx 可以找到 api/node_modules
+export NODE_PATH="$PROJECT_ROOT/api/node_modules:$PROJECT_ROOT/node_modules:$NODE_PATH"
+
+# 執行 CLI
+cd "$PROJECT_ROOT/api" && tsx "$SCRIPT_DIR/users.ts" "$@"