musicland/functions/src/orders.js

const admin = require("firebase-admin");
const { FieldValue } = require("firebase-admin/firestore");
const { onDocumentCreated } = require("firebase-functions/firestore");
const { HttpsError, onCall } = require("firebase-functions/https");

const { REGION, ALERT_TYPE } = require("../index");
const { sendNotification } = require("./notifications");
const {
  ORDER_TYPES,
  ORDER_STATUS,
  ORDERS_COLLECTION,
  createOrderDocument,
  normalizeAmount,
} = require("./helpers/orders");

const USERS_COLLECTION = "users";

const formatCoinsText = (value) => {
  if (typeof value !== "number" || Number.isNaN(value)) {
    return null;
  }

  const absoluteValue = Math.abs(value);
  if (!Number.isFinite(absoluteValue)) {
    return null;
  }

  const formatted = Number.isInteger(absoluteValue)
    ? `${absoluteValue}`
    : absoluteValue.toFixed(2);
  const suffix = absoluteValue === 1 ? "crédit" : "crédits";

  return `${formatted} ${suffix}`;
};

const buildOrderNotificationContent = ({ amount, orderType, balanceAfter }) => {
  if (typeof amount !== "number" || Number.isNaN(amount) || amount === 0) {
    return null;
  }

  const coinsText = formatCoinsText(amount);
  if (!coinsText) {
    return null;
  }

  const balanceText = formatCoinsText(balanceAfter);
  const balanceSentence = balanceText
    ? ` Ton solde est maintenant de ${balanceText}.`
    : "";

  if (amount > 0) {
    if (orderType === ORDER_TYPES.COINS) {
      return {
        title: "Crédits achetés",
        message: `Ton achat de ${coinsText} est confirmé.${balanceSentence}`,
        action: "PURCHASED",
      };
    }

    if (orderType === ORDER_TYPES.GIFT) {
      return {
        title: "Crédits reçus",
        message: `Tu as reçu ${coinsText}.${balanceSentence}`,
        action: "EARNED",
      };
    }

    return {
      title: "Crédits ajoutés",
      message: `Ton solde augmente de ${coinsText}.${balanceSentence}`,
      action: "CREDITED",
    };
  }

  const reason =
    orderType === ORDER_TYPES.SONG ? " pour générer un nouveau son" : "";

  return {
    title: "Crédits dépensés",
    message: `Tu as dépensé ${coinsText}${reason}.${balanceSentence}`,
    action: "SPENT",
  };
};

const notifyOrderApplied = async ({
  userId,
  orderId,
  amount,
  orderType,
  balanceBefore,
  balanceAfter,
  metadata = {},
}) => {
  const content = buildOrderNotificationContent({
    amount,
    orderType,
    balanceAfter,
  });

  if (!content || !userId) {
    return;
  }

  try {
    await sendNotification({
      sender: "SYSTEM",
      receiver: userId,
      receiverCollection: USERS_COLLECTION,
      title: content.title,
      message: content.message,
      data: {
        type: ALERT_TYPE?.CREDITS_UPDATED || "CREDITS_UPDATED",
        orderId,
        orderType: orderType || null,
        amount,
        balanceBefore,
        balanceAfter,
        action: content.action,
        source:
          typeof metadata?.source === "string" ? metadata.source : null,
        metadata: metadata || {},
      },
    });
  } catch (error) {
    console.error(
      "[orders-onOrderCreated] Failed to send notification",
      orderId,
      error,
    );
  }
};

