@exaudeus/workrail 3.32.0 → 3.34.0

package/dist/cli/commands/worktrain-daemon.js

@@ -0,0 +1,272 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.executeWorktrainDaemonCommand = executeWorktrainDaemonCommand;
+const cli_result_js_1 = require("../types/cli-result.js");
+const LAUNCHD_LABEL = 'io.worktrain.daemon';
+const PLIST_FILENAME = `${LAUNCHD_LABEL}.plist`;
+const CAPTURED_ENV_VARS = [
+    'AWS_PROFILE',
+    'AWS_ACCESS_KEY_ID',
+    'AWS_SECRET_ACCESS_KEY',
+    'AWS_SESSION_TOKEN',
+    'ANTHROPIC_API_KEY',
+    'WORKRAIL_TRIGGERS_ENABLED',
+    'WORKRAIL_DEFAULT_WORKSPACE',
+    'GITHUB_TOKEN',
+    'GITLAB_TOKEN',
+    'HOME',
+    'USER',
+    'PATH',
+    'WORKRAIL_DEV',
+    'WORKRAIL_LOG_LEVEL',
+    'WORKRAIL_VERBOSE_LOGGING',
+];
+function buildPlist(nodeBinPath, worktrainBinPath, envVars, logDir, homeDir) {
+    const envEntries = Object.entries(envVars)
+        .map(([k, v]) => `    <key>${escapeXml(k)}</key>\n    <string>${escapeXml(v)}</string>`)
+        .join('\n');
+    const stdoutLog = `${logDir}/daemon.stdout.log`;
+    const stderrLog = `${logDir}/daemon.stderr.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>${LAUNCHD_LABEL}</string>
+
+  <key>ProgramArguments</key>
+  <array>
+    <string>${escapeXml(nodeBinPath)}</string>
+    <string>${escapeXml(worktrainBinPath)}</string>
+    <string>daemon</string>
+  </array>
+
+  <key>WorkingDirectory</key>
+  <string>${escapeXml(homeDir)}</string>
+
+  <key>EnvironmentVariables</key>
+  <dict>
+${envEntries}
+  </dict>
+
+  <key>StandardOutPath</key>
+  <string>${escapeXml(stdoutLog)}</string>
+
+  <key>StandardErrorPath</key>
+  <string>${escapeXml(stderrLog)}</string>
+
+  <key>RunAtLoad</key>
+  <true/>
+
+  <key>KeepAlive</key>
+  <true/>
+
+  <!--
+    ThrottleInterval: minimum seconds between launchd restarts.
+    WHY 30s: prevents launchd from spinning in a tight restart loop if the daemon
+    exits immediately (e.g., missing credentials or invalid workspace path).
+    Without this, a misconfigured service consumes CPU and spams logs.
+  -->
+  <key>ThrottleInterval</key>
+  <integer>30</integer>
+</dict>
+</plist>
+`;
+}
+function escapeXml(s) {
+    return s
+        .replace(/&/g, '&amp;')
+        .replace(/</g, '&lt;')
+        .replace(/>/g, '&gt;')
+        .replace(/"/g, '&quot;')
+        .replace(/'/g, '&apos;');
+}
+function captureEnvVars(env, warn) {
+    const captured = {};
+    for (const key of CAPTURED_ENV_VARS) {
+        const value = env[key];
+        if (value !== undefined && value !== '') {
+            captured[key] = value;
+        }
+    }
+    const existing = captured['WORKRAIL_TRIGGERS_ENABLED'];
+    if (!existing) {
+        captured['WORKRAIL_TRIGGERS_ENABLED'] = 'true';
+    }
+    else if (existing !== 'true') {
+        warn(`[worktrain daemon --install] WORKRAIL_TRIGGERS_ENABLED is set to '${existing}' in your environment. ` +
+            `The plist will override this with 'true' so the daemon can start. ` +
+            `Remove WORKRAIL_TRIGGERS_ENABLED from your shell environment if you do not want this warning.`);
+        captured['WORKRAIL_TRIGGERS_ENABLED'] = 'true';
+    }
+    return captured;
+}
+function parseLaunchctlList(stdout, exitCode) {
+    if (exitCode !== 0) {
+        return { running: false, pid: null, loaded: false };
+    }
+    try {
+        const parsed = JSON.parse(stdout);
+        const pid = typeof parsed['PID'] === 'number' ? parsed['PID'] : null;
+        return { running: pid !== null, pid, loaded: true };
+    }
+    catch {
+        return { running: false, pid: null, loaded: false };
+    }
+}
+async function runInstall(deps) {
+    const home = deps.homedir();
+    const plistDir = deps.joinPath(home, 'Library', 'LaunchAgents');
+    const plistPath = deps.joinPath(plistDir, PLIST_FILENAME);
+    const logDir = deps.joinPath(home, '.workrail', 'logs');
+    const env = deps.env;
+    const hasBedrock = !!(env['AWS_PROFILE'] || env['AWS_ACCESS_KEY_ID']);
+    const hasAnthropic = !!env['ANTHROPIC_API_KEY'];
+    if (!hasBedrock && !hasAnthropic) {
+        return (0, cli_result_js_1.failure)('No LLM credentials found in the current environment. ' +
+            'Set AWS_PROFILE (for Bedrock) or ANTHROPIC_API_KEY (for Anthropic) ' +
+            'before running --install so the daemon can authenticate.', {
+            suggestions: [
+                'export AWS_PROFILE=your-sso-profile',
+                'export ANTHROPIC_API_KEY=sk-ant-...',
+            ],
+        });
+    }
+    deps.print('Installing WorkTrain daemon as a launchd service...');
+    await deps.mkdir(plistDir, { recursive: true });
+    await deps.mkdir(logDir, { recursive: true });
+    const alreadyInstalled = await deps.exists(plistPath);
+    if (alreadyInstalled) {
+        deps.print('  Existing service found -- unloading before reinstall...');
+        await deps.exec('launchctl', ['unload', plistPath]);
+    }
+    const capturedEnv = captureEnvVars(env, (msg) => console.warn(msg));
+    const plist = buildPlist(deps.nodeBinPath, deps.worktrainBinPath, capturedEnv, logDir, home);
+    await deps.writeFile(plistPath, plist);
+    await deps.chmod(plistPath, 0o600);
+    deps.print(`  Plist written: ${plistPath}`);
+    const loadResult = await deps.exec('launchctl', ['load', plistPath]);
+    if (loadResult.exitCode !== 0) {
+        return (0, cli_result_js_1.failure)(`launchctl load failed (exit ${loadResult.exitCode}): ${loadResult.stderr.trim() || loadResult.stdout.trim()}`, {
+            suggestions: [
+                `Check the plist manually: plutil -lint ${plistPath}`,
+                `View daemon logs: tail -f ${logDir}/daemon.stderr.log`,
+            ],
+        });
+    }
+    await deps.sleep(1500);
+    const listResult = await deps.exec('launchctl', ['list', LAUNCHD_LABEL]);
+    const status = parseLaunchctlList(listResult.stdout, listResult.exitCode);
+    if (!status.loaded) {
+        return (0, cli_result_js_1.failure)(`Service loaded but launchctl cannot find it. This may be a transient issue.`, {
+            suggestions: [
+                `Check: launchctl list ${LAUNCHD_LABEL}`,
+                `View daemon logs: tail -f ${logDir}/daemon.stderr.log`,
+            ],
+        });
+    }
+    deps.print('');
+    if (status.running) {
+        deps.print(`WorkTrain daemon installed and running (PID ${status.pid}).`);
+    }
+    else {
+        deps.print(`WorkTrain daemon installed. Service loaded but not yet running.`);
+        deps.print(`This may be normal if WORKRAIL_TRIGGERS_ENABLED was not set.`);
+    }
+    deps.print(`Logs: ${logDir}/daemon.stdout.log`);
+    deps.print(`      ${logDir}/daemon.stderr.log`);
+    return (0, cli_result_js_1.success)({
+        message: status.running
+            ? `WorkTrain daemon installed and running (PID ${status.pid})`
+            : 'WorkTrain daemon installed (service loaded, not yet running)',
+        details: [
+            `Plist: ${plistPath}`,
+            `Logs:  ${logDir}/daemon.stdout.log`,
+            `       ${logDir}/daemon.stderr.log`,
+        ],
+    });
+}
+async function runUninstall(deps) {
+    const home = deps.homedir();
+    const plistPath = deps.joinPath(home, 'Library', 'LaunchAgents', PLIST_FILENAME);
+    const exists = await deps.exists(plistPath);
+    if (!exists) {
+        return (0, cli_result_js_1.failure)('WorkTrain daemon is not installed (plist not found).', { suggestions: [`Expected: ${plistPath}`] });
+    }
+    deps.print('Uninstalling WorkTrain daemon...');
+    const unloadResult = await deps.exec('launchctl', ['unload', plistPath]);
+    if (unloadResult.exitCode !== 0) {
+        deps.print(`  Warning: launchctl unload returned non-zero: ${unloadResult.stderr.trim()}`);
+    }
+    else {
+        deps.print('  Service unloaded.');
+    }
+    await deps.removeFile(plistPath);
+    deps.print(`  Plist removed: ${plistPath}`);
+    return (0, cli_result_js_1.success)({ message: 'WorkTrain daemon uninstalled successfully.' });
+}
+async function runStatus(deps) {
+    const home = deps.homedir();
+    const plistPath = deps.joinPath(home, 'Library', 'LaunchAgents', PLIST_FILENAME);
+    const logDir = deps.joinPath(home, '.workrail', 'logs');
+    const plistExists = await deps.exists(plistPath);
+    const listResult = await deps.exec('launchctl', ['list', LAUNCHD_LABEL]);
+    const status = parseLaunchctlList(listResult.stdout, listResult.exitCode);
+    deps.print('');
+    deps.print('WorkTrain daemon status:');
+    deps.print(`  Plist installed : ${plistExists ? `yes (${plistPath})` : 'no'}`);
+    deps.print(`  Service loaded  : ${status.loaded ? 'yes' : 'no'}`);
+    deps.print(`  Running         : ${status.running ? `yes (PID ${status.pid})` : 'no'}`);
+    if (plistExists || status.loaded) {
+        deps.print(`  Logs (stdout)   : ${logDir}/daemon.stdout.log`);
+        deps.print(`  Logs (stderr)   : ${logDir}/daemon.stderr.log`);
+    }
+    if (!plistExists && !status.loaded) {
+        deps.print('');
+        deps.print('Daemon is not installed. Run: worktrain daemon --install');
+    }
+    else if (plistExists && !status.running) {
+        deps.print('');
+        deps.print(`Daemon installed but not running. Check logs: tail -f ${logDir}/daemon.stderr.log`);
+    }
+    deps.print('');
+    return (0, cli_result_js_1.success)({
+        message: status.running
+            ? `WorkTrain daemon is running (PID ${status.pid})`
+            : plistExists
+                ? 'WorkTrain daemon is installed but not running'
+                : 'WorkTrain daemon is not installed',
+    });
+}
+async function executeWorktrainDaemonCommand(deps, opts) {
+    const flagCount = [opts.install, opts.uninstall, opts.status].filter(Boolean).length;
+    if (flagCount === 0) {
+        if (deps.startDaemon) {
+            await deps.startDaemon();
+            return (0, cli_result_js_1.success)({ message: 'WorkTrain daemon stopped.' });
+        }
+        return (0, cli_result_js_1.misuse)('Specify one of: --install, --uninstall, or --status', [
+            'worktrain daemon --install    Install and start as a launchd service',
+            'worktrain daemon --uninstall  Stop and remove the launchd service',
+            'worktrain daemon --status     Show service status',
+        ]);
+    }
+    if (flagCount > 1) {
+        return (0, cli_result_js_1.misuse)('--install, --uninstall, and --status are mutually exclusive. Specify only one.');
+    }
+    if (deps.platform !== 'darwin') {
+        return (0, cli_result_js_1.failure)(`worktrain daemon --install requires macOS (launchd). ` +
+            `Current platform: ${deps.platform}.`, {
+            suggestions: [
+                'On Linux, use systemd: create a user service with systemctl --user.',
+                'See docs/daemon-service.md for platform-specific instructions.',
+            ],
+        });
+    }
+    if (opts.install)
+        return runInstall(deps);
+    if (opts.uninstall)
+        return runUninstall(deps);
+    return runStatus(deps);
+}

package/dist/cli/commands/worktrain-spawn.js

@@ -3,20 +3,22 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.executeWorktrainSpawnCommand = executeWorktrainSpawnCommand;
 const cli_result_js_1 = require("../types/cli-result.js");
 const DEFAULT_CONSOLE_PORT = 3456;
-const LOCK_FILE_NAME = 'dashboard.lock';
+const LOCK_FILE_NAMES = ['daemon-console.lock', 'dashboard.lock'];
 async function discoverConsolePort(deps, portOverride) {
     if (portOverride !== undefined && portOverride > 0) {
         return portOverride;
     }
-    const lockPath = deps.joinPath(deps.homedir(), '.workrail', LOCK_FILE_NAME);
-    try {
-        const raw = await deps.readFile(lockPath);
-        const parsed = JSON.parse(raw);
-        if (typeof parsed.port === 'number' && parsed.port > 0) {
-            return parsed.port;
+    for (const lockFileName of LOCK_FILE_NAMES) {
+        const lockPath = deps.joinPath(deps.homedir(), '.workrail', lockFileName);
+        try {
+            const raw = await deps.readFile(lockPath);
+            const parsed = JSON.parse(raw);
+            if (typeof parsed.port === 'number' && parsed.port > 0) {
+                return parsed.port;
+            }
+        }
+        catch {
         }
-    }
-    catch {
     }
     return DEFAULT_CONSOLE_PORT;
 }