@yeaft/webchat-agent 0.0.234 → 0.0.236

package/service/linux.js

@@ -0,0 +1,111 @@
+/**
+ * Service — Linux (systemd) platform implementation
+ */
+import { execSync } from 'child_process';
+import { existsSync, writeFileSync, unlinkSync, mkdirSync } from 'fs';
+import { join, dirname } from 'path';
+import { homedir } from 'os';
+import { SERVICE_NAME, getLogDir, getNodePath, getCliPath } from './config.js';
+
+export function getSystemdServicePath() {
+  const dir = join(homedir(), '.config', 'systemd', 'user');
+  mkdirSync(dir, { recursive: true });
+  return join(dir, `${SERVICE_NAME}.service`);
+}
+
+function generateSystemdUnit(config) {
+  const nodePath = getNodePath();
+  const cliPath = getCliPath();
+  const envLines = [];
+  if (config.serverUrl) envLines.push(`Environment=SERVER_URL=${config.serverUrl}`);
+  if (config.agentName) envLines.push(`Environment=AGENT_NAME=${config.agentName}`);
+  if (config.agentSecret) envLines.push(`Environment=AGENT_SECRET=${config.agentSecret}`);
+  if (config.workDir) envLines.push(`Environment=WORK_DIR=${config.workDir}`);
+
+  // Include node's bin dir in PATH for claude CLI access
+  const nodeBinDir = dirname(nodePath);
+
+  return `[Unit]
+Description=Yeaft WebChat Agent
+After=network-online.target
+Wants=network-online.target
+
+[Service]
+Type=simple
+ExecStart=${nodePath} ${cliPath}
+WorkingDirectory=${config.workDir || homedir()}
+Restart=on-failure
+RestartSec=10
+KillMode=process
+${envLines.join('\n')}
+Environment=PATH=${nodeBinDir}:${homedir()}/.local/bin:${homedir()}/.npm-global/bin:/usr/local/bin:/usr/bin:/bin
+
+StandardOutput=append:${getLogDir()}/out.log
+StandardError=append:${getLogDir()}/error.log
+
+[Install]
+WantedBy=default.target
+`;
+}
+
+export function linuxInstall(config) {
+  const servicePath = getSystemdServicePath();
+  writeFileSync(servicePath, generateSystemdUnit(config));
+  execSync('systemctl --user daemon-reload');
+  execSync(`systemctl --user enable ${SERVICE_NAME}`);
+  execSync(`systemctl --user start ${SERVICE_NAME}`);
+  console.log(`Service installed and started.`);
+  console.log(`\nManage with:`);
+  console.log(`  yeaft-agent status`);
+  console.log(`  yeaft-agent logs`);
+  console.log(`  yeaft-agent restart`);
+  console.log(`  yeaft-agent uninstall`);
+  console.log(`\nTo run when not logged in:`);
+  console.log(`  sudo loginctl enable-linger $(whoami)`);
+}
+
+export function linuxUninstall() {
+  try { execSync(`systemctl --user stop ${SERVICE_NAME} 2>/dev/null`); } catch {}
+  try { execSync(`systemctl --user disable ${SERVICE_NAME} 2>/dev/null`); } catch {}
+  const servicePath = getSystemdServicePath();
+  if (existsSync(servicePath)) unlinkSync(servicePath);
+  try { execSync('systemctl --user daemon-reload'); } catch {}
+  console.log('Service uninstalled.');
+}
+
+export function linuxStart() {
+  execSync(`systemctl --user start ${SERVICE_NAME}`, { stdio: 'inherit' });
+  console.log('Service started.');
+}
+
+export function linuxStop() {
+  execSync(`systemctl --user stop ${SERVICE_NAME}`, { stdio: 'inherit' });
+  console.log('Service stopped.');
+}
+
+export function linuxRestart() {
+  execSync(`systemctl --user restart ${SERVICE_NAME}`, { stdio: 'inherit' });
+  console.log('Service restarted.');
+}
+
+export function linuxStatus() {
+  try {
+    execSync(`systemctl --user status ${SERVICE_NAME}`, { stdio: 'inherit' });
+  } catch {
+    // systemctl status returns non-zero when service is stopped
+  }
+}
+
+export function linuxLogs() {
+  try {
+    execSync(`journalctl --user -u ${SERVICE_NAME} -f --no-pager -n 100`, { stdio: 'inherit' });
+  } catch {
+    // Fallback to log files
+    const logFile = join(getLogDir(), 'out.log');
+    if (existsSync(logFile)) {
+      execSync(`tail -f -n 100 ${logFile}`, { stdio: 'inherit' });
+    } else {
+      console.log('No logs found.');
+    }
+  }
+}

