@rudderhq/server 0.1.0-canary.0

package/dist/services/workspace-operation-log-store.js

@@ -0,0 +1,115 @@
+import { createReadStream, promises as fs } from "node:fs";
+import path from "node:path";
+import { createHash } from "node:crypto";
+import { notFound } from "../errors.js";
+import { resolveRudderInstanceRoot } from "../home-paths.js";
+function safeSegments(...segments) {
+    return segments.map((segment) => segment.replace(/[^a-zA-Z0-9._-]/g, "_"));
+}
+function resolveWithin(basePath, relativePath) {
+    const resolved = path.resolve(basePath, relativePath);
+    const base = path.resolve(basePath) + path.sep;
+    if (!resolved.startsWith(base) && resolved !== path.resolve(basePath)) {
+        throw new Error("Invalid log path");
+    }
+    return resolved;
+}
+function createLocalFileWorkspaceOperationLogStore(basePath) {
+    async function ensureDir(relativeDir) {
+        const dir = resolveWithin(basePath, relativeDir);
+        await fs.mkdir(dir, { recursive: true });
+    }
+    async function readFileRange(filePath, offset, limitBytes) {
+        const stat = await fs.stat(filePath).catch(() => null);
+        if (!stat)
+            throw notFound("Workspace operation log not found");
+        const start = Math.max(0, Math.min(offset, stat.size));
+        if (stat.size === 0 || start >= stat.size) {
+            return { content: "", endOffset: start, eof: true };
+        }
+        const end = Math.max(start, Math.min(start + limitBytes - 1, stat.size - 1));
+        if (start > end) {
+            return { content: "", endOffset: start, eof: true };
+        }
+        const chunks = [];
+        await new Promise((resolve, reject) => {
+            const stream = createReadStream(filePath, { start, end });
+            stream.on("data", (chunk) => {
+                chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
+            });
+            stream.on("error", reject);
+            stream.on("end", () => resolve());
+        });
+        const content = Buffer.concat(chunks).toString("utf8");
+        const endOffset = end + 1;
+        const eof = endOffset >= stat.size;
+        const nextOffset = eof ? undefined : endOffset;
+        return { content, endOffset, eof, nextOffset };
+    }
+    async function sha256File(filePath) {
+        return new Promise((resolve, reject) => {
+            const hash = createHash("sha256");
+            const stream = createReadStream(filePath);
+            stream.on("data", (chunk) => hash.update(chunk));
+            stream.on("error", reject);
+            stream.on("end", () => resolve(hash.digest("hex")));
+        });
+    }
+    return {
+        async begin(input) {
+            const [orgId] = safeSegments(input.orgId);
+            const operationId = safeSegments(input.operationId)[0];
+            const relDir = orgId;
+            const relPath = path.join(relDir, `${operationId}.ndjson`);
+            await ensureDir(relDir);
+            const absPath = resolveWithin(basePath, relPath);
+            await fs.writeFile(absPath, "", "utf8");
+            return { store: "local_file", logRef: relPath };
+        },
+        async append(handle, event) {
+            if (handle.store !== "local_file")
+                return;
+            const absPath = resolveWithin(basePath, handle.logRef);
+            const line = JSON.stringify({
+                ts: event.ts,
+                stream: event.stream,
+                chunk: event.chunk,
+            });
+            await fs.appendFile(absPath, `${line}\n`, "utf8");
+        },
+        async finalize(handle) {
+            if (handle.store !== "local_file") {
+                return { bytes: 0, compressed: false };
+            }
+            const absPath = resolveWithin(basePath, handle.logRef);
+            const stat = await fs.stat(absPath).catch(() => null);
+            if (!stat)
+                throw notFound("Workspace operation log not found");
+            const hash = await sha256File(absPath);
+            return {
+                bytes: stat.size,
+                sha256: hash,
+                compressed: false,
+            };
+        },
+        async read(handle, opts) {
+            if (handle.store !== "local_file") {
+                throw notFound("Workspace operation log not found");
+            }
+            const absPath = resolveWithin(basePath, handle.logRef);
+            const offset = opts?.offset ?? 0;
+            const limitBytes = opts?.limitBytes ?? 256_000;
+            return readFileRange(absPath, offset, limitBytes);
+        },
+    };
+}
+let cachedStore = null;
+export function getWorkspaceOperationLogStore() {
+    if (cachedStore)
+        return cachedStore;
+    const basePath = process.env.WORKSPACE_OPERATION_LOG_BASE_PATH
+        ?? path.resolve(resolveRudderInstanceRoot(), "data", "workspace-operation-logs");
+    cachedStore = createLocalFileWorkspaceOperationLogStore(basePath);
+    return cachedStore;
+}
+//# sourceMappingURL=workspace-operation-log-store.js.map

