shennian 0.2.6 → 0.2.8

package/dist/src/commands/agent.d.ts

@@ -0,0 +1,2 @@
+import type { Command } from 'commander';
+export declare function registerAgentCommand(program: Command): void;

package/dist/src/commands/agent.js

@@ -0,0 +1,131 @@
+import chalk from 'chalk';
+import { loadConfig, saveConfig } from '../config/index.js';
+import { spawnCommandString } from '../agents/command-spec.js';
+function getCapsModels(caps) {
+    if (caps.models?.length)
+        return caps.models;
+    if (caps.model)
+        return [caps.model];
+    return [];
+}
+/**
+ * Runs `<command> /caps` and returns the parsed caps JSON.
+ */
+async function probeCaps(command) {
+    return new Promise((resolve, reject) => {
+        const proc = spawnCommandString(command, ['/caps'], { stdio: ['ignore', 'pipe', 'pipe'] });
+        let stdout = '';
+        let stderr = '';
+        proc.stdout?.on('data', (d) => { stdout += d.toString(); });
+        proc.stderr?.on('data', (d) => { stderr += d.toString(); });
+        proc.on('error', (err) => reject(new Error(`Failed to run command: ${err.message}`)));
+        proc.on('close', (_code) => {
+            const line = stdout.trim().split('\n').find((l) => l.trim().startsWith('{'));
+            if (!line) {
+                reject(new Error(`Command did not output valid caps JSON.\nstdout: ${stdout.trim()}\nstderr: ${stderr.trim()}`));
+                return;
+            }
+            try {
+                const caps = JSON.parse(line);
+                if (!caps.name || !caps.mode) {
+                    reject(new Error('Caps JSON must contain "name" and "mode" fields'));
+                    return;
+                }
+                if (getCapsModels(caps).length === 0) {
+                    reject(new Error('Caps JSON must contain either "model" or a non-empty "models" array'));
+                    return;
+                }
+                if (caps.mode !== 'spawn' && caps.mode !== 'stdio') {
+                    reject(new Error(`Caps "mode" must be "spawn" or "stdio", got: ${caps.mode}`));
+                    return;
+                }
+                resolve(caps);
+            }
+            catch {
+                reject(new Error(`Failed to parse caps JSON: ${line}`));
+            }
+        });
+        setTimeout(() => {
+            proc.kill('SIGTERM');
+            reject(new Error('Timed out waiting for caps response (5s)'));
+        }, 5000);
+    });
+}
+export function registerAgentCommand(program) {
+    const agent = program
+        .command('agent')
+        .description('Manage custom agents');
+    // ── add ──────────────────────────────────────────────────────────────────
+    agent
+        .command('add <name>')
+        .description('Register a custom agent')
+        .requiredOption('--command <cmd>', 'Command to run the agent (e.g. "python /path/agent.py")')
+        .action(async (name, opts) => {
+        if (!/^[a-zA-Z0-9_-]+$/.test(name)) {
+            console.error(chalk.red('✗ Agent name must contain only letters, numbers, hyphens, and underscores'));
+            process.exit(1);
+        }
+        console.log(chalk.gray(`Probing caps: ${opts.command} /caps ...`));
+        let caps;
+        try {
+            caps = await probeCaps(opts.command);
+        }
+        catch (err) {
+            console.error(chalk.red(`✗ ${err instanceof Error ? err.message : String(err)}`));
+            process.exit(1);
+        }
+        const config = loadConfig();
+        config.customAgents = config.customAgents ?? {};
+        config.customAgents[name] = { command: opts.command, caps };
+        saveConfig(config);
+        console.log(chalk.green(`✓ Agent "${name}" registered`));
+        console.log(chalk.gray(`  Name:    ${caps.name}`));
+        console.log(chalk.gray(`  Models:  ${getCapsModels(caps).join(', ')}`));
+        console.log(chalk.gray(`  Mode:    ${caps.mode}`));
+        if (caps.version)
+            console.log(chalk.gray(`  Version: ${caps.version}`));
+        console.log();
+        console.log(chalk.yellow(`  The agent will appear as "custom:${name}" in the Shennian app.`));
+        console.log(chalk.yellow('  Refresh Agents in the app to hot-reload this daemon, or restart the daemon as a fallback.'));
+    });
+    // ── list ─────────────────────────────────────────────────────────────────
+    agent
+        .command('list')
+        .description('List registered custom agents')
+        .action(() => {
+        const config = loadConfig();
+        const entries = Object.entries(config.customAgents ?? {});
+        if (entries.length === 0) {
+            console.log(chalk.gray('No custom agents registered.'));
+            console.log(chalk.gray('Add one with: shennian agent add <name> --command "<cmd>"'));
+            return;
+        }
+        console.log(chalk.bold('\nCustom agents:\n'));
+        for (const [name, entry] of entries) {
+            const models = entry.caps.models?.length
+                ? entry.caps.models.join(', ')
+                : entry.caps.model ?? '(none)';
+            console.log(`  ${chalk.cyan(name)}`);
+            console.log(`    ${chalk.gray('display:')}  ${entry.caps.name}`);
+            console.log(`    ${chalk.gray('models:')}   ${models}`);
+            console.log(`    ${chalk.gray('mode:')}     ${entry.caps.mode}`);
+            console.log(`    ${chalk.gray('command:')}  ${entry.command}`);
+            console.log();
+        }
+    });
+    // ── remove ────────────────────────────────────────────────────────────────
+    agent
+        .command('remove <name>')
+        .description('Unregister a custom agent')
+        .action((name) => {
+        const config = loadConfig();
+        if (!config.customAgents?.[name]) {
+            console.error(chalk.red(`✗ No agent named "${name}"`));
+            process.exit(1);
+        }
+        delete config.customAgents[name];
+        saveConfig(config);
+        console.log(chalk.green(`✓ Agent "${name}" removed`));
+        console.log(chalk.yellow('  Refresh Agents in the app to hot-reload this daemon, or restart the daemon as a fallback.'));
+    });
+}

package/dist/src/commands/daemon.d.ts

@@ -0,0 +1,35 @@
+import type { Command } from 'commander';
+type ServiceLaunchMode = 'direct' | 'global-shim' | 'npx';
+export type ServiceLaunchSpec = {
+    command: string;
+    args: string[];
+    mode: ServiceLaunchMode;
+};
+export declare function isEphemeralCliPath(candidate: string): boolean;
+export declare function resolveServiceLaunchSpec(input: {
+    nodeExec: string;
+    scriptPath: string;
+    shennianCommandPath?: string | null;
+    npxPath?: string | null;
+}): ServiceLaunchSpec;
+export declare function buildWindowsLauncherCommand(spec: ServiceLaunchSpec, logFile: string): string;
+export declare function buildWindowsStartupVbs(cmdPath: string): string;
+export declare function buildWindowsScheduledTaskXml(input: {
+    userId: string;
+    command: string;
+    arguments: string[];
+}): string;
+export declare function captureEnvForService(): Record<string, string>;
+/**
+ * Save current env vars so the daemon can load them at startup.
+ * This is critical for launchd/systemd which start with minimal env.
+ */
+export declare function saveEnvSnapshot(): void;
+export declare function startDaemonProcess(): void;
+/**
+ * Install the platform-native auto-start service.
+ * Returns true if the service was immediately started (so caller can skip startDaemonProcess).
+ */
+export declare function installService(): boolean;
+export declare function registerDaemonCommand(program: Command): void;
+export {};