@kronos-ts/drizzle 0.1.0

package/dist/drizzle-token-store.d.ts

@@ -0,0 +1,62 @@
+import type { TokenStore } from "@kronos-ts/messaging";
+import type { DrizzleDatabaseLike } from "./drizzle-transaction-manager.js";
+/**
+ * Token table configuration. Users pass their Drizzle table reference
+ * and the `eq`, `and`, `or`, `lt` operators from drizzle-orm.
+ *
+ * The table must match the Kronos `TokenEntry` schema:
+ *
+ * ```typescript
+ * import { pgTable, varchar, integer, primaryKey } from "drizzle-orm/pg-core"
+ *
+ * export const kronosTokenEntries = pgTable("kronos_token_entries", {
+ *   processorName: varchar("processor_name", { length: 255 }).notNull(),
+ *   segment: integer("segment").notNull(),
+ *   mask: integer("mask").notNull().default(0),
+ *   tokenType: varchar("token_type", { length: 255 }),
+ *   token: varchar("token", { length: 10000 }),
+ *   timestamp: varchar("timestamp", { length: 255 }),
+ *   owner: varchar("owner", { length: 255 }),
+ * }, (table) => ({
+ *   pk: primaryKey({ columns: [table.processorName, table.segment] }),
+ * }))
+ * ```
+ */
+export interface DrizzleTokenStoreConfig {
+    /** The Drizzle database instance. */
+    db: DrizzleDatabaseLike;
+    /** The Drizzle table reference for `kronos_token_entries`. */
+    table: any;
+    /** Drizzle `eq` operator. */
+    eq: (column: any, value: any) => any;
+    /** Drizzle `and` operator. */
+    and: (...conditions: any[]) => any;
+    /** Drizzle `or` operator. */
+    or: (...conditions: any[]) => any;
+    /** Drizzle `lt` operator. */
+    lt: (column: any, value: any) => any;
+    /** Drizzle `isNull` operator. */
+    isNull: (column: any) => any;
+    /** Claim timeout in ms. Default: 10000. */
+    claimTimeoutMs?: number;
+}
+/**
+ * Creates a TokenStore backed by Drizzle ORM.
+ *
+ * Participates in the active transaction via `getActiveTransaction()`.
+ *
+ * ```typescript
+ * import { eq, and, or, lt, isNull } from "drizzle-orm"
+ * import { drizzleTokenStore } from "@kronos-ts/drizzle"
+ * import { kronosTokenEntries } from "./schema"
+ *
+ * // tokenStore wiring to a kronos() App is pending a typed `tokenStore` slot
+ * // (Phase 9). For now, construct the store and pass it directly to the
+ * // tracking processor that owns it:
+ * const tokenStore = drizzleTokenStore({
+ *   db, table: kronosTokenEntries, eq, and, or, lt, isNull,
+ * })
+ * ```
+ */
+export declare function drizzleTokenStore(config: DrizzleTokenStoreConfig): TokenStore;
+//# sourceMappingURL=drizzle-token-store.d.ts.map

package/dist/drizzle-token-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"drizzle-token-store.d.ts","sourceRoot":"","sources":["../src/drizzle-token-store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAiB,MAAM,sBAAsB,CAAA;AAErE,OAAO,KAAK,EAAE,mBAAmB,EAAsB,MAAM,kCAAkC,CAAA;AAE/F;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,WAAW,uBAAuB;IACtC,qCAAqC;IACrC,EAAE,EAAE,mBAAmB,CAAA;IACvB,8DAA8D;IAC9D,KAAK,EAAE,GAAG,CAAA;IACV,6BAA6B;IAC7B,EAAE,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,KAAK,GAAG,CAAA;IACpC,8BAA8B;IAC9B,GAAG,EAAE,CAAC,GAAG,UAAU,EAAE,GAAG,EAAE,KAAK,GAAG,CAAA;IAClC,6BAA6B;IAC7B,EAAE,EAAE,CAAC,GAAG,UAAU,EAAE,GAAG,EAAE,KAAK,GAAG,CAAA;IACjC,6BAA6B;IAC7B,EAAE,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,KAAK,GAAG,CAAA;IACpC,iCAAiC;IACjC,MAAM,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,GAAG,CAAA;IAC5B,2CAA2C;IAC3C,cAAc,CAAC,EAAE,MAAM,CAAA;CACxB;AAmBD;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,uBAAuB,GAAG,UAAU,CAiH7E"}

package/dist/drizzle-token-store.js