package/dist/services/workspace-operation-log-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace-operation-log-store.js","sourceRoot":"","sources":["../../src/services/workspace-operation-log-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AAC3D,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,yBAAyB,EAAE,MAAM,kBAAkB,CAAC;AAqC7D,SAAS,YAAY,CAAC,GAAG,QAAkB;IACzC,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,EAAE,GAAG,CAAC,CAAC,CAAC;AAC7E,CAAC;AAED,SAAS,aAAa,CAAC,QAAgB,EAAE,YAAoB;IAC3D,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IACtD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC;IAC/C,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,QAAQ,KAAK,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACtE,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;IACtC,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,yCAAyC,CAAC,QAAgB;IACjE,KAAK,UAAU,SAAS,CAAC,WAAmB;QAC1C,MAAM,GAAG,GAAG,aAAa,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QACjD,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,UAAU,aAAa,CAAC,QAAgB,EAAE,MAAc,EAAE,UAAkB;QAC/E,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;QACvD,IAAI,CAAC,IAAI;YAAE,MAAM,QAAQ,CAAC,mCAAmC,CAAC,CAAC;QAE/D,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACvD,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC1C,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;QACtD,CAAC;QACD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,UAAU,GAAG,CAAC,EAAE,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;QAE7E,IAAI,KAAK,GAAG,GAAG,EAAE,CAAC;YAChB,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;QACtD,CAAC;QAED,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC1C,MAAM,MAAM,GAAG,gBAAgB,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;YAC1D,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;gBAC1B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YACnE,CAAC,CAAC,CAAC;YACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAC3B,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACvD,MAAM,SAAS,GAAG,GAAG,GAAG,CAAC,CAAC;QAC1B,MAAM,GAAG,GAAG,SAAS,IAAI,IAAI,CAAC,IAAI,CAAC;QACnC,MAAM,UAAU,GAAG,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QAC/C,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC;IACjD,CAAC;IAED,KAAK,UAAU,UAAU,CAAC,QAAgB;QACxC,OAAO,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC7C,MAAM,IAAI,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;YAClC,MAAM,MAAM,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAC1C,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YACjD,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAC3B,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC;IAED,OAAO;QACL,KAAK,CAAC,KAAK,CAAC,KAAK;YACf,MAAM,CAAC,KAAK,CAAC,GAAG,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC1C,MAAM,WAAW,GAAG,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,CAAE,CAAC;YACxD,MAAM,MAAM,GAAG,KAAK,CAAC;YACrB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,WAAW,SAAS,CAAC,CAAC;YAC3D,MAAM,SAAS,CAAC,MAAM,CAAC,CAAC;YAExB,MAAM,OAAO,GAAG,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACjD,MAAM,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC;YAExC,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;QAClD,CAAC;QAED,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK;YACxB,IAAI,MAAM,CAAC,KAAK,KAAK,YAAY;gBAAE,OAAO;YAC1C,MAAM,OAAO,GAAG,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YACvD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC;gBAC1B,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,KAAK,EAAE,KAAK,CAAC,KAAK;aACnB,CAAC,CAAC;YACH,MAAM,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,EAAE,MAAM,CAAC,CAAC;QACpD,CAAC;QAED,KAAK,CAAC,QAAQ,CAAC,MAAM;YACnB,IAAI,MAAM,CAAC,KAAK,KAAK,YAAY,EAAE,CAAC;gBAClC,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;YACzC,CAAC;YACD,MAAM,OAAO,GAAG,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YACvD,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;YACtD,IAAI,CAAC,IAAI;gBAAE,MAAM,QAAQ,CAAC,mCAAmC,CAAC,CAAC;YAE/D,MAAM,IAAI,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,CAAC;YACvC,OAAO;gBACL,KAAK,EAAE,IAAI,CAAC,IAAI;gBAChB,MAAM,EAAE,IAAI;gBACZ,UAAU,EAAE,KAAK;aAClB,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI;YACrB,IAAI,MAAM,CAAC,KAAK,KAAK,YAAY,EAAE,CAAC;gBAClC,MAAM,QAAQ,CAAC,mCAAmC,CAAC,CAAC;YACtD,CAAC;YACD,MAAM,OAAO,GAAG,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YACvD,MAAM,MAAM,GAAG,IAAI,EAAE,MAAM,IAAI,CAAC,CAAC;YACjC,MAAM,UAAU,GAAG,IAAI,EAAE,UAAU,IAAI,OAAO,CAAC;YAC/C,OAAO,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACpD,CAAC;KACF,CAAC;AACJ,CAAC;AAED,IAAI,WAAW,GAAsC,IAAI,CAAC;AAE1D,MAAM,UAAU,6BAA6B;IAC3C,IAAI,WAAW;QAAE,OAAO,WAAW,CAAC;IACpC,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,iCAAiC;WACzD,IAAI,CAAC,OAAO,CAAC,yBAAyB,EAAE,EAAE,MAAM,EAAE,0BAA0B,CAAC,CAAC;IACnF,WAAW,GAAG,yCAAyC,CAAC,QAAQ,CAAC,CAAC;IAClE,OAAO,WAAW,CAAC;AACrB,CAAC"}

package/dist/services/workspace-operations.d.ts

@@ -0,0 +1,46 @@
+import type { Db } from "@rudderhq/db";
+import { workspaceOperations } from "@rudderhq/db";
+import type { WorkspaceOperation, WorkspaceOperationPhase, WorkspaceOperationStatus } from "@rudderhq/shared";
+type WorkspaceOperationRow = typeof workspaceOperations.$inferSelect;
+declare function toWorkspaceOperation(row: WorkspaceOperationRow): WorkspaceOperation;
+export interface WorkspaceOperationRecorder {
+    attachExecutionWorkspaceId(executionWorkspaceId: string | null): Promise<void>;
+    recordOperation(input: {
+        phase: WorkspaceOperationPhase;
+        command?: string | null;
+        cwd?: string | null;
+        metadata?: Record<string, unknown> | null;
+        run: () => Promise<{
+            status?: WorkspaceOperationStatus;
+            exitCode?: number | null;
+            stdout?: string | null;
+            stderr?: string | null;
+            system?: string | null;
+            metadata?: Record<string, unknown> | null;
+        }>;
+    }): Promise<WorkspaceOperation>;
+}
+export declare function workspaceOperationService(db: Db): {
+    getById: (id: string) => Promise<WorkspaceOperation | null>;
+    createRecorder(input: {
+        orgId: string;
+        heartbeatRunId?: string | null;
+        executionWorkspaceId?: string | null;
+    }): WorkspaceOperationRecorder;
+    listForRun: (runId: string, executionWorkspaceId?: string | null) => Promise<WorkspaceOperation[]>;
+    listForExecutionWorkspace: (executionWorkspaceId: string) => Promise<WorkspaceOperation[]>;
+    readLog: (operationId: string, opts?: {
+        offset?: number;
+        limitBytes?: number;
+    }) => Promise<{
+        content: string;
+        endOffset: number;
+        eof: boolean;
+        nextOffset?: number;
+        operationId: string;
+        store: string;
+        logRef: string;
+    }>;
+};
+export { toWorkspaceOperation };
+//# sourceMappingURL=workspace-operations.d.ts.map

