@bbigbang/core 0.1.0

package/dist/services/workspaceToolRunLedger.js

@@ -0,0 +1,239 @@
+import { randomUUID } from 'node:crypto';
+import { WorkspaceToolServiceError } from './workspaceToolErrors.js';
+import { normalizeWorkspaceToolRunEventEnvelope, runRowToInfo, serializeJson, } from './workspaceToolExecutionUtils.js';
+import { workspaceToolRunRowToInfoForViewer } from './workspaceToolReadModel.js';
+export class WorkspaceToolRunLedger {
+    deps;
+    constructor(deps) {
+        this.deps = deps;
+    }
+    hasRunRow(runId) {
+        const row = this.deps.db.prepare(`SELECT 1 as present
+         FROM workspace_tool_runs
+        WHERE run_id = ?
+        LIMIT 1`).get(runId);
+        return Boolean(row);
+    }
+    insertRun(run) {
+        this.deps.db.prepare(`INSERT INTO workspace_tool_runs(
+         run_id,
+         tool_id,
+         action_id,
+         terminal_id,
+         status,
+         params_json,
+         exit_code,
+         signal,
+         started_at,
+         ended_at,
+         requested_by_user_id,
+         requested_by_agent_id,
+         tool_revision,
+         source_bundle_root,
+         action_cwd_absolute,
+         runtime_log_dir,
+         execution_target_kind,
+         execution_agent_id,
+         execution_node_id,
+         execution_hostname,
+         execution_workspace_path
+       )
+       VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`).run(run.runId, run.toolId, run.actionId, run.terminalId, run.status, run.paramsJson, run.exitCode, run.signal, run.startedAt, run.endedAt, run.requestedByUserId, run.requestedByAgentId ?? null, run.toolRevision ?? null, run.sourceBundleRoot ?? null, run.actionCwdAbsolute ?? null, run.runtimeLogDir ?? null, run.executionTargetKind ?? null, run.executionAgentId ?? null, run.executionNodeId ?? null, run.executionHostname ?? null, run.executionWorkspacePath ?? null);
+    }
+    completeRun(runId, params) {
+        this.deps.db.prepare(`UPDATE workspace_tool_runs
+          SET status = ?,
+              ended_at = ?,
+              exit_code = ?,
+              signal = ?
+        WHERE run_id = ?`).run(params.status, params.endedAt, params.exitCode ?? null, params.signal ?? null, runId);
+    }
+    completeRunAndClearActiveTerminal(runId, toolId, completion) {
+        this.completeRun(runId, completion);
+        this.deps.db.prepare(`UPDATE workspace_tools
+          SET active_terminal_id = NULL,
+              updated_at = ?
+        WHERE tool_id = ?`).run(Date.now(), toolId);
+    }
+    requireRun(runId, actionKind, viewerUserId, ownerUserId) {
+        const row = this.deps.db.prepare(`SELECT run_id as runId,
+              wtr.tool_id as toolId,
+              action_id as actionId,
+              terminal_id as terminalId,
+              status,
+              params_json as paramsJson,
+              exit_code as exitCode,
+              signal,
+              started_at as startedAt,
+              ended_at as endedAt,
+              requested_by_user_id as requestedByUserId,
+              u.username as requestedByUsername,
+              requested_by_agent_id as requestedByAgentId,
+              a.name as requestedByAgentName,
+              execution_target_kind as executionTargetKind,
+              runtime_log_dir as runtimeLogDir,
+              execution_agent_id as executionAgentId,
+              execution_node_id as executionNodeId,
+              execution_hostname as executionHostname,
+              execution_workspace_path as executionWorkspacePath
+         FROM workspace_tool_runs wtr
+         LEFT JOIN users u ON u.id = wtr.requested_by_user_id
+         LEFT JOIN agents a ON a.agent_id = wtr.requested_by_agent_id
+        WHERE run_id = ?`).get(runId);
+        if (!row) {
+            throw new WorkspaceToolServiceError('Tool run not found.', 500);
+        }
+        return this.runInfoForActionViewer(row, actionKind, viewerUserId, ownerUserId);
+    }
+    runInfoForActionViewer(row, actionKind, viewerUserId, ownerUserId) {
+        const canViewOwnerFields = viewerUserId === ownerUserId || this.deps.isAdminUser(viewerUserId);
+        return workspaceToolRunRowToInfoForViewer(row, actionKind, {
+            viewerUserId,
+            ownerUserId,
+            canViewOwnerFields,
+        });
+    }
+    runRowToInfo(row, actionKind) {
+        return runRowToInfo(row, actionKind);
+    }
+    recordRunEventsFromOutput(runId, toolId, outputText) {
+        for (const line of outputText.split('\n')) {
+            const trimmed = line.trim();
+            if (!trimmed)
+                continue;
+            let parsed;
+            try {
+                parsed = JSON.parse(trimmed);
+            }
+            catch {
+                continue;
+            }
+            this.insertRunEventFromEnvelope(runId, toolId, parsed);
+        }
+    }
+    insertRunEventFromEnvelope(runId, toolId, value) {
+        const event = normalizeWorkspaceToolRunEventEnvelope(value);
+        if (!event)
+            return false;
+        this.deps.db.prepare(`INSERT INTO workspace_tool_run_events(
+         event_id,
+         run_id,
+         tool_id,
+         sequence,
+         event_type,
+         level,
+         summary,
+         payload_json,
+         artifacts_json,
+         created_at
+       )
+       VALUES(
+         ?,
+         ?,
+         ?,
+         COALESCE((SELECT MAX(sequence) + 1 FROM workspace_tool_run_events WHERE run_id = ?), 1),
+         ?,
+         ?,
+         ?,
+         ?,
+         ?,
+         ?
+       )`).run(randomUUID(), runId, toolId, runId, event.eventType, event.level, event.summary, event.payloadJson, event.artifactsJson, Date.now());
+        return true;
+    }
+    insertCleanupRunEvent(runId, toolId, type, payload) {
+        const payloadJson = serializeJson({
+            actionId: payload.actionId,
+            timeoutMs: payload.timeoutMs,
+            exitCode: payload.exitCode,
+            signal: payload.signal,
+            error: payload.error,
+        });
+        this.deps.db.prepare(`INSERT INTO workspace_tool_run_events(
+         event_id,
+         run_id,
+         tool_id,
+         sequence,
+         event_type,
+         level,
+         summary,
+         payload_json,
+         artifacts_json,
+         created_at
+       )
+       VALUES(
+         ?,
+         ?,
+         ?,
+         COALESCE((SELECT MAX(sequence) + 1 FROM workspace_tool_run_events WHERE run_id = ?), 1),
+         ?,
+         ?,
+         ?,
+         ?,
+         NULL,
+         ?
+       )`).run(randomUUID(), runId, toolId, runId, type, type === 'cleanup_completed' || type === 'cleanup_started' ? 'info' : 'error', payload.summary, payloadJson, Date.now());
+    }
+    insertPersistentRuntimeActionTimeoutRunEvent(runId, toolId, payload) {
+        const payloadJson = serializeJson({
+            actionId: payload.actionId,
+            timeoutMs: payload.timeoutMs,
+            error: payload.error,
+            expectedResultPath: payload.expectedResultPath,
+        });
+        this.deps.db.prepare(`INSERT INTO workspace_tool_run_events(
+         event_id,
+         run_id,
+         tool_id,
+         sequence,
+         event_type,
+        level,
+         summary,
+         payload_json,
+         artifacts_json,
+         created_at
+       )
+       VALUES(
+         ?,
+         ?,
+         ?,
+         COALESCE((SELECT MAX(sequence) + 1 FROM workspace_tool_run_events WHERE run_id = ?), 1),
+         'runtime_action_timed_out',
+         'error',
+         ?,
+         ?,
+         NULL,
+         ?
+       )`).run(randomUUID(), runId, toolId, runId, `Persistent runtime action "${payload.actionId}" timed out: did not write a result or error event per protocol.`, payloadJson, Date.now());
+    }
+    insertMissingTerminalRunEvent(runId, toolId, payload) {
+        const payloadJson = serializeJson({
+            actionId: payload.actionId,
+            error: payload.error,
+        });
+        this.deps.db.prepare(`INSERT INTO workspace_tool_run_events(
+         event_id,
+         run_id,
+         tool_id,
+         sequence,
+         event_type,
+         level,
+         summary,
+         payload_json,
+         artifacts_json,
+         created_at
+       )
+       VALUES(
+         ?,
+         ?,
+         ?,
+         COALESCE((SELECT MAX(sequence) + 1 FROM workspace_tool_run_events WHERE run_id = ?), 1),
+         'runtime_terminal_missing',
+         'error',
+         ?,
+         ?,
+         NULL,
+         ?
+       )`).run(randomUUID(), runId, toolId, runId, `Tool run terminal was lost before the action completed: ${payload.error}`, payloadJson, Date.now());
+    }
+}