antigravity-tc 1.1.0

package/src/autostart.js

@@ -0,0 +1,115 @@
+const fs = require('fs');
+const os = require('os');
+const path = require('path');
+const AutoLaunch = require('auto-launch');
+const { ensureAppDirs, LOGS_DIR } = require('./paths');
+
+// ============================================================
+// macOS: LaunchAgent native.
+// auto-launch tidak cocok untuk app berbasis node CLI: plist yang
+// dihasilnya hanya berisi [node, --hidden] tanpa path script, jadi
+// tidak pernah benar-benar menjalankan app saat login. Karena itu
+// di darwin kita tulis plist sendiri dengan command lengkap.
+// ============================================================
+const PLIST_LABEL = 'com.antigravity-tc.launcher';
+const LAUNCH_AGENTS_DIR = path.join(os.homedir(), 'Library', 'LaunchAgents');
+const PLIST_PATH = path.join(LAUNCH_AGENTS_DIR, `${PLIST_LABEL}.plist`);
+// Plist rusak buatan versi lama (auto-launch, label "node").
+const LEGACY_PLIST_PATH = path.join(LAUNCH_AGENTS_DIR, 'node.plist');
+
+const IS_DARWIN = process.platform === 'darwin';
+
+// Fallback untuk platform non-macOS.
+const launcher = new AutoLaunch({
+  name: 'AntigravityTokenCollector',
+  path: process.execPath,
+  args: ['--hidden'],
+  isHidden: true,
+  mac: {
+    useLaunchAgent: true,
+  },
+});
+
+function escapeXml(value) {
+  return String(value).replace(/&/g, '&amp;').replace(/</g, '&lt;').replace(/>/g, '&gt;');
+}
+
+function buildPlistContent() {
+  const entryScript = path.join(__dirname, '..', 'index.js');
+  const programArguments = [process.execPath, entryScript, '--hidden'];
+  const argumentsXml = programArguments
+    .map((argument) => `    <string>${escapeXml(argument)}</string>`)
+    .join('\n');
+  const logPath = path.join(LOGS_DIR, 'launcher.log');
+  return `<?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>${PLIST_LABEL}</string>
+  <key>ProgramArguments</key>
+  <array>
+${argumentsXml}
+  </array>
+  <key>RunAtLoad</key>
+  <true/>
+  <key>StandardOutPath</key>
+  <string>${escapeXml(logPath)}</string>
+  <key>StandardErrorPath</key>
+  <string>${escapeXml(logPath)}</string>
+</dict>
+</plist>`;
+}
+
+// Hapus plist rusak buatan versi lama kalau terdeteksi (label "node" +
+// argumen --hidden adalah signature milik app ini).
+function removeLegacyPlistIfPresent() {
+  try {
+    if (!fs.existsSync(LEGACY_PLIST_PATH)) return;
+    const content = fs.readFileSync(LEGACY_PLIST_PATH, 'utf-8');
+    if (content.includes('<string>node</string>') && content.includes('--hidden')) {
+      fs.unlinkSync(LEGACY_PLIST_PATH);
+    }
+  } catch (e) {}
+}
+
+async function enableAutoStart() {
+  if (!IS_DARWIN) {
+    await launcher.enable();
+    return true;
+  }
+  ensureAppDirs();
+  removeLegacyPlistIfPresent();
+  fs.mkdirSync(LAUNCH_AGENTS_DIR, { recursive: true });
+  fs.writeFileSync(PLIST_PATH, buildPlistContent(), 'utf-8');
+  return true;
+}
+
+async function disableAutoStart() {
+  if (!IS_DARWIN) {
+    await launcher.disable();
+    return false;
+  }
+  removeLegacyPlistIfPresent();
+  try {
+    if (fs.existsSync(PLIST_PATH)) fs.unlinkSync(PLIST_PATH);
+  } catch (e) {}
+  return false;
+}
+
+async function isAutoStartEnabled() {
+  if (!IS_DARWIN) {
+    return launcher.isEnabled();
+  }
+  try {
+    return fs.existsSync(PLIST_PATH);
+  } catch (e) {
+    return false;
+  }
+}
+
+module.exports = {
+  enableAutoStart,
+  disableAutoStart,
+  isAutoStartEnabled,
+};

package/src/cli.js