@@ -0,0 +1,125 @@
+import { getActiveTransaction, UnableToClaimTokenError, globalSequenceToken } from "@kronos-ts/messaging";
+function serializeToken(token) {
+    return {
+        tokenType: "GlobalSequenceToken",
+        token: JSON.stringify({ position: token.position().toString() }),
+    };
+}
+function deserializeToken(tokenType, token) {
+    if (!token || !tokenType)
+        return undefined;
+    const data = JSON.parse(token);
+    return globalSequenceToken(BigInt(data.position));
+}
+function nowIso() {
+    return new Date().toISOString();
+}
+/**
+ * Creates a TokenStore backed by Drizzle ORM.
+ *
+ * Participates in the active transaction via `getActiveTransaction()`.
+ *
+ * ```typescript
+ * import { eq, and, or, lt, isNull } from "drizzle-orm"
+ * import { drizzleTokenStore } from "@kronos-ts/drizzle"
+ * import { kronosTokenEntries } from "./schema"
+ *
+ * // tokenStore wiring to a kronos() App is pending a typed `tokenStore` slot
+ * // (Phase 9). For now, construct the store and pass it directly to the
+ * // tracking processor that owns it:
+ * const tokenStore = drizzleTokenStore({
+ *   db, table: kronosTokenEntries, eq, and, or, lt, isNull,
+ * })
+ * ```
+ */
+export function drizzleTokenStore(config) {
+    const { table, eq, and, or, lt, isNull } = config;
+    const claimTimeoutMs = config.claimTimeoutMs ?? 10000;
+    function getDb() {
+        return getActiveTransaction() ?? config.db;
+    }
+    return {
+        async store(processorName, segment, token) {
+            const db = getDb();
+            const { tokenType, token: tokenData } = serializeToken(token);
+            await db.insert(table)
+                .values({ processorName, segment, mask: 0, tokenType, token: tokenData, timestamp: nowIso(), owner: null })
+                .onConflictDoUpdate({
+                target: [table.processorName, table.segment],
+                set: { tokenType, token: tokenData, timestamp: nowIso() },
+            });
+        },
+        async get(processorName, segment) {
+            const db = getDb();
+            const rows = await db.select().from(table)
+                .where(and(eq(table.processorName, processorName), eq(table.segment, segment)))
+                .limit(1);
+            if (rows.length === 0)
+                return undefined;
+            return deserializeToken(rows[0].tokenType, rows[0].token);
+        },
+        async initializeSegments(processorName, segmentCount) {
+            const db = getDb();
+            for (let i = 0; i < segmentCount; i++) {
+                await db.insert(table)
+                    .values({ processorName, segment: i, mask: 0, tokenType: null, token: null, timestamp: null, owner: null })
+                    .onConflictDoNothing();
+            }
+        },
+        async claimToken(processorName, segment, ownerId) {
+            const db = getDb();
+            const rows = await db.select().from(table)
+                .where(and(eq(table.processorName, processorName), eq(table.segment, segment)))
+                .limit(1);
+            if (rows.length === 0) {
+                await db.insert(table).values({
+                    processorName, segment, mask: 0, tokenType: null, token: null, timestamp: nowIso(), owner: ownerId,
+                });
+                return undefined;
+            }
+            const row = rows[0];
+            const isExpired = !row.owner || !row.timestamp ||
+                (Date.now() - new Date(row.timestamp).getTime() > claimTimeoutMs);
+            if (row.owner === ownerId || isExpired) {
+                await db.update(table)
+                    .set({ owner: ownerId, timestamp: nowIso() })
+                    .where(and(eq(table.processorName, processorName), eq(table.segment, segment)));
+                return deserializeToken(row.tokenType, row.token);
+            }
+            throw new UnableToClaimTokenError(processorName, segment);
+        },
+        async extendClaim(processorName, segment, ownerId) {
+            const db = getDb();
+            await db.update(table)
+                .set({ timestamp: nowIso() })
+                .where(and(eq(table.processorName, processorName), eq(table.segment, segment), eq(table.owner, ownerId)));
+        },
+        async releaseClaim(processorName, segment, ownerId) {
+            const db = getDb();
+            await db.update(table)
+                .set({ owner: null, timestamp: null })
+                .where(and(eq(table.processorName, processorName), eq(table.segment, segment), eq(table.owner, ownerId)));
+        },
+        async fetchSegments(processorName) {
+            const db = getDb();
+            const rows = await db.select({ segment: table.segment }).from(table)
+                .where(eq(table.processorName, processorName))
+                .orderBy(table.segment);
+            return rows.map((r) => r.segment);
+        },
+        async fetchAvailableSegments(processorName) {
+            const db = getDb();
+            const cutoff = new Date(Date.now() - claimTimeoutMs).toISOString();
+            const rows = await db.select({ segment: table.segment }).from(table)
+                .where(and(eq(table.processorName, processorName), or(isNull(table.owner), lt(table.timestamp, cutoff))))
+                .orderBy(table.segment);
+            return rows.map((r) => r.segment);
+        },
+        async deleteToken(processorName, segment) {
+            const db = getDb();
+            await db.delete(table)
+                .where(and(eq(table.processorName, processorName), eq(table.segment, segment)));
+        },
+    };
+}
+//# sourceMappingURL=drizzle-token-store.js.map

