const admin = require("firebase-admin");
const { BATCH_TYPE } = require("../config/types");

async function deleteFolder(path) {
  try {
    console.log(`Deleting folder: ${path}`);
    const bucket = admin.storage().bucket();
    await bucket.deleteFiles({ prefix: path, force: true });
    console.log(`Folder ${path} deleted successfully`);
  } catch (e) {
    console.log(e);
  }
}

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;