atm-droid 1.0.7 → 1.0.8

package/package.json

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

package/src/autostart.js

@@ -0,0 +1,163 @@
+const os = require('os');
+const fs = require('fs');
+const path = require('path');
+const { execSync } = require('child_process');
+
+const APP_NAME = 'ATM-Droid';
+
+// 获取启动命令
+function getStartCommand() {
+  const daemonPath = path.join(__dirname, 'daemon-runner.js');
+  return `"${process.execPath}" "${daemonPath}"`;
+}
+
+// Windows: 注册表
+function enableWindows() {
+  try {
+    const cmd = getStartCommand();
+    execSync(`reg add "HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Run" /v "${APP_NAME}" /t REG_SZ /d "${cmd}" /f`, { stdio: 'ignore' });
+    return true;
+  } catch (e) {
+    return false;
+  }
+}
+
+function disableWindows() {
+  try {
+    execSync(`reg delete "HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Run" /v "${APP_NAME}" /f`, { stdio: 'ignore' });
+    return true;
+  } catch (e) {
+    return false;
+  }
+}
+
+function isEnabledWindows() {
+  try {
+    execSync(`reg query "HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Run" /v "${APP_NAME}"`, { stdio: 'ignore' });
+    return true;
+  } catch (e) {
+    return false;
+  }
+}
+
+// macOS: LaunchAgents
+function getMacPlistPath() {
+  return path.join(os.homedir(), 'Library', 'LaunchAgents', 'com.atm-droid.daemon.plist');
+}
+
+function enableMac() {
+  try {
+    const plistPath = getMacPlistPath();
+    const daemonPath = path.join(__dirname, 'daemon-runner.js');
+    
+    const plist = `<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+    <key>Label</key>
+    <string>com.atm-droid.daemon</string>
+    <key>ProgramArguments</key>
+    <array>
+        <string>${process.execPath}</string>
+        <string>${daemonPath}</string>
+    </array>
+    <key>RunAtLoad</key>
+    <true/>
+    <key>KeepAlive</key>
+    <false/>
+</dict>
+</plist>`;
+    
+    fs.mkdirSync(path.dirname(plistPath), { recursive: true });
+    fs.writeFileSync(plistPath, plist);
+    return true;
+  } catch (e) {
+    return false;
+  }
+}
+
+function disableMac() {
+  try {
+    const plistPath = getMacPlistPath();
+    if (fs.existsSync(plistPath)) {
+      fs.unlinkSync(plistPath);
+    }
+    return true;
+  } catch (e) {
+    return false;
+  }
+}
+
+function isEnabledMac() {
+  return fs.existsSync(getMacPlistPath());
+}
+
+// Linux: ~/.config/autostart
+function getLinuxDesktopPath() {
+  return path.join(os.homedir(), '.config', 'autostart', 'atm-droid.desktop');
+}
+
+function enableLinux() {
+  try {
+    const desktopPath = getLinuxDesktopPath();
+    const daemonPath = path.join(__dirname, 'daemon-runner.js');
+    
+    const desktop = `[Desktop Entry]
+Type=Application
+Name=ATM Droid Daemon
+Exec=${process.execPath} ${daemonPath}
+Hidden=false
+NoDisplay=true
+X-GNOME-Autostart-enabled=true
+`;
+    
+    fs.mkdirSync(path.dirname(desktopPath), { recursive: true });
+    fs.writeFileSync(desktopPath, desktop);
+    return true;
+  } catch (e) {
+    return false;
+  }
+}
+
+function disableLinux() {
+  try {
+    const desktopPath = getLinuxDesktopPath();
+    if (fs.existsSync(desktopPath)) {
+      fs.unlinkSync(desktopPath);
+    }
+    return true;
+  } catch (e) {
+    return false;
+  }
+}
+
+function isEnabledLinux() {
+  return fs.existsSync(getLinuxDesktopPath());
+}
+
+// 统一接口
+function enable() {
+  const platform = os.platform();
+  if (platform === 'win32') return enableWindows();
+  if (platform === 'darwin') return enableMac();
+  if (platform === 'linux') return enableLinux();
+  return false;
+}
+
+function disable() {
+  const platform = os.platform();
+  if (platform === 'win32') return disableWindows();
+  if (platform === 'darwin') return disableMac();
+  if (platform === 'linux') return disableLinux();
+  return false;
+}
+
+function isEnabled() {
+  const platform = os.platform();
+  if (platform === 'win32') return isEnabledWindows();
+  if (platform === 'darwin') return isEnabledMac();
+  if (platform === 'linux') return isEnabledLinux();
+  return false;
+}
+
+module.exports = { enable, disable, isEnabled };

