atm-droid 1.0.0

package/README.md

@@ -0,0 +1,85 @@
+# ATM Client CLI
+
+跨平台 Factory Token 管理工具，支持 Windows / macOS / Linux。
+
+## 安装
+
+```bash
+# 全局安装
+npm install -g atm-client
+
+# 或本地安装后链接
+cd atm-cli
+npm install
+npm link
+```
+
+## 使用
+
+```bash
+# 查看帮助
+atm --help
+
+# 登录（使用激活码）
+atm login ABC-DEF-123
+
+# 查看账号列表
+atm list
+
+# 切换账号（交互式选择）
+atm switch
+
+# 切换到指定账号
+atm switch 1
+
+# 查看当前状态
+atm status
+
+# 检查余额并自动切换
+atm check
+
+# 启动后台服务（实时同步）
+atm start
+
+# 停止后台服务
+atm stop
+
+# 退出登录
+atm logout
+```
+
+## 功能
+
+- ✅ 激活码登录
+- ✅ 多账号管理
+- ✅ 手动切换账号
+- ✅ 自动切换（余额用完时）
+- ✅ 后台服务（WebSocket 实时同步）
+- ✅ 自动写入 ~/.factory/auth.json
+
+## 后台服务
+
+启动后台服务后，会保持与服务器的 WebSocket 连接：
+- 实时接收 Token 更新
+- 自动刷新 auth.json
+- 余额用完自动切换
+
+```bash
+atm start   # 启动
+atm stop    # 停止
+atm status  # 查看状态
+```
+
+## 配合 crontab 使用（Linux/macOS）
+
+如果不想运行后台服务，可以用 crontab 定时检查：
+
+```bash
+# 每分钟检查一次
+* * * * * /usr/local/bin/atm check
+```
+
+## 配置文件位置
+
+- 配置: `~/.config/atm-client/config.json`
+- Factory 认证: `~/.factory/auth.json`

package/bin/atm.js

@@ -0,0 +1,62 @@
+#!/usr/bin/env node
+
+const { program } = require('commander');
+const chalk = require('chalk');
+const pkg = require('../package.json');
+const { login, logout, list, switchToken, status, start, stop, check } = require('../src/commands');
+
+program
+  .name('atm')
+  .description('ATM Token Manager - Factory Token 管理工具')
+  .version(pkg.version);
+
+program
+  .command('login <code>')
+  .description('使用激活码登录')
+  .action(login);
+
+program
+  .command('logout')
+  .description('退出登录并清除数据')
+  .action(logout);
+
+program
+  .command('list')
+  .alias('ls')
+  .description('列出所有可用账号')
+  .action(list);
+
+program
+  .command('switch [index]')
+  .alias('sw')
+  .description('切换到指定账号')
+  .action(switchToken);
+
+program
+  .command('status')
+  .alias('st')
+  .description('查看当前状态')
+  .action(status);
+
+program
+  .command('start')
+  .description('启动后台服务（实时同步）')
+  .action(start);
+
+program
+  .command('stop')
+  .description('停止后台服务')
+  .action(stop);
+
+program
+  .command('check')
+  .description('检查并自动切换（余额不足时）')
+  .action(check);
+
+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();
+}

package/package.json

@@ -0,0 +1,24 @@
+{
+  "name": "atm-droid",
+  "version": "1.0.0",
+  "description": "ATM Token Manager CLI - 跨平台 Factory Token 管理工具",
+  "main": "src/index.js",
+  "bin": {
+    "atm": "./bin/atm.js"
+  },
+  "scripts": {
+    "start": "node bin/atm.js"
+  },
+  "keywords": ["factory", "token", "manager", "cli"],
+  "author": "ATM Team",
+  "license": "MIT",
+  "dependencies": {
+    "commander": "^11.1.0",
+    "chalk": "^4.1.2",
+    "inquirer": "^8.2.6",
+    "ws": "^8.14.2",
+    "node-fetch": "^2.7.0",
+    "ora": "^5.4.1",
+    "conf": "^10.2.0"
+  }
+}

package/src/api.js