package/dist/services/workspace-operations.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace-operations.d.ts","sourceRoot":"","sources":["../../src/services/workspace-operations.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AACnD,OAAO,KAAK,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,MAAM,kBAAkB,CAAC;AAQ9G,KAAK,qBAAqB,GAAG,OAAO,mBAAmB,CAAC,YAAY,CAAC;AAErE,iBAAS,oBAAoB,CAAC,GAAG,EAAE,qBAAqB,GAAG,kBAAkB,CAwB5E;AAiBD,MAAM,WAAW,0BAA0B;IACzC,0BAA0B,CAAC,oBAAoB,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/E,eAAe,CAAC,KAAK,EAAE;QACrB,KAAK,EAAE,uBAAuB,CAAC;QAC/B,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACxB,GAAG,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACpB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;QAC1C,GAAG,EAAE,MAAM,OAAO,CAAC;YACjB,MAAM,CAAC,EAAE,wBAAwB,CAAC;YAClC,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;YACzB,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;YACvB,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;YACvB,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;YACvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;SAC3C,CAAC,CAAC;KACJ,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;CACjC;AAED,wBAAgB,yBAAyB,CAAC,EAAE,EAAE,EAAE;kBAInB,MAAM;0BAYT;QACpB,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC/B,oBAAoB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KACtC,GAAG,0BAA0B;wBAgJJ,MAAM,yBAAyB,MAAM,GAAG,IAAI;sDAmBd,MAAM;2BASjC,MAAM,SAAS;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE;;;;;;;;;EAwBvF;AAED,OAAO,EAAE,oBAAoB,EAAE,CAAC"}

package/dist/services/workspace-operations.js

@@ -0,0 +1,237 @@
+import { randomUUID } from "node:crypto";
+import { workspaceOperations } from "@rudderhq/db";
+import { asc, desc, eq, inArray, isNull, or, and } from "drizzle-orm";
+import { notFound } from "../errors.js";
+import { withExecutionObservation } from "../langfuse.js";
+import { redactCurrentUserText, redactCurrentUserValue } from "../log-redaction.js";
+import { instanceSettingsService } from "./instance-settings.js";
+import { getWorkspaceOperationLogStore } from "./workspace-operation-log-store.js";
+function toWorkspaceOperation(row) {
+    return {
+        id: row.id,
+        orgId: row.orgId,
+        executionWorkspaceId: row.executionWorkspaceId ?? null,
+        heartbeatRunId: row.heartbeatRunId ?? null,
+        phase: row.phase,
+        command: row.command ?? null,
+        cwd: row.cwd ?? null,
+        status: row.status,
+        exitCode: row.exitCode ?? null,
+        logStore: row.logStore ?? null,
+        logRef: row.logRef ?? null,
+        logBytes: row.logBytes ?? null,
+        logSha256: row.logSha256 ?? null,
+        logCompressed: row.logCompressed,
+        stdoutExcerpt: row.stdoutExcerpt ?? null,
+        stderrExcerpt: row.stderrExcerpt ?? null,
+        metadata: row.metadata ?? null,
+        startedAt: row.startedAt,
+        finishedAt: row.finishedAt ?? null,
+        createdAt: row.createdAt,
+        updatedAt: row.updatedAt,
+    };
+}
+function appendExcerpt(current, chunk) {
+    return `${current}${chunk}`.slice(-4096);
+}
+function combineMetadata(base, patch) {
+    if (!base && !patch)
+        return null;
+    return {
+        ...(base ?? {}),
+        ...(patch ?? {}),
+    };
+}
+export function workspaceOperationService(db) {
+    const instanceSettings = instanceSettingsService(db);
+    const logStore = getWorkspaceOperationLogStore();
+    async function getById(id) {
+        const row = await db
+            .select()
+            .from(workspaceOperations)
+            .where(eq(workspaceOperations.id, id))
+            .then((rows) => rows[0] ?? null);
+        return row ? toWorkspaceOperation(row) : null;
+    }
+    return {
+        getById,
+        createRecorder(input) {
+            let executionWorkspaceId = input.executionWorkspaceId ?? null;
+            const createdIds = [];
+            return {
+                async attachExecutionWorkspaceId(nextExecutionWorkspaceId) {
+                    executionWorkspaceId = nextExecutionWorkspaceId ?? null;
+                    if (!executionWorkspaceId || createdIds.length === 0)
+                        return;
+                    await db
+                        .update(workspaceOperations)
+                        .set({
+                        executionWorkspaceId,
+                        updatedAt: new Date(),
+                    })
+                        .where(inArray(workspaceOperations.id, createdIds));
+                },
+                async recordOperation(recordInput) {
+                    const currentUserRedactionOptions = {
+                        enabled: (await instanceSettings.getGeneral()).censorUsernameInLogs,
+                    };
+                    const startedAt = new Date();
+                    const id = randomUUID();
+                    const handle = await logStore.begin({
+                        orgId: input.orgId,
+                        operationId: id,
+                    });
+                    let stdoutExcerpt = "";
+                    let stderrExcerpt = "";
+                    const append = async (stream, chunk) => {
+                        if (!chunk)
+                            return;
+                        const sanitizedChunk = redactCurrentUserText(chunk, currentUserRedactionOptions);
+                        if (stream === "stdout")
+                            stdoutExcerpt = appendExcerpt(stdoutExcerpt, sanitizedChunk);
+                        if (stream === "stderr")
+                            stderrExcerpt = appendExcerpt(stderrExcerpt, sanitizedChunk);
+                        await logStore.append(handle, {
+                            stream,
+                            chunk: sanitizedChunk,
+                            ts: new Date().toISOString(),
+                        });
+                    };
+                    await db.insert(workspaceOperations).values({
+                        id,
+                        orgId: input.orgId,
+                        executionWorkspaceId,
+                        heartbeatRunId: input.heartbeatRunId ?? null,
+                        phase: recordInput.phase,
+                        command: recordInput.command ?? null,
+                        cwd: recordInput.cwd ?? null,
+                        status: "running",
+                        logStore: handle.store,
+                        logRef: handle.logRef,
+                        metadata: redactCurrentUserValue(recordInput.metadata ?? null, currentUserRedactionOptions),
+                        startedAt,
+                    });
+                    createdIds.push(id);
+                    return withExecutionObservation({
+                        surface: "workspace_operation",
+                        rootExecutionId: input.heartbeatRunId ?? id,
+                        orgId: input.orgId,
+                        sessionKey: executionWorkspaceId,
+                        status: "running",
+                        metadata: {
+                            operationId: id,
+                            phase: recordInput.phase,
+                            command: recordInput.command ?? null,
+                            executionWorkspaceId,
+                            standalone: input.heartbeatRunId == null,
+                        },
+                    }, {
+                        name: input.heartbeatRunId
+                            ? `workspace.${recordInput.phase}`
+                            : `workspace_operation:${recordInput.phase}:${id}`,
+                        asType: "tool",
+                        input: {
+                            command: recordInput.command ?? null,
+                            cwd: recordInput.cwd ?? null,
+                        },
+                        metadata: recordInput.metadata ?? undefined,
+                    }, async () => {
+                        try {
+                            const result = await recordInput.run();
+                            await append("system", result.system ?? null);
+                            await append("stdout", result.stdout ?? null);
+                            await append("stderr", result.stderr ?? null);
+                            const finalized = await logStore.finalize(handle);
+                            const finishedAt = new Date();
+                            const row = await db
+                                .update(workspaceOperations)
+                                .set({
+                                executionWorkspaceId,
+                                status: result.status ?? "succeeded",
+                                exitCode: result.exitCode ?? null,
+                                stdoutExcerpt: stdoutExcerpt || null,
+                                stderrExcerpt: stderrExcerpt || null,
+                                logBytes: finalized.bytes,
+                                logSha256: finalized.sha256,
+                                logCompressed: finalized.compressed,
+                                metadata: redactCurrentUserValue(combineMetadata(recordInput.metadata, result.metadata), currentUserRedactionOptions),
+                                finishedAt,
+                                updatedAt: finishedAt,
+                            })
+                                .where(eq(workspaceOperations.id, id))
+                                .returning()
+                                .then((rows) => rows[0] ?? null);
+                            if (!row)
+                                throw notFound("Workspace operation not found");
+                            return toWorkspaceOperation(row);
+                        }
+                        catch (error) {
+                            await append("stderr", error instanceof Error ? error.message : String(error));
+                            const finalized = await logStore.finalize(handle).catch(() => null);
+                            const finishedAt = new Date();
+                            await db
+                                .update(workspaceOperations)
+                                .set({
+                                executionWorkspaceId,
+                                status: "failed",
+                                stdoutExcerpt: stdoutExcerpt || null,
+                                stderrExcerpt: stderrExcerpt || null,
+                                logBytes: finalized?.bytes ?? null,
+                                logSha256: finalized?.sha256 ?? null,
+                                logCompressed: finalized?.compressed ?? false,
+                                finishedAt,
+                                updatedAt: finishedAt,
+                            })
+                                .where(eq(workspaceOperations.id, id));
+                            throw error;
+                        }
+                    });
+                },
+            };
+        },
+        listForRun: async (runId, executionWorkspaceId) => {
+            const conditions = [eq(workspaceOperations.heartbeatRunId, runId)];
+            if (executionWorkspaceId) {
+                const cleanupCondition = and(eq(workspaceOperations.executionWorkspaceId, executionWorkspaceId), isNull(workspaceOperations.heartbeatRunId));
+                if (cleanupCondition)
+                    conditions.push(cleanupCondition);
+            }
+            const rows = await db
+                .select()
+                .from(workspaceOperations)
+                .where(conditions.length === 1 ? conditions[0] : or(...conditions))
+                .orderBy(asc(workspaceOperations.startedAt), asc(workspaceOperations.createdAt), asc(workspaceOperations.id));
+            return rows.map(toWorkspaceOperation);
+        },
+        listForExecutionWorkspace: async (executionWorkspaceId) => {
+            const rows = await db
+                .select()
+                .from(workspaceOperations)
+                .where(eq(workspaceOperations.executionWorkspaceId, executionWorkspaceId))
+                .orderBy(desc(workspaceOperations.startedAt), desc(workspaceOperations.createdAt));
+            return rows.map(toWorkspaceOperation);
+        },
+        readLog: async (operationId, opts) => {
+            const operation = await getById(operationId);
+            if (!operation)
+                throw notFound("Workspace operation not found");
+            if (!operation.logStore || !operation.logRef)
+                throw notFound("Workspace operation log not found");
+            const result = await logStore.read({
+                store: operation.logStore,
+                logRef: operation.logRef,
+            }, opts);
+            return {
+                operationId,
+                store: operation.logStore,
+                logRef: operation.logRef,
+                ...result,
+                content: redactCurrentUserText(result.content, {
+                    enabled: (await instanceSettings.getGeneral()).censorUsernameInLogs,
+                }),
+            };
+        },
+    };
+}
+export { toWorkspaceOperation };
+//# sourceMappingURL=workspace-operations.js.map