package/dist/drizzle-token-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"drizzle-token-store.js","sourceRoot":"","sources":["../src/drizzle-token-store.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,uBAAuB,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAA;AA4CzG,SAAS,cAAc,CAAC,KAAoB;IAC1C,OAAO;QACL,SAAS,EAAE,qBAAqB;QAChC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC;KACjE,CAAA;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,SAAwB,EAAE,KAAoB;IACtE,IAAI,CAAC,KAAK,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAA;IAC1C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;IAC9B,OAAO,mBAAmB,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAA;AACnD,CAAC;AAED,SAAS,MAAM;IACb,OAAO,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAA;AACjC,CAAC;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,UAAU,iBAAiB,CAAC,MAA+B;IAC/D,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,MAAM,CAAA;IACjD,MAAM,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,KAAK,CAAA;IAErD,SAAS,KAAK;QACZ,OAAO,oBAAoB,EAAsB,IAAI,MAAM,CAAC,EAAE,CAAA;IAChE,CAAC;IAED,OAAO;QACL,KAAK,CAAC,KAAK,CAAC,aAAa,EAAE,OAAO,EAAE,KAAK;YACvC,MAAM,EAAE,GAAG,KAAK,EAAE,CAAA;YAClB,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,cAAc,CAAC,KAAK,CAAC,CAAA;YAC7D,MAAM,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC;iBACnB,MAAM,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;iBAC1G,kBAAkB,CAAC;gBAClB,MAAM,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,KAAK,CAAC,OAAO,CAAC;gBAC5C,GAAG,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE;aAC1D,CAAC,CAAA;QACN,CAAC;QAED,KAAK,CAAC,GAAG,CAAC,aAAa,EAAE,OAAO;YAC9B,MAAM,EAAE,GAAG,KAAK,EAAE,CAAA;YAClB,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC;iBACvC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,aAAa,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;iBAC9E,KAAK,CAAC,CAAC,CAAC,CAAA;YACX,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,SAAS,CAAA;YACvC,OAAO,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAA;QAC3D,CAAC;QAED,KAAK,CAAC,kBAAkB,CAAC,aAAa,EAAE,YAAY;YAClD,MAAM,EAAE,GAAG,KAAK,EAAE,CAAA;YAClB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtC,MAAM,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC;qBACnB,MAAM,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;qBAC1G,mBAAmB,EAAE,CAAA;YAC1B,CAAC;QACH,CAAC;QAED,KAAK,CAAC,UAAU,CAAC,aAAa,EAAE,OAAO,EAAE,OAAO;YAC9C,MAAM,EAAE,GAAG,KAAK,EAAE,CAAA;YAClB,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC;iBACvC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,aAAa,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;iBAC9E,KAAK,CAAC,CAAC,CAAC,CAAA;YAEX,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACtB,MAAM,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;oBAC5B,aAAa,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,OAAO;iBACnG,CAAC,CAAA;gBACF,OAAO,SAAS,CAAA;YAClB,CAAC;YAED,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAA;YACnB,MAAM,SAAS,GAAG,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,GAAG,CAAC,SAAS;gBAC5C,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,GAAG,cAAc,CAAC,CAAA;YAEnE,IAAI,GAAG,CAAC,KAAK,KAAK,OAAO,IAAI,SAAS,EAAE,CAAC;gBACvC,MAAM,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC;qBACnB,GAAG,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC;qBAC5C,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,aAAa,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAA;gBACjF,OAAO,gBAAgB,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,KAAK,CAAC,CAAA;YACnD,CAAC;YAED,MAAM,IAAI,uBAAuB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAA;QAC3D,CAAC;QAED,KAAK,CAAC,WAAW,CAAC,aAAa,EAAE,OAAO,EAAE,OAAO;YAC/C,MAAM,EAAE,GAAG,KAAK,EAAE,CAAA;YAClB,MAAM,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC;iBACnB,GAAG,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC;iBAC5B,KAAK,CAAC,GAAG,CACR,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,aAAa,CAAC,EACtC,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,EAC1B,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC,CACzB,CAAC,CAAA;QACN,CAAC;QAED,KAAK,CAAC,YAAY,CAAC,aAAa,EAAE,OAAO,EAAE,OAAO;YAChD,MAAM,EAAE,GAAG,KAAK,EAAE,CAAA;YAClB,MAAM,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC;iBACnB,GAAG,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;iBACrC,KAAK,CAAC,GAAG,CACR,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,aAAa,CAAC,EACtC,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,EAC1B,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC,CACzB,CAAC,CAAA;QACN,CAAC;QAED,KAAK,CAAC,aAAa,CAAC,aAAa;YAC/B,MAAM,EAAE,GAAG,KAAK,EAAE,CAAA;YAClB,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;iBACjE,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;iBAC7C,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;YACzB,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAA;QACxC,CAAC;QAED,KAAK,CAAC,sBAAsB,CAAC,aAAa;YACxC,MAAM,EAAE,GAAG,KAAK,EAAE,CAAA;YAClB,MAAM,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc,CAAC,CAAC,WAAW,EAAE,CAAA;YAClE,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;iBACjE,KAAK,CAAC,GAAG,CACR,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,aAAa,CAAC,EACtC,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CACrD,CAAC;iBACD,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;YACzB,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAA;QACxC,CAAC;QAED,KAAK,CAAC,WAAW,CAAC,aAAa,EAAE,OAAO;YACtC,MAAM,EAAE,GAAG,KAAK,EAAE,CAAA;YAClB,MAAM,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC;iBACnB,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,aAAa,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAA;QACnF,CAAC;KACF,CAAA;AACH,CAAC"}

