@omega-flow/store-aws 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) omega-flow contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/index.d.mts

@@ -0,0 +1,125 @@
+import { DynamoDBClient } from '@aws-sdk/client-dynamodb';
+import { WorkflowStore, WorkflowMemory, WorkflowScheduler } from '@omega-flow/engine';
+import { Workflow, Context, Event } from '@omega-flow/types';
+import { SchedulerClient } from '@aws-sdk/client-scheduler';
+
+interface DynamoDBWorkflowStoreConfig {
+    client: DynamoDBClient;
+    tableName: string;
+}
+/**
+ * DynamoDB-backed implementation of WorkflowStore.
+ *
+ * Table layout (dedicated workflows table):
+ *   domain (pk) = tenant identifier (e.g. organization id, company id)
+ *   workflowId (sk) = workflow id
+ *   data = full Workflow JSON
+ *   createdAt, updatedAt = epoch ms
+ */
+declare class DynamoDBWorkflowStore implements WorkflowStore {
+    private docClient;
+    private tableName;
+    constructor(config: DynamoDBWorkflowStoreConfig);
+    getWorkflow(domain: string, workflowId: string): Promise<Workflow | null>;
+    getAllWorkflows(domain: string): Promise<Workflow[]>;
+    setWorkflow(domain: string, workflow: Workflow): Promise<void>;
+    createWorkflow(domain: string, workflowData: Omit<Workflow, "id">): Promise<Workflow>;
+    deleteWorkflow(domain: string, workflowId: string): Promise<boolean>;
+}
+
+interface DynamoDBWorkflowMemoryConfig {
+    client: DynamoDBClient;
+    tableName: string;
+    /**
+     * Name of the GSI with `domain` as partition key and `subjectId` as sort key.
+     * Required for `getAllContexts` and `getAllContextsForSubject`.
+     * Defaults to `domain-subjectId-index`.
+     */
+    gsiName?: string;
+}
+/**
+ * DynamoDB-backed implementation of WorkflowMemory.
+ *
+ * Table layout (dedicated contexts table):
+ *   contextKey (pk) = `${domain}#${workflowId}#${subjectId}`
+ *   instanceId (sk) = workflow instance id
+ *   domain = tenant identifier (denormalised for GSI)
+ *   subjectId = subject identifier (denormalised for GSI)
+ *   data = full Context JSON
+ *   isCompleted, startedAt = mirrored from Context for filtering/sorting
+ *   updatedAt = epoch ms, set on every save
+ *
+ * GSI (domain-subjectId-index):
+ *   domain (pk), subjectId (sk) — enables getAllContexts and getAllContextsForSubject queries.
+ *
+ * Assumes domain / workflowId / subjectId do not contain '#'.
+ */
+declare class DynamoDBWorkflowMemory implements WorkflowMemory {
+    private docClient;
+    private tableName;
+    private gsiName;
+    constructor(config: DynamoDBWorkflowMemoryConfig);
+    private buildContextKey;
+    getContexts(domain: string, workflowId: string, subjectId: string): Promise<Context[]>;
+    saveContext(domain: string, workflowId: string, subjectId: string, context: Context): Promise<void>;
+    deleteContext(domain: string, workflowId: string, subjectId: string, instanceId: string): Promise<void>;
+    /**
+     * Fetch a single Context by instance id.
+     */
+    getContext(domain: string, workflowId: string, subjectId: string, instanceId: string): Promise<Context | null>;
+    /**
+     * List all contexts for a given subject across all workflows in a domain.
+     * Uses the `domain-subjectId-index` GSI.
+     */
+    getAllContextsForSubject(domain: string, subjectId: string): Promise<Context[]>;
+    /**
+     * List all contexts across all subjects and workflows in a domain.
+     * Uses the `domain-subjectId-index` GSI.
+     * Returns contexts annotated with `subjectId`.
+     */
+    getAllContexts(domain: string): Promise<Array<Context & {
+        subjectId: string;
+    }>>;
+}
+
+interface EventBusWorkflowSchedulerConfig {
+    client: SchedulerClient;
+    /** ARN of the EventBridge bus that receives the scheduled event. */
+    eventBusArn: string;
+    /** IAM role assumed by EventBridge Scheduler to invoke the bus target. */
+    roleArn: string;
+    /** Schedule group name. Defaults to `default`. */
+    scheduleGroupName?: string;
+    /** EventBridge `Source` for the published event. Defaults to `omega-flow`. */
+    source?: string;
+    /** EventBridge `DetailType` for the published event. Defaults to `workflow.event`. */
+    detailType?: string;
+}
+/**
+ * EventBridge Scheduler-backed implementation of WorkflowScheduler.
+ *
+ * Creates one-time schedules that publish the workflow event to an EventBridge
+ * bus when they fire. A downstream consumer (e.g. a Lambda subscribed to the
+ * bus) is expected to deserialize the event and call `WorkflowManager.processEvent`.
+ *
+ * Schedules use `ActionAfterCompletion: DELETE` so AWS removes them once fired.
+ */
+declare class EventBusWorkflowScheduler implements WorkflowScheduler {
+    private client;
+    private eventBusArn;
+    private roleArn;
+    private scheduleGroupName;
+    private source;
+    private detailType;
+    constructor(config: EventBusWorkflowSchedulerConfig);
+    schedule(event: Event, delayMs: number): Promise<string>;
+    cancel(scheduleId: string): Promise<boolean>;
+}
+
+declare class WorkflowAlreadyExistsError extends Error {
+    readonly domain: string;
+    readonly workflowId: string;
+    constructor(domain: string, workflowId: string);
+}
+
+export { DynamoDBWorkflowMemory, type DynamoDBWorkflowMemoryConfig, DynamoDBWorkflowStore, type DynamoDBWorkflowStoreConfig, EventBusWorkflowScheduler, type EventBusWorkflowSchedulerConfig, WorkflowAlreadyExistsError };