const onOrderCreated = onDocumentCreated(
  `${ORDERS_COLLECTION}/{orderId}`,
  async (event) => {
    const orderRef = event?.data?.ref;
    const orderData = event?.data?.data();

    if (!orderRef || !orderData) {
      return;
    }

    if (orderData?.processedAt) {
      return;
    }

    const userId =
      typeof orderData.userId === "string" ? orderData.userId.trim() : "";
    const amount = normalizeAmount(orderData.amount);

    if (!userId) {
      await orderRef.set(
        {
          status: ORDER_STATUS.REJECTED,
          processedAt: FieldValue.serverTimestamp(),
          failureReason: "USER_NOT_FOUND",
        },
        { merge: true },
      );
      return;
    }

    if (amount === null || amount === 0) {
      await orderRef.set(
        {
          status: ORDER_STATUS.REJECTED,
          processedAt: FieldValue.serverTimestamp(),
          failureReason: "INVALID_AMOUNT",
        },
        { merge: true },
      );
      return;
    }

    const userRef = admin.firestore().collection(USERS_COLLECTION).doc(userId);

    let notificationContext = null;

    try {
      await admin.firestore().runTransaction(async (transaction) => {
        const userSnapshot = await transaction.get(userRef);
        const userData = userSnapshot?.data() || {};
        const currentBalanceValue = normalizeAmount(userData?.coins);
        const currentBalance =
          currentBalanceValue !== null ? currentBalanceValue : 0;

        const nextBalance = currentBalance + amount;

        if (
          amount < 0 &&
          nextBalance < 0 &&
          orderData?.type === ORDER_TYPES.SONG
        ) {
          transaction.set(
            orderRef,
            {
              status: ORDER_STATUS.REJECTED,
              processedAt: FieldValue.serverTimestamp(),
              failureReason: "INSUFFICIENT_FUNDS",
              balanceBefore: currentBalance,
              balanceAfter: currentBalance,
            },
            { merge: true },
          );
          return;
        }

        if (userSnapshot?.exists) {
          transaction.update(userRef, {
            coins: nextBalance,
            updatedAt: FieldValue.serverTimestamp(),
          });
        } else {
          transaction.set(
            userRef,
            {
              coins: nextBalance,
              createdAt: FieldValue.serverTimestamp(),
              updatedAt: FieldValue.serverTimestamp(),
            },
            { merge: true },
          );
        }

        transaction.set(
          orderRef,
          {
            status: ORDER_STATUS.APPLIED,
            processedAt: FieldValue.serverTimestamp(),
            balanceBefore: currentBalance,
            balanceAfter: nextBalance,
          },
          { merge: true },
        );

        notificationContext = {
          balanceBefore: currentBalance,
          balanceAfter: nextBalance,
        };
      });
    } catch (error) {
      console.error(
        "[orders-onOrderCreated] Failed to process order",
        orderRef.id,
        error,
      );

      await orderRef.set(
        {
          status: ORDER_STATUS.REJECTED,
          processedAt: FieldValue.serverTimestamp(),
          failureReason: "PROCESSING_ERROR",
          errorMessage: error?.message || String(error),
        },
        { merge: true },
      );
      return;
    }

    if (notificationContext) {
      await notifyOrderApplied({
        userId,
        orderId: orderRef.id,
        amount,
        orderType:
          typeof orderData?.type === "string" ? orderData.type : null,
        balanceBefore: notificationContext.balanceBefore,
        balanceAfter: notificationContext.balanceAfter,
        metadata: orderData?.metadata || {},
      });
    }
  },
);

const createSongOrder = onCall({ region: REGION }, async (request) => {
  const { auth, data } = request || {};

  if (!auth?.uid) {
    throw new HttpsError("unauthenticated", "Authentification requise.");
  }

  const amount = normalizeAmount(data?.amount);

  if (amount === null || amount >= 0) {
    throw new HttpsError(
      "invalid-argument",
      "Le montant doit être négatif pour un achat de musique.",
    );
  }

  const songId =
    typeof data?.songId === "string" && data.songId.trim()
      ? data.songId.trim()
      : null;

  const userRef = admin.firestore().collection(USERS_COLLECTION).doc(auth.uid);
  const userSnapshot = await userRef.get();
  const currentCoinsValue = normalizeAmount(userSnapshot?.data()?.coins);
  const currentCoins =
    currentCoinsValue !== null ? currentCoinsValue : 0;

  if (currentCoins + amount < 0) {
    throw new HttpsError(
      "failed-precondition",
      "Crédits insuffisants pour finaliser l'opération.",
    );
  }

  const metadata = {
    source:
      typeof data?.source === "string" && data.source.trim()
        ? data.source.trim()
        : "music_generation",
  };

  if (typeof data?.requestId === "string" && data.requestId.trim()) {
    metadata.requestId = data.requestId.trim();
  }

  const { orderId } = await createOrderDocument({
    userId: auth.uid,
    type: ORDER_TYPES.SONG,
    amount,
    songId,
    createdBy: auth.uid,
    metadata,
  });

  return { orderId };
});

module.exports = {
  onOrderCreated,
  createSongOrder,
};