@@ -0,0 +1,82 @@
+const { exec } = require('child_process');
+const { select } = require('@inquirer/prompts');
+const { version } = require('../package.json');
+
+function openBrowser(url) {
+  const platform = process.platform;
+  let cmd;
+
+  if (platform === 'darwin') {
+    cmd = `open "${url}"`;
+  } else if (platform === 'win32') {
+    cmd = `start "" "${url}"`;
+  } else {
+    cmd = `xdg-open "${url}"`;
+  }
+
+  return new Promise((resolve) => {
+    exec(cmd, { windowsHide: true }, (err) => {
+      if (err) {
+        console.log(`Open browser manually: ${url}`);
+      }
+      resolve();
+    });
+  });
+}
+
+function printHeader(appUrl) {
+  console.clear();
+  console.log('========================================');
+  console.log(`  Antigravity Token Collector v${version || '0.0.0'}`);
+  console.log(`  Server: ${appUrl}`);
+  console.log('========================================');
+  console.log('');
+}
+async function showMainMenu({ trayManager, appUrl, onHideToTray, onExit }) {
+  printHeader(appUrl);
+  let choice;
+
+  try {
+    choice = await select({
+      message: 'Menu:',
+      choices: [
+        { name: 'Open Dashboard', value: 'dashboard' },
+        { name: 'Hide to Tray (Background)', value: 'hide' },
+        { name: 'Exit', value: 'exit' },
+      ],
+    });
+  } catch (error) {
+    if (error && error.name === 'ExitPromptError') {
+      choice = 'exit';
+    } else {
+      throw error;
+    }
+  }
+
+  if (choice === 'dashboard') {
+    const dashboardUrl = appUrl || 'http://localhost:3456/dashboard';
+    await openBrowser(dashboardUrl);
+    return showMainMenu({ trayManager, appUrl, onHideToTray, onExit });
+  }
+
+  if (choice === 'hide') {
+    if (typeof onHideToTray === 'function') {
+      await onHideToTray();
+      return;
+    }
+    console.log('✓ Hidden to tray. Klik icon di menubar macOS.');
+    return;
+  }
+
+  if (typeof onExit === 'function') {
+    await onExit();
+    return;
+  }
+
+  if (trayManager && trayManager.tray) {
+    await trayManager.stop();
+  }
+  process.exit(0);
+}
+
+module.exports = { showMainMenu };

package/src/config.js

@@ -0,0 +1,23 @@
+const ConfImport = require('conf');
+const Conf = ConfImport.default || ConfImport;
+
+let configInstance = null;
+
+function getConfig() {
+  if (!configInstance) {
+    configInstance = new Conf({
+      projectName: 'antigravity-token-collector',
+      defaults: {
+        port: 3456,
+        autoStart: false,
+        nineRouterDbPath: null,
+        tunnelEnabled: false,
+        adminPassword: null,
+      },
+    });
+  }
+
+  return configInstance;
+}
+
+module.exports = { getConfig };

package/src/jwt.js

@@ -0,0 +1,37 @@
+const crypto = require('crypto');
+const jwt = require('jsonwebtoken');
+
+const TOKEN_MAX_AGE_SECONDS = 7 * 24 * 60 * 60;
+
+// Secret JWT dibuat random sekali per instalasi lalu disimpan di config,
+// sehingga token buatan instalasi A tidak bisa dipakai di instalasi B.
+function getJwtSecret(config) {
+  let secret = config.get('jwtSecret');
+  if (!secret || typeof secret !== 'string' || secret.length < 32) {
+    secret = crypto.randomBytes(48).toString('hex');
+    config.set('jwtSecret', secret);
+  }
+  return secret;
+}
+
+function signAdminToken(config) {
+  return jwt.sign({ role: 'admin' }, getJwtSecret(config), {
+    expiresIn: TOKEN_MAX_AGE_SECONDS,
+  });
+}
+
+function verifyAdminToken(config, token) {
+  if (!token || typeof token !== 'string') return false;
+  try {
+    const payload = jwt.verify(token, getJwtSecret(config));
+    return payload && payload.role === 'admin';
+  } catch (e) {
+    return false;
+  }
+}
+
+module.exports = {
+  TOKEN_MAX_AGE_SECONDS,
+  signAdminToken,
+  verifyAdminToken,
+};

package/src/paths.js

