@opengeni/db 4.0.0-canary.1 → 4.0.0-canary.3

package/dist/canonical-human-identities.js

@@ -11,9 +11,9 @@ import {
   getCanonicalHumanIdentityProjection,
   synchronizeCanonicalHumanLoginBindings,
   validateCanonicalHumanSession
-} from "./chunk-47LKLIIK.js";
-import "./chunk-ZQ32COTC.js";
-import "./chunk-XAKQCOF3.js";
+} from "./chunk-CZXYNVEU.js";
+import "./chunk-GPKHEPVX.js";
+import "./chunk-WU3MSPZS.js";
 import "./chunk-PZ5AY32C.js";
 export {
   CanonicalHumanIdentityAuthorityError,

package/dist/{chunk-2HGYCAXN.js → chunk-3JGOYOWD.js}

@@ -5,7 +5,7 @@ import {
   withWorkspaceRls,
   withWorkspaceSubjectRls,
   withWorkspaceSubjectSessionActivityRls
-} from "./chunk-ZQ32COTC.js";
+} from "./chunk-GPKHEPVX.js";
 
 // src/session-tenancy.ts
 import { createHash } from "crypto";
@@ -481,4 +481,4 @@ export {
   forkSessionContent,
   replayAppliedSessionFork
 };
-//# sourceMappingURL=chunk-2HGYCAXN.js.map
+//# sourceMappingURL=chunk-3JGOYOWD.js.map

package/dist/{chunk-47LKLIIK.js → chunk-CZXYNVEU.js}

@@ -1,7 +1,7 @@
 import {
   nestedPostgresSqlState,
   rawRows
-} from "./chunk-ZQ32COTC.js";
+} from "./chunk-GPKHEPVX.js";
 
 // src/canonical-human-identities.ts
 import { createHash } from "crypto";
@@ -309,4 +309,4 @@ export {
   getCanonicalHumanIdentityProjection,
   applyCanonicalHumanIdentityOperation
 };
-//# sourceMappingURL=chunk-47LKLIIK.js.map
+//# sourceMappingURL=chunk-CZXYNVEU.js.map

package/dist/{chunk-YVDQDD33.js → chunk-DIFDLF2C.js}

@@ -1,7 +1,7 @@
 import {
   rawRows,
   withRlsContext
-} from "./chunk-ZQ32COTC.js";
+} from "./chunk-GPKHEPVX.js";
 
 // src/editable-artifacts.ts
 import { createHash } from "crypto";
@@ -3540,4 +3540,4 @@ export {
   PostgresEditableArtifactStore,
   PostgresEditableArtifactLiveReadStore
 };
-//# sourceMappingURL=chunk-YVDQDD33.js.map
+//# sourceMappingURL=chunk-DIFDLF2C.js.map

package/dist/{chunk-ZQ32COTC.js → chunk-GPKHEPVX.js}

@@ -4,7 +4,7 @@ import {
   sessions,
   workspaceSessionActivityRevisions,
   workspaces
-} from "./chunk-XAKQCOF3.js";
+} from "./chunk-WU3MSPZS.js";
 
 // src/persistence-errors.ts
 var SQLSTATE_KEYS = ["sqlState", "sqlstate", "code"];
@@ -755,4 +755,4 @@ export {
   withAccountRls,
   dbBindingFor
 };
-//# sourceMappingURL=chunk-ZQ32COTC.js.map
+//# sourceMappingURL=chunk-GPKHEPVX.js.map

package/dist/{chunk-A5RFBEZR.js → chunk-HUCU5BDD.js}

