sqlite-zod-orm 3.5.2 → 3.6.1

package/dist/index.js

@@ -4714,7 +4714,13 @@ class _Database {
         upsert: (conditions, data) => this.upsert(entityName, data, conditions),
         delete: (id) => this.delete(entityName, id),
         select: (...cols) => this._createQueryBuilder(entityName, cols),
-        on: (callback, options2) => this._createOnStream(entityName, callback, options2?.interval),
+        on: (event, callback, options2) => {
+          if (event === "insert")
+            return this._createOnStream(entityName, callback, options2?.interval);
+          if (event === "update" || event === "delete")
+            return this._createChangeStream(entityName, event, callback, options2?.interval);
+          throw new Error(`Unknown event type: '${event}'. Supported: 'insert', 'update', 'delete'`);
+        },
         _tableName: entityName
       };
       this[key] = accessor;
@@ -4793,6 +4799,61 @@ class _Database {
       stopped = true;
     };
   }
+  _createChangeStream(entityName, eventType, callback, intervalOverride) {
+    const interval = intervalOverride ?? this.pollInterval;
+    const allRows = this.db.query(`SELECT * FROM "${entityName}" ORDER BY id ASC`).all();
+    let snapshot = new Map;
+    let snapshotEntities = new Map;
+    for (const row of allRows) {
+      snapshot.set(row.id, JSON.stringify(row));
+      snapshotEntities.set(row.id, row);
+    }
+    let lastRevision = this._getRevision(entityName);
+    let stopped = false;
+    const poll = async () => {
+      if (stopped)
+        return;
+      const currentRevision = this._getRevision(entityName);
+      if (currentRevision !== lastRevision) {
+        lastRevision = currentRevision;
+        const currentRows = this.db.query(`SELECT * FROM "${entityName}" ORDER BY id ASC`).all();
+        const currentMap = new Map;
+        const currentEntities = new Map;
+        for (const row of currentRows) {
+          currentMap.set(row.id, JSON.stringify(row));
+          currentEntities.set(row.id, row);
+        }
+        if (eventType === "update") {
+          for (const [id, json] of currentMap) {
+            if (stopped)
+              return;
+            if (snapshot.has(id) && snapshot.get(id) !== json) {
+              const entity = this._attachMethods(entityName, transformFromStorage(currentEntities.get(id), this.schemas[entityName]));
+              const oldEntity = this._attachMethods(entityName, transformFromStorage(snapshotEntities.get(id), this.schemas[entityName]));
+              await callback(entity, oldEntity);
+            }
+          }
+        } else if (eventType === "delete") {
+          for (const [id] of snapshot) {
+            if (stopped)
+              return;
+            if (!currentMap.has(id)) {
+              const oldEntity = this._attachMethods(entityName, transformFromStorage(snapshotEntities.get(id), this.schemas[entityName]));
+              await callback(oldEntity);
+            }
+          }
+        }
+        snapshot = currentMap;
+        snapshotEntities = currentEntities;
+      }
+      if (!stopped)
+        setTimeout(poll, interval);
+    };
+    setTimeout(poll, interval);
+    return () => {
+      stopped = true;
+    };
+  }
   insert(entityName, data) {
     const schema = this.schemas[entityName];
     const validatedData = asZodObject(schema).passthrough().parse(data);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sqlite-zod-orm",
-  "version": "3.5.2",
+  "version": "3.6.1",
   "description": "Type-safe SQLite ORM for Bun — Zod schemas, fluent queries, auto relationships, zero SQL",
   "type": "module",
   "main": "./dist/index.js",

package/src/database.ts

@@ -59,8 +59,11 @@ class _Database<Schemas extends SchemaMap> {
                 upsert: (conditions, data) => this.upsert(entityName, data, conditions),
                 delete: (id) => this.delete(entityName, id),
                 select: (...cols: string[]) => this._createQueryBuilder(entityName, cols),
-                on: (callback: (row: any) => void | Promise<void>, options?: { interval?: number }) =>
-                    this._createOnStream(entityName, callback, options?.interval),
+                on: (event: string, callback: (...args: any[]) => void | Promise<void>, options?: { interval?: number }) => {
+                    if (event === 'insert') return this._createOnStream(entityName, callback, options?.interval);
+                    if (event === 'update' || event === 'delete') return this._createChangeStream(entityName, event, callback, options?.interval);
+                    throw new Error(`Unknown event type: '${event}'. Supported: 'insert', 'update', 'delete'`);
+                },
                 _tableName: entityName,
             };
             (this as any)[key] = accessor;
@@ -201,6 +204,92 @@ class _Database<Schemas extends SchemaMap> {
         return () => { stopped = true; };
     }
 
+    /**
+     * Stream specific mutations (update or delete) with natural callback signatures.
+     *
+     * Maintains a full snapshot and diffs on each poll.
+     * Only fires for the subscribed event type.
+     *
+     * - 'update': callback(row, oldRow)
+     * - 'delete': callback(row)
+     */
+    public _createChangeStream(
+        entityName: string,
+        eventType: 'update' | 'delete',
+        callback: (...args: any[]) => void | Promise<void>,
+        intervalOverride?: number,
+    ): () => void {
+        const interval = intervalOverride ?? this.pollInterval;
+
+        // Build initial snapshot: Map<id, serialized row>
+        const allRows = this.db.query(`SELECT * FROM "${entityName}" ORDER BY id ASC`).all() as any[];
+        let snapshot = new Map<number, string>();
+        let snapshotEntities = new Map<number, any>();
+        for (const row of allRows) {
+            snapshot.set(row.id, JSON.stringify(row));
+            snapshotEntities.set(row.id, row);
+        }
+
+        let lastRevision: string = this._getRevision(entityName);
+        let stopped = false;
+
+        const poll = async () => {
+            if (stopped) return;
+
+            const currentRevision = this._getRevision(entityName);
+            if (currentRevision !== lastRevision) {
+                lastRevision = currentRevision;
+
+                const currentRows = this.db.query(`SELECT * FROM "${entityName}" ORDER BY id ASC`).all() as any[];
+                const currentMap = new Map<number, string>();
+                const currentEntities = new Map<number, any>();
+                for (const row of currentRows) {
+                    currentMap.set(row.id, JSON.stringify(row));
+                    currentEntities.set(row.id, row);
+                }
+
+                if (eventType === 'update') {
+                    // Detect updates: existing rows whose JSON changed
+                    for (const [id, json] of currentMap) {
+                        if (stopped) return;
+                        if (snapshot.has(id) && snapshot.get(id) !== json) {
+                            const entity = this._attachMethods(
+                                entityName,
+                                transformFromStorage(currentEntities.get(id), this.schemas[entityName]!)
+                            );
+                            const oldEntity = this._attachMethods(
+                                entityName,
+                                transformFromStorage(snapshotEntities.get(id), this.schemas[entityName]!)
+                            );
+                            await callback(entity, oldEntity);
+                        }
+                    }
+                } else if (eventType === 'delete') {
+                    // Detect deletes: rows in snapshot but not in current
+                    for (const [id] of snapshot) {
+                        if (stopped) return;
+                        if (!currentMap.has(id)) {
+                            const oldEntity = this._attachMethods(
+                                entityName,
+                                transformFromStorage(snapshotEntities.get(id), this.schemas[entityName]!)
+                            );
+                            await callback(oldEntity);
+                        }
+                    }
+                }
+
+                snapshot = currentMap;
+                snapshotEntities = currentEntities;
+            }
+
+            if (!stopped) setTimeout(poll, interval);
+        };
+
+        setTimeout(poll, interval);
+
+        return () => { stopped = true; };
+    }
+
     // ===========================================================================
     // CRUD
     // ===========================================================================

package/src/types.ts

@@ -45,6 +45,7 @@ export type Relationship = {
     foreignKey: string;
 };
 
+
 // =============================================================================
 // Type Helpers
 // =============================================================================
@@ -149,12 +150,20 @@ export type NavEntityAccessor<
     delete: (id: number) => void;
     select: (...cols: (keyof z.infer<S[Table & keyof S]> & string)[]) => QueryBuilder<NavEntity<S, R, Table>>;
     /**
-     * Stream new rows one at a time, in insertion order.
-     * Only emits rows inserted AFTER subscription starts.
+     * Listen for table events.
+     *
+     * `'insert'` — streams new rows, one at a time.
+     * `'update'` — fires on row changes with (newRow, oldRow).
+     * `'delete'` — fires when a row is removed.
+     *
      * Callbacks are awaited — strict ordering guaranteed even with async handlers.
      * @returns Unsubscribe function.
      */
-    on: (callback: (row: NavEntity<S, R, Table>) => void | Promise<void>, options?: { interval?: number }) => () => void;
+    on: {
+        (event: 'insert', callback: (row: NavEntity<S, R, Table>) => void | Promise<void>, options?: { interval?: number }): () => void;
+        (event: 'update', callback: (row: NavEntity<S, R, Table>, oldRow: NavEntity<S, R, Table>) => void | Promise<void>, options?: { interval?: number }): () => void;
+        (event: 'delete', callback: (row: NavEntity<S, R, Table>) => void | Promise<void>, options?: { interval?: number }): () => void;
+    };
     _tableName: string;
     readonly _schema?: S[Table & keyof S];
 };