package/dist/drizzle-transaction-manager.d.ts

@@ -0,0 +1,36 @@
+import type { TransactionManager } from "@kronos-ts/messaging";
+/**
+ * A Drizzle database instance that supports transactions.
+ * Works with any Drizzle driver (postgres-js, node-postgres, better-sqlite3, etc.).
+ */
+export interface DrizzleDatabaseLike {
+    transaction<T>(fn: (tx: any) => Promise<T>): Promise<T>;
+}
+/**
+ * The Drizzle transaction object — has the same query API as the database.
+ */
+export type DrizzleTransaction = any;
+/**
+ * Creates a TransactionManager for Drizzle ORM.
+ *
+ * Drizzle's `db.transaction()` uses a callback pattern. This manager
+ * bridges it to the framework's `begin/commit/rollback` lifecycle
+ * using deferred promises — the same pattern as the Prisma extension.
+ *
+ * The `tx` is made available via `getActiveTransaction()` so projections
+ * and token stores participate in the same transaction.
+ *
+ * ```typescript
+ * import { drizzle } from "drizzle-orm/postgres-js"
+ * import { drizzleTransactionManager } from "@kronos-ts/drizzle"
+ *
+ * const db = drizzle(sql)
+ *
+ * // transactionManager wiring to a kronos() App is pending a typed
+ * // `transactionManager` slot (Phase 9). For now, construct the manager
+ * // and pass it directly into the unitOfWorkFactory composition:
+ * const txManager = drizzleTransactionManager(db)
+ * ```
+ */
+export declare function drizzleTransactionManager(db: DrizzleDatabaseLike): TransactionManager<DrizzleTransaction>;
+//# sourceMappingURL=drizzle-transaction-manager.d.ts.map

package/dist/drizzle-transaction-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"drizzle-transaction-manager.d.ts","sourceRoot":"","sources":["../src/drizzle-transaction-manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAA;AAE9D;;;GAGG;AACH,MAAM,WAAW,mBAAmB;IAClC,WAAW,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAA;CACxD;AAED;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG,GAAG,CAAA;AAEpC;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,wBAAgB,yBAAyB,CACvC,EAAE,EAAE,mBAAmB,GACtB,kBAAkB,CAAC,kBAAkB,CAAC,CA2CxC"}

package/dist/drizzle-transaction-manager.js

