@naturalcycles/firestore-lib 1.4.1 → 1.4.2

package/dist/firestore.db.d.ts

@@ -23,4 +23,5 @@ export declare class FirestoreDB extends BaseCommonDB implements CommonDB {
     private querySnapshotToArray;
     commitTransaction(tx: DBTransaction, _opt?: CommonDBSaveOptions): Promise<void>;
     ping(): Promise<void>;
+    getTables(): Promise<string[]>;
 }

package/dist/firestore.db.js

@@ -41,6 +41,13 @@ class FirestoreDB extends db_lib_1.BaseCommonDB {
     }
     // QUERY
     async runQuery(q, opt) {
+        const idFilter = q._filters.find(f => f.name === 'id');
+        if (idFilter) {
+            const ids = Array.isArray(idFilter.val) ? idFilter.val : [idFilter.val];
+            return {
+                rows: await this.getByIds(q.table, ids, opt),
+            };
+        }
         const firestoreQuery = (0, query_util_1.dbQueryToFirestoreQuery)(q, this.cfg.firestore.collection(q.table));
         let rows = await this.runFirestoreQuery(firestoreQuery, opt);
         // Special case when projection query didn't specify 'id'
@@ -85,8 +92,15 @@ class FirestoreDB extends db_lib_1.BaseCommonDB {
     }
     // DELETE
     async deleteByQuery(q, opt) {
-        const firestoreQuery = (0, query_util_1.dbQueryToFirestoreQuery)(q.select([]), this.cfg.firestore.collection(q.table));
-        const ids = (await this.runFirestoreQuery(firestoreQuery)).map(obj => obj.id);
+        let ids;
+        const idFilter = q._filters.find(f => f.name === 'id');
+        if (idFilter) {
+            ids = Array.isArray(idFilter.val) ? idFilter.val : [idFilter.val];
+        }
+        else {
+            const firestoreQuery = (0, query_util_1.dbQueryToFirestoreQuery)(q.select([]), this.cfg.firestore.collection(q.table));
+            ids = (await this.runFirestoreQuery(firestoreQuery)).map(obj => obj.id);
+        }
         await this.deleteByIds(q.table, ids, opt);
         return ids.length;
     }
@@ -133,5 +147,8 @@ class FirestoreDB extends db_lib_1.BaseCommonDB {
     async ping() {
         // no-op now
     }
+    async getTables() {
+        return [];
+    }
 }
 exports.FirestoreDB = FirestoreDB;

package/package.json

@@ -37,7 +37,7 @@
   "engines": {
     "node": ">=14.15.0"
   },
-  "version": "1.4.1",
+  "version": "1.4.2",
   "description": "Firestore implementation of CommonDB interface",
   "author": "Natural Cycles Team",
   "license": "MIT"

package/src/firestore.db.ts

@@ -45,7 +45,7 @@ export class FirestoreDB extends BaseCommonDB implements CommonDB {
   }
 
   // GET
-  override async getByIds<ROW extends ObjectWithId>(
+  async getByIds<ROW extends ObjectWithId>(
     table: string,
     ids: ROW['id'][],
     _opt?: FirestoreDBOptions,
@@ -77,6 +77,14 @@ export class FirestoreDB extends BaseCommonDB implements CommonDB {
     q: DBQuery<ROW>,
     opt?: FirestoreDBOptions,
   ): Promise<RunQueryResult<ROW>> {
+    const idFilter = q._filters.find(f => f.name === 'id')
+    if (idFilter) {
+      const ids: string[] = Array.isArray(idFilter.val) ? idFilter.val : [idFilter.val]
+      return {
+        rows: await this.getByIds(q.table, ids, opt),
+      }
+    }
+
     const firestoreQuery = dbQueryToFirestoreQuery(q, this.cfg.firestore.collection(q.table))
 
     let rows = await this.runFirestoreQuery<ROW>(firestoreQuery, opt)
@@ -165,18 +173,25 @@ export class FirestoreDB extends BaseCommonDB implements CommonDB {
     q: DBQuery<ROW>,
     opt?: FirestoreDBOptions,
   ): Promise<number> {
-    const firestoreQuery = dbQueryToFirestoreQuery(
-      q.select([]),
-      this.cfg.firestore.collection(q.table),
-    )
-    const ids = (await this.runFirestoreQuery<ROW>(firestoreQuery)).map(obj => obj.id)
+    let ids: ROW['id'][]
+
+    const idFilter = q._filters.find(f => f.name === 'id')
+    if (idFilter) {
+      ids = Array.isArray(idFilter.val) ? idFilter.val : [idFilter.val]
+    } else {
+      const firestoreQuery = dbQueryToFirestoreQuery(
+        q.select([]),
+        this.cfg.firestore.collection(q.table),
+      )
+      ids = (await this.runFirestoreQuery<ROW>(firestoreQuery)).map(obj => obj.id)
+    }
 
     await this.deleteByIds(q.table, ids, opt)
 
     return ids.length
   }
 
-  override async deleteByIds<ROW extends ObjectWithId>(
+  async deleteByIds<ROW extends ObjectWithId>(
     table: string,
     ids: ROW['id'][],
     _opt?: FirestoreDBOptions,
@@ -233,4 +248,8 @@ export class FirestoreDB extends BaseCommonDB implements CommonDB {
   override async ping(): Promise<void> {
     // no-op now
   }
+
+  override async getTables(): Promise<string[]> {
+    return []
+  }
 }