@@ -0,0 +1,59 @@
+const os = require('os');
+const path = require('path');
+const fs = require('fs');
+
+const APP_DIR = path.join(os.homedir(), '.antigravity-tc');
+const LOGS_DIR = path.join(APP_DIR, 'logs');
+const BIN_DIR = path.join(APP_DIR, 'bin');
+
+// Default lokasi database 9Router mengikuti OS tempat app di-install:
+// - Windows  : %APPDATA%/9router/db/data.sqlite
+// - Docker   : /app/data/db/data.sqlite (mount volume untuk persist)
+// - lainnya  : ~/.9router/db/data.sqlite (macOS/Linux)
+function getDefault9RouterDb() {
+  if (process.platform === 'win32') {
+    const appData = process.env.APPDATA || path.join(os.homedir(), 'AppData', 'Roaming');
+    return path.join(appData, '9router', 'db', 'data.sqlite');
+  }
+  try {
+    if (fs.existsSync('/.dockerenv')) {
+      return '/app/data/db/data.sqlite';
+    }
+  } catch (e) {}
+  return path.join(os.homedir(), '.9router', 'db', 'data.sqlite');
+}
+
+const DEFAULT_9ROUTER_DB = getDefault9RouterDb();
+
+function ensureAppDirs() {
+  for (const dir of [APP_DIR, LOGS_DIR, BIN_DIR]) {
+    if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
+  }
+}
+
+function migrateLegacyLogDb(projectRoot) {
+  const targetPath = path.join(LOGS_DIR, 'donation-logs.sqlite');
+  if (fs.existsSync(targetPath)) return;
+  const legacyPath = path.join(projectRoot, 'donation-logs.sqlite');
+  if (!fs.existsSync(legacyPath)) return;
+  try {
+    fs.renameSync(legacyPath, targetPath);
+    console.log(`[migrate] Moved donation-logs.sqlite → ${targetPath}`);
+  } catch (e) {
+    try {
+      fs.copyFileSync(legacyPath, targetPath);
+      console.log(`[migrate] Copied donation-logs.sqlite → ${targetPath}`);
+    } catch (e2) {}
+  }
+}
+
+module.exports = {
+  APP_DIR,
+  LOGS_DIR,
+  BIN_DIR,
+  LOG_DB_PATH: path.join(LOGS_DIR, 'donation-logs.sqlite'),
+  DEFAULT_9ROUTER_DB,
+  CLOUDFLARED_BIN: path.join(BIN_DIR, process.platform === 'win32' ? 'cloudflared.exe' : 'cloudflared'),
+  ensureAppDirs,
+  migrateLegacyLogDb,
+};

package/src/tray.js