@@ -2,7 +2,7 @@ import {
   rawRows,
   withRlsContext,
   withWorkspaceRls
-} from "./chunk-ZQ32COTC.js";
+} from "./chunk-GPKHEPVX.js";
 import {
   connections,
   creditLedgerEntries,
@@ -14,7 +14,7 @@ import {
   workspaceVideoGenerationPolicies,
   workspaceVideoGenerationQuotas,
   workspaces
-} from "./chunk-XAKQCOF3.js";
+} from "./chunk-WU3MSPZS.js";
 
 // src/video-generation.ts
 import {
@@ -992,4 +992,4 @@ export {
   mediaGenerationResultForStoredOperation,
   getVideoGenerationOperationSummary
 };
-//# sourceMappingURL=chunk-A5RFBEZR.js.map
+//# sourceMappingURL=chunk-HUCU5BDD.js.map

package/dist/{chunk-I6YPQMAL.js → chunk-OWZ36ST7.js}

@@ -1,7 +1,7 @@
 import {
   nestedPostgresSqlState,
   rawRows
-} from "./chunk-ZQ32COTC.js";
+} from "./chunk-GPKHEPVX.js";
 
 // src/managed-auth-session-sets.ts
 import {
@@ -296,4 +296,4 @@ export {
   reapManagedAuthIsolatedSessions,
   reapExpiredManagedAuthSessionSets
 };
-//# sourceMappingURL=chunk-I6YPQMAL.js.map
+//# sourceMappingURL=chunk-OWZ36ST7.js.map

package/dist/{chunk-FVZINOBB.js → chunk-R6LIJYG6.js}

@@ -1,11 +1,11 @@
 import {
   cancelUnacceptedVideoGenerationsForToolCallsInTransaction
-} from "./chunk-A5RFBEZR.js";
+} from "./chunk-HUCU5BDD.js";
 import {
   nestedPostgresSqlState,
   withRlsContext,
   withSessionActivityRlsContext
-} from "./chunk-ZQ32COTC.js";
+} from "./chunk-GPKHEPVX.js";
 import {
   LOSSLESS_CONTENT_CODEC_VERSION,
   auditEvents,
@@ -32,7 +32,7 @@ import {
   workspaceControlEvents,
   workspaceInferenceControls,
   workspaces
-} from "./chunk-XAKQCOF3.js";
+} from "./chunk-WU3MSPZS.js";
 
 // src/session-background-commands.ts
 import { SessionCommandFailure } from "@opengeni/contracts";