package/dist/services/workspace-operations.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace-operations.js","sourceRoot":"","sources":["../../src/services/workspace-operations.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AAEnD,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AACtE,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,wBAAwB,EAAE,MAAM,gBAAgB,CAAC;AAC1D,OAAO,EAAE,qBAAqB,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AACpF,OAAO,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,6BAA6B,EAAE,MAAM,oCAAoC,CAAC;AAInF,SAAS,oBAAoB,CAAC,GAA0B;IACtD,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,oBAAoB,EAAE,GAAG,CAAC,oBAAoB,IAAI,IAAI;QACtD,cAAc,EAAE,GAAG,CAAC,cAAc,IAAI,IAAI;QAC1C,KAAK,EAAE,GAAG,CAAC,KAAgC;QAC3C,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,IAAI;QAC5B,GAAG,EAAE,GAAG,CAAC,GAAG,IAAI,IAAI;QACpB,MAAM,EAAE,GAAG,CAAC,MAAkC;QAC9C,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,IAAI;QAC9B,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,IAAI;QAC9B,MAAM,EAAE,GAAG,CAAC,MAAM,IAAI,IAAI;QAC1B,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,IAAI;QAC9B,SAAS,EAAE,GAAG,CAAC,SAAS,IAAI,IAAI;QAChC,aAAa,EAAE,GAAG,CAAC,aAAa;QAChC,aAAa,EAAE,GAAG,CAAC,aAAa,IAAI,IAAI;QACxC,aAAa,EAAE,GAAG,CAAC,aAAa,IAAI,IAAI;QACxC,QAAQ,EAAG,GAAG,CAAC,QAA2C,IAAI,IAAI;QAClE,SAAS,EAAE,GAAG,CAAC,SAAS;QACxB,UAAU,EAAE,GAAG,CAAC,UAAU,IAAI,IAAI;QAClC,SAAS,EAAE,GAAG,CAAC,SAAS;QACxB,SAAS,EAAE,GAAG,CAAC,SAAS;KACzB,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,OAAe,EAAE,KAAa;IACnD,OAAO,GAAG,OAAO,GAAG,KAAK,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC;AAC3C,CAAC;AAED,SAAS,eAAe,CACtB,IAAgD,EAChD,KAAiD;IAEjD,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACjC,OAAO;QACL,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC;QACf,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;KACjB,CAAC;AACJ,CAAC;AAoBD,MAAM,UAAU,yBAAyB,CAAC,EAAM;IAC9C,MAAM,gBAAgB,GAAG,uBAAuB,CAAC,EAAE,CAAC,CAAC;IACrD,MAAM,QAAQ,GAAG,6BAA6B,EAAE,CAAC;IAEjD,KAAK,UAAU,OAAO,CAAC,EAAU;QAC/B,MAAM,GAAG,GAAG,MAAM,EAAE;aACjB,MAAM,EAAE;aACR,IAAI,CAAC,mBAAmB,CAAC;aACzB,KAAK,CAAC,EAAE,CAAC,mBAAmB,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;aACrC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC;QACnC,OAAO,GAAG,CAAC,CAAC,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAChD,CAAC;IAED,OAAO;QACL,OAAO;QAEP,cAAc,CAAC,KAId;YACC,IAAI,oBAAoB,GAAG,KAAK,CAAC,oBAAoB,IAAI,IAAI,CAAC;YAC9D,MAAM,UAAU,GAAa,EAAE,CAAC;YAEhC,OAAO;gBACL,KAAK,CAAC,0BAA0B,CAAC,wBAAwB;oBACvD,oBAAoB,GAAG,wBAAwB,IAAI,IAAI,CAAC;oBACxD,IAAI,CAAC,oBAAoB,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;wBAAE,OAAO;oBAC7D,MAAM,EAAE;yBACL,MAAM,CAAC,mBAAmB,CAAC;yBAC3B,GAAG,CAAC;wBACH,oBAAoB;wBACpB,SAAS,EAAE,IAAI,IAAI,EAAE;qBACtB,CAAC;yBACD,KAAK,CAAC,OAAO,CAAC,mBAAmB,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC,CAAC;gBACxD,CAAC;gBAED,KAAK,CAAC,eAAe,CAAC,WAAW;oBAC/B,MAAM,2BAA2B,GAAG;wBAClC,OAAO,EAAE,CAAC,MAAM,gBAAgB,CAAC,UAAU,EAAE,CAAC,CAAC,oBAAoB;qBACpE,CAAC;oBACF,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;oBAC7B,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;oBACxB,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC;wBAClC,KAAK,EAAE,KAAK,CAAC,KAAK;wBAClB,WAAW,EAAE,EAAE;qBAChB,CAAC,CAAC;oBAEH,IAAI,aAAa,GAAG,EAAE,CAAC;oBACvB,IAAI,aAAa,GAAG,EAAE,CAAC;oBACvB,MAAM,MAAM,GAAG,KAAK,EAAE,MAAsC,EAAE,KAAgC,EAAE,EAAE;wBAChG,IAAI,CAAC,KAAK;4BAAE,OAAO;wBACnB,MAAM,cAAc,GAAG,qBAAqB,CAAC,KAAK,EAAE,2BAA2B,CAAC,CAAC;wBACjF,IAAI,MAAM,KAAK,QAAQ;4BAAE,aAAa,GAAG,aAAa,CAAC,aAAa,EAAE,cAAc,CAAC,CAAC;wBACtF,IAAI,MAAM,KAAK,QAAQ;4BAAE,aAAa,GAAG,aAAa,CAAC,aAAa,EAAE,cAAc,CAAC,CAAC;wBACtF,MAAM,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE;4BAC5B,MAAM;4BACN,KAAK,EAAE,cAAc;4BACrB,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;yBAC7B,CAAC,CAAC;oBACL,CAAC,CAAC;oBAEF,MAAM,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,MAAM,CAAC;wBAC1C,EAAE;wBACF,KAAK,EAAE,KAAK,CAAC,KAAK;wBAClB,oBAAoB;wBACpB,cAAc,EAAE,KAAK,CAAC,cAAc,IAAI,IAAI;wBAC5C,KAAK,EAAE,WAAW,CAAC,KAAK;wBACxB,OAAO,EAAE,WAAW,CAAC,OAAO,IAAI,IAAI;wBACpC,GAAG,EAAE,WAAW,CAAC,GAAG,IAAI,IAAI;wBAC5B,MAAM,EAAE,SAAS;wBACjB,QAAQ,EAAE,MAAM,CAAC,KAAK;wBACtB,MAAM,EAAE,MAAM,CAAC,MAAM;wBACrB,QAAQ,EAAE,sBAAsB,CAC9B,WAAW,CAAC,QAAQ,IAAI,IAAI,EAC5B,2BAA2B,CACM;wBACnC,SAAS;qBACV,CAAC,CAAC;oBACH,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAEpB,OAAO,wBAAwB,CAC7B;wBACE,OAAO,EAAE,qBAAqB;wBAC9B,eAAe,EAAE,KAAK,CAAC,cAAc,IAAI,EAAE;wBAC3C,KAAK,EAAE,KAAK,CAAC,KAAK;wBAClB,UAAU,EAAE,oBAAoB;wBAChC,MAAM,EAAE,SAAS;wBACjB,QAAQ,EAAE;4BACR,WAAW,EAAE,EAAE;4BACf,KAAK,EAAE,WAAW,CAAC,KAAK;4BACxB,OAAO,EAAE,WAAW,CAAC,OAAO,IAAI,IAAI;4BACpC,oBAAoB;4BACpB,UAAU,EAAE,KAAK,CAAC,cAAc,IAAI,IAAI;yBACzC;qBACF,EACD;wBACE,IAAI,EAAE,KAAK,CAAC,cAAc;4BACxB,CAAC,CAAC,aAAa,WAAW,CAAC,KAAK,EAAE;4BAClC,CAAC,CAAC,uBAAuB,WAAW,CAAC,KAAK,IAAI,EAAE,EAAE;wBACpD,MAAM,EAAE,MAAM;wBACd,KAAK,EAAE;4BACL,OAAO,EAAE,WAAW,CAAC,OAAO,IAAI,IAAI;4BACpC,GAAG,EAAE,WAAW,CAAC,GAAG,IAAI,IAAI;yBAC7B;wBACD,QAAQ,EAAE,WAAW,CAAC,QAAQ,IAAI,SAAS;qBAC5C,EACD,KAAK,IAAI,EAAE;wBACT,IAAI,CAAC;4BACH,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,GAAG,EAAE,CAAC;4BACvC,MAAM,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;4BAC9C,MAAM,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;4BAC9C,MAAM,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;4BAC9C,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;4BAClD,MAAM,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC;4BAC9B,MAAM,GAAG,GAAG,MAAM,EAAE;iCACjB,MAAM,CAAC,mBAAmB,CAAC;iCAC3B,GAAG,CAAC;gCACH,oBAAoB;gCACpB,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,WAAW;gCACpC,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,IAAI;gCACjC,aAAa,EAAE,aAAa,IAAI,IAAI;gCACpC,aAAa,EAAE,aAAa,IAAI,IAAI;gCACpC,QAAQ,EAAE,SAAS,CAAC,KAAK;gCACzB,SAAS,EAAE,SAAS,CAAC,MAAM;gCAC3B,aAAa,EAAE,SAAS,CAAC,UAAU;gCACnC,QAAQ,EAAE,sBAAsB,CAC9B,eAAe,CAAC,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,EACtD,2BAA2B,CACM;gCACnC,UAAU;gCACV,SAAS,EAAE,UAAU;6BACtB,CAAC;iCACD,KAAK,CAAC,EAAE,CAAC,mBAAmB,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;iCACrC,SAAS,EAAE;iCACX,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC;4BACnC,IAAI,CAAC,GAAG;gCAAE,MAAM,QAAQ,CAAC,+BAA+B,CAAC,CAAC;4BAC1D,OAAO,oBAAoB,CAAC,GAAG,CAAC,CAAC;wBACnC,CAAC;wBAAC,OAAO,KAAK,EAAE,CAAC;4BACf,MAAM,MAAM,CAAC,QAAQ,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;4BAC/E,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;4BACpE,MAAM,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC;4BAC9B,MAAM,EAAE;iCACL,MAAM,CAAC,mBAAmB,CAAC;iCAC3B,GAAG,CAAC;gCACH,oBAAoB;gCACpB,MAAM,EAAE,QAAQ;gCAChB,aAAa,EAAE,aAAa,IAAI,IAAI;gCACpC,aAAa,EAAE,aAAa,IAAI,IAAI;gCACpC,QAAQ,EAAE,SAAS,EAAE,KAAK,IAAI,IAAI;gCAClC,SAAS,EAAE,SAAS,EAAE,MAAM,IAAI,IAAI;gCACpC,aAAa,EAAE,SAAS,EAAE,UAAU,IAAI,KAAK;gCAC7C,UAAU;gCACV,SAAS,EAAE,UAAU;6BACtB,CAAC;iCACD,KAAK,CAAC,EAAE,CAAC,mBAAmB,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;4BACzC,MAAM,KAAK,CAAC;wBACd,CAAC;oBACH,CAAC,CACF,CAAC;gBACJ,CAAC;aACF,CAAC;QACJ,CAAC;QAED,UAAU,EAAE,KAAK,EAAE,KAAa,EAAE,oBAAoC,EAAE,EAAE;YACxE,MAAM,UAAU,GAAG,CAAC,EAAE,CAAC,mBAAmB,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC,CAAC;YACnE,IAAI,oBAAoB,EAAE,CAAC;gBACzB,MAAM,gBAAgB,GAAG,GAAG,CAC1B,EAAE,CAAC,mBAAmB,CAAC,oBAAoB,EAAE,oBAAoB,CAAE,EACnE,MAAM,CAAC,mBAAmB,CAAC,cAAc,CAAC,CAC1C,CAAC;gBACH,IAAI,gBAAgB;oBAAE,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAC1D,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,EAAE;iBAClB,MAAM,EAAE;iBACR,IAAI,CAAC,mBAAmB,CAAC;iBACzB,KAAK,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,UAAU,CAAE,CAAC;iBACpE,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEhH,OAAO,IAAI,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC;QAED,yBAAyB,EAAE,KAAK,EAAE,oBAA4B,EAAE,EAAE;YAChE,MAAM,IAAI,GAAG,MAAM,EAAE;iBAClB,MAAM,EAAE;iBACR,IAAI,CAAC,mBAAmB,CAAC;iBACzB,KAAK,CAAC,EAAE,CAAC,mBAAmB,CAAC,oBAAoB,EAAE,oBAAoB,CAAC,CAAC;iBACzE,OAAO,CAAC,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,CAAC;YACrF,OAAO,IAAI,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC;QAED,OAAO,EAAE,KAAK,EAAE,WAAmB,EAAE,IAA+C,EAAE,EAAE;YACtF,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,CAAC;YAC7C,IAAI,CAAC,SAAS;gBAAE,MAAM,QAAQ,CAAC,+BAA+B,CAAC,CAAC;YAChE,IAAI,CAAC,SAAS,CAAC,QAAQ,IAAI,CAAC,SAAS,CAAC,MAAM;gBAAE,MAAM,QAAQ,CAAC,mCAAmC,CAAC,CAAC;YAElG,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAChC;gBACE,KAAK,EAAE,SAAS,CAAC,QAAwB;gBACzC,MAAM,EAAE,SAAS,CAAC,MAAM;aACzB,EACD,IAAI,CACL,CAAC;YAEF,OAAO;gBACL,WAAW;gBACX,KAAK,EAAE,SAAS,CAAC,QAAQ;gBACzB,MAAM,EAAE,SAAS,CAAC,MAAM;gBACxB,GAAG,MAAM;gBACT,OAAO,EAAE,qBAAqB,CAAC,MAAM,CAAC,OAAO,EAAE;oBAC7C,OAAO,EAAE,CAAC,MAAM,gBAAgB,CAAC,UAAU,EAAE,CAAC,CAAC,oBAAoB;iBACpE,CAAC;aACH,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,OAAO,EAAE,oBAAoB,EAAE,CAAC"}

