botrun-horse 2.29.2 → 2.31.0

package/botrun-c/api/scripts/query-user-info.ts

@@ -0,0 +1,304 @@
+#!/usr/bin/env tsx
+/**
+ * 查詢特定使用者的訂閱和 Rate Limit 資訊
+ */
+
+import admin from 'firebase-admin';
+import { Firestore } from '@google-cloud/firestore';
+import { readFileSync, existsSync } from 'fs';
+import { join } from 'path';
+import { Plan, type PlanType } from '../src/rate-limit/domain/entities/Plan.js';
+import { CounterKey } from '../src/rate-limit/domain/entities/UsageCounter.js';
+
+// 找到認證金鑰
+const possiblePaths = [
+  join(process.cwd(), '..', 'keys', 'firebase-admin-key.json'),
+  join(process.cwd(), '..', 'keys', 'compute-sa-key.json'),
+  join(process.cwd(), 'keys', 'firebase-admin-key.json'),
+  join(process.cwd(), 'keys', 'compute-sa-key.json'),
+];
+
+let keyFilePath: string | undefined;
+for (const path of possiblePaths) {
+  if (existsSync(path)) {
+    keyFilePath = path;
+    break;
+  }
+}
+
+// 初始化 Firebase Admin
+if (!admin.apps.length) {
+  try {
+    if (keyFilePath) {
+      const serviceAccount = JSON.parse(readFileSync(keyFilePath, 'utf-8'));
+      admin.initializeApp({ credential: admin.credential.cert(serviceAccount) });
+    } else {
+      admin.initializeApp(); // Use Application Default Credentials
+    }
+  } catch (error) {
+    console.error('⚠️  Firebase Admin 初始化失敗:', error);
+    throw error;
+  }
+}
+
+// 初始化 Firestore（使用相同的金鑰）
+const db = keyFilePath
+  ? new Firestore({ keyFilename: keyFilePath })
+  : new Firestore();
+
+async function getUserIdFromEmail(email: string): Promise<string | null> {
+  try {
+    const userRecord = await admin.auth().getUserByEmail(email);
+    return userRecord.uid;
+  } catch (error) {
+    console.error(`❌ 找不到 email 對應的使用者: ${email}`);
+    return null;
+  }
+}
+
+async function getSubscriptionInfo(userId: string) {
+  try {
+    const subscriptionDoc = await db.collection('user_subscriptions').doc(userId).get();
+
+    if (subscriptionDoc.exists) {
+      return subscriptionDoc.data();
+    } else {
+      return null;
+    }
+  } catch (error) {
+    console.error('❌ 讀取訂閱資訊失敗:', error);
+    return null;
+  }
+}
+
+async function getRateLimitCounters(userId: string) {
+  try {
+    const countersSnapshot = await db.collection('rate_limit_counters')
+      .where('__name__', '>=', `rpm:${userId}`)
+      .where('__name__', '<=', `rpm:${userId}\uf8ff`)
+      .get();
+
+    const tpmSnapshot = await db.collection('rate_limit_counters')
+      .where('__name__', '>=', `tpm:${userId}`)
+      .where('__name__', '<=', `tpm:${userId}\uf8ff`)
+      .get();
+
+    const rpdSnapshot = await db.collection('rate_limit_counters')
+      .where('__name__', '>=', `rpd:${userId}`)
+      .where('__name__', '<=', `rpd:${userId}\uf8ff`)
+      .get();
+
+    const counters: any[] = [];
+
+    countersSnapshot.forEach(doc => {
+      counters.push({ id: doc.id, ...doc.data() });
+    });
+
+    tpmSnapshot.forEach(doc => {
+      counters.push({ id: doc.id, ...doc.data() });
+    });
+
+    rpdSnapshot.forEach(doc => {
+      counters.push({ id: doc.id, ...doc.data() });
+    });
+
+    return counters;
+  } catch (error) {
+    console.error('❌ 讀取 Rate Limit 計數器失敗:', error);
+    return [];
+  }
+}
+
+async function getUserPlan(userId: string): Promise<PlanType> {
+  try {
+    const userDoc = await db.collection('users').doc(userId).get();
+    if (userDoc.exists) {
+      const data = userDoc.data();
+      const plan = data?.plan as PlanType;
+      if (plan && ['Lite', 'Max-1', 'Max-2'].includes(plan)) {
+        return plan;
+      }
+    }
+  } catch (error) {
+    // 忽略錯誤，使用預設方案
+  }
+  return 'Lite';
+}
+
+async function getCustomLimits(userId: string) {
+  try {
+    const userDoc = await db.collection('users').doc(userId).get();
+    if (userDoc.exists) {
+      const data = userDoc.data();
+      return data?.customLimits || null;
+    }
+  } catch (error) {
+    console.error('❌ 讀取 customLimits 失敗:', error);
+  }
+  return null;
+}
+
+async function getCurrentCounters(userId: string) {
+  const now = new Date();
+  const rpmKey = CounterKey.create(userId, 'rpm', now);
+  const tpmKey = CounterKey.create(userId, 'tpm', now);
+  const rpdKey = CounterKey.create(userId, 'rpd', now);
+
+  const results = {
+    rpm: 0,
+    tpm: 0,
+    rpd: 0
+  };
+
+  try {
+    const rpmDoc = await db.collection('rate_limit_counters').doc(rpmKey).get();
+    if (rpmDoc.exists) {
+      results.rpm = rpmDoc.data()?.count || 0;
+    }
+
+    const tpmDoc = await db.collection('rate_limit_counters').doc(tpmKey).get();
+    if (tpmDoc.exists) {
+      results.tpm = tpmDoc.data()?.count || 0;
+    }
+
+    const rpdDoc = await db.collection('rate_limit_counters').doc(rpdKey).get();
+    if (rpdDoc.exists) {
+      results.rpd = rpdDoc.data()?.count || 0;
+    }
+  } catch (error) {
+    console.error('❌ 讀取當前計數器失敗:', error);
+  }
+
+  return results;
+}
+
+async function queryUserInfo(emailOrUid: string) {
+  console.log('==================================================');
+  console.log('  使用者資訊查詢');
+  console.log('==================================================\n');
+
+  let userId: string;
+  let email: string;
+
+  // 判斷輸入是 email 還是 UID
+  if (emailOrUid.includes('@')) {
+    email = emailOrUid;
+    console.log(`📧 輸入 Email: ${email}\n`);
+    console.log('🔍 正在從 Firebase Auth 查詢 UID...');
+
+    const uid = await getUserIdFromEmail(email);
+    if (!uid) {
+      console.log('\n❌ 找不到對應的使用者');
+      process.exit(1);
+    }
+    userId = uid;
+    console.log(`✅ 找到使用者 UID: ${userId}\n`);
+  } else {
+    userId = emailOrUid;
+    console.log(`👤 使用者 UID: ${userId}\n`);
+
+    // 嘗試從 Firebase Auth 取得 email
+    try {
+      const userRecord = await admin.auth().getUser(userId);
+      email = userRecord.email || '未設定';
+      console.log(`📧 Email: ${email}\n`);
+    } catch {
+      email = '未知';
+    }
+  }
+
+  // 1. 查詢訂閱資訊
+  console.log('==================================================');
+  console.log('📋 訂閱資訊 (Subscription Info)');
+  console.log('==================================================\n');
+
+  const subscription = await getSubscriptionInfo(userId);
+  if (subscription) {
+    console.log('✅ 找到訂閱記錄\n');
+    console.log('訂閱資料:');
+    console.log(JSON.stringify(subscription, null, 2));
+  } else {
+    console.log('❌ 沒有訂閱記錄');
+  }
+
+  // 2. 查詢使用者方案與自訂限制
+  console.log('\n==================================================');
+  console.log('📊 Rate Limit 方案與當前使用狀況');
+  console.log('==================================================\n');
+
+  const planType = await getUserPlan(userId);
+  const customLimits = await getCustomLimits(userId);
+  const plan = Plan.create(planType, customLimits);
+
+  console.log(`當前方案: ${planType}`);
+
+  if (customLimits) {
+    console.log(`\n⚙️  自訂限制:`);
+    if (customLimits.rpm !== undefined) {
+      console.log(`  - RPM: ${customLimits.rpm} (覆寫預設值)`);
+    }
+    if (customLimits.tpm !== undefined) {
+      console.log(`  - TPM: ${customLimits.tpm.toLocaleString()} (覆寫預設值)`);
+    }
+    if (customLimits.rpd !== undefined) {
+      console.log(`  - RPD: ${customLimits.rpd} (覆寫預設值)`);
+    }
+  }
+
+  console.log(`\n${customLimits ? '實際生效限制' : '方案限制'}:`);
+  console.log(`  - RPM (每分鐘請求數): ${plan.limits.rpm}`);
+  console.log(`  - TPM (每分鐘 Token 數): ${plan.limits.tpm.toLocaleString()}`);
+  console.log(`  - RPD (每日請求數): ${plan.limits.rpd}`);
+
+  // 3. 查詢當前計數
+  console.log('\n當前使用狀況 (目前時段):');
+  const currentCounters = await getCurrentCounters(userId);
+
+  console.log(`  - RPM: ${currentCounters.rpm} / ${plan.limits.rpm} (${Math.round((currentCounters.rpm / plan.limits.rpm) * 100)}%)`);
+  console.log(`  - TPM: ${currentCounters.tpm.toLocaleString()} / ${plan.limits.tpm.toLocaleString()} (${Math.round((currentCounters.tpm / plan.limits.tpm) * 100)}%)`);
+  console.log(`  - RPD: ${currentCounters.rpd} / ${plan.limits.rpd} (${Math.round((currentCounters.rpd / plan.limits.rpd) * 100)}%)`);
+
+  // 判斷是否被限制
+  const isBlocked =
+    currentCounters.rpm >= plan.limits.rpm ||
+    currentCounters.tpm >= plan.limits.tpm ||
+    currentCounters.rpd >= plan.limits.rpd;
+
+  if (isBlocked) {
+    console.log('\n⛔ 狀態: 已達限制 (BLOCKED)');
+  } else {
+    console.log('\n✅ 狀態: 正常 (OK)');
+  }
+
+  // 4. 查詢所有歷史計數器
+  console.log('\n==================================================');
+  console.log('📊 所有 Rate Limit 計數器記錄');
+  console.log('==================================================\n');
+
+  const allCounters = await getRateLimitCounters(userId);
+
+  if (allCounters.length > 0) {
+    console.log(`找到 ${allCounters.length} 個計數器記錄:\n`);
+
+    allCounters.forEach(counter => {
+      const parsed = CounterKey.parse(counter.id);
+      if (parsed) {
+        console.log(`  🔸 ${parsed.type.toUpperCase()}: ${counter.count} (時段: ${parsed.timestamp})`);
+      } else {
+        console.log(`  🔸 ${counter.id}: ${counter.count}`);
+      }
+    });
+  } else {
+    console.log('❌ 沒有找到任何計數器記錄');
+  }
+
+  console.log('\n==================================================\n');
+}
+
+// 從命令列參數取得要查詢的使用者
+const targetUser = process.argv[2] || 'aidoer@cameo.tw';
+
+queryUserInfo(targetUser).catch(error => {
+  console.error('❌ 執行失敗:', error);
+  process.exit(1);
+});