@@ -0,0 +1,210 @@
+const path = require('path');
+const fs = require('fs');
+const EventEmitter = require('events');
+const { exec } = require('child_process');
+const SysTrayImport = require('systray2');
+const {
+  enableAutoStart,
+  disableAutoStart,
+  isAutoStartEnabled,
+} = require('./autostart');
+
+const SysTray = SysTrayImport.default || SysTrayImport;
+
+function openBrowser(url) {
+  const platform = process.platform;
+  let cmd;
+
+  if (platform === 'darwin') {
+    cmd = `open "${url}"`;
+  } else if (platform === 'win32') {
+    cmd = `start "" "${url}"`;
+  } else {
+    cmd = `xdg-open "${url}"`;
+  }
+
+  return new Promise((resolve) => {
+    exec(cmd, { windowsHide: true }, (err) => {
+      if (err) {
+        console.log(`Open browser manually: ${url}`);
+      }
+      resolve();
+    });
+  });
+}
+
+class TrayManager extends EventEmitter {
+  constructor({ config, port }) {
+    super();
+    this.config = config;
+    this.port = Number(port) || 3456;
+    this.autoStartEnabled = Boolean(this.config.get('autoStart'));
+    this.tray = null;
+    this.menu = null;
+    this.autoStartItem = null;
+    this.autoStartPollTimer = null;
+  }
+
+  get appUrl() {
+    return `http://localhost:${this.port}/dashboard`;
+  }
+
+  get tooltip() {
+    return `Antigravity Token Collector - running on port ${this.port}`;
+  }
+
+  // Menu dibangun SEKALI di start(). Objek item (termasuk autoStartItem)
+  // dipertahankan agar __id internal systray2 stabil — klik selalu ter-resolve.
+  buildMenu() {
+    this.autoStartItem = {
+      title: this.autoStartItemTitle(),
+      tooltip: 'Run On OS startup',
+      checked: this.autoStartEnabled,
+      enabled: true,
+    };
+    return {
+      icon: this.getTrayIconBase64(),
+      isTemplateIcon: process.platform === 'darwin',
+      tooltip: this.tooltip,
+      items: [
+        {
+          title: `Running on :${this.port}`,
+          tooltip: this.tooltip,
+          enabled: false,
+        },
+        SysTray.separator,
+        {
+          title: 'Open Dashboard',
+          tooltip: 'Open dashboard in browser',
+          enabled: true,
+        },
+        SysTray.separator,
+        this.autoStartItem,
+        SysTray.separator,
+        {
+          title: 'Exit',
+          tooltip: 'Exit application',
+          enabled: true,
+        },
+      ],
+    };
+  }
+
+  autoStartItemTitle() {
+    return `Auto Start: ${this.autoStartEnabled ? 'ON' : 'OFF'}`;
+  }
+
+  getTrayIconBase64() {
+    const iconPath = path.join(__dirname, '..', 'web', 'public', 'favicon.ico');
+    return fs.readFileSync(iconPath).toString('base64');
+  }
+
+  async start() {
+    this.autoStartEnabled = await isAutoStartEnabled().catch(() => Boolean(this.config.get('autoStart')));
+    this.config.set('autoStart', this.autoStartEnabled);
+    this.ensureTrayBinaryExecutable();
+
+    this.menu = this.buildMenu();
+    this.tray = new SysTray({
+      menu: this.menu,
+      debug: false,
+      copyDir: false,
+    });
+
+    await this.tray.ready();
+    await this.tray.onClick(async (action) => {
+      await this.handleClick(action && action.item ? action.item : null);
+    });
+
+    // Auto Start bisa berubah dari dashboard; sinkronkan tiap 5 detik
+    // supaya status di tray selalu realtime.
+    this.autoStartPollTimer = setInterval(() => {
+      this.syncAutoStartState().catch(() => {});
+    }, 5000);
+  }
+
+  async syncAutoStartState(forceUpdate = false) {
+    const enabled = await isAutoStartEnabled().catch(() => null);
+    if (enabled === null) return;
+    if (!forceUpdate && enabled === this.autoStartEnabled) return;
+    this.autoStartEnabled = enabled;
+    this.config.set('autoStart', enabled);
+    await this.refreshAutoStartMenuItem();
+  }
+
+  // Update item Auto Start TANPA membangun ulang menu. PENTING: jangan pernah
+  // memakai action 'update-menu' — systray2 tidak memperbarui internalIdMap
+  // untuk menu baru, sehingga __id melenceng dan klik berikutnya gagal
+  // resolve (toggle terlihat "mati"). Dengan 'update-item' pada objek yang
+  // sama, __id tetap stabil dan checked/title ter-update di menubar.
+  async refreshAutoStartMenuItem() {
+    if (!this.tray || !this.autoStartItem) return;
+    this.autoStartItem.title = this.autoStartItemTitle();
+    this.autoStartItem.checked = this.autoStartEnabled;
+    try {
+      await this.tray.sendAction({
+        type: 'update-item',
+        item: this.autoStartItem,
+      });
+    } catch (error) {
+      console.error('[tray] update autostart item failed:', error.message);
+    }
+  }
+
+  ensureTrayBinaryExecutable() {
+    if (process.platform !== 'darwin') return;
+
+    const candidates = [
+      path.join(__dirname, '..', 'node_modules', 'systray2', 'traybin', 'tray_darwin_release'),
+      path.join(__dirname, '..', 'node_modules', 'systray2', 'traybin', 'tray_darwin'),
+    ];
+
+    for (const filePath of candidates) {
+      if (!fs.existsSync(filePath)) continue;
+      try {
+        fs.chmodSync(filePath, 0o755);
+      } catch (error) {}
+    }
+  }
+
+  async handleClick(item) {
+    if (!item || !item.title) return;
+
+    if (item.title === 'Open Dashboard') {
+      await openBrowser(this.appUrl || 'http://localhost:3456/dashboard');
+      return;
+    }
+
+    if (item.title.startsWith('Auto Start')) {
+      try {
+        const currentlyEnabled = await isAutoStartEnabled().catch(() => this.autoStartEnabled);
+        if (currentlyEnabled) await disableAutoStart();
+        else await enableAutoStart();
+      } catch (error) {
+        console.error('[tray] toggle autostart failed:', error.message);
+      }
+      // Baca ulang status sebenarnya lalu update item di menubar.
+      await this.syncAutoStartState(true);
+      return;
+    }
+
+    if (item.title === 'Exit') {
+      await this.stop();
+      // index.js mendengarkan event ini untuk cleanup (lock, tunnel) sebelum exit.
+      this.emit('exit');
+      if (this.listenerCount('exit') === 0) process.exit(0);
+    }
+  }
+
+  async stop() {
+    if (this.autoStartPollTimer) {
+      clearInterval(this.autoStartPollTimer);
+      this.autoStartPollTimer = null;
+    }
+    if (!this.tray) return;
+    await this.tray.kill(false);
+    this.tray = null;
+  }
+}
+
+module.exports = { TrayManager };