@@ -0,0 +1,104 @@
+const fetch = require('node-fetch');
+const { config, getDeviceId } = require('./config');
+
+const getServerUrl = () => config.get('serverUrl');
+
+// 激活码登录
+async function activate(code) {
+  const deviceId = getDeviceId();
+  const response = await fetch(`${getServerUrl()}/api/activate`, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({ code, device_id: deviceId })
+  });
+  return response.json();
+}
+
+// 获取 Token 列表
+async function getTokens() {
+  const sessionToken = config.get('sessionToken');
+  const deviceId = getDeviceId();
+  
+  if (!sessionToken) {
+    throw new Error('未登录');
+  }
+  
+  const response = await fetch(`${getServerUrl()}/api/tokens`, {
+    headers: {
+      'Authorization': `Bearer ${sessionToken}`,
+      'X-Device-ID': deviceId
+    }
+  });
+  
+  const result = await response.json();
+  
+  // 如果返回加密数据，需要解密
+  if (result.success && result.data && result.iv) {
+    // 简化版：假设服务器返回明文（或添加解密逻辑）
+    return result;
+  }
+  
+  return result;
+}
+
+// 激活指定 Token
+async function activateToken(tokenId) {
+  const sessionToken = config.get('sessionToken');
+  const deviceId = getDeviceId();
+  
+  const response = await fetch(`${getServerUrl()}/api/tokens/activate`, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+      'Authorization': `Bearer ${sessionToken}`,
+      'X-Device-ID': deviceId
+    },
+    body: JSON.stringify({ token_id: tokenId })
+  });
+  
+  return response.json();
+}
+
+// 心跳
+async function heartbeat() {
+  const sessionToken = config.get('sessionToken');
+  const deviceId = getDeviceId();
+  
+  if (!sessionToken) return { valid: false };
+  
+  const response = await fetch(`${getServerUrl()}/api/heartbeat`, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+      'Authorization': `Bearer ${sessionToken}`,
+      'X-Device-ID': deviceId
+    }
+  });
+  
+  return response.json();
+}
+
+// 解绑设备
+async function unbind() {
+  const sessionToken = config.get('sessionToken');
+  const deviceId = getDeviceId();
+  
+  const response = await fetch(`${getServerUrl()}/api/unbind`, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+      'Authorization': `Bearer ${sessionToken}`,
+      'X-Device-ID': deviceId
+    }
+  });
+  
+  return response.json();
+}
+
+module.exports = {
+  activate,
+  getTokens,
+  activateToken,
+  heartbeat,
+  unbind
+};

package/src/commands.js