package/src/commands.js

@@ -6,6 +6,7 @@ const path = require('path');
 const { config, writeFactoryAuth, readFactoryAuth, getFactoryAuthPath } = require('./config');
 const api = require('./api');
 const daemon = require('./daemon');
+const autostart = require('./autostart');
 
 // 格式化数字
 function formatNumber(num) {
@@ -45,6 +46,13 @@ async function login(code) {
         console.log(chalk.green('  ✓ 后台服务已启动'));
       }
       
+      // 默认启用开机自启动
+      if (!autostart.isEnabled()) {
+        if (autostart.enable()) {
+          console.log(chalk.green('  ✓ 已设置开机自启动'));
+        }
+      }
+      
       console.log(chalk.yellow('\n  提示: 使用 atm list 查看账号列表'));
     } else {
       spinner.fail(chalk.red('激活失败: ' + (result.error || '未知错误')));

package/src/menu.js

@@ -4,6 +4,7 @@ const fetch = require('node-fetch');
 const { config, writeFactoryAuth, getFactoryAuthPath } = require('./config');
 const api = require('./api');
 const daemon = require('./daemon');
+const autostart = require('./autostart');
 const pkg = require('../package.json');
 
 let latestVersion = null;
@@ -79,6 +80,7 @@ async function showStatus() {
       console.log(chalk.gray(`  账号: ${tokens.length} 个可用`));
       console.log(chalk.gray(`  自动切换: ${autoSwitch ? '开启' : '关闭'}`));
       console.log(chalk.gray(`  后台服务: ${daemonStatus.running ? '运行中' : '未运行'}`));
+      console.log(chalk.gray(`  开机自启: ${autostart.isEnabled() ? '开启' : '关闭'}`));
       console.log();
       
       return tokens;
@@ -93,6 +95,10 @@ async function showStatus() {
 // 主菜单
 async function mainMenu() {
   clearScreen();
+  
+  // 每次刷新菜单时检查新版本（后台执行，不阻塞）
+  checkNewVersion();
+  
   showHeader();
   
   const tokens = await showStatus();
@@ -115,6 +121,13 @@ async function mainMenu() {
       choices.push({ name: '▶️  启动后台服务', value: 'start' });
     }
     
+    // 开机自启动开关
+    if (autostart.isEnabled()) {
+      choices.push({ name: '🔕 关闭开机自启', value: 'autostart_off' });
+    } else {
+      choices.push({ name: '🔔 开启开机自启', value: 'autostart_on' });
+    }
+    
     choices.push(new inquirer.Separator());
     choices.push({ name: '🚪 退出登录', value: 'logout' });
   }
@@ -482,6 +495,22 @@ async function menu() {
       case 'logout':
         await doLogout();
         break;
+      case 'autostart_on':
+        if (autostart.enable()) {
+          console.log(chalk.green('\n  ✓ 已开启开机自启动\n'));
+        } else {
+          console.log(chalk.red('\n  ✗ 设置失败\n'));
+        }
+        await pause();
+        break;
+      case 'autostart_off':
+        if (autostart.disable()) {
+          console.log(chalk.yellow('\n  ✓ 已关闭开机自启动\n'));
+        } else {
+          console.log(chalk.red('\n  ✗ 设置失败\n'));
+        }
+        await pause();
+        break;
       case 'update':
         await doUpdate();
         break;