package/botrun-c/api/scripts/reset-rate-limit.ts

@@ -0,0 +1,206 @@
+#!/usr/bin/env tsx
+/**
+ * Rate Limit Reset Script
+ * 清除指定使用者的所有 Rate Limit 計數器
+ * 支援 email 或 UID 作為輸入
+ */
+
+import admin from 'firebase-admin';
+import { readFileSync, existsSync } from 'fs';
+import { join } from 'path';
+import { CounterKey } from '../src/rate-limit/domain/entities/UsageCounter.js';
+
+// 找到認證金鑰
+const possiblePaths = [
+  join(process.cwd(), '..', 'keys', 'firebase-admin-key.json'),
+  join(process.cwd(), '..', 'keys', 'compute-sa-key.json'),
+  join(process.cwd(), 'keys', 'firebase-admin-key.json'),
+  join(process.cwd(), 'keys', 'compute-sa-key.json'),
+];
+
+let keyFilePath: string | undefined;
+for (const path of possiblePaths) {
+  if (existsSync(path)) {
+    keyFilePath = path;
+    break;
+  }
+}
+
+// 初始化 Firebase Admin
+if (!admin.apps.length) {
+  try {
+    if (keyFilePath) {
+      const serviceAccount = JSON.parse(readFileSync(keyFilePath, 'utf-8'));
+      admin.initializeApp({ credential: admin.credential.cert(serviceAccount) });
+    } else {
+      admin.initializeApp(); // Use Application Default Credentials
+    }
+  } catch (error) {
+    console.error('⚠️  Firebase Admin 初始化失敗:', error);
+  }
+}
+
+// 使用 Firebase Admin 的 Firestore
+const db = admin.firestore();
+
+/**
+ * 將 email 轉換成 UID
+ * 如果輸入已經是 UID，直接回傳
+ */
+async function resolveUserIdFromEmailOrUid(input: string): Promise<string> {
+  // 如果包含 @ 符號，視為 email
+  if (input.includes('@')) {
+    try {
+      const userRecord = await admin.auth().getUserByEmail(input);
+      console.log(`✅ 從 email 找到使用者: ${input} → ${userRecord.uid}`);
+      console.log('');
+      return userRecord.uid;
+    } catch (error: any) {
+      if (error.code === 'auth/user-not-found') {
+        console.error(`❌ 找不到 email 對應的使用者: ${input}`);
+      } else {
+        console.error(`❌ 查詢使用者失敗:`, error);
+      }
+      throw error;
+    }
+  }
+
+  // 否則視為 UID，直接回傳
+  return input;
+}
+
+/**
+ * 取得指定使用者的所有計數器文件
+ */
+async function getUserCounters(userId: string): Promise<string[]> {
+  const keys: string[] = [];
+
+  try {
+    const countersSnapshot = await db.collection('rate_limit_counters').get();
+
+    countersSnapshot.forEach(doc => {
+      const key = doc.id;
+      const parsed = CounterKey.parse(key);
+      if (parsed && parsed.userId === userId) {
+        keys.push(key);
+      }
+    });
+  } catch (error) {
+    console.error('❌ 無法讀取計數器:', error);
+    throw error;
+  }
+
+  return keys;
+}
+
+/**
+ * 刪除指定的計數器文件
+ */
+async function deleteCounter(key: string): Promise<void> {
+  try {
+    await db.collection('rate_limit_counters').doc(key).delete();
+  } catch (error) {
+    console.error(`  ❌ 無法刪除 ${key}:`, error);
+    throw error;
+  }
+}
+
+/**
+ * 顯示計數器資訊
+ */
+async function displayCounterInfo(key: string): Promise<void> {
+  const parsed = CounterKey.parse(key);
+  if (!parsed) {
+    console.log(`  ⚠️  無法解析鍵值: ${key}`);
+    return;
+  }
+
+  try {
+    const doc = await db.collection('rate_limit_counters').doc(key).get();
+    if (doc.exists) {
+      const data = doc.data();
+      const count = data?.count || 0;
+      const typeLabel = parsed.type.toUpperCase();
+      console.log(`  🗑️  ${typeLabel}: ${count} (鍵值: ${key})`);
+    }
+  } catch (error) {
+    console.log(`  ⚠️  無法讀取 ${key}`);
+  }
+}
+
+/**
+ * 主程式
+ */
+async function main() {
+  try {
+    // 取得使用者輸入（從命令列參數）
+    const userInput = process.argv[2];
+
+    if (!userInput) {
+      console.error('❌ 錯誤: 請提供使用者 email 或 UID');
+      console.error('');
+      console.error('使用方式:');
+      console.error('  tsx reset-rate-limit.ts <user_email_or_uid>');
+      console.error('');
+      console.error('範例:');
+      console.error('  tsx reset-rate-limit.ts cbh@cameo.tw');
+      console.error('  tsx reset-rate-limit.ts bVD7t0Fb7SMX3LPpMzVzCLCkd4o2');
+      process.exit(1);
+    }
+
+    console.log(`🔍 正在搜尋使用者...`);
+    console.log(`📧 輸入: ${userInput}`);
+    console.log('');
+
+    // 將 email 或 UID 轉換成 UID
+    const userId = await resolveUserIdFromEmailOrUid(userInput);
+
+    console.log(`🔎 正在搜尋 Rate Limit 計數器...`);
+    console.log(`👤 UID: ${userId}`);
+    console.log('');
+
+    // 取得所有計數器
+    const keys = await getUserCounters(userId);
+
+    if (keys.length === 0) {
+      console.log('✅ 此使用者沒有任何 Rate Limit 記錄');
+      console.log('');
+      return;
+    }
+
+    console.log(`找到 ${keys.length} 個計數器:`);
+    console.log('');
+
+    // 顯示所有計數器資訊
+    for (const key of keys) {
+      await displayCounterInfo(key);
+    }
+
+    console.log('');
+    console.log(`🗑️  準備刪除 ${keys.length} 個計數器...`);
+    console.log('');
+
+    // 刪除所有計數器
+    let deleted = 0;
+    for (const key of keys) {
+      await deleteCounter(key);
+      deleted++;
+    }
+
+    console.log('');
+    console.log('✅ Rate Limit 已成功清除！');
+    console.log(`📊 共刪除 ${deleted} 個計數器`);
+    console.log('');
+    console.log('🎉 使用者現在可以重新開始使用服務');
+    console.log('');
+
+  } catch (error) {
+    console.error('');
+    console.error('❌ 執行失敗:', error);
+    console.error('');
+    process.exit(1);
+  }
+}
+
+// 執行
+main();

