atm-droid 1.0.3 → 1.0.5

package/bin/atm.js

@@ -53,10 +53,14 @@ program
   .description('检查并自动切换（余额不足时）')
   .action(check);
 
+program
+  .command('menu')
+  .description('进入交互式菜单')
+  .action(require('../src/menu'));
+
 program.parse();
 
 if (!process.argv.slice(2).length) {
-  console.log(chalk.cyan('\n  ATM Token Manager v' + pkg.version));
-  console.log(chalk.gray('  跨平台 Factory Token 管理工具\n'));
-  program.outputHelp();
+  // 没有参数时进入交互式菜单
+  require('../src/menu')();
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "atm-droid",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "description": "ATM Token Manager CLI - 跨平台 Factory Token 管理工具",
   "main": "src/index.js",
   "bin": {

package/src/commands.js

@@ -173,7 +173,7 @@ async function switchToken(index) {
     if (activateResult.success) {
       // 写入 auth.json
       if (activateResult.access_token && activateResult.refresh_token) {
-        writeFactoryAuth(activateResult.access_token, activateResult.refresh_token);
+        writeFactoryAuth(activateResult.access_token, activateResult.refresh_token, targetToken.id);
         config.set('currentTokenId', targetToken.id);
         
         switchSpinner.succeed(chalk.green(`已切换到 ${targetToken.email}`));
@@ -349,7 +349,7 @@ async function check() {
     const activateResult = await api.activateToken(target.id);
     
     if (activateResult.success && activateResult.access_token) {
-      writeFactoryAuth(activateResult.access_token, activateResult.refresh_token);
+      writeFactoryAuth(activateResult.access_token, activateResult.refresh_token, target.id);
       config.set('currentTokenId', target.id);
       spinner.succeed(chalk.green(`已自动切换到 ${target.email}`));
     } else {

package/src/config.js

@@ -42,14 +42,15 @@ function readFactoryAuth() {
   return null;
 }
 
-// 写入 Factory auth.json
-function writeFactoryAuth(accessToken, refreshToken) {
+// 写入 Factory auth.json (格式与 Tauri 客户端一致)
+function writeFactoryAuth(accessToken, refreshToken, tokenId) {
   ensureFactoryDir();
   const authPath = getFactoryAuthPath();
   const data = {
-    accessToken,
-    refreshToken,
-    expiresAt: Date.now() + 3600000 // 1小时后过期
+    access_token: accessToken,
+    refresh_token: refreshToken,
+    token_id: tokenId || null,
+    updated_at: Math.floor(Date.now() / 1000)
   };
   fs.writeFileSync(authPath, JSON.stringify(data, null, 2));
   return true;

package/src/daemon.js

@@ -86,7 +86,7 @@ async function refreshCurrentToken() {
     
     const result = await api.activateToken(currentTokenId);
     if (result.success && result.access_token && result.refresh_token) {
-      writeFactoryAuth(result.access_token, result.refresh_token);
+      writeFactoryAuth(result.access_token, result.refresh_token, currentTokenId);
       log(`Token 已更新: ${currentTokenId}`);
     }
   } catch (e) {
@@ -121,7 +121,7 @@ async function autoSwitch() {
       
       if (activateResult.success) {
         config.set('currentTokenId', target.id);
-        writeFactoryAuth(activateResult.access_token, activateResult.refresh_token);
+        writeFactoryAuth(activateResult.access_token, activateResult.refresh_token, target.id);
         log(`自动切换到: ${target.email}`);
       }
     } else {

package/src/menu.js

@@ -0,0 +1,442 @@
+const chalk = require('chalk');
+const inquirer = require('inquirer');
+const { config, writeFactoryAuth, getFactoryAuthPath } = require('./config');
+const api = require('./api');
+const daemon = require('./daemon');
+
+// 格式化数字
+function formatNumber(num) {
+  if (num >= 1000000) return (num / 1000000).toFixed(1) + 'M';
+  if (num >= 1000) return (num / 1000).toFixed(1) + 'K';
+  return num.toString();
+}
+
+// 清屏
+function clearScreen() {
+  console.clear();
+}
+
+// 显示标题
+function showHeader() {
+  console.log(chalk.cyan('╔════════════════════════════════════╗'));
+  console.log(chalk.cyan('║') + chalk.white.bold('     ATM Token Manager v1.0.3     ') + chalk.cyan('║'));
+  console.log(chalk.cyan('╚════════════════════════════════════╝'));
+  console.log();
+}
+
+// 显示当前状态
+async function showStatus() {
+  const sessionToken = config.get('sessionToken');
+  const currentId = config.get('currentTokenId');
+  const autoSwitch = config.get('autoSwitch');
+  const daemonStatus = daemon.getDaemonStatus();
+  
+  if (!sessionToken) {
+    console.log(chalk.yellow('  状态: 未登录\n'));
+    return null;
+  }
+  
+  console.log(chalk.green('  状态: 已登录'));
+  
+  try {
+    const result = await api.getTokens();
+    if (result.success && result.data) {
+      const tokens = result.data;
+      const current = tokens.find(t => t.id === currentId);
+      
+      if (current) {
+        const remaining = (current.quota_total || 0) - (current.quota_used || 0);
+        console.log(chalk.white(`  当前: ${current.email}`));
+        console.log(chalk.white(`  余额: ${formatNumber(remaining)}`));
+      } else {
+        console.log(chalk.yellow('  当前: 未选择账号'));
+      }
+      
+      console.log(chalk.gray(`  账号: ${tokens.length} 个可用`));
+      console.log(chalk.gray(`  自动切换: ${autoSwitch ? '开启' : '关闭'}`));
+      console.log(chalk.gray(`  后台服务: ${daemonStatus.running ? '运行中' : '未运行'}`));
+      console.log();
+      
+      return tokens;
+    }
+  } catch (e) {
+    console.log(chalk.red(`  错误: ${e.message}\n`));
+  }
+  
+  return null;
+}
+
+// 主菜单
+async function mainMenu() {
+  clearScreen();
+  showHeader();
+  
+  const tokens = await showStatus();
+  const sessionToken = config.get('sessionToken');
+  
+  const choices = [];
+  
+  if (!sessionToken) {
+    choices.push({ name: '🔑 登录 (输入激活码)', value: 'login' });
+  } else {
+    choices.push({ name: '📋 查看账号列表', value: 'list' });
+    choices.push({ name: '🔄 切换账号', value: 'switch' });
+    choices.push({ name: '✅ 检查并自动切换', value: 'check' });
+    choices.push(new inquirer.Separator());
+    
+    const daemonStatus = daemon.getDaemonStatus();
+    if (daemonStatus.running) {
+      choices.push({ name: '⏹️  停止后台服务', value: 'stop' });
+    } else {
+      choices.push({ name: '▶️  启动后台服务', value: 'start' });
+    }
+    
+    choices.push(new inquirer.Separator());
+    choices.push({ name: '🚪 退出登录', value: 'logout' });
+  }
+  
+  choices.push(new inquirer.Separator());
+  choices.push({ name: '❌ 退出程序', value: 'exit' });
+  
+  const { action } = await inquirer.prompt([{
+    type: 'list',
+    name: 'action',
+    message: '请选择操作:',
+    choices,
+    pageSize: 15
+  }]);
+  
+  return action;
+}
+
+// 登录
+async function doLogin() {
+  clearScreen();
+  showHeader();
+  
+  const { code } = await inquirer.prompt([{
+    type: 'input',
+    name: 'code',
+    message: '请输入激活码:',
+    validate: input => input.length > 0 || '激活码不能为空'
+  }]);
+  
+  console.log(chalk.gray('\n  正在激活...\n'));
+  
+  try {
+    const result = await api.activate(code);
+    
+    if (result.success) {
+      config.set('sessionToken', result.session_token);
+      config.set('autoSwitch', result.auto_switch || false);
+      
+      console.log(chalk.green('  ✓ 激活成功！'));
+      console.log(chalk.gray(`    可用账号数: ${result.quota || 0}`));
+      
+      if (result.auto_switch) {
+        console.log(chalk.cyan('    自动切换: 已启用'));
+      }
+    } else {
+      console.log(chalk.red(`  ✗ 激活失败: ${result.error || '未知错误'}`));
+    }
+  } catch (e) {
+    console.log(chalk.red(`  ✗ 激活失败: ${e.message}`));
+  }
+  
+  await pause();
+}
+
+// 查看账号列表
+async function doList() {
+  clearScreen();
+  showHeader();
+  console.log(chalk.white.bold('  账号列表:\n'));
+  
+  try {
+    const result = await api.getTokens();
+    
+    if (!result.success) {
+      console.log(chalk.red(`  获取失败: ${result.error}`));
+      await pause();
+      return;
+    }
+    
+    const tokens = result.data || [];
+    const currentId = config.get('currentTokenId');
+    
+    if (tokens.length === 0) {
+      console.log(chalk.yellow('  暂无可用账号'));
+    } else {
+      tokens.forEach((token, index) => {
+        const total = token.quota_total || 0;
+        const used = token.quota_used || 0;
+        const remaining = Math.max(0, total - used);
+        const isCurrent = token.id === currentId;
+        
+        const prefix = isCurrent ? chalk.green('  ► ') : '    ';
+        const indexStr = chalk.gray(`${index + 1}.`);
+        const email = isCurrent ? chalk.green(token.email) : chalk.white(token.email);
+        const quota = remaining > 0 
+          ? chalk.cyan(`(${formatNumber(remaining)})`)
+          : chalk.red('(已用完)');
+        const current = isCurrent ? chalk.green(' [当前]') : '';
+        
+        console.log(`${prefix}${indexStr} ${email} ${quota}${current}`);
+      });
+    }
+    
+    console.log();
+  } catch (e) {
+    console.log(chalk.red(`  获取失败: ${e.message}`));
+  }
+  
+  await pause();
+}
+
+// 切换账号
+async function doSwitch() {
+  clearScreen();
+  showHeader();
+  console.log(chalk.white.bold('  切换账号:\n'));
+  
+  try {
+    const result = await api.getTokens();
+    
+    if (!result.success) {
+      console.log(chalk.red(`  获取失败: ${result.error}`));
+      await pause();
+      return;
+    }
+    
+    const tokens = result.data || [];
+    
+    if (tokens.length === 0) {
+      console.log(chalk.yellow('  暂无可用账号'));
+      await pause();
+      return;
+    }
+    
+    const currentId = config.get('currentTokenId');
+    
+    const choices = tokens.map((t, i) => {
+      const remaining = (t.quota_total || 0) - (t.quota_used || 0);
+      const isCurrent = t.id === currentId;
+      return {
+        name: `${isCurrent ? '► ' : '  '}${t.email} (${formatNumber(remaining)})${isCurrent ? ' [当前]' : ''}`,
+        value: t,
+        short: t.email
+      };
+    });
+    
+    choices.push(new inquirer.Separator());
+    choices.push({ name: '← 返回', value: null });
+    
+    const { selected } = await inquirer.prompt([{
+      type: 'list',
+      name: 'selected',
+      message: '选择要切换的账号:',
+      choices,
+      pageSize: 15
+    }]);
+    
+    if (!selected) return;
+    
+    console.log(chalk.gray(`\n  正在切换到 ${selected.email}...\n`));
+    
+    const activateResult = await api.activateToken(selected.id);
+    
+    if (activateResult.success && activateResult.access_token) {
+      writeFactoryAuth(activateResult.access_token, activateResult.refresh_token, selected.id);
+      config.set('currentTokenId', selected.id);
+      
+      console.log(chalk.green(`  ✓ 已切换到 ${selected.email}`));
+      console.log(chalk.gray(`    auth.json 已更新`));
+    } else {
+      console.log(chalk.red(`  ✗ 切换失败: ${activateResult.error || '未知错误'}`));
+    }
+    
+  } catch (e) {
+    console.log(chalk.red(`  切换失败: ${e.message}`));
+  }
+  
+  await pause();
+}
+
+// 检查并自动切换
+async function doCheck() {
+  clearScreen();
+  showHeader();
+  console.log(chalk.white.bold('  检查账号状态...\n'));
+  
+  try {
+    const result = await api.getTokens();
+    
+    if (!result.success) {
+      console.log(chalk.red(`  检查失败: ${result.error}`));
+      await pause();
+      return;
+    }
+    
+    const tokens = result.data || [];
+    const currentId = config.get('currentTokenId');
+    const current = tokens.find(t => t.id === currentId);
+    
+    if (current) {
+      const remaining = (current.quota_total || 0) - (current.quota_used || 0);
+      
+      if (remaining > 0) {
+        console.log(chalk.green(`  ✓ 当前账号 ${current.email} 余额充足`));
+        console.log(chalk.gray(`    剩余: ${formatNumber(remaining)}`));
+        await pause();
+        return;
+      }
+      
+      console.log(chalk.yellow(`  当前账号余额不足，正在寻找可用账号...\n`));
+    }
+    
+    const available = tokens.filter(t => {
+      const r = (t.quota_total || 0) - (t.quota_used || 0);
+      return r > 0 && t.id !== currentId;
+    });
+    
+    if (available.length === 0) {
+      console.log(chalk.red('  ✗ 所有账号余额已用完'));
+      await pause();
+      return;
+    }
+    
+    // 切换到余额最少的
+    available.sort((a, b) => {
+      const ra = (a.quota_total || 0) - (a.quota_used || 0);
+      const rb = (b.quota_total || 0) - (b.quota_used || 0);
+      return ra - rb;
+    });
+    
+    const target = available[0];
+    const activateResult = await api.activateToken(target.id);
+    
+    if (activateResult.success && activateResult.access_token) {
+      writeFactoryAuth(activateResult.access_token, activateResult.refresh_token, target.id);
+      config.set('currentTokenId', target.id);
+      console.log(chalk.green(`  ✓ 已自动切换到 ${target.email}`));
+    } else {
+      console.log(chalk.red(`  ✗ 切换失败: ${activateResult.error || '未知错误'}`));
+    }
+    
+  } catch (e) {
+    console.log(chalk.red(`  检查失败: ${e.message}`));
+  }
+  
+  await pause();
+}
+
+// 启动后台服务
+async function doStart() {
+  const { spawn } = require('child_process');
+  const path = require('path');
+  
+  console.log(chalk.gray('\n  正在启动后台服务...\n'));
+  
+  const child = spawn(process.execPath, [
+    path.join(__dirname, 'daemon-runner.js')
+  ], {
+    detached: true,
+    stdio: 'ignore'
+  });
+  
+  child.unref();
+  
+  await new Promise(resolve => setTimeout(resolve, 1000));
+  
+  const status = daemon.getDaemonStatus();
+  if (status.running) {
+    console.log(chalk.green(`  ✓ 后台服务已启动 (PID: ${status.pid})`));
+  } else {
+    console.log(chalk.red('  ✗ 启动失败'));
+  }
+  
+  await pause();
+}
+
+// 停止后台服务
+async function doStop() {
+  console.log(chalk.gray('\n  正在停止后台服务...\n'));
+  
+  const result = daemon.stopDaemon();
+  
+  if (result) {
+    console.log(chalk.green('  ✓ 后台服务已停止'));
+  } else {
+    console.log(chalk.yellow('  后台服务未运行'));
+  }
+  
+  await pause();
+}
+
+// 退出登录
+async function doLogout() {
+  const { confirm } = await inquirer.prompt([{
+    type: 'confirm',
+    name: 'confirm',
+    message: '确定要退出登录吗?',
+    default: false
+  }]);
+  
+  if (!confirm) return;
+  
+  try {
+    await api.unbind();
+  } catch (e) {
+    // 忽略
+  }
+  
+  config.clear();
+  console.log(chalk.green('\n  ✓ 已退出登录\n'));
+  await pause();
+}
+
+// 暂停等待用户按键
+async function pause() {
+  await inquirer.prompt([{
+    type: 'input',
+    name: 'continue',
+    message: '按回车键继续...'
+  }]);
+}
+
+// 主循环
+async function menu() {
+  while (true) {
+    const action = await mainMenu();
+    
+    switch (action) {
+      case 'login':
+        await doLogin();
+        break;
+      case 'list':
+        await doList();
+        break;
+      case 'switch':
+        await doSwitch();
+        break;
+      case 'check':
+        await doCheck();
+        break;
+      case 'start':
+        await doStart();
+        break;
+      case 'stop':
+        await doStop();
+        break;
+      case 'logout':
+        await doLogout();
+        break;
+      case 'exit':
+        clearScreen();
+        console.log(chalk.cyan('\n  再见！\n'));
+        process.exit(0);
+    }
+  }
+}
+
+module.exports = menu;