@@ -0,0 +1,63 @@
+/**
+ * Creates a TransactionManager for Drizzle ORM.
+ *
+ * Drizzle's `db.transaction()` uses a callback pattern. This manager
+ * bridges it to the framework's `begin/commit/rollback` lifecycle
+ * using deferred promises — the same pattern as the Prisma extension.
+ *
+ * The `tx` is made available via `getActiveTransaction()` so projections
+ * and token stores participate in the same transaction.
+ *
+ * ```typescript
+ * import { drizzle } from "drizzle-orm/postgres-js"
+ * import { drizzleTransactionManager } from "@kronos-ts/drizzle"
+ *
+ * const db = drizzle(sql)
+ *
+ * // transactionManager wiring to a kronos() App is pending a typed
+ * // `transactionManager` slot (Phase 9). For now, construct the manager
+ * // and pass it directly into the unitOfWorkFactory composition:
+ * const txManager = drizzleTransactionManager(db)
+ * ```
+ */
+export function drizzleTransactionManager(db) {
+    return {
+        async begin() {
+            let resolveTx;
+            let resolveCompletion;
+            let rejectCompletion;
+            const txReady = new Promise((resolve) => {
+                resolveTx = resolve;
+            });
+            const completionSignal = new Promise((resolve, reject) => {
+                resolveCompletion = resolve;
+                rejectCompletion = reject;
+            });
+            const txPromise = db.transaction(async (tx) => {
+                resolveTx(tx);
+                await completionSignal;
+            });
+            const tx = await txReady;
+            tx.__kronos_commit = resolveCompletion;
+            tx.__kronos_rollback = rejectCompletion;
+            tx.__kronos_txPromise = txPromise;
+            return tx;
+        },
+        async commit(tx) {
+            const commit = tx.__kronos_commit;
+            const txPromise = tx.__kronos_txPromise;
+            commit();
+            await txPromise;
+        },
+        async rollback(tx) {
+            const rollback = tx.__kronos_rollback;
+            const txPromise = tx.__kronos_txPromise;
+            rollback(new Error("Transaction rolled back"));
+            try {
+                await txPromise;
+            }
+            catch { /* expected */ }
+        },
+    };
+}
+//# sourceMappingURL=drizzle-transaction-manager.js.map

package/dist/drizzle-transaction-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"drizzle-transaction-manager.js","sourceRoot":"","sources":["../src/drizzle-transaction-manager.ts"],"names":[],"mappings":"AAeA;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,UAAU,yBAAyB,CACvC,EAAuB;IAEvB,OAAO;QACL,KAAK,CAAC,KAAK;YACT,IAAI,SAA4C,CAAA;YAChD,IAAI,iBAA8B,CAAA;YAClC,IAAI,gBAA2C,CAAA;YAE/C,MAAM,OAAO,GAAG,IAAI,OAAO,CAAqB,CAAC,OAAO,EAAE,EAAE;gBAC1D,SAAS,GAAG,OAAO,CAAA;YACrB,CAAC,CAAC,CAAA;YAEF,MAAM,gBAAgB,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC7D,iBAAiB,GAAG,OAAO,CAAA;gBAC3B,gBAAgB,GAAG,MAAM,CAAA;YAC3B,CAAC,CAAC,CAAA;YAEF,MAAM,SAAS,GAAG,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE;gBAC5C,SAAS,CAAC,EAAE,CAAC,CAAA;gBACb,MAAM,gBAAgB,CAAA;YACxB,CAAC,CAAC,CAAA;YAEF,MAAM,EAAE,GAAG,MAAM,OAAO,CACvB;YAAC,EAAU,CAAC,eAAe,GAAG,iBAAiB,CAC/C;YAAC,EAAU,CAAC,iBAAiB,GAAG,gBAAgB,CAChD;YAAC,EAAU,CAAC,kBAAkB,GAAG,SAAS,CAAA;YAE3C,OAAO,EAAE,CAAA;QACX,CAAC;QAED,KAAK,CAAC,MAAM,CAAC,EAAsB;YACjC,MAAM,MAAM,GAAI,EAAU,CAAC,eAA6B,CAAA;YACxD,MAAM,SAAS,GAAI,EAAU,CAAC,kBAAmC,CAAA;YACjE,MAAM,EAAE,CAAA;YACR,MAAM,SAAS,CAAA;QACjB,CAAC;QAED,KAAK,CAAC,QAAQ,CAAC,EAAsB;YACnC,MAAM,QAAQ,GAAI,EAAU,CAAC,iBAA6C,CAAA;YAC1E,MAAM,SAAS,GAAI,EAAU,CAAC,kBAAmC,CAAA;YACjE,QAAQ,CAAC,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC,CAAA;YAC9C,IAAI,CAAC;gBAAC,MAAM,SAAS,CAAA;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,cAAc,CAAC,CAAC;QAClD,CAAC;KACF,CAAA;AACH,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+export { drizzleTransactionManager, type DrizzleDatabaseLike, type DrizzleTransaction, } from "./drizzle-transaction-manager.js";
+export { drizzleTokenStore, type DrizzleTokenStoreConfig, } from "./drizzle-token-store.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,EACzB,KAAK,mBAAmB,EACxB,KAAK,kBAAkB,GACxB,MAAM,kCAAkC,CAAA;AAEzC,OAAO,EACL,iBAAiB,EACjB,KAAK,uBAAuB,GAC7B,MAAM,0BAA0B,CAAA"}