package/botrun-c/api/scripts/reset-trial-subscription.ts

@@ -0,0 +1,166 @@
+#!/usr/bin/env tsx
+/**
+ * Reset Trial Subscription Script
+ * 重置指定使用者的試用期訂閱（刪除後自動重建 14 天試用）
+ * 支援 email 或 UID 作為輸入
+ */
+
+import { Firestore } from '@google-cloud/firestore';
+import admin from 'firebase-admin';
+import { readFileSync, existsSync } from 'fs';
+import { join } from 'path';
+
+const db = new Firestore();
+
+// 初始化 Firebase Admin
+if (!admin.apps.length) {
+  try {
+    const serviceAccountPath = join(process.cwd(), 'keys', 'firebase-admin-key.json');
+    if (existsSync(serviceAccountPath)) {
+      const serviceAccount = JSON.parse(readFileSync(serviceAccountPath, 'utf-8'));
+      admin.initializeApp({ credential: admin.credential.cert(serviceAccount) });
+    } else {
+      admin.initializeApp(); // Use Application Default Credentials
+    }
+  } catch (error) {
+    console.error('⚠️  Firebase Admin 初始化失敗:', error);
+  }
+}
+
+/**
+ * 將 email 轉換成 UID
+ * 如果輸入已經是 UID，直接回傳
+ */
+async function resolveUserIdFromEmailOrUid(input: string): Promise<string> {
+  // 如果包含 @ 符號，視為 email
+  if (input.includes('@')) {
+    try {
+      const userRecord = await admin.auth().getUserByEmail(input);
+      console.log(`✅ 從 email 找到使用者: ${input} → ${userRecord.uid}`);
+      console.log('');
+      return userRecord.uid;
+    } catch (error: any) {
+      if (error.code === 'auth/user-not-found') {
+        console.error(`❌ 找不到 email 對應的使用者: ${input}`);
+      } else {
+        console.error(`❌ 查詢使用者失敗:`, error);
+      }
+      throw error;
+    }
+  }
+
+  // 否則視為 UID，直接回傳
+  return input;
+}
+
+/**
+ * 顯示訂閱資訊
+ */
+async function displaySubscriptionInfo(userId: string): Promise<void> {
+  try {
+    const docRef = db.collection('user_subscriptions').doc(userId);
+    const docSnap = await docRef.get();
+
+    if (!docSnap.exists) {
+      console.log('  ℹ️  此使用者目前沒有訂閱記錄');
+      return;
+    }
+
+    const data = docSnap.data();
+    console.log('  📋 目前訂閱資訊:');
+    console.log(`     狀態: ${data?.status}`);
+    console.log(`     方案: ${data?.plan}`);
+
+    if (data?.trialStartDate) {
+      console.log(`     試用開始: ${data.trialStartDate.toDate().toISOString()}`);
+    }
+
+    if (data?.trialEndDate) {
+      console.log(`     試用結束: ${data.trialEndDate.toDate().toISOString()}`);
+    }
+
+    if (data?.subscriptionStartDate) {
+      console.log(`     訂閱開始: ${data.subscriptionStartDate.toDate().toISOString()}`);
+    }
+
+    if (data?.subscriptionEndDate) {
+      console.log(`     訂閱結束: ${data.subscriptionEndDate.toDate().toISOString()}`);
+    }
+  } catch (error) {
+    console.log('  ⚠️  無法讀取訂閱資訊');
+  }
+}
+
+/**
+ * 刪除訂閱記錄
+ */
+async function deleteSubscription(userId: string): Promise<void> {
+  try {
+    const docRef = db.collection('user_subscriptions').doc(userId);
+    await docRef.delete();
+  } catch (error) {
+    console.error('  ❌ 無法刪除訂閱記錄:', error);
+    throw error;
+  }
+}
+
+/**
+ * 主程式
+ */
+async function main() {
+  try {
+    // 取得使用者輸入（從命令列參數）
+    const userInput = process.argv[2];
+
+    if (!userInput) {
+      console.error('❌ 錯誤: 請提供使用者 email 或 UID');
+      console.error('');
+      console.error('使用方式:');
+      console.error('  tsx reset-trial-subscription.ts <user_email_or_uid>');
+      console.error('');
+      console.error('範例:');
+      console.error('  tsx reset-trial-subscription.ts cbh@cameo.tw');
+      console.error('  tsx reset-trial-subscription.ts bVD7t0Fb7SMX3LPpMzVzCLCkd4o2');
+      process.exit(1);
+    }
+
+    console.log('================================================');
+    console.log('  重置試用期訂閱');
+    console.log('================================================');
+    console.log('');
+    console.log(`🔍 正在搜尋使用者...`);
+    console.log(`📧 輸入: ${userInput}`);
+    console.log('');
+
+    // 將 email 或 UID 轉換成 UID
+    const userId = await resolveUserIdFromEmailOrUid(userInput);
+
+    console.log(`📋 檢查目前訂閱狀態...`);
+    console.log(`👤 UID: ${userId}`);
+    console.log('');
+
+    // 顯示目前訂閱資訊
+    await displaySubscriptionInfo(userId);
+    console.log('');
+
+    // 刪除訂閱記錄
+    console.log('🗑️  刪除訂閱記錄...');
+    await deleteSubscription(userId);
+    console.log('');
+
+    console.log('✅ 訂閱記錄已成功刪除！');
+    console.log('');
+    console.log('🎉 使用者下次使用時將自動建立新的 14 天試用期');
+    console.log('   （透過 SubscriptionService.getOrCreateTrialSubscription）');
+    console.log('');
+
+  } catch (error) {
+    console.error('');
+    console.error('❌ 執行失敗:', error);
+    console.error('');
+    process.exit(1);
+  }
+}
+
+// 執行
+main();