package/dist/index.d.ts

@@ -0,0 +1,125 @@
+import { DynamoDBClient } from '@aws-sdk/client-dynamodb';
+import { WorkflowStore, WorkflowMemory, WorkflowScheduler } from '@omega-flow/engine';
+import { Workflow, Context, Event } from '@omega-flow/types';
+import { SchedulerClient } from '@aws-sdk/client-scheduler';
+
+interface DynamoDBWorkflowStoreConfig {
+    client: DynamoDBClient;
+    tableName: string;
+}
+/**
+ * DynamoDB-backed implementation of WorkflowStore.
+ *
+ * Table layout (dedicated workflows table):
+ *   domain (pk) = tenant identifier (e.g. organization id, company id)
+ *   workflowId (sk) = workflow id
+ *   data = full Workflow JSON
+ *   createdAt, updatedAt = epoch ms
+ */
+declare class DynamoDBWorkflowStore implements WorkflowStore {
+    private docClient;
+    private tableName;
+    constructor(config: DynamoDBWorkflowStoreConfig);
+    getWorkflow(domain: string, workflowId: string): Promise<Workflow | null>;
+    getAllWorkflows(domain: string): Promise<Workflow[]>;
+    setWorkflow(domain: string, workflow: Workflow): Promise<void>;
+    createWorkflow(domain: string, workflowData: Omit<Workflow, "id">): Promise<Workflow>;
+    deleteWorkflow(domain: string, workflowId: string): Promise<boolean>;
+}
+
+interface DynamoDBWorkflowMemoryConfig {
+    client: DynamoDBClient;
+    tableName: string;
+    /**
+     * Name of the GSI with `domain` as partition key and `subjectId` as sort key.
+     * Required for `getAllContexts` and `getAllContextsForSubject`.
+     * Defaults to `domain-subjectId-index`.
+     */
+    gsiName?: string;
+}
+/**
+ * DynamoDB-backed implementation of WorkflowMemory.
+ *
+ * Table layout (dedicated contexts table):
+ *   contextKey (pk) = `${domain}#${workflowId}#${subjectId}`
+ *   instanceId (sk) = workflow instance id
+ *   domain = tenant identifier (denormalised for GSI)
+ *   subjectId = subject identifier (denormalised for GSI)
+ *   data = full Context JSON
+ *   isCompleted, startedAt = mirrored from Context for filtering/sorting
+ *   updatedAt = epoch ms, set on every save
+ *
+ * GSI (domain-subjectId-index):
+ *   domain (pk), subjectId (sk) — enables getAllContexts and getAllContextsForSubject queries.
+ *
+ * Assumes domain / workflowId / subjectId do not contain '#'.
+ */
+declare class DynamoDBWorkflowMemory implements WorkflowMemory {
+    private docClient;
+    private tableName;
+    private gsiName;
+    constructor(config: DynamoDBWorkflowMemoryConfig);
+    private buildContextKey;
+    getContexts(domain: string, workflowId: string, subjectId: string): Promise<Context[]>;
+    saveContext(domain: string, workflowId: string, subjectId: string, context: Context): Promise<void>;
+    deleteContext(domain: string, workflowId: string, subjectId: string, instanceId: string): Promise<void>;
+    /**
+     * Fetch a single Context by instance id.
+     */
+    getContext(domain: string, workflowId: string, subjectId: string, instanceId: string): Promise<Context | null>;
+    /**
+     * List all contexts for a given subject across all workflows in a domain.
+     * Uses the `domain-subjectId-index` GSI.
+     */
+    getAllContextsForSubject(domain: string, subjectId: string): Promise<Context[]>;
+    /**
+     * List all contexts across all subjects and workflows in a domain.
+     * Uses the `domain-subjectId-index` GSI.
+     * Returns contexts annotated with `subjectId`.
+     */
+    getAllContexts(domain: string): Promise<Array<Context & {
+        subjectId: string;
+    }>>;
+}
+
+interface EventBusWorkflowSchedulerConfig {
+    client: SchedulerClient;
+    /** ARN of the EventBridge bus that receives the scheduled event. */
+    eventBusArn: string;
+    /** IAM role assumed by EventBridge Scheduler to invoke the bus target. */
+    roleArn: string;
+    /** Schedule group name. Defaults to `default`. */
+    scheduleGroupName?: string;
+    /** EventBridge `Source` for the published event. Defaults to `omega-flow`. */
+    source?: string;
+    /** EventBridge `DetailType` for the published event. Defaults to `workflow.event`. */
+    detailType?: string;
+}
+/**
+ * EventBridge Scheduler-backed implementation of WorkflowScheduler.
+ *
+ * Creates one-time schedules that publish the workflow event to an EventBridge
+ * bus when they fire. A downstream consumer (e.g. a Lambda subscribed to the
+ * bus) is expected to deserialize the event and call `WorkflowManager.processEvent`.
+ *
+ * Schedules use `ActionAfterCompletion: DELETE` so AWS removes them once fired.
+ */
+declare class EventBusWorkflowScheduler implements WorkflowScheduler {
+    private client;
+    private eventBusArn;
+    private roleArn;
+    private scheduleGroupName;
+    private source;
+    private detailType;
+    constructor(config: EventBusWorkflowSchedulerConfig);
+    schedule(event: Event, delayMs: number): Promise<string>;
+    cancel(scheduleId: string): Promise<boolean>;
+}
+
+declare class WorkflowAlreadyExistsError extends Error {
+    readonly domain: string;
+    readonly workflowId: string;
+    constructor(domain: string, workflowId: string);
+}
+
+export { DynamoDBWorkflowMemory, type DynamoDBWorkflowMemoryConfig, DynamoDBWorkflowStore, type DynamoDBWorkflowStoreConfig, EventBusWorkflowScheduler, type EventBusWorkflowSchedulerConfig, WorkflowAlreadyExistsError };