package/dist/index.js

@@ -0,0 +1,3 @@
+export { drizzleTransactionManager, } from "./drizzle-transaction-manager.js";
+export { drizzleTokenStore, } from "./drizzle-token-store.js";
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,GAG1B,MAAM,kCAAkC,CAAA;AAEzC,OAAO,EACL,iBAAiB,GAElB,MAAM,0BAA0B,CAAA"}

package/package.json

@@ -0,0 +1,58 @@
+{
+  "name": "@kronos-ts/drizzle",
+  "version": "0.1.0",
+  "description": "Drizzle ORM extension for Kronos.",
+  "license": "Apache-2.0",
+  "type": "module",
+  "author": "Theo Emanuelsson",
+  "homepage": "https://github.com/KronosDB/kronos-ts/tree/main/packages/extensions/drizzle#readme",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/KronosDB/kronos-ts.git",
+    "directory": "packages/extensions/drizzle"
+  },
+  "bugs": {
+    "url": "https://github.com/KronosDB/kronos-ts/issues"
+  },
+  "keywords": [
+    "kronos",
+    "event-sourcing",
+    "cqrs",
+    "dcb",
+    "typescript",
+    "drizzle",
+    "orm"
+  ],
+  "sideEffects": false,
+  "main": "src/index.ts",
+  "types": "src/index.ts",
+  "files": [
+    "dist",
+    "src",
+    "!src/**/__tests__",
+    "!src/**/*.test.ts",
+    "!src/**/*.bench.ts"
+  ],
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "clean": "rm -rf dist *.tsbuildinfo"
+  },
+  "publishConfig": {
+    "access": "public",
+    "main": "./dist/index.js",
+    "types": "./dist/index.d.ts",
+    "exports": {
+      ".": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      }
+    }
+  },
+  "dependencies": {
+    "@kronos-ts/common": "workspace:*",
+    "@kronos-ts/messaging": "workspace:*"
+  },
+  "peerDependencies": {
+    "drizzle-orm": ">=0.30.0"
+  }
+}

package/src/drizzle-token-store.ts