@@ -866,6 +866,9 @@ async function lockWorkspaceInferenceControl(db, workspaceId, mode, options = {}
         revision,
         workspace_state as "workspaceState",
         workspace_pause_revision as "workspacePauseRevision",
+        timer_id as "timerId", timer_action as "timerAction",
+        timer_due_at as "timerDueAt", timer_pause_for_seconds as "timerPauseForSeconds",
+        timer_pause_revision as "timerPauseRevision",
         reason,
         changed_by as "changedBy",
         changed_at as "changedAt"
@@ -3181,13 +3184,13 @@ async function mutateWorkspaceControlInTransaction(db, input) {
     workspace.workspacePauseRevision,
     "workspace pause revision"
   );
-  const changed = input.action === "pause" ? workspace.workspaceState !== "paused" || workspacePauseRevision === null || await workspacePauseEffectsNeeded(db, {
+  const changed = !input.timerExecution && workspace.timerId != null || (input.action === "pause" ? workspace.workspaceState !== "paused" || workspacePauseRevision === null || await workspacePauseEffectsNeeded(db, {
     workspaceId: input.workspaceId,
     workspacePauseRevision
   }) : workspace.workspaceState !== "active" || await continuableWakeRepairNeeded(db, {
     workspaceId: input.workspaceId,
     rootSessionId: null
-  });
+  }));
   if (!changed) {
     const receipt2 = await updateSessionCommandReceiptResult(db, reserved.receipt.id, {
       result: {
@@ -3216,6 +3219,7 @@ async function mutateWorkspaceControlInTransaction(db, input) {
     subjectId: actor
   }) : 0;
   const [updated] = await db.update(workspaceInferenceControls).set({
+    ...!input.timerExecution ? clearWorkspaceTimerFields : {},
     revision,
     workspaceState: input.action === "pause" ? "paused" : "active",
     workspacePauseRevision: input.action === "pause" ? revision : null,
@@ -3239,7 +3243,7 @@ async function mutateWorkspaceControlInTransaction(db, input) {
     scope: "workspace",
     rootSessionId: null,
     action: input.action,
-    automatic: false,
+    automatic: input.timerExecution === true,
     reason: input.reason ?? null,
     actor
   });
@@ -3308,6 +3312,131 @@ async function insertWorkspaceControlEventInTransaction(db, input) {
   }
   return event.id;
 }
+var WorkspacePauseTimerInputError = class extends Error {
+};
+var clearWorkspaceTimerFields = {
+  timerId: null,
+  timerAction: null,
+  timerDueAt: null,
+  timerPauseForSeconds: null,
+  timerPauseRevision: null
+};
+async function setWorkspacePauseTimerInTransaction(db, input) {
+  let workspace = await lockWorkspaceInferenceControl(db, input.workspaceId, "update", {
+    lockTimeoutMs: workspaceControlRequestLockTimeoutMs()
+  });
+  const reserved = await reserveSessionCommandReceipt(db, {
+    accountId: input.accountId,
+    workspaceId: input.workspaceId,
+    actor: { type: "human", subjectId: input.subjectId },
+    action: `workspace.timer.${input.action}`,
+    targetSessionId: null,
+    targetTurnId: null,
+    operationKey: input.clientEventId,
+    canonicalRequestHash: canonicalSessionCommandHash({
+      action: input.action,
+      pauseInSeconds: input.pauseInSeconds ?? 0,
+      pauseForSeconds: input.pauseForSeconds ?? null,
+      expectedRevision: input.expectedRevision
+    })
+  });
+  if (reserved.replay)
+    return {
+      workspaceControlEventId: reserved.receipt.result.workspaceControlEventId ?? null
+    };
+  if (Number(workspace.revision) !== input.expectedRevision)
+    throw new SessionControlConflictError();
+  let action = "pause";
+  let delay = input.pauseInSeconds ?? 0;
+  let pauseRevision = null;
+  if (input.action === "set") {
+    if (workspace.workspaceState === "paused") {
+      if (delay !== 0 || input.pauseForSeconds == null) {
+        throw new WorkspacePauseTimerInputError("A paused workspace requires a resume duration");
+      }
+      action = "resume";
+      delay = input.pauseForSeconds;
+      pauseRevision = Number(workspace.workspacePauseRevision);
+    } else if (delay === 0) {
+      await mutateWorkspaceControlInTransaction(db, {
+        accountId: input.accountId,
+        workspaceId: input.workspaceId,
+        actor: { type: "human", subjectId: input.subjectId },
+        operationKey: `timer-pause:${reserved.receipt.id}`,
+        action: "pause"
+      });
+      workspace = await lockWorkspaceInferenceControl(db, input.workspaceId, "update");
+      action = "resume";
+      delay = input.pauseForSeconds ?? 0;
+      pauseRevision = Number(workspace.workspacePauseRevision);
+    }
+  }
+  const hasTimer = input.action === "set" && delay > 0;
+  const [clock] = await db.execute(sql3`select clock_timestamp() as now`);
+  const revision = nextRevision(workspace);
+  await db.update(workspaceInferenceControls).set({
+    ...clearWorkspaceTimerFields,
+    ...hasTimer ? {
+      timerId: reserved.receipt.id,
+      timerAction: action,
+      timerDueAt: new Date(new Date(clock.now).getTime() + delay * 1e3),
+      timerPauseForSeconds: action === "pause" ? input.pauseForSeconds ?? null : null,
+      timerPauseRevision: pauseRevision
+    } : {},
+    revision,
+    updatedAt: new Date(clock.now)
+  }).where(eq3(workspaceInferenceControls.workspaceId, input.workspaceId));
+  const workspaceControlEventId = await insertWorkspaceControlEventInTransaction(db, {
+    accountId: input.accountId,
+    workspaceId: input.workspaceId,
+    revision,
+    scope: "workspace",
+    rootSessionId: null,
+    action: hasTimer ? "timer_set" : "timer_cancelled",
+    automatic: false,
+    reason: null,
+    actor: input.subjectId
+  });
+  await updateSessionCommandReceiptResult(db, reserved.receipt.id, {
+    controlRevision: revision,
+    result: { workspaceControlEventId }
+  });
+  return { workspaceControlEventId };
+}
+async function fireWorkspacePauseTimerInTransaction(db, input) {
+  const workspace = await lockWorkspaceInferenceControl(db, input.workspaceId, "update", {
+    lockTimeoutMs: 1e3
+  });
+  const [clock] = await db.execute(sql3`select clock_timestamp() as now`);
+  if (workspace.timerId !== input.timerId || !workspace.timerDueAt || new Date(workspace.timerDueAt).getTime() > new Date(clock.now).getTime())
+    return null;
+  const action = workspace.timerAction;
+  if (action === "resume" && (workspace.workspaceState !== "paused" || Number(workspace.workspacePauseRevision) !== Number(workspace.timerPauseRevision))) {
+    await db.update(workspaceInferenceControls).set(clearWorkspaceTimerFields).where(eq3(workspaceInferenceControls.workspaceId, input.workspaceId));
+    return null;
+  }
+  const result = await mutateWorkspaceControlInTransaction(db, {
+    accountId: workspace.accountId,
+    workspaceId: input.workspaceId,
+    actor: { type: "service", subjectId: "service:workspace-pause-timer" },
+    operationKey: `timer:${input.timerId}:${action}`,
+    action,
+    timerExecution: true
+  });
+  const [applied] = await db.execute(sql3`select clock_timestamp() as now`);
+  await db.update(workspaceInferenceControls).set({
+    ...clearWorkspaceTimerFields,
+    ...action === "pause" && workspace.timerPauseForSeconds !== null ? {
+      timerId: input.timerId,
+      timerAction: "resume",
+      timerDueAt: new Date(
+        new Date(applied.now).getTime() + workspace.timerPauseForSeconds * 1e3
+      ),
+      timerPauseRevision: result.revision
+    } : {}
+  }).where(eq3(workspaceInferenceControls.workspaceId, input.workspaceId));
+  return result;
+}
 
 // src/session-background-commands.ts
 var SESSION_BACKGROUND_COMMAND_REASON_MAX_BYTES = 512;
@@ -3409,7 +3538,12 @@ async function insertManagedSessionBackgroundCommandInTransaction(db, input) {
     accountId: sandboxRetainedProcesses.accountId,
     workspaceId: sandboxRetainedProcesses.workspaceId,
     sessionId: sandboxRetainedProcesses.sessionId,
-    state: sandboxRetainedProcesses.state
+    state: sandboxRetainedProcesses.state,
+    ownerActorKind: sandboxRetainedProcesses.ownerActorKind,
+    ownerActorId: sandboxRetainedProcesses.ownerActorId,
+    ownerTurnId: sandboxRetainedProcesses.ownerTurnId,
+    ownerAttemptId: sandboxRetainedProcesses.ownerAttemptId,
+    ownerExecutionGeneration: sandboxRetainedProcesses.ownerExecutionGeneration
   }).from(sandboxRetainedProcesses).where(eq4(sandboxRetainedProcesses.id, input.retainedProcessId)).for("update").limit(1);
   if (!process || process.accountId !== input.accountId || process.workspaceId !== input.workspaceId || process.sessionId !== input.sessionId || process.state !== "active") {
     throw new Error("Managed background command requires its exact active retained process");
@@ -3422,14 +3556,18 @@ async function insertManagedSessionBackgroundCommandInTransaction(db, input) {
     provider: "managed",
     state: "running",
     retainedProcessId: input.retainedProcessId,
-    commandPreview: commandPreview(input.command)
+    commandPreview: commandPreview(input.command),
+    launchTurnId: input.turnId ?? null,
+    launchAttemptId: input.attemptId ?? null,
+    launchExecutionGeneration: input.executionGeneration ?? null
   }).onConflictDoUpdate({
     target: sessionBackgroundCommands.retainedProcessId,
     targetWhere: sql4`${sessionBackgroundCommands.retainedProcessId} is not null`,
     set: { updatedAt: /* @__PURE__ */ new Date() }
   }).returning();
   if (!row) throw new Error("Managed background command adoption returned no row");
-  if (row.accountId !== input.accountId || row.workspaceId !== input.workspaceId || row.sessionId !== input.sessionId || row.provider !== "managed" || row.retainedProcessId !== input.retainedProcessId) {
+  const legacyReplayMatches = row.launchTurnId === null && row.launchAttemptId === null && row.launchExecutionGeneration === null && process.ownerActorKind === "turn" && process.ownerActorId === input.attemptId && process.ownerTurnId === input.turnId && process.ownerAttemptId === input.attemptId && process.ownerExecutionGeneration === input.executionGeneration;
+  if (row.accountId !== input.accountId || row.workspaceId !== input.workspaceId || row.sessionId !== input.sessionId || !legacyReplayMatches && (row.launchTurnId !== (input.turnId ?? null) || row.launchAttemptId !== (input.attemptId ?? null) || row.launchExecutionGeneration !== (input.executionGeneration ?? null)) || row.provider !== "managed" || row.retainedProcessId !== input.retainedProcessId) {
     throw new Error("Managed background command adoption conflicted with another identity");
   }
   return mapCommand(row);
@@ -3446,7 +3584,10 @@ async function insertConnectedMachineSessionBackgroundCommandInTransaction(db, i
     enrollmentId: input.enrollmentId,
     connectionInstanceId: input.connectionInstanceId,
     opId: input.opId,
-    commandPreview: commandPreview(input.command)
+    commandPreview: commandPreview(input.command),
+    launchTurnId: input.turnId ?? null,
+    launchAttemptId: input.attemptId ?? null,
+    launchExecutionGeneration: input.executionGeneration ?? null
   }).onConflictDoUpdate({
     target: [
       sessionBackgroundCommands.controlWorkspaceId,
@@ -3458,7 +3599,7 @@ async function insertConnectedMachineSessionBackgroundCommandInTransaction(db, i
     set: { updatedAt: /* @__PURE__ */ new Date() }
   }).returning();
   if (!row) throw new Error("Connected Machine background command adoption returned no row");
-  if (row.accountId !== input.accountId || row.workspaceId !== input.workspaceId || row.sessionId !== input.sessionId || row.provider !== "connected_machine" || row.controlWorkspaceId !== input.controlWorkspaceId || row.enrollmentId !== input.enrollmentId || row.connectionInstanceId !== input.connectionInstanceId || row.opId !== input.opId) {
+  if (row.accountId !== input.accountId || row.workspaceId !== input.workspaceId || row.sessionId !== input.sessionId || row.launchTurnId !== (input.turnId ?? null) || row.launchAttemptId !== (input.attemptId ?? null) || row.launchExecutionGeneration !== (input.executionGeneration ?? null) || row.provider !== "connected_machine" || row.controlWorkspaceId !== input.controlWorkspaceId || row.enrollmentId !== input.enrollmentId || row.connectionInstanceId !== input.connectionInstanceId || row.opId !== input.opId) {
     throw new Error(
       "Connected Machine background command adoption conflicted with another identity"
     );
@@ -4128,6 +4269,9 @@ export {
   updateSessionCommandReceiptResult,
   mutateSessionControlInTransaction,
   autoResumeSessionBranchInTransaction,
-  mutateWorkspaceControlInTransaction
+  mutateWorkspaceControlInTransaction,
+  WorkspacePauseTimerInputError,
+  setWorkspacePauseTimerInTransaction,
+  fireWorkspacePauseTimerInTransaction
 };
-//# sourceMappingURL=chunk-FVZINOBB.js.map
+//# sourceMappingURL=chunk-R6LIJYG6.js.map