package/dist/index.js

@@ -0,0 +1,329 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var index_exports = {};
+__export(index_exports, {
+  DynamoDBWorkflowMemory: () => DynamoDBWorkflowMemory,
+  DynamoDBWorkflowStore: () => DynamoDBWorkflowStore,
+  EventBusWorkflowScheduler: () => EventBusWorkflowScheduler,
+  WorkflowAlreadyExistsError: () => WorkflowAlreadyExistsError
+});
+module.exports = __toCommonJS(index_exports);
+
+// src/stores/DynamoDBWorkflowStore.ts
+var import_client_dynamodb = require("@aws-sdk/client-dynamodb");
+var import_lib_dynamodb = require("@aws-sdk/lib-dynamodb");
+var import_nanoid = require("nanoid");
+
+// src/errors.ts
+var WorkflowAlreadyExistsError = class extends Error {
+  constructor(domain, workflowId) {
+    super(
+      `Workflow already exists: domain=${domain} workflowId=${workflowId}`
+    );
+    this.domain = domain;
+    this.workflowId = workflowId;
+    this.name = "WorkflowAlreadyExistsError";
+  }
+};
+
+// src/stores/DynamoDBWorkflowStore.ts
+var DynamoDBWorkflowStore = class {
+  constructor(config) {
+    this.docClient = import_lib_dynamodb.DynamoDBDocumentClient.from(config.client);
+    this.tableName = config.tableName;
+  }
+  async getWorkflow(domain, workflowId) {
+    const result = await this.docClient.send(
+      new import_lib_dynamodb.GetCommand({
+        TableName: this.tableName,
+        Key: { domain, workflowId }
+      })
+    );
+    if (!result.Item) {
+      return null;
+    }
+    return result.Item.data;
+  }
+  async getAllWorkflows(domain) {
+    const workflows = [];
+    let lastKey;
+    do {
+      const result = await this.docClient.send(
+        new import_lib_dynamodb.QueryCommand({
+          TableName: this.tableName,
+          KeyConditionExpression: "#domain = :domain",
+          ExpressionAttributeNames: { "#domain": "domain" },
+          ExpressionAttributeValues: { ":domain": domain },
+          ExclusiveStartKey: lastKey
+        })
+      );
+      for (const item of result.Items ?? []) {
+        workflows.push(item.data);
+      }
+      lastKey = result.LastEvaluatedKey;
+    } while (lastKey);
+    return workflows;
+  }
+  async setWorkflow(domain, workflow) {
+    const now = Date.now();
+    await this.docClient.send(
+      new import_lib_dynamodb.UpdateCommand({
+        TableName: this.tableName,
+        Key: { domain, workflowId: workflow.id },
+        UpdateExpression: "SET #data = :data, createdAt = if_not_exists(createdAt, :now), updatedAt = :now",
+        ExpressionAttributeNames: { "#data": "data" },
+        ExpressionAttributeValues: { ":data": workflow, ":now": now }
+      })
+    );
+  }
+  async createWorkflow(domain, workflowData) {
+    const id = (0, import_nanoid.nanoid)(8);
+    const workflow = { ...workflowData, id };
+    const now = Date.now();
+    const item = {
+      domain,
+      workflowId: id,
+      data: workflow,
+      createdAt: now,
+      updatedAt: now
+    };
+    try {
+      await this.docClient.send(
+        new import_lib_dynamodb.PutCommand({
+          TableName: this.tableName,
+          Item: item,
+          ConditionExpression: "attribute_not_exists(#domain)",
+          ExpressionAttributeNames: { "#domain": "domain" }
+        })
+      );
+    } catch (err) {
+      if (err instanceof import_client_dynamodb.ConditionalCheckFailedException) {
+        throw new WorkflowAlreadyExistsError(domain, id);
+      }
+      throw err;
+    }
+    return workflow;
+  }
+  async deleteWorkflow(domain, workflowId) {
+    const result = await this.docClient.send(
+      new import_lib_dynamodb.DeleteCommand({
+        TableName: this.tableName,
+        Key: { domain, workflowId },
+        ReturnValues: "ALL_OLD"
+      })
+    );
+    return result.Attributes !== void 0;
+  }
+};
+
+// src/memories/DynamoDBWorkflowMemory.ts
+var import_lib_dynamodb2 = require("@aws-sdk/lib-dynamodb");
+var DynamoDBWorkflowMemory = class {
+  constructor(config) {
+    this.docClient = import_lib_dynamodb2.DynamoDBDocumentClient.from(config.client);
+    this.tableName = config.tableName;
+    this.gsiName = config.gsiName ?? "domain-subjectId-index";
+  }
+  buildContextKey(domain, workflowId, subjectId) {
+    return `${domain}#${workflowId}#${subjectId}`;
+  }
+  async getContexts(domain, workflowId, subjectId) {
+    const contextKey = this.buildContextKey(domain, workflowId, subjectId);
+    const contexts = [];
+    let lastKey;
+    do {
+      const result = await this.docClient.send(
+        new import_lib_dynamodb2.QueryCommand({
+          TableName: this.tableName,
+          KeyConditionExpression: "contextKey = :contextKey",
+          ExpressionAttributeValues: { ":contextKey": contextKey },
+          ExclusiveStartKey: lastKey
+        })
+      );
+      for (const item of result.Items ?? []) {
+        contexts.push(item.data);
+      }
+      lastKey = result.LastEvaluatedKey;
+    } while (lastKey);
+    return contexts;
+  }
+  async saveContext(domain, workflowId, subjectId, context) {
+    const item = {
+      contextKey: this.buildContextKey(domain, workflowId, subjectId),
+      instanceId: context.instanceId,
+      domain,
+      subjectId,
+      data: context,
+      isCompleted: !!context.isCompleted,
+      startedAt: context.startedAt,
+      updatedAt: Date.now()
+    };
+    await this.docClient.send(
+      new import_lib_dynamodb2.PutCommand({
+        TableName: this.tableName,
+        Item: item
+      })
+    );
+  }
+  async deleteContext(domain, workflowId, subjectId, instanceId) {
+    await this.docClient.send(
+      new import_lib_dynamodb2.DeleteCommand({
+        TableName: this.tableName,
+        Key: {
+          contextKey: this.buildContextKey(domain, workflowId, subjectId),
+          instanceId
+        }
+      })
+    );
+  }
+  /**
+   * Fetch a single Context by instance id.
+   */
+  async getContext(domain, workflowId, subjectId, instanceId) {
+    const result = await this.docClient.send(
+      new import_lib_dynamodb2.GetCommand({
+        TableName: this.tableName,
+        Key: {
+          contextKey: this.buildContextKey(domain, workflowId, subjectId),
+          instanceId
+        }
+      })
+    );
+    if (!result.Item) {
+      return null;
+    }
+    return result.Item.data;
+  }
+  /**
+   * List all contexts for a given subject across all workflows in a domain.
+   * Uses the `domain-subjectId-index` GSI.
+   */
+  async getAllContextsForSubject(domain, subjectId) {
+    const contexts = [];
+    let lastKey;
+    do {
+      const result = await this.docClient.send(
+        new import_lib_dynamodb2.QueryCommand({
+          TableName: this.tableName,
+          IndexName: this.gsiName,
+          KeyConditionExpression: "#domain = :domain AND subjectId = :subjectId",
+          ExpressionAttributeNames: { "#domain": "domain" },
+          ExpressionAttributeValues: { ":domain": domain, ":subjectId": subjectId },
+          ExclusiveStartKey: lastKey
+        })
+      );
+      for (const item of result.Items ?? []) {
+        contexts.push(item.data);
+      }
+      lastKey = result.LastEvaluatedKey;
+    } while (lastKey);
+    return contexts;
+  }
+  /**
+   * List all contexts across all subjects and workflows in a domain.
+   * Uses the `domain-subjectId-index` GSI.
+   * Returns contexts annotated with `subjectId`.
+   */
+  async getAllContexts(domain) {
+    const contexts = [];
+    let lastKey;
+    do {
+      const result = await this.docClient.send(
+        new import_lib_dynamodb2.QueryCommand({
+          TableName: this.tableName,
+          IndexName: this.gsiName,
+          KeyConditionExpression: "#domain = :domain",
+          ExpressionAttributeNames: { "#domain": "domain" },
+          ExpressionAttributeValues: { ":domain": domain },
+          ExclusiveStartKey: lastKey
+        })
+      );
+      for (const item of result.Items ?? []) {
+        const ci = item;
+        contexts.push({ ...ci.data, subjectId: ci.subjectId });
+      }
+      lastKey = result.LastEvaluatedKey;
+    } while (lastKey);
+    return contexts;
+  }
+};
+
+// src/schedulers/EventBusWorkflowScheduler.ts
+var import_client_scheduler = require("@aws-sdk/client-scheduler");
+var import_nanoid2 = require("nanoid");
+var EventBusWorkflowScheduler = class {
+  constructor(config) {
+    this.client = config.client;
+    this.eventBusArn = config.eventBusArn;
+    this.roleArn = config.roleArn;
+    this.scheduleGroupName = config.scheduleGroupName ?? "default";
+    this.source = config.source ?? "omega-flow";
+    this.detailType = config.detailType ?? "workflow.event";
+  }
+  async schedule(event, delayMs) {
+    const fireAt = new Date(Date.now() + delayMs);
+    const scheduleExpression = `at(${fireAt.toISOString().slice(0, 19)})`;
+    const name = `omf-${(0, import_nanoid2.nanoid)(16)}`;
+    await this.client.send(
+      new import_client_scheduler.CreateScheduleCommand({
+        Name: name,
+        GroupName: this.scheduleGroupName,
+        ScheduleExpression: scheduleExpression,
+        ScheduleExpressionTimezone: "UTC",
+        FlexibleTimeWindow: { Mode: "OFF" },
+        ActionAfterCompletion: "DELETE",
+        Target: {
+          Arn: this.eventBusArn,
+          RoleArn: this.roleArn,
+          Input: JSON.stringify(event),
+          EventBridgeParameters: {
+            Source: this.source,
+            DetailType: this.detailType
+          }
+        }
+      })
+    );
+    return name;
+  }
+  async cancel(scheduleId) {
+    try {
+      await this.client.send(
+        new import_client_scheduler.DeleteScheduleCommand({
+          Name: scheduleId,
+          GroupName: this.scheduleGroupName
+        })
+      );
+      return true;
+    } catch (err) {
+      if (err instanceof import_client_scheduler.ResourceNotFoundException) {
+        return false;
+      }
+      throw err;
+    }
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  DynamoDBWorkflowMemory,
+  DynamoDBWorkflowStore,
+  EventBusWorkflowScheduler,
+  WorkflowAlreadyExistsError
+});

