@proletariat/cli 0.3.43 → 0.3.44

package/dist/commands/orchestrator/stop.js

@@ -0,0 +1,104 @@
+import { Flags } from '@oclif/core';
+import { execSync } from 'node:child_process';
+import * as path from 'node:path';
+import * as fs from 'node:fs';
+import Database from 'better-sqlite3';
+import { PromptCommand } from '../../lib/prompt-command.js';
+import { machineOutputFlags } from '../../lib/pmo/index.js';
+import { findHQRoot } from '../../lib/workspace.js';
+import { shouldOutputJson, outputErrorAsJson, outputSuccessAsJson, createMetadata, } from '../../lib/prompt-json.js';
+import { styles } from '../../lib/styles.js';
+import { getHostTmuxSessionNames } from '../../lib/execution/session-utils.js';
+import { ExecutionStorage } from '../../lib/execution/storage.js';
+import { ORCHESTRATOR_SESSION_NAME } from './start.js';
+export default class OrchestratorStop extends PromptCommand {
+    static description = 'Stop the running orchestrator';
+    static examples = [
+        '<%= config.bin %> <%= command.id %>',
+        '<%= config.bin %> <%= command.id %> --force',
+    ];
+    static flags = {
+        ...machineOutputFlags,
+        force: Flags.boolean({
+            char: 'f',
+            description: 'Skip confirmation',
+            default: false,
+        }),
+    };
+    async run() {
+        const { flags } = await this.parse(OrchestratorStop);
+        const jsonMode = shouldOutputJson(flags);
+        // Check if orchestrator session exists
+        const hostSessions = getHostTmuxSessionNames();
+        if (!hostSessions.includes(ORCHESTRATOR_SESSION_NAME)) {
+            if (jsonMode) {
+                outputErrorAsJson('NOT_RUNNING', 'Orchestrator is not running.', createMetadata('orchestrator stop', flags));
+                return;
+            }
+            this.log('');
+            this.log(styles.muted('Orchestrator is not running.'));
+            this.log('');
+            return;
+        }
+        // Confirm unless --force
+        if (!flags.force && !jsonMode) {
+            const { confirmed } = await this.prompt([{
+                    type: 'list',
+                    name: 'confirmed',
+                    message: 'Stop the orchestrator?',
+                    choices: [
+                        { name: 'Yes', value: true },
+                        { name: 'No', value: false },
+                    ],
+                }]);
+            if (!confirmed) {
+                this.log(styles.muted('Cancelled.'));
+                return;
+            }
+        }
+        // Kill the tmux session
+        try {
+            execSync(`tmux kill-session -t "${ORCHESTRATOR_SESSION_NAME}"`, { stdio: 'pipe' });
+        }
+        catch (error) {
+            if (jsonMode) {
+                outputErrorAsJson('KILL_FAILED', `Failed to stop orchestrator: ${error instanceof Error ? error.message : error}`, createMetadata('orchestrator stop', flags));
+                return;
+            }
+            this.error(`Failed to stop orchestrator: ${error instanceof Error ? error.message : error}`);
+        }
+        // Update execution record to stopped
+        const hqPath = findHQRoot(process.cwd());
+        if (hqPath) {
+            const dbPath = path.join(hqPath, '.proletariat', 'workspace.db');
+            if (fs.existsSync(dbPath)) {
+                let db = null;
+                try {
+                    db = new Database(dbPath);
+                    const executionStorage = new ExecutionStorage(db);
+                    const running = executionStorage.listExecutions({ agentName: 'orchestrator', status: 'running' });
+                    const starting = executionStorage.listExecutions({ agentName: 'orchestrator', status: 'starting' });
+                    for (const exec of [...running, ...starting]) {
+                        executionStorage.updateStatus(exec.id, 'stopped');
+                    }
+                }
+                catch {
+                    // Non-fatal
+                }
+                finally {
+                    db?.close();
+                }
+            }
+        }
+        if (jsonMode) {
+            outputSuccessAsJson({
+                sessionId: ORCHESTRATOR_SESSION_NAME,
+                status: 'stopped',
+            }, createMetadata('orchestrator stop', flags));
+            return;
+        }
+        this.log('');
+        this.log(styles.success('Orchestrator stopped.'));
+        this.log('');
+    }
+}