@@ -181,12 +190,20 @@ export type EntityAccessor<S extends z.ZodType<any>> = {
     delete: (id: number) => void;
     select: (...cols: (keyof InferSchema<S> & string)[]) => QueryBuilder<AugmentedEntity<S>>;
     /**
-     * Stream new rows one at a time, in insertion order.
-     * Only emits rows inserted AFTER subscription starts.
+     * Listen for table events.
+     *
+     * `'insert'` — streams new rows, one at a time.
+     * `'update'` — fires on row changes with (newRow, oldRow).
+     * `'delete'` — fires when a row is removed.
+     *
      * Callbacks are awaited — strict ordering guaranteed even with async handlers.
      * @returns Unsubscribe function.
      */
-    on: (callback: (row: AugmentedEntity<S>) => void | Promise<void>, options?: { interval?: number }) => () => void;
+    on: {
+        (event: 'insert', callback: (row: AugmentedEntity<S>) => void | Promise<void>, options?: { interval?: number }): () => void;
+        (event: 'update', callback: (row: AugmentedEntity<S>, oldRow: AugmentedEntity<S>) => void | Promise<void>, options?: { interval?: number }): () => void;
+        (event: 'delete', callback: (row: AugmentedEntity<S>) => void | Promise<void>, options?: { interval?: number }): () => void;
+    };
     _tableName: string;
     readonly _schema?: S;
 };