package/dist/index.mjs

@@ -0,0 +1,318 @@
+// src/stores/DynamoDBWorkflowStore.ts
+import {
+  ConditionalCheckFailedException
+} from "@aws-sdk/client-dynamodb";
+import {
+  DeleteCommand,
+  DynamoDBDocumentClient,
+  GetCommand,
+  PutCommand,
+  QueryCommand,
+  UpdateCommand
+} from "@aws-sdk/lib-dynamodb";
+import { nanoid } from "nanoid";
+
+// src/errors.ts
+var WorkflowAlreadyExistsError = class extends Error {
+  constructor(domain, workflowId) {
+    super(
+      `Workflow already exists: domain=${domain} workflowId=${workflowId}`
+    );
+    this.domain = domain;
+    this.workflowId = workflowId;
+    this.name = "WorkflowAlreadyExistsError";
+  }
+};
+
+// src/stores/DynamoDBWorkflowStore.ts
+var DynamoDBWorkflowStore = class {
+  constructor(config) {
+    this.docClient = DynamoDBDocumentClient.from(config.client);
+    this.tableName = config.tableName;
+  }
+  async getWorkflow(domain, workflowId) {
+    const result = await this.docClient.send(
+      new GetCommand({
+        TableName: this.tableName,
+        Key: { domain, workflowId }
+      })
+    );
+    if (!result.Item) {
+      return null;
+    }
+    return result.Item.data;
+  }
+  async getAllWorkflows(domain) {
+    const workflows = [];
+    let lastKey;
+    do {
+      const result = await this.docClient.send(
+        new QueryCommand({
+          TableName: this.tableName,
+          KeyConditionExpression: "#domain = :domain",
+          ExpressionAttributeNames: { "#domain": "domain" },
+          ExpressionAttributeValues: { ":domain": domain },
+          ExclusiveStartKey: lastKey
+        })
+      );
+      for (const item of result.Items ?? []) {
+        workflows.push(item.data);
+      }
+      lastKey = result.LastEvaluatedKey;
+    } while (lastKey);
+    return workflows;
+  }
+  async setWorkflow(domain, workflow) {
+    const now = Date.now();
+    await this.docClient.send(
+      new UpdateCommand({
+        TableName: this.tableName,
+        Key: { domain, workflowId: workflow.id },
+        UpdateExpression: "SET #data = :data, createdAt = if_not_exists(createdAt, :now), updatedAt = :now",
+        ExpressionAttributeNames: { "#data": "data" },
+        ExpressionAttributeValues: { ":data": workflow, ":now": now }
+      })
+    );
+  }
+  async createWorkflow(domain, workflowData) {
+    const id = nanoid(8);
+    const workflow = { ...workflowData, id };
+    const now = Date.now();
+    const item = {
+      domain,
+      workflowId: id,
+      data: workflow,
+      createdAt: now,
+      updatedAt: now
+    };
+    try {
+      await this.docClient.send(
+        new PutCommand({
+          TableName: this.tableName,
+          Item: item,
+          ConditionExpression: "attribute_not_exists(#domain)",
+          ExpressionAttributeNames: { "#domain": "domain" }
+        })
+      );
+    } catch (err) {
+      if (err instanceof ConditionalCheckFailedException) {
+        throw new WorkflowAlreadyExistsError(domain, id);
+      }
+      throw err;
+    }
+    return workflow;
+  }
+  async deleteWorkflow(domain, workflowId) {
+    const result = await this.docClient.send(
+      new DeleteCommand({
+        TableName: this.tableName,
+        Key: { domain, workflowId },
+        ReturnValues: "ALL_OLD"
+      })
+    );
+    return result.Attributes !== void 0;
+  }
+};
+
+// src/memories/DynamoDBWorkflowMemory.ts
+import {
+  DeleteCommand as DeleteCommand2,
+  DynamoDBDocumentClient as DynamoDBDocumentClient2,
+  GetCommand as GetCommand2,
+  PutCommand as PutCommand2,
+  QueryCommand as QueryCommand2
+} from "@aws-sdk/lib-dynamodb";
+var DynamoDBWorkflowMemory = class {
+  constructor(config) {
+    this.docClient = DynamoDBDocumentClient2.from(config.client);
+    this.tableName = config.tableName;
+    this.gsiName = config.gsiName ?? "domain-subjectId-index";
+  }
+  buildContextKey(domain, workflowId, subjectId) {
+    return `${domain}#${workflowId}#${subjectId}`;
+  }
+  async getContexts(domain, workflowId, subjectId) {
+    const contextKey = this.buildContextKey(domain, workflowId, subjectId);
+    const contexts = [];
+    let lastKey;
+    do {
+      const result = await this.docClient.send(
+        new QueryCommand2({
+          TableName: this.tableName,
+          KeyConditionExpression: "contextKey = :contextKey",
+          ExpressionAttributeValues: { ":contextKey": contextKey },
+          ExclusiveStartKey: lastKey
+        })
+      );
+      for (const item of result.Items ?? []) {
+        contexts.push(item.data);
+      }
+      lastKey = result.LastEvaluatedKey;
+    } while (lastKey);
+    return contexts;
+  }
+  async saveContext(domain, workflowId, subjectId, context) {
+    const item = {
+      contextKey: this.buildContextKey(domain, workflowId, subjectId),
+      instanceId: context.instanceId,
+      domain,
+      subjectId,
+      data: context,
+      isCompleted: !!context.isCompleted,
+      startedAt: context.startedAt,
+      updatedAt: Date.now()
+    };
+    await this.docClient.send(
+      new PutCommand2({
+        TableName: this.tableName,
+        Item: item
+      })
+    );
+  }
+  async deleteContext(domain, workflowId, subjectId, instanceId) {
+    await this.docClient.send(
+      new DeleteCommand2({
+        TableName: this.tableName,
+        Key: {
+          contextKey: this.buildContextKey(domain, workflowId, subjectId),
+          instanceId
+        }
+      })
+    );
+  }
+  /**
+   * Fetch a single Context by instance id.
+   */
+  async getContext(domain, workflowId, subjectId, instanceId) {
+    const result = await this.docClient.send(
+      new GetCommand2({
+        TableName: this.tableName,
+        Key: {
+          contextKey: this.buildContextKey(domain, workflowId, subjectId),
+          instanceId
+        }
+      })
+    );
+    if (!result.Item) {
+      return null;
+    }
+    return result.Item.data;
+  }
+  /**
+   * List all contexts for a given subject across all workflows in a domain.
+   * Uses the `domain-subjectId-index` GSI.
+   */
+  async getAllContextsForSubject(domain, subjectId) {
+    const contexts = [];
+    let lastKey;
+    do {
+      const result = await this.docClient.send(
+        new QueryCommand2({
+          TableName: this.tableName,
+          IndexName: this.gsiName,
+          KeyConditionExpression: "#domain = :domain AND subjectId = :subjectId",
+          ExpressionAttributeNames: { "#domain": "domain" },
+          ExpressionAttributeValues: { ":domain": domain, ":subjectId": subjectId },
+          ExclusiveStartKey: lastKey
+        })
+      );
+      for (const item of result.Items ?? []) {
+        contexts.push(item.data);
+      }
+      lastKey = result.LastEvaluatedKey;
+    } while (lastKey);
+    return contexts;
+  }
+  /**
+   * List all contexts across all subjects and workflows in a domain.
+   * Uses the `domain-subjectId-index` GSI.
+   * Returns contexts annotated with `subjectId`.
+   */
+  async getAllContexts(domain) {
+    const contexts = [];
+    let lastKey;
+    do {
+      const result = await this.docClient.send(
+        new QueryCommand2({
+          TableName: this.tableName,
+          IndexName: this.gsiName,
+          KeyConditionExpression: "#domain = :domain",
+          ExpressionAttributeNames: { "#domain": "domain" },
+          ExpressionAttributeValues: { ":domain": domain },
+          ExclusiveStartKey: lastKey
+        })
+      );
+      for (const item of result.Items ?? []) {
+        const ci = item;
+        contexts.push({ ...ci.data, subjectId: ci.subjectId });
+      }
+      lastKey = result.LastEvaluatedKey;
+    } while (lastKey);
+    return contexts;
+  }
+};
+
+// src/schedulers/EventBusWorkflowScheduler.ts
+import {
+  CreateScheduleCommand,
+  DeleteScheduleCommand,
+  ResourceNotFoundException
+} from "@aws-sdk/client-scheduler";
+import { nanoid as nanoid2 } from "nanoid";
+var EventBusWorkflowScheduler = class {
+  constructor(config) {
+    this.client = config.client;
+    this.eventBusArn = config.eventBusArn;
+    this.roleArn = config.roleArn;
+    this.scheduleGroupName = config.scheduleGroupName ?? "default";
+    this.source = config.source ?? "omega-flow";
+    this.detailType = config.detailType ?? "workflow.event";
+  }
+  async schedule(event, delayMs) {
+    const fireAt = new Date(Date.now() + delayMs);
+    const scheduleExpression = `at(${fireAt.toISOString().slice(0, 19)})`;
+    const name = `omf-${nanoid2(16)}`;
+    await this.client.send(
+      new CreateScheduleCommand({
+        Name: name,
+        GroupName: this.scheduleGroupName,
+        ScheduleExpression: scheduleExpression,
+        ScheduleExpressionTimezone: "UTC",
+        FlexibleTimeWindow: { Mode: "OFF" },
+        ActionAfterCompletion: "DELETE",
+        Target: {
+          Arn: this.eventBusArn,
+          RoleArn: this.roleArn,
+          Input: JSON.stringify(event),
+          EventBridgeParameters: {
+            Source: this.source,
+            DetailType: this.detailType
+          }
+        }
+      })
+    );
+    return name;
+  }
+  async cancel(scheduleId) {
+    try {
+      await this.client.send(
+        new DeleteScheduleCommand({
+          Name: scheduleId,
+          GroupName: this.scheduleGroupName
+        })
+      );
+      return true;
+    } catch (err) {
+      if (err instanceof ResourceNotFoundException) {
+        return false;
+      }
+      throw err;
+    }
+  }
+};
+export {
+  DynamoDBWorkflowMemory,
+  DynamoDBWorkflowStore,
+  EventBusWorkflowScheduler,
+  WorkflowAlreadyExistsError
+};