@@ -0,0 +1,194 @@
+import type { TokenStore, TrackingToken } from "@kronos-ts/messaging"
+import { getActiveTransaction, UnableToClaimTokenError, globalSequenceToken } from "@kronos-ts/messaging"
+import type { DrizzleDatabaseLike, DrizzleTransaction } from "./drizzle-transaction-manager.js"
+
+/**
+ * Token table configuration. Users pass their Drizzle table reference
+ * and the `eq`, `and`, `or`, `lt` operators from drizzle-orm.
+ *
+ * The table must match the Kronos `TokenEntry` schema:
+ *
+ * ```typescript
+ * import { pgTable, varchar, integer, primaryKey } from "drizzle-orm/pg-core"
+ *
+ * export const kronosTokenEntries = pgTable("kronos_token_entries", {
+ *   processorName: varchar("processor_name", { length: 255 }).notNull(),
+ *   segment: integer("segment").notNull(),
+ *   mask: integer("mask").notNull().default(0),
+ *   tokenType: varchar("token_type", { length: 255 }),
+ *   token: varchar("token", { length: 10000 }),
+ *   timestamp: varchar("timestamp", { length: 255 }),
+ *   owner: varchar("owner", { length: 255 }),
+ * }, (table) => ({
+ *   pk: primaryKey({ columns: [table.processorName, table.segment] }),
+ * }))
+ * ```
+ */
+export interface DrizzleTokenStoreConfig {
+  /** The Drizzle database instance. */
+  db: DrizzleDatabaseLike
+  /** The Drizzle table reference for `kronos_token_entries`. */
+  table: any
+  /** Drizzle `eq` operator. */
+  eq: (column: any, value: any) => any
+  /** Drizzle `and` operator. */
+  and: (...conditions: any[]) => any
+  /** Drizzle `or` operator. */
+  or: (...conditions: any[]) => any
+  /** Drizzle `lt` operator. */
+  lt: (column: any, value: any) => any
+  /** Drizzle `isNull` operator. */
+  isNull: (column: any) => any
+  /** Claim timeout in ms. Default: 10000. */
+  claimTimeoutMs?: number
+}
+
+function serializeToken(token: TrackingToken): { tokenType: string; token: string } {
+  return {
+    tokenType: "GlobalSequenceToken",
+    token: JSON.stringify({ position: token.position().toString() }),
+  }
+}
+
+function deserializeToken(tokenType: string | null, token: string | null): TrackingToken | undefined {
+  if (!token || !tokenType) return undefined
+  const data = JSON.parse(token)
+  return globalSequenceToken(BigInt(data.position))
+}
+
+function nowIso(): string {
+  return new Date().toISOString()
+}
+
+/**
+ * Creates a TokenStore backed by Drizzle ORM.
+ *
+ * Participates in the active transaction via `getActiveTransaction()`.
+ *
+ * ```typescript
+ * import { eq, and, or, lt, isNull } from "drizzle-orm"
+ * import { drizzleTokenStore } from "@kronos-ts/drizzle"
+ * import { kronosTokenEntries } from "./schema"
+ *
+ * // tokenStore wiring to a kronos() App is pending a typed `tokenStore` slot
+ * // (Phase 9). For now, construct the store and pass it directly to the
+ * // tracking processor that owns it:
+ * const tokenStore = drizzleTokenStore({
+ *   db, table: kronosTokenEntries, eq, and, or, lt, isNull,
+ * })
+ * ```
+ */
+export function drizzleTokenStore(config: DrizzleTokenStoreConfig): TokenStore {
+  const { table, eq, and, or, lt, isNull } = config
+  const claimTimeoutMs = config.claimTimeoutMs ?? 10000
+
+  function getDb(): any {
+    return getActiveTransaction<DrizzleTransaction>() ?? config.db
+  }
+
+  return {
+    async store(processorName, segment, token) {
+      const db = getDb()
+      const { tokenType, token: tokenData } = serializeToken(token)
+      await db.insert(table)
+        .values({ processorName, segment, mask: 0, tokenType, token: tokenData, timestamp: nowIso(), owner: null })
+        .onConflictDoUpdate({
+          target: [table.processorName, table.segment],
+          set: { tokenType, token: tokenData, timestamp: nowIso() },
+        })
+    },
+
+    async get(processorName, segment) {
+      const db = getDb()
+      const rows = await db.select().from(table)
+        .where(and(eq(table.processorName, processorName), eq(table.segment, segment)))
+        .limit(1)
+      if (rows.length === 0) return undefined
+      return deserializeToken(rows[0].tokenType, rows[0].token)
+    },
+
+    async initializeSegments(processorName, segmentCount) {
+      const db = getDb()
+      for (let i = 0; i < segmentCount; i++) {
+        await db.insert(table)
+          .values({ processorName, segment: i, mask: 0, tokenType: null, token: null, timestamp: null, owner: null })
+          .onConflictDoNothing()
+      }
+    },
+
+    async claimToken(processorName, segment, ownerId) {
+      const db = getDb()
+      const rows = await db.select().from(table)
+        .where(and(eq(table.processorName, processorName), eq(table.segment, segment)))
+        .limit(1)
+
+      if (rows.length === 0) {
+        await db.insert(table).values({
+          processorName, segment, mask: 0, tokenType: null, token: null, timestamp: nowIso(), owner: ownerId,
+        })
+        return undefined
+      }
+
+      const row = rows[0]
+      const isExpired = !row.owner || !row.timestamp ||
+        (Date.now() - new Date(row.timestamp).getTime() > claimTimeoutMs)
+
+      if (row.owner === ownerId || isExpired) {
+        await db.update(table)
+          .set({ owner: ownerId, timestamp: nowIso() })
+          .where(and(eq(table.processorName, processorName), eq(table.segment, segment)))
+        return deserializeToken(row.tokenType, row.token)
+      }
+
+      throw new UnableToClaimTokenError(processorName, segment)
+    },
+
+    async extendClaim(processorName, segment, ownerId) {
+      const db = getDb()
+      await db.update(table)
+        .set({ timestamp: nowIso() })
+        .where(and(
+          eq(table.processorName, processorName),
+          eq(table.segment, segment),
+          eq(table.owner, ownerId),
+        ))
+    },
+
+    async releaseClaim(processorName, segment, ownerId) {
+      const db = getDb()
+      await db.update(table)
+        .set({ owner: null, timestamp: null })
+        .where(and(
+          eq(table.processorName, processorName),
+          eq(table.segment, segment),
+          eq(table.owner, ownerId),
+        ))
+    },
+
+    async fetchSegments(processorName) {
+      const db = getDb()
+      const rows = await db.select({ segment: table.segment }).from(table)
+        .where(eq(table.processorName, processorName))
+        .orderBy(table.segment)
+      return rows.map((r: any) => r.segment)
+    },
+
+    async fetchAvailableSegments(processorName) {
+      const db = getDb()
+      const cutoff = new Date(Date.now() - claimTimeoutMs).toISOString()
+      const rows = await db.select({ segment: table.segment }).from(table)
+        .where(and(
+          eq(table.processorName, processorName),
+          or(isNull(table.owner), lt(table.timestamp, cutoff)),
+        ))
+        .orderBy(table.segment)
+      return rows.map((r: any) => r.segment)
+    },
+
+    async deleteToken(processorName, segment) {
+      const db = getDb()
+      await db.delete(table)
+        .where(and(eq(table.processorName, processorName), eq(table.segment, segment)))
+    },
+  }
+}