package/dist/services/workspace-runtime.d.ts

@@ -0,0 +1,164 @@
+import type { AgentRuntimeServiceReport } from "@rudderhq/agent-runtime-utils";
+import type { Db } from "@rudderhq/db";
+import type { WorkspaceOperationRecorder } from "./workspace-operations.js";
+export interface ExecutionWorkspaceInput {
+    baseCwd: string;
+    source: "project_primary" | "task_session" | "agent_home";
+    projectId: string | null;
+    workspaceId: string | null;
+    repoUrl: string | null;
+    repoRef: string | null;
+}
+export interface ExecutionWorkspaceIssueRef {
+    id: string;
+    identifier: string | null;
+    title: string | null;
+}
+export interface ExecutionWorkspaceAgentRef {
+    id: string;
+    name: string;
+    orgId: string;
+}
+export interface RealizedExecutionWorkspace extends ExecutionWorkspaceInput {
+    strategy: "project_primary" | "git_worktree";
+    cwd: string;
+    branchName: string | null;
+    worktreePath: string | null;
+    warnings: string[];
+    created: boolean;
+}
+export interface RuntimeServiceRef {
+    id: string;
+    orgId: string;
+    projectId: string | null;
+    projectWorkspaceId: string | null;
+    executionWorkspaceId: string | null;
+    issueId: string | null;
+    serviceName: string;
+    status: "starting" | "running" | "stopped" | "failed";
+    lifecycle: "shared" | "ephemeral";
+    scopeType: "project_workspace" | "execution_workspace" | "run" | "agent";
+    scopeId: string | null;
+    reuseKey: string | null;
+    command: string | null;
+    cwd: string | null;
+    port: number | null;
+    url: string | null;
+    provider: "local_process" | "adapter_managed";
+    providerRef: string | null;
+    ownerAgentId: string | null;
+    startedByRunId: string | null;
+    lastUsedAt: string;
+    startedAt: string;
+    stoppedAt: string | null;
+    stopPolicy: Record<string, unknown> | null;
+    healthStatus: "unknown" | "healthy" | "unhealthy";
+    reused: boolean;
+}
+export declare function sanitizeRuntimeServiceBaseEnv(baseEnv: NodeJS.ProcessEnv): NodeJS.ProcessEnv;
+export declare function realizeExecutionWorkspace(input: {
+    base: ExecutionWorkspaceInput;
+    config: Record<string, unknown>;
+    issue: ExecutionWorkspaceIssueRef | null;
+    agent: ExecutionWorkspaceAgentRef;
+    recorder?: WorkspaceOperationRecorder | null;
+}): Promise<RealizedExecutionWorkspace>;
+export declare function cleanupExecutionWorkspaceArtifacts(input: {
+    workspace: {
+        id: string;
+        cwd: string | null;
+        providerType: string;
+        providerRef: string | null;
+        branchName: string | null;
+        repoUrl: string | null;
+        baseRef: string | null;
+        projectId: string | null;
+        projectWorkspaceId: string | null;
+        sourceIssueId: string | null;
+        metadata?: Record<string, unknown> | null;
+    };
+    projectWorkspace?: {
+        cwd: string | null;
+        cleanupCommand: string | null;
+    } | null;
+    teardownCommand?: string | null;
+    recorder?: WorkspaceOperationRecorder | null;
+}): Promise<{
+    cleanedPath: string | null;
+    cleaned: boolean;
+    warnings: string[];
+}>;
+export declare function normalizeAdapterManagedRuntimeServices(input: {
+    agentRuntimeType: string;
+    runId: string;
+    agent: ExecutionWorkspaceAgentRef;
+    issue: ExecutionWorkspaceIssueRef | null;
+    workspace: RealizedExecutionWorkspace;
+    executionWorkspaceId?: string | null;
+    reports: AgentRuntimeServiceReport[];
+    now?: Date;
+}): RuntimeServiceRef[];
+export declare function ensureRuntimeServicesForRun(input: {
+    db?: Db;
+    runId: string;
+    agent: ExecutionWorkspaceAgentRef;
+    issue: ExecutionWorkspaceIssueRef | null;
+    workspace: RealizedExecutionWorkspace;
+    executionWorkspaceId?: string | null;
+    config: Record<string, unknown>;
+    adapterEnv: Record<string, string>;
+    onLog?: (stream: "stdout" | "stderr", chunk: string) => Promise<void>;
+}): Promise<RuntimeServiceRef[]>;
+export declare function releaseRuntimeServicesForRun(runId: string): Promise<void>;
+export declare function stopRuntimeServicesForExecutionWorkspace(input: {
+    db?: Db;
+    executionWorkspaceId: string;
+    workspaceCwd?: string | null;
+}): Promise<void>;
+export declare function listWorkspaceRuntimeServicesForProjectWorkspaces(db: Db, orgId: string, projectWorkspaceIds: string[]): Promise<Map<string, {
+    url: string | null;
+    id: string;
+    status: string;
+    updatedAt: Date;
+    orgId: string;
+    projectId: string | null;
+    projectWorkspaceId: string | null;
+    executionWorkspaceId: string | null;
+    startedAt: Date;
+    createdAt: Date;
+    scopeType: string;
+    scopeId: string | null;
+    provider: string;
+    cwd: string | null;
+    healthStatus: string;
+    ownerAgentId: string | null;
+    issueId: string | null;
+    command: string | null;
+    port: number | null;
+    lastUsedAt: Date;
+    providerRef: string | null;
+    serviceName: string;
+    lifecycle: string;
+    reuseKey: string | null;
+    startedByRunId: string | null;
+    stoppedAt: Date | null;
+    stopPolicy: Record<string, unknown> | null;
+}[]>>;
+export declare function reconcilePersistedRuntimeServicesOnStartup(db: Db): Promise<{
+    reconciled: number;
+}>;
+export declare function persistAdapterManagedRuntimeServices(input: {
+    db: Db;
+    agentRuntimeType: string;
+    runId: string;
+    agent: ExecutionWorkspaceAgentRef;
+    issue: ExecutionWorkspaceIssueRef | null;
+    workspace: RealizedExecutionWorkspace;
+    executionWorkspaceId?: string | null;
+    reports: AgentRuntimeServiceReport[];
+}): Promise<RuntimeServiceRef[]>;
+export declare function buildWorkspaceReadyComment(input: {
+    workspace: RealizedExecutionWorkspace;
+    runtimeServices: RuntimeServiceRef[];
+}): string;
+//# sourceMappingURL=workspace-runtime.d.ts.map