package/package.json

@@ -0,0 +1,40 @@
+{
+  "name": "@omega-flow/store-aws",
+  "version": "0.1.0",
+  "description": "AWS-backed storage and scheduler implementations for Omega Flow",
+  "main": "dist/index.js",
+  "module": "dist/index.esm.js",
+  "types": "dist/index.d.ts",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/omega-flow/omega-flow.git",
+    "directory": "packages/store-aws"
+  },
+  "homepage": "https://omega-flow.github.io/omega-flow/",
+  "bugs": "https://github.com/omega-flow/omega-flow/issues",
+  "publishConfig": {
+    "access": "public"
+  },
+  "files": [
+    "dist"
+  ],
+  "devDependencies": {
+    "aws-sdk-client-mock": "^4.1.0"
+  },
+  "dependencies": {
+    "@aws-sdk/client-dynamodb": "^3.700.0",
+    "@aws-sdk/client-scheduler": "^3.1045.0",
+    "@aws-sdk/lib-dynamodb": "^3.700.0",
+    "nanoid": "^3.3.7",
+    "@omega-flow/types": "0.1.0",
+    "@omega-flow/engine": "0.1.0"
+  },
+  "scripts": {
+    "build": "tsup src/index.ts --format cjs,esm --dts",
+    "dev": "tsup src/index.ts --format cjs,esm --dts --watch",
+    "test": "jest --verbose --passWithNoTests",
+    "test:watch": "jest --watchAll --verbose --passWithNoTests --detectOpenHandles",
+    "test:coverage": "jest --verbose --coverage"
+  }
+}