@@ -0,0 +1,373 @@
+const chalk = require('chalk');
+const ora = require('ora');
+const inquirer = require('inquirer');
+const { spawn } = require('child_process');
+const path = require('path');
+const { config, writeFactoryAuth, readFactoryAuth, 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();
+}
+
+// 登录
+async function login(code) {
+  const spinner = ora('正在激活...').start();
+  
+  try {
+    const result = await api.activate(code);
+    
+    if (result.success) {
+      config.set('sessionToken', result.session_token);
+      config.set('autoSwitch', result.auto_switch || false);
+      
+      spinner.succeed(chalk.green('激活成功！'));
+      console.log(chalk.gray(`  会话有效期: ${new Date(result.expires_at * 1000).toLocaleString()}`));
+      console.log(chalk.gray(`  可用账号数: ${result.quota || 0}`));
+      
+      if (result.auto_switch) {
+        console.log(chalk.cyan('  自动切换: 已启用'));
+      }
+      
+      console.log(chalk.yellow('\n  提示: 使用 atm list 查看账号列表'));
+    } else {
+      spinner.fail(chalk.red('激活失败: ' + (result.error || '未知错误')));
+    }
+  } catch (e) {
+    spinner.fail(chalk.red('激活失败: ' + e.message));
+  }
+}
+
+// 退出
+async function logout() {
+  const { confirm } = await inquirer.prompt([{
+    type: 'confirm',
+    name: 'confirm',
+    message: '确定要退出登录吗？',
+    default: false
+  }]);
+  
+  if (!confirm) return;
+  
+  const spinner = ora('正在退出...').start();
+  
+  try {
+    await api.unbind();
+  } catch (e) {
+    // 忽略错误
+  }
+  
+  config.clear();
+  spinner.succeed(chalk.green('已退出登录'));
+}
+
+// 列出账号
+async function list() {
+  const spinner = ora('获取账号列表...').start();
+  
+  try {
+    const result = await api.getTokens();
+    
+    if (!result.success) {
+      spinner.fail(chalk.red('获取失败: ' + (result.error || '未知错误')));
+      return;
+    }
+    
+    const tokens = result.data || [];
+    const currentId = config.get('currentTokenId');
+    
+    spinner.stop();
+    
+    if (tokens.length === 0) {
+      console.log(chalk.yellow('\n  暂无可用账号'));
+      return;
+    }
+    
+    console.log(chalk.cyan('\n  可用账号列表:\n'));
+    
+    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) : 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(chalk.gray('\n  使用 atm switch <序号> 切换账号'));
+    
+  } catch (e) {
+    spinner.fail(chalk.red('获取失败: ' + e.message));
+  }
+}
+
+// 切换账号
+async function switchToken(index) {
+  const spinner = ora('获取账号列表...').start();
+  
+  try {
+    const result = await api.getTokens();
+    
+    if (!result.success) {
+      spinner.fail(chalk.red('获取失败: ' + (result.error || '未知错误')));
+      return;
+    }
+    
+    const tokens = result.data || [];
+    
+    if (tokens.length === 0) {
+      spinner.fail(chalk.yellow('暂无可用账号'));
+      return;
+    }
+    
+    spinner.stop();
+    
+    let targetToken;
+    
+    if (index) {
+      // 指定序号
+      const idx = parseInt(index) - 1;
+      if (idx < 0 || idx >= tokens.length) {
+        console.log(chalk.red('无效的序号'));
+        return;
+      }
+      targetToken = tokens[idx];
+    } else {
+      // 交互式选择
+      const choices = tokens.map((t, i) => {
+        const remaining = (t.quota_total || 0) - (t.quota_used || 0);
+        return {
+          name: `${t.email} (${formatNumber(remaining)})`,
+          value: t,
+          short: t.email
+        };
+      });
+      
+      const { selected } = await inquirer.prompt([{
+        type: 'list',
+        name: 'selected',
+        message: '选择要切换的账号:',
+        choices
+      }]);
+      
+      targetToken = selected;
+    }
+    
+    // 执行切换
+    const switchSpinner = ora(`正在切换到 ${targetToken.email}...`).start();
+    
+    const activateResult = await api.activateToken(targetToken.id);
+    
+    if (activateResult.success) {
+      // 写入 auth.json
+      if (activateResult.access_token && activateResult.refresh_token) {
+        writeFactoryAuth(activateResult.access_token, activateResult.refresh_token);
+        config.set('currentTokenId', targetToken.id);
+        
+        switchSpinner.succeed(chalk.green(`已切换到 ${targetToken.email}`));
+        console.log(chalk.gray(`  auth.json 已更新: ${getFactoryAuthPath()}`));
+      } else {
+        switchSpinner.fail(chalk.red('切换失败: 未获取到 Token'));
+      }
+    } else {
+      switchSpinner.fail(chalk.red('切换失败: ' + (activateResult.error || '未知错误')));
+    }
+    
+  } catch (e) {
+    spinner.fail(chalk.red('切换失败: ' + e.message));
+  }
+}
+
+// 查看状态
+async function status() {
+  const sessionToken = config.get('sessionToken');
+  
+  console.log(chalk.cyan('\n  ATM Token Manager 状态\n'));
+  
+  // 登录状态
+  if (sessionToken) {
+    console.log(chalk.green('  ✓ 已登录'));
+  } else {
+    console.log(chalk.red('  ✗ 未登录'));
+    console.log(chalk.gray('    使用 atm login <激活码> 登录'));
+    return;
+  }
+  
+  // 当前账号
+  const currentId = config.get('currentTokenId');
+  if (currentId) {
+    try {
+      const result = await api.getTokens();
+      if (result.success && result.data) {
+        const current = result.data.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)}`));
+        }
+      }
+    } catch (e) {
+      // 忽略
+    }
+  } else {
+    console.log(chalk.yellow('  当前账号: 未选择'));
+    console.log(chalk.gray('    使用 atm switch 选择账号'));
+  }
+  
+  // 自动切换
+  const autoSwitch = config.get('autoSwitch');
+  console.log(chalk.white(`  自动切换: ${autoSwitch ? '已启用' : '未启用'}`));
+  
+  // 后台服务
+  const daemonStatus = daemon.getDaemonStatus();
+  if (daemonStatus.running) {
+    console.log(chalk.green(`  后台服务: 运行中 (PID: ${daemonStatus.pid})`));
+  } else {
+    console.log(chalk.gray('  后台服务: 未运行'));
+    console.log(chalk.gray('    使用 atm start 启动后台服务'));
+  }
+  
+  // auth.json 状态
+  const auth = readFactoryAuth();
+  if (auth && auth.accessToken) {
+    console.log(chalk.green('  auth.json: 已配置'));
+  } else {
+    console.log(chalk.yellow('  auth.json: 未配置'));
+  }
+  
+  console.log();
+}
+
+// 启动后台服务
+async function start() {
+  const daemonStatus = daemon.getDaemonStatus();
+  
+  if (daemonStatus.running) {
+    console.log(chalk.yellow('后台服务已在运行 (PID: ' + daemonStatus.pid + ')'));
+    return;
+  }
+  
+  const sessionToken = config.get('sessionToken');
+  if (!sessionToken) {
+    console.log(chalk.red('请先登录: atm login <激活码>'));
+    return;
+  }
+  
+  console.log(chalk.cyan('正在启动后台服务...'));
+  
+  // 使用 spawn 启动独立进程
+  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 newStatus = daemon.getDaemonStatus();
+  if (newStatus.running) {
+    console.log(chalk.green('✓ 后台服务已启动 (PID: ' + newStatus.pid + ')'));
+    console.log(chalk.gray('  日志文件: ' + daemon.LOG_FILE));
+  } else {
+    console.log(chalk.red('启动失败，请检查日志: ' + daemon.LOG_FILE));
+  }
+}
+
+// 停止后台服务
+async function stop() {
+  const result = daemon.stopDaemon();
+  
+  if (result) {
+    console.log(chalk.green('✓ 后台服务已停止'));
+  }
+}
+
+// 检查并自动切换
+async function check() {
+  const spinner = ora('检查账号状态...').start();
+  
+  try {
+    const result = await api.getTokens();
+    
+    if (!result.success) {
+      spinner.fail(chalk.red('检查失败: ' + (result.error || '未知错误')));
+      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) {
+        spinner.succeed(chalk.green(`当前账号 ${current.email} 余额充足 (${formatNumber(remaining)})`));
+        return;
+      }
+      
+      spinner.text = '当前账号余额不足，正在切换...';
+    }
+    
+    // 找到有余额的账号
+    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) {
+      spinner.fail(chalk.red('所有账号余额已用完'));
+      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);
+      config.set('currentTokenId', target.id);
+      spinner.succeed(chalk.green(`已自动切换到 ${target.email}`));
+    } else {
+      spinner.fail(chalk.red('切换失败'));
+    }
+    
+  } catch (e) {
+    spinner.fail(chalk.red('检查失败: ' + e.message));
+  }
+}
+
+module.exports = {
+  login,
+  logout,
+  list,
+  switchToken,
+  status,
+  start,
+  stop,
+  check
+};

package/src/config.js

@@ -0,0 +1,77 @@
+const Conf = require('conf');
+const os = require('os');
+const path = require('path');
+const fs = require('fs');
+
+// 配置存储
+const config = new Conf({
+  projectName: 'atm-client',
+  defaults: {
+    serverUrl: 'https://dd.776523718.xyz',
+    sessionToken: null,
+    deviceId: null,
+    currentTokenId: null,
+    autoSwitch: false
+  }
+});
+
+// Factory auth.json 路径
+function getFactoryAuthPath() {
+  const home = os.homedir();
+  return path.join(home, '.factory', 'auth.json');
+}
+
+// 确保 .factory 目录存在
+function ensureFactoryDir() {
+  const dir = path.join(os.homedir(), '.factory');
+  if (!fs.existsSync(dir)) {
+    fs.mkdirSync(dir, { recursive: true });
+  }
+}
+
+// 读取 Factory auth.json
+function readFactoryAuth() {
+  const authPath = getFactoryAuthPath();
+  if (fs.existsSync(authPath)) {
+    try {
+      return JSON.parse(fs.readFileSync(authPath, 'utf8'));
+    } catch (e) {
+      return null;
+    }
+  }
+  return null;
+}
+
+// 写入 Factory auth.json
+function writeFactoryAuth(accessToken, refreshToken) {
+  ensureFactoryDir();
+  const authPath = getFactoryAuthPath();
+  const data = {
+    accessToken,
+    refreshToken,
+    expiresAt: Date.now() + 3600000 // 1小时后过期
+  };
+  fs.writeFileSync(authPath, JSON.stringify(data, null, 2));
+  return true;
+}
+
+// 生成设备ID
+function getDeviceId() {
+  let deviceId = config.get('deviceId');
+  if (!deviceId) {
+    const crypto = require('crypto');
+    const machineInfo = `${os.hostname()}-${os.platform()}-${os.arch()}-${os.userInfo().username}`;
+    deviceId = crypto.createHash('sha256').update(machineInfo).digest('hex').substring(0, 32);
+    config.set('deviceId', deviceId);
+  }
+  return deviceId;
+}
+
+module.exports = {
+  config,
+  getFactoryAuthPath,
+  ensureFactoryDir,
+  readFactoryAuth,
+  writeFactoryAuth,
+  getDeviceId
+};

package/src/daemon-runner.js

@@ -0,0 +1,7 @@
+// 独立运行的守护进程
+const daemon = require('./daemon');
+
+daemon.startDaemon();
+
+// 保持进程运行
+setInterval(() => {}, 1000 * 60 * 60); // 每小时空循环一次，保持进程存活

package/src/daemon.js

@@ -0,0 +1,257 @@
+const WebSocket = require('ws');
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const { config, writeFactoryAuth, getDeviceId } = require('./config');
+const api = require('./api');
+
+const PID_FILE = path.join(os.tmpdir(), 'atm-daemon.pid');
+const LOG_FILE = path.join(os.tmpdir(), 'atm-daemon.log');
+
+let ws = null;
+let reconnectTimer = null;
+let heartbeatTimer = null;
+
+function log(message) {
+  const timestamp = new Date().toISOString();
+  const line = `[${timestamp}] ${message}\n`;
+  fs.appendFileSync(LOG_FILE, line);
+  if (process.env.ATM_DEBUG) {
+    console.log(line.trim());
+  }
+}
+
+function connectWebSocket() {
+  const serverUrl = config.get('serverUrl').replace('https://', 'wss://').replace('http://', 'ws://');
+  const sessionToken = config.get('sessionToken');
+  const deviceId = getDeviceId();
+  
+  if (!sessionToken) {
+    log('未登录，无法连接 WebSocket');
+    return;
+  }
+  
+  const wsUrl = `${serverUrl}/ws?token=${sessionToken}&device=${deviceId}`;
+  
+  log(`连接 WebSocket: ${serverUrl}/ws`);
+  
+  ws = new WebSocket(wsUrl);
+  
+  ws.on('open', () => {
+    log('WebSocket 已连接');
+    // 订阅 Token 更新
+    ws.send(JSON.stringify({ type: 'subscribe', channel: 'tokens' }));
+  });
+  
+  ws.on('message', async (data) => {
+    try {
+      const message = JSON.parse(data.toString());
+      log(`收到消息: ${message.type}`);
+      
+      if (message.type === 'token_updated') {
+        // Token 已更新，重新获取并写入 auth.json
+        await refreshCurrentToken();
+      } else if (message.type === 'quota_depleted') {
+        // 余额用完，自动切换
+        if (config.get('autoSwitch')) {
+          await autoSwitch();
+        }
+      }
+    } catch (e) {
+      log(`解析消息失败: ${e.message}`);
+    }
+  });
+  
+  ws.on('close', () => {
+    log('WebSocket 断开，5秒后重连...');
+    scheduleReconnect();
+  });
+  
+  ws.on('error', (err) => {
+    log(`WebSocket 错误: ${err.message}`);
+  });
+}
+
+function scheduleReconnect() {
+  if (reconnectTimer) clearTimeout(reconnectTimer);
+  reconnectTimer = setTimeout(() => {
+    connectWebSocket();
+  }, 5000);
+}
+
+async function refreshCurrentToken() {
+  try {
+    const currentTokenId = config.get('currentTokenId');
+    if (!currentTokenId) return;
+    
+    const result = await api.activateToken(currentTokenId);
+    if (result.success && result.access_token && result.refresh_token) {
+      writeFactoryAuth(result.access_token, result.refresh_token);
+      log(`Token 已更新: ${currentTokenId}`);
+    }
+  } catch (e) {
+    log(`刷新 Token 失败: ${e.message}`);
+  }
+}
+
+async function autoSwitch() {
+  try {
+    const result = await api.getTokens();
+    if (!result.success || !result.data) return;
+    
+    const tokens = result.data;
+    const currentId = config.get('currentTokenId');
+    
+    // 找到有余额的 Token
+    const available = tokens.filter(t => {
+      const remaining = (t.quota_total || 0) - (t.quota_used || 0);
+      return remaining > 0 && t.id !== currentId;
+    });
+    
+    if (available.length > 0) {
+      // 切换到余额最少的（用完一个再用下一个）
+      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) {
+        config.set('currentTokenId', target.id);
+        writeFactoryAuth(activateResult.access_token, activateResult.refresh_token);
+        log(`自动切换到: ${target.email}`);
+      }
+    } else {
+      log('所有账号余额已用完');
+    }
+  } catch (e) {
+    log(`自动切换失败: ${e.message}`);
+  }
+}
+
+function startHeartbeat() {
+  heartbeatTimer = setInterval(async () => {
+    try {
+      const result = await api.heartbeat();
+      if (!result.valid) {
+        log('会话已失效');
+      }
+    } catch (e) {
+      log(`心跳失败: ${e.message}`);
+    }
+  }, 30000); // 每30秒心跳一次
+}
+
+function writePidFile() {
+  fs.writeFileSync(PID_FILE, process.pid.toString());
+}
+
+function removePidFile() {
+  if (fs.existsSync(PID_FILE)) {
+    fs.unlinkSync(PID_FILE);
+  }
+}
+
+function getPid() {
+  if (fs.existsSync(PID_FILE)) {
+    return parseInt(fs.readFileSync(PID_FILE, 'utf8'));
+  }
+  return null;
+}
+
+function isRunning(pid) {
+  try {
+    process.kill(pid, 0);
+    return true;
+  } catch (e) {
+    return false;
+  }
+}
+
+// 启动守护进程
+function startDaemon() {
+  const existingPid = getPid();
+  if (existingPid && isRunning(existingPid)) {
+    console.log('后台服务已在运行 (PID: ' + existingPid + ')');
+    return false;
+  }
+  
+  // 清空日志
+  fs.writeFileSync(LOG_FILE, '');
+  
+  writePidFile();
+  log('守护进程启动');
+  
+  connectWebSocket();
+  startHeartbeat();
+  
+  // 优雅退出
+  process.on('SIGTERM', () => {
+    log('收到 SIGTERM，正在退出...');
+    cleanup();
+    process.exit(0);
+  });
+  
+  process.on('SIGINT', () => {
+    log('收到 SIGINT，正在退出...');
+    cleanup();
+    process.exit(0);
+  });
+  
+  return true;
+}
+
+function cleanup() {
+  if (ws) ws.close();
+  if (reconnectTimer) clearTimeout(reconnectTimer);
+  if (heartbeatTimer) clearInterval(heartbeatTimer);
+  removePidFile();
+}
+
+// 停止守护进程
+function stopDaemon() {
+  const pid = getPid();
+  if (!pid) {
+    console.log('后台服务未运行');
+    return false;
+  }
+  
+  if (!isRunning(pid)) {
+    removePidFile();
+    console.log('后台服务未运行');
+    return false;
+  }
+  
+  try {
+    process.kill(pid, 'SIGTERM');
+    removePidFile();
+    return true;
+  } catch (e) {
+    console.log('停止失败:', e.message);
+    return false;
+  }
+}
+
+// 获取守护进程状态
+function getDaemonStatus() {
+  const pid = getPid();
+  if (!pid) return { running: false };
+  
+  return {
+    running: isRunning(pid),
+    pid,
+    logFile: LOG_FILE
+  };
+}
+
+module.exports = {
+  startDaemon,
+  stopDaemon,
+  getDaemonStatus,
+  refreshCurrentToken,
+  autoSwitch,
+  LOG_FILE
+};

package/src/index.js

@@ -0,0 +1,2 @@
+// ATM Client CLI - 主入口
+module.exports = require('./commands');