package/src/drizzle-transaction-manager.ts

@@ -0,0 +1,83 @@
+import type { TransactionManager } from "@kronos-ts/messaging"
+
+/**
+ * A Drizzle database instance that supports transactions.
+ * Works with any Drizzle driver (postgres-js, node-postgres, better-sqlite3, etc.).
+ */
+export interface DrizzleDatabaseLike {
+  transaction<T>(fn: (tx: any) => Promise<T>): Promise<T>
+}
+
+/**
+ * The Drizzle transaction object — has the same query API as the database.
+ */
+export type DrizzleTransaction = any
+
+/**
+ * Creates a TransactionManager for Drizzle ORM.
+ *
+ * Drizzle's `db.transaction()` uses a callback pattern. This manager
+ * bridges it to the framework's `begin/commit/rollback` lifecycle
+ * using deferred promises — the same pattern as the Prisma extension.
+ *
+ * The `tx` is made available via `getActiveTransaction()` so projections
+ * and token stores participate in the same transaction.
+ *
+ * ```typescript
+ * import { drizzle } from "drizzle-orm/postgres-js"
+ * import { drizzleTransactionManager } from "@kronos-ts/drizzle"
+ *
+ * const db = drizzle(sql)
+ *
+ * // transactionManager wiring to a kronos() App is pending a typed
+ * // `transactionManager` slot (Phase 9). For now, construct the manager
+ * // and pass it directly into the unitOfWorkFactory composition:
+ * const txManager = drizzleTransactionManager(db)
+ * ```
+ */
+export function drizzleTransactionManager(
+  db: DrizzleDatabaseLike,
+): TransactionManager<DrizzleTransaction> {
+  return {
+    async begin(): Promise<DrizzleTransaction> {
+      let resolveTx!: (tx: DrizzleTransaction) => void
+      let resolveCompletion!: () => void
+      let rejectCompletion!: (error: unknown) => void
+
+      const txReady = new Promise<DrizzleTransaction>((resolve) => {
+        resolveTx = resolve
+      })
+
+      const completionSignal = new Promise<void>((resolve, reject) => {
+        resolveCompletion = resolve
+        rejectCompletion = reject
+      })
+
+      const txPromise = db.transaction(async (tx) => {
+        resolveTx(tx)
+        await completionSignal
+      })
+
+      const tx = await txReady
+      ;(tx as any).__kronos_commit = resolveCompletion
+      ;(tx as any).__kronos_rollback = rejectCompletion
+      ;(tx as any).__kronos_txPromise = txPromise
+
+      return tx
+    },
+
+    async commit(tx: DrizzleTransaction): Promise<void> {
+      const commit = (tx as any).__kronos_commit as () => void
+      const txPromise = (tx as any).__kronos_txPromise as Promise<void>
+      commit()
+      await txPromise
+    },
+
+    async rollback(tx: DrizzleTransaction): Promise<void> {
+      const rollback = (tx as any).__kronos_rollback as (error: unknown) => void
+      const txPromise = (tx as any).__kronos_txPromise as Promise<void>
+      rollback(new Error("Transaction rolled back"))
+      try { await txPromise } catch { /* expected */ }
+    },
+  }
+}

package/src/index.ts

@@ -0,0 +1,10 @@
+export {
+  drizzleTransactionManager,
+  type DrizzleDatabaseLike,
+  type DrizzleTransaction,
+} from "./drizzle-transaction-manager.js"
+
+export {
+  drizzleTokenStore,
+  type DrizzleTokenStoreConfig,
+} from "./drizzle-token-store.js"