package/dist/services/workspace-runtime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace-runtime.d.ts","sourceRoot":"","sources":["../../src/services/workspace-runtime.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AAC/E,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,cAAc,CAAC;AAKvC,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,2BAA2B,CAAC;AAE5E,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,iBAAiB,GAAG,cAAc,GAAG,YAAY,CAAC;IAC1D,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;CACxB;AAED,MAAM,WAAW,0BAA0B;IACzC,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;CACtB;AAED,MAAM,WAAW,0BAA0B;IACzC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,0BAA2B,SAAQ,uBAAuB;IACzE,QAAQ,EAAE,iBAAiB,GAAG,cAAc,CAAC;IAC7C,GAAG,EAAE,MAAM,CAAC;IACZ,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,kBAAkB,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC,oBAAoB,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,UAAU,GAAG,SAAS,GAAG,SAAS,GAAG,QAAQ,CAAC;IACtD,SAAS,EAAE,QAAQ,GAAG,WAAW,CAAC;IAClC,SAAS,EAAE,mBAAmB,GAAG,qBAAqB,GAAG,KAAK,GAAG,OAAO,CAAC;IACzE,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,GAAG,EAAE,MAAM,GAAG,IAAI,CAAC;IACnB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IACpB,GAAG,EAAE,MAAM,GAAG,IAAI,CAAC;IACnB,QAAQ,EAAE,eAAe,GAAG,iBAAiB,CAAC;IAC9C,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,cAAc,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IAC3C,YAAY,EAAE,SAAS,GAAG,SAAS,GAAG,WAAW,CAAC;IAClD,MAAM,EAAE,OAAO,CAAC;CACjB;AAyBD,wBAAgB,6BAA6B,CAAC,OAAO,EAAE,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAS3F;AAqbD,wBAAsB,yBAAyB,CAAC,KAAK,EAAE;IACrD,IAAI,EAAE,uBAAuB,CAAC;IAC9B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,KAAK,EAAE,0BAA0B,GAAG,IAAI,CAAC;IACzC,KAAK,EAAE,0BAA0B,CAAC;IAClC,QAAQ,CAAC,EAAE,0BAA0B,GAAG,IAAI,CAAC;CAC9C,GAAG,OAAO,CAAC,0BAA0B,CAAC,CAwItC;AAED,wBAAsB,kCAAkC,CAAC,KAAK,EAAE;IAC9D,SAAS,EAAE;QACT,EAAE,EAAE,MAAM,CAAC;QACX,GAAG,EAAE,MAAM,GAAG,IAAI,CAAC;QACnB,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;QAC3B,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;QAC1B,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QACvB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QACvB,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;QACzB,kBAAkB,EAAE,MAAM,GAAG,IAAI,CAAC;QAClC,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;KAC3C,CAAC;IACF,gBAAgB,CAAC,EAAE;QACjB,GAAG,EAAE,MAAM,GAAG,IAAI,CAAC;QACnB,cAAc,EAAE,MAAM,GAAG,IAAI,CAAC;KAC/B,GAAG,IAAI,CAAC;IACT,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,QAAQ,CAAC,EAAE,0BAA0B,GAAG,IAAI,CAAC;CAC9C;;;;GAkIA;AAgLD,wBAAgB,sCAAsC,CAAC,KAAK,EAAE;IAC5D,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,0BAA0B,CAAC;IAClC,KAAK,EAAE,0BAA0B,GAAG,IAAI,CAAC;IACzC,SAAS,EAAE,0BAA0B,CAAC;IACtC,oBAAoB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACrC,OAAO,EAAE,yBAAyB,EAAE,CAAC;IACrC,GAAG,CAAC,EAAE,IAAI,CAAC;CACZ,GAAG,iBAAiB,EAAE,CA0DtB;AA6LD,wBAAsB,2BAA2B,CAAC,KAAK,EAAE;IACvD,EAAE,CAAC,EAAE,EAAE,CAAC;IACR,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,0BAA0B,CAAC;IAClC,KAAK,EAAE,0BAA0B,GAAG,IAAI,CAAC;IACzC,SAAS,EAAE,0BAA0B,CAAC;IACtC,oBAAoB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACrC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnC,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,QAAQ,GAAG,QAAQ,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACvE,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAoE/B;AAED,wBAAsB,4BAA4B,CAAC,KAAK,EAAE,MAAM,iBAkB/D;AAED,wBAAsB,wCAAwC,CAAC,KAAK,EAAE;IACpE,EAAE,CAAC,EAAE,EAAE,CAAC;IACR,oBAAoB,EAAE,MAAM,CAAC;IAC7B,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC9B,iBAwBA;AAED,wBAAsB,gDAAgD,CACpE,EAAE,EAAE,EAAE,EACN,KAAK,EAAE,MAAM,EACb,mBAAmB,EAAE,MAAM,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;MAsB9B;AAED,wBAAsB,0CAA0C,CAAC,EAAE,EAAE,EAAE;;GA+BtE;AAED,wBAAsB,oCAAoC,CAAC,KAAK,EAAE;IAChE,EAAE,EAAE,EAAE,CAAC;IACP,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,0BAA0B,CAAC;IAClC,KAAK,EAAE,0BAA0B,GAAG,IAAI,CAAC;IACzC,SAAS,EAAE,0BAA0B,CAAC;IACtC,oBAAoB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACrC,OAAO,EAAE,yBAAyB,EAAE,CAAC;CACtC,gCA6EA;AAED,wBAAgB,0BAA0B,CAAC,KAAK,EAAE;IAChD,SAAS,EAAE,0BAA0B,CAAC;IACtC,eAAe,EAAE,iBAAiB,EAAE,CAAC;CACtC,UAcA"}