functions

functions/helpers/firebase.js

@@ -1,4 +1,5 @@
 const admin = require("firebase-admin");
+const { BATCH_TYPE } = require("../config/types");
 
 async function deleteFolder(path) {
   try {
@@ -11,4 +12,77 @@ async function deleteFolder(path) {
   }
 }
 
+async function batchFirestore({
+  path = "", // ADD only
+  docs = [], // not for ADD
+  data = {}, // not for DELETE
+  type = BATCH_TYPE.UPDATE,
+}) {
+  try {
+    if (docs?.length === 0 && type !== BATCH_TYPE.ADD) {
+      console.log(`Aucun document trouvé dans la collection ${path}.`);
+      return;
+    }
+
+    // Préparer des lots pour les opérations
+    const batches = [];
+    let currentBatch = admin.firestore().batch();
+    let operationCounter = 0;
+
+    docs.forEach((doc) => {
+      const ref =
+        doc?.ref ||
+        (typeof doc?.path === "string"
+          ? admin.firestore().doc(doc.path)
+          : null);
+      const payload =
+        doc && typeof doc.data === "object" && doc.data !== null && !Array.isArray(doc.data)
+          ? doc.data
+          : data;
+
+      if (type === BATCH_TYPE.ADD) {
+        // Pour ADD, créer un nouveau document avec ID automatique
+        const newDocRef = admin.firestore().collection(path).doc();
+        currentBatch.set(newDocRef, data);
+      } else if (type === BATCH_TYPE.UPDATE) {
+        if (!ref) {
+          throw new Error("batchFirestore UPDATE nécessite une référence de document.");
+        }
+        if (!payload || Object.keys(payload).length === 0) {
+          throw new Error("batchFirestore UPDATE nécessite des données à écrire.");
+        }
+        currentBatch.set(ref, payload, { merge: true });
+      } else if (type === BATCH_TYPE.DELETE) {
+        if (!ref) {
+          throw new Error("batchFirestore DELETE nécessite une référence de document.");
+        }
+        currentBatch.delete(ref);
+      } else {
+        throw new Error(`Opération non supportée : ${type}`);
+      }
+      operationCounter++;
+      // Si le lot atteint la limite de 500 opérations, le sauvegarder et en créer un nouveau
+      if (operationCounter === 500) {
+        batches.push(currentBatch);
+        currentBatch = admin.firestore().batch();
+        operationCounter = 0;
+      }
+    });
+
+    // Ajouter le dernier lot si des opérations y sont présentes
+    if (operationCounter > 0) {
+      batches.push(currentBatch);
+    }
+
+    // Exécuter tous les lots en parallèle
+    await Promise.all(batches.map((batch) => batch.commit()));
+  } catch (error) {
+    console.error(
+      `Erreur lors des opérations ${type} pour la collection ${path} :`,
+      error,
+    );
+  }
+}
+
+exports.batchFirestore = batchFirestore;
 exports.deleteFolder = deleteFolder;