package/service/macos.js

@@ -0,0 +1,137 @@
+/**
+ * Service — macOS (launchd) platform implementation
+ */
+import { execSync } from 'child_process';
+import { existsSync, writeFileSync, unlinkSync, mkdirSync } from 'fs';
+import { join } from 'path';
+import { homedir } from 'os';
+import { getLogDir, getNodePath, getCliPath } from './config.js';
+
+export function getLaunchdPlistPath() {
+  const dir = join(homedir(), 'Library', 'LaunchAgents');
+  mkdirSync(dir, { recursive: true });
+  return join(dir, 'com.yeaft.agent.plist');
+}
+
+function generateLaunchdPlist(config) {
+  const nodePath = getNodePath();
+  const cliPath = getCliPath();
+  const logDir = getLogDir();
+
+  const envDict = [];
+  if (config.serverUrl) envDict.push(`      <key>SERVER_URL</key>\n      <string>${config.serverUrl}</string>`);
+  if (config.agentName) envDict.push(`      <key>AGENT_NAME</key>\n      <string>${config.agentName}</string>`);
+  if (config.agentSecret) envDict.push(`      <key>AGENT_SECRET</key>\n      <string>${config.agentSecret}</string>`);
+  if (config.workDir) envDict.push(`      <key>WORK_DIR</key>\n      <string>${config.workDir}</string>`);
+
+  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>com.yeaft.agent</string>
+    <key>ProgramArguments</key>
+    <array>
+        <string>${nodePath}</string>
+        <string>${cliPath}</string>
+    </array>
+    <key>WorkingDirectory</key>
+    <string>${config.workDir || homedir()}</string>
+    <key>EnvironmentVariables</key>
+    <dict>
+${envDict.join('\n')}
+    </dict>
+    <key>RunAtLoad</key>
+    <true/>
+    <key>KeepAlive</key>
+    <dict>
+        <key>SuccessfulExit</key>
+        <false/>
+    </dict>
+    <key>ThrottleInterval</key>
+    <integer>10</integer>
+    <key>StandardOutPath</key>
+    <string>${logDir}/out.log</string>
+    <key>StandardErrorPath</key>
+    <string>${logDir}/error.log</string>
+</dict>
+</plist>
+`;
+}
+
+export function macInstall(config) {
+  const plistPath = getLaunchdPlistPath();
+  // Unload first if exists
+  if (existsSync(plistPath)) {
+    try { execSync(`launchctl unload ${plistPath} 2>/dev/null`); } catch {}
+  }
+  writeFileSync(plistPath, generateLaunchdPlist(config));
+  execSync(`launchctl load ${plistPath}`);
+  console.log('Service installed and started.');
+  console.log(`\nManage with:`);
+  console.log(`  yeaft-agent status`);
+  console.log(`  yeaft-agent logs`);
+  console.log(`  yeaft-agent restart`);
+  console.log(`  yeaft-agent uninstall`);
+}
+
+export function macUninstall() {
+  const plistPath = getLaunchdPlistPath();
+  if (existsSync(plistPath)) {
+    try { execSync(`launchctl unload ${plistPath}`); } catch {}
+    unlinkSync(plistPath);
+  }
+  console.log('Service uninstalled.');
+}
+
+export function macStart() {
+  const plistPath = getLaunchdPlistPath();
+  if (!existsSync(plistPath)) {
+    console.error('Service not installed. Run "yeaft-agent install" first.');
+    process.exit(1);
+  }
+  execSync(`launchctl load ${plistPath}`);
+  console.log('Service started.');
+}
+
+export function macStop() {
+  const plistPath = getLaunchdPlistPath();
+  if (existsSync(plistPath)) {
+    execSync(`launchctl unload ${plistPath}`);
+  }
+  console.log('Service stopped.');
+}
+
+export function macRestart() {
+  macStop();
+  macStart();
+}
+
+export function macStatus() {
+  try {
+    const output = execSync(`launchctl list | grep com.yeaft.agent`, { encoding: 'utf-8' });
+    if (output.trim()) {
+      const parts = output.trim().split(/\s+/);
+      const pid = parts[0];
+      const exitCode = parts[1];
+      if (pid !== '-') {
+        console.log(`Service is running (PID: ${pid})`);
+      } else {
+        console.log(`Service is stopped (last exit code: ${exitCode})`);
+      }
+    } else {
+      console.log('Service is not installed.');
+    }
+  } catch {
+    console.log('Service is not installed.');
+  }
+}
+
+export function macLogs() {
+  const logFile = join(getLogDir(), 'out.log');
+  if (existsSync(logFile)) {
+    execSync(`tail -f -n 100 ${logFile}`, { stdio: 'inherit' });
+  } else {
+    console.log('No logs found.');
+  }
+}

package/service/windows.js

@@ -0,0 +1,181 @@
+/**
+ * Service — Windows (pm2) platform implementation
+ */
+import { execSync, spawn } from 'child_process';
+import { existsSync, mkdirSync, writeFileSync, readFileSync, unlinkSync } from 'fs';
+import { join, dirname } from 'path';
+import { SERVICE_NAME, getConfigDir, getLogDir, getNodePath, getCliPath } from './config.js';
+
+const WIN_TASK_NAME = 'YeaftAgent';
+const PM2_APP_NAME = 'yeaft-agent';
+
+// Legacy paths for cleanup
+function getWinWrapperPath() { return join(getConfigDir(), 'run.vbs'); }
+function getWinBatPath() { return join(getConfigDir(), 'run.bat'); }
+
+function ensurePm2() {
+  try {
+    execSync('pm2 --version', { stdio: 'pipe' });
+  } catch {
+    console.log('Installing pm2...');
+    execSync('npm install -g pm2', { stdio: 'inherit' });
+  }
+}
+
+function getEcosystemPath() {
+  return join(getConfigDir(), 'ecosystem.config.cjs');
+}
+
+function generateEcosystem(config) {
+  const nodePath = getNodePath();
+  const cliPath = getCliPath();
+  const cliDir = dirname(cliPath);
+  const logDir = getLogDir();
+
+  const env = {};
+  if (config.serverUrl) env.SERVER_URL = config.serverUrl;
+  if (config.agentName) env.AGENT_NAME = config.agentName;
+  if (config.agentSecret) env.AGENT_SECRET = config.agentSecret;
+  if (config.workDir) env.WORK_DIR = config.workDir;
+
+  return `module.exports = {
+  apps: [{
+    name: '${PM2_APP_NAME}',
+    script: '${cliPath.replace(/\\/g, '\\\\')}',
+    interpreter: '${nodePath.replace(/\\/g, '\\\\')}',
+    cwd: '${cliDir.replace(/\\/g, '\\\\')}',
+    env: ${JSON.stringify(env, null, 6)},
+    autorestart: true,
+    watch: false,
+    max_restarts: 10,
+    restart_delay: 5000,
+    log_date_format: 'YYYY-MM-DD HH:mm:ss',
+    error_file: '${join(logDir, 'error.log').replace(/\\/g, '\\\\')}',
+    out_file: '${join(logDir, 'out.log').replace(/\\/g, '\\\\')}',
+    merge_logs: true,
+    max_memory_restart: '500M',
+  }]
+};
+`;
+}
+
+export function winInstall(config) {
+  ensurePm2();
+  const logDir = getLogDir();
+  mkdirSync(logDir, { recursive: true });
+
+  // Generate ecosystem config
+  const ecoPath = getEcosystemPath();
+  writeFileSync(ecoPath, generateEcosystem(config));
+
+  // Stop existing instance if any
+  try { execSync(`pm2 delete ${PM2_APP_NAME}`, { stdio: 'pipe' }); } catch {}
+
+  // Start with pm2
+  execSync(`pm2 start "${ecoPath}"`, { stdio: 'inherit' });
+
+  // Save pm2 process list for resurrection
+  execSync('pm2 save', { stdio: 'pipe' });
+
+  // Setup auto-start: create startup script in Windows Startup folder
+  // pm2-startup doesn't work well on Windows, use Startup folder approach
+  const trayScript = join(dirname(getCliPath()), 'scripts', 'agent-tray.ps1');
+  const startupDir = join(process.env.APPDATA, 'Microsoft', 'Windows', 'Start Menu', 'Programs', 'Startup');
+  const startupBat = join(startupDir, `${PM2_APP_NAME}.bat`);
+  // Resurrect pm2 processes + launch tray icon
+  let batContent = `@echo off\r\npm2 resurrect\r\n`;
+  if (existsSync(trayScript)) {
+    batContent += `start "" powershell -WindowStyle Hidden -ExecutionPolicy Bypass -File "${trayScript}"\r\n`;
+  }
+  writeFileSync(startupBat, batContent);
+
+  // Launch tray now
+  if (existsSync(trayScript)) {
+    spawn('powershell', ['-WindowStyle', 'Hidden', '-ExecutionPolicy', 'Bypass', '-File', trayScript], {
+      detached: true, stdio: 'ignore'
+    }).unref();
+  }
+
+  console.log(`\nService installed and started.`);
+  console.log(`  Ecosystem: ${ecoPath}`);
+  console.log(`  Startup:   ${startupBat}`);
+  console.log(`\nManage with:`);
+  console.log(`  yeaft-agent status`);
+  console.log(`  yeaft-agent logs`);
+  console.log(`  yeaft-agent restart`);
+  console.log(`  yeaft-agent uninstall`);
+}
+
+export function winUninstall() {
+  try { execSync(`pm2 delete ${PM2_APP_NAME}`, { stdio: 'pipe' }); } catch {}
+  try { execSync('pm2 save', { stdio: 'pipe' }); } catch {}
+  // Clean up ecosystem config
+  const ecoPath = getEcosystemPath();
+  if (existsSync(ecoPath)) unlinkSync(ecoPath);
+  // Clean up Startup bat
+  const startupBat = join(process.env.APPDATA, 'Microsoft', 'Windows', 'Start Menu', 'Programs', 'Startup', `${PM2_APP_NAME}.bat`);
+  if (existsSync(startupBat)) unlinkSync(startupBat);
+  // Clean up legacy files
+  const vbsPath = getWinWrapperPath();
+  const batPath = getWinBatPath();
+  if (existsSync(vbsPath)) unlinkSync(vbsPath);
+  if (existsSync(batPath)) unlinkSync(batPath);
+  const startupVbs = join(process.env.APPDATA, 'Microsoft', 'Windows', 'Start Menu', 'Programs', 'Startup', `${WIN_TASK_NAME}.vbs`);
+  if (existsSync(startupVbs)) unlinkSync(startupVbs);
+  console.log('Service uninstalled.');
+}
+
+export function winStart() {
+  try {
+    execSync(`pm2 start ${PM2_APP_NAME}`, { stdio: 'inherit' });
+  } catch {
+    // Try ecosystem file
+    const ecoPath = getEcosystemPath();
+    if (existsSync(ecoPath)) {
+      execSync(`pm2 start "${ecoPath}"`, { stdio: 'inherit' });
+    } else {
+      console.error('Service not installed. Run "yeaft-agent install" first.');
+      process.exit(1);
+    }
+  }
+}
+
+export function winStop() {
+  try {
+    execSync(`pm2 stop ${PM2_APP_NAME}`, { stdio: 'inherit' });
+  } catch {
+    console.error('Service not running or not installed.');
+  }
+}
+
+export function winRestart() {
+  try {
+    execSync(`pm2 restart ${PM2_APP_NAME}`, { stdio: 'inherit' });
+  } catch {
+    console.error('Service not running. Use "yeaft-agent start" to start.');
+  }
+}
+
+export function winStatus() {
+  try {
+    execSync(`pm2 describe ${PM2_APP_NAME}`, { stdio: 'inherit' });
+  } catch {
+    console.log('Service is not installed.');
+  }
+}
+
+export function winLogs() {
+  const child = spawn('pm2', ['logs', PM2_APP_NAME, '--lines', '100'], {
+    stdio: 'inherit',
+    shell: true
+  });
+  child.on('error', () => {
+    // Fallback to reading log file directly
+    const logFile = join(getLogDir(), 'out.log');
+    if (existsSync(logFile)) {
+      console.log(readFileSync(logFile, 'utf-8'));
+    } else {
+      console.log('No logs found.');
+    }
+  });
+}