@sixb/core 0.1.20 → 0.1.21

package/src/storage/internal.ts

@@ -1,8 +1,38 @@
 /** Implementation exports for Sixb packages only. */
 
 export { normalizeRecordActionRunInput, resolveActionRunEffects } from "./action-runs/record"
+export type { ActivitySourceState, StoredActivityEvent } from "./activity/policy"
+export {
+  ACTIVITY_BATCH_LIMIT,
+  ACTIVITY_EVENT_ID_PATTERN,
+  ACTIVITY_HOSTNAME_PATTERN,
+  ACTIVITY_ID_MAX_LENGTH,
+  ACTIVITY_INTERACTION_NAME_MAX_LENGTH,
+  ACTIVITY_INTERACTION_NAME_PATTERN,
+  ACTIVITY_LABEL_MAX_LENGTH,
+  ACTIVITY_LABEL_PATTERN,
+  ACTIVITY_MAX_PAGE_SIZE,
+  ACTIVITY_MAX_PERIOD_MS,
+  ACTIVITY_MAX_VISIT_MS,
+  ACTIVITY_REFERRER_MAX_LENGTH,
+  ACTIVITY_ROUTE_MAX_LENGTH,
+  ACTIVITY_VISIT_ID_PATTERN,
+  activityActor,
+  activityCalendarDays,
+  isActivityRoutePattern,
+  parseActivityAttribution,
+  planActivityAppend,
+  serializeActivityAttribution,
+  validateActivityAppend,
+} from "./activity/policy"
 export { assertAiUsageExecutionId, normalizeAiModelCallRecord } from "./ai-usage/record"
 export { normalizeAiModelCallUsage } from "./ai-usage/usage"
+export {
+  artifactStorageErrors,
+  parseArtifactFrameGrantRow,
+  parseArtifactRow,
+  parseArtifactVersionRow,
+} from "./artifacts/record"
 export { parseFileDownloadGrantRow } from "./file-download-grants/record"
 export { createFileUploadId, createUploadExpiresAt } from "./file-upload-sessions/utils"
 export { canRequeuePipelineRunAfterEnqueueFailure } from "./pipeline-runs/idempotency"

package/src/storage/ontology/in-memory/sources.ts

@@ -19,6 +19,7 @@ import type {
   AbandonRunSourceMaterializationInput,
   AbandonSourceMaterializationCandidateInput,
   AbandonSourceMaterializationInput,
+  ActiveOntologySource,
   AdoptedSourceMaterialization,
   AdoptSourceMaterializationInput,
   AssertSourceMaterializationExecution,
@@ -26,6 +27,7 @@ import type {
   CleanupTerminalSourceMaterializationsInput,
   CleanupTerminalSourceMaterializationsResult,
   GetActiveOntologySourceInput,
+  ListActiveOntologySourcesInput,
   MarkSourceMaterializationReadyInput,
   OntologySourceRecord,
   OntologySourceStorage,
@@ -193,6 +195,27 @@ export class InMemoryOntologySourceStorage implements OntologySourceStorage {
     })
   }
 
+  async listActive(
+    input: ListActiveOntologySourcesInput
+  ): Promise<readonly ActiveOntologySource[]> {
+    return this.runRootOperation(() => {
+      assertNonblank(input.projectId, "Source project id")
+
+      const active: ActiveOntologySource[] = []
+      for (const materialization of this.state.sourceMaterializations.values()) {
+        if (materialization.projectId !== input.projectId) continue
+        if (materialization.status !== "active") continue
+        active.push({
+          source: { projectionId: materialization.source.projectionId },
+          datasetVersion: { ...materialization.datasetVersion },
+          projectionRevision: materialization.projectionRevision,
+        })
+      }
+
+      return active
+    })
+  }
+
   async adopt(
     input: AdoptSourceMaterializationInput
   ): Promise<AdoptedSourceMaterialization | null> {

package/src/storage/ontology/index.ts

@@ -107,6 +107,7 @@ export type {
   AbandonRunSourceMaterializationInput,
   AbandonSourceMaterializationCandidateInput,
   AbandonSourceMaterializationInput,
+  ActiveOntologySource,
   AdoptedSourceMaterialization,
   AdoptSourceMaterializationInput,
   AssertSourceMaterializationExecution,
@@ -115,6 +116,7 @@ export type {
   CleanupTerminalSourceMaterializationsInput,
   CleanupTerminalSourceMaterializationsResult,
   GetActiveOntologySourceInput,
+  ListActiveOntologySourcesInput,
   MarkSourceMaterializationReadyInput,
   OntologySourceMaterializationStatus,
   OntologySourceRecord,

package/src/storage/ontology/sources.ts

@@ -69,6 +69,21 @@ export interface GetActiveOntologySourceInput {
   readonly source: ProjectionSourceRef
 }
 
+export interface ListActiveOntologySourcesInput {
+  readonly projectId: string
+}
+
+/**
+ * What one source applied: the dataset version its active materialization replaced the source
+ * with, under which projection revision. A delta applies like a full replacement by its version,
+ * so the version describes every value of the source.
+ */
+export interface ActiveOntologySource {
+  readonly source: ProjectionSourceRef
+  readonly datasetVersion: PinnedDatasetVersion
+  readonly projectionRevision: string
+}
+
 export interface BeginSourceMaterializationInput {
   readonly projectId: string
   readonly source: ProjectionSourceRef
@@ -210,6 +225,8 @@ export interface OntologySourceStorage {
   stageRows(input: StageSourceRowsInput): Promise<StageSourceRowsResult>
   markReady(input: MarkSourceMaterializationReadyInput): Promise<OntologySourceRecord>
   getActive(input: GetActiveOntologySourceInput): Promise<OntologySourceRecord | null>
+  /** Every source of the project with an active materialization, in no particular order. */
+  listActive(input: ListActiveOntologySourcesInput): Promise<readonly ActiveOntologySource[]>
   /** Hands a redelivered run's candidate to its current execution; null when it has none. */
   adopt(input: AdoptSourceMaterializationInput): Promise<AdoptedSourceMaterialization | null>
   abandon(input: AbandonSourceMaterializationCandidateInput): Promise<OntologySourceRecord>

package/src/storage/share-grants/in-memory.ts

@@ -10,6 +10,7 @@ import {
 import type {
   CreateShareGrantInput,
   GetShareGrantByIdInput,
+  GetShareGrantsByIdsInput,
   ListShareGrantsInput,
   ListShareGrantsResult,
   RevokeShareGrantInput,
@@ -43,6 +44,13 @@ export class InMemoryShareGrantStorage implements ShareGrantStorage {
     return record ? cloneShareGrantRecord(record) : null
   }
 
+  async getByIds(input: GetShareGrantsByIdsInput): Promise<readonly ShareGrantRecord[]> {
+    return [...new Set(input.ids)].flatMap((id) => {
+      const record = this.rows.get(rowKey(input.projectId, id))
+      return record ? [cloneShareGrantRecord(record)] : []
+    })
+  }
+
   async list(input: ListShareGrantsInput): Promise<ListShareGrantsResult> {
     const normalized = normalizeListShareGrantsInput(input)
     const matching = [...this.rows.values()]

package/src/storage/share-grants/index.ts

@@ -4,6 +4,7 @@ export { InMemoryShareGrantStorage } from "./in-memory"
 export type {
   CreateShareGrantInput,
   GetShareGrantByIdInput,
+  GetShareGrantsByIdsInput,
   ListShareGrantsInput,
   ListShareGrantsResult,
   RevokeShareGrantInput,

package/src/storage/share-grants/types.ts

@@ -43,6 +43,11 @@ export interface GetShareGrantByIdInput {
   readonly id: string
 }
 
+export interface GetShareGrantsByIdsInput {
+  readonly projectId: string
+  readonly ids: readonly string[]
+}
+
 export interface ListShareGrantsInput {
   readonly projectId: string
   readonly definitionId?: string
@@ -96,6 +101,9 @@ export interface ShareGrantStorage {
   /** Load one project-scoped grant, including expired or revoked grants, or return null. */
   getById(input: GetShareGrantByIdInput): Promise<ShareGrantRecord | null>
 
+  /** The grants found, revoked and expired ones included, in first-occurrence input order. */
+  getByIds(input: GetShareGrantsByIdsInput): Promise<readonly ShareGrantRecord[]>
+
   /**
    * List in deterministic `createdAt DESC, id DESC` order. Expired and revoked rows are excluded
    * unless requested; `total` is computed after filtering and before pagination. A root call may

package/src/storage/types.ts

@@ -1,8 +1,10 @@
 import type { ActionRunStorage } from "./action-runs"
+import type { ActivityStorage } from "./activity"
 import type { AgentStorage } from "./agents"
 import type { AiCostStorage } from "./ai-cost"
 import type { AiLimitStorage } from "./ai-limits"
 import type { AiUsageStorage } from "./ai-usage"
+import type { ArtifactStorage } from "./artifacts"
 import type { AuthStorage } from "./auth"
 import type { ConnectorConnectionStorage } from "./connector-connections"
 import type { ExecutionStorage } from "./executions"
@@ -355,6 +357,7 @@ export interface Storage {
   executions: ExecutionStorage
   auth?: AuthStorage
   agents?: AgentStorage
+  artifacts?: ArtifactStorage
   aiUsage?: AiUsageStorage
   aiCosts?: AiCostStorage
   aiLimits?: AiLimitStorage
@@ -371,6 +374,7 @@ export interface Storage {
   fileUploadSessions?: FileUploadSessionStore
   fileDownloadGrants?: FileDownloadGrantStorage
   connectorConnections?: ConnectorConnectionStorage
+  activity?: ActivityStorage
 
   /** Lightweight reachability probe. It must not open a write transaction or run migrations. */
   ping(): Promise<void>

package/src/storage/workflow-runs/in-memory.ts

@@ -267,6 +267,22 @@ export class InMemoryWorkflowRunStorage implements WorkflowRunStorage {
     return record ? cloneRecord(record) : null
   }
 
+  async getByExecutionIds(params: {
+    readonly projectId: string
+    readonly executionIds: readonly string[]
+  }): Promise<readonly WorkflowRunRecord[]> {
+    const byExecution = new Map(
+      [...this.runs.values()]
+        .filter((record) => record.projectId === params.projectId)
+        .map((record) => [record.executionId, record])
+    )
+
+    return [...new Set(params.executionIds)].flatMap((executionId) => {
+      const record = byExecution.get(executionId)
+      return record ? [cloneRecord(record)] : []
+    })
+  }
+
   async list(input: ListWorkflowRunsInput): Promise<ListWorkflowRunsResult> {
     if (hasEmptyStatuses(input) || input.workflowIds?.length === 0) {
       return {

package/src/storage/workflow-runs/types.ts

@@ -328,6 +328,11 @@ export interface WorkflowRunStorage {
   resume(input: ResumeWorkflowRunInput): Promise<WorkflowRunRecord>
   finish(input: FinishWorkflowRunInput): Promise<WorkflowRunRecord>
   getById(params: { projectId: string; id: string }): Promise<WorkflowRunRecord | null>
+  /** The runs of these executions, in first-occurrence input order; a missing one is omitted. */
+  getByExecutionIds(params: {
+    readonly projectId: string
+    readonly executionIds: readonly string[]
+  }): Promise<readonly WorkflowRunRecord[]>
   list(input: ListWorkflowRunsInput): Promise<ListWorkflowRunsResult>
   listLatestByWorkflowIds(input: ListLatestWorkflowRunsInput): Promise<ListLatestWorkflowRunsResult>
 }

package/src/testing/action-run-storage-contract.ts

@@ -310,6 +310,31 @@ export function runActionRunStorageContractSuite<TStorage extends ActionRunStora
       })
     })
 
+    test("reads the runs of several executions at once", async () => {
+      await withStorage(async (storage) => {
+        const first = await recordTestActionRun(storage, run("act_1"))
+        const second = await recordTestActionRun(storage, run("act_2"))
+        // An execution whose run is still in flight has no record.
+        const inFlight = await createTestActionRunRecord(storage.executions, run("act_in_flight"))
+
+        const runs = await storage.actionRuns.getByExecutionIds({
+          projectId,
+          executionIds: [second.executionId, inFlight.executionId, first.executionId, "missing"],
+        })
+        expect(runs).toEqual([second, first])
+
+        await expect(
+          storage.actionRuns.getByExecutionIds({
+            projectId: "other-project",
+            executionIds: [first.executionId],
+          })
+        ).resolves.toEqual([])
+        await expect(
+          storage.actionRuns.getByExecutionIds({ projectId, executionIds: [] })
+        ).resolves.toEqual([])
+      })
+    })
+
     test("drops a record written in a transaction that rolls back", async () => {
       await withStorage(async (storage) => {
         const input = await createTestActionRunRecord(storage.executions, run("act_rolled_back"))