import { OrderStatus } from "@polar-sh/sdk/models/components/orderstatus.js";
import {
  type CreateCheckoutSessionArgs,
  type FetchCustomerPortalUrlArgs,
  type PaymentProcessor,
} from "../paymentProcessor";
import { getPaymentProcessorPlanId } from "../paymentProcessorPlans";
import {
  fetchUserPaymentProcessorUserId,
  updateUserPaymentProcessorUserId,
} from "../user";
import {
  createPolarCheckoutSession,
  ensurePolarCustomer,
} from "./checkoutUtils";
import { polarClient } from "./polarClient";
import { polarMiddlewareConfigFn, polarWebhook } from "./webhook";

export const polarPaymentProcessor: PaymentProcessor = {
  id: "polar",
  createCheckoutSession: async ({
    userId,
    userEmail,
    paymentPlan,
    prismaUserDelegate,
  }: CreateCheckoutSessionArgs) => {
    const customer = await ensurePolarCustomer(userId, userEmail);

    await updateUserPaymentProcessorUserId(
      { userId, paymentProcessorUserId: customer.id },
      prismaUserDelegate,
    );

    const checkoutSession = await createPolarCheckoutSession({
      productId: getPaymentProcessorPlanId(paymentPlan),
      customerId: customer.id,
    });

    return {
      session: {
        id: checkoutSession.id,
        url: checkoutSession.url,
      },
    };
  },
  fetchCustomerPortalUrl: async ({
    userId,
    prismaUserDelegate,
  }: FetchCustomerPortalUrlArgs) => {
    const paymentProcessorUserId = await fetchUserPaymentProcessorUserId(
      userId,
      prismaUserDelegate,
    );

    if (!paymentProcessorUserId) {
      return null;
    }

    const customerSession = await polarClient.customerSessions.create({
      customerId: paymentProcessorUserId,
    });

    return customerSession.customerPortalUrl;
  },
  webhook: polarWebhook,
  webhookMiddlewareConfigFn: polarMiddlewareConfigFn,
  fetchTotalRevenue: async () => {
    let totalRevenue = 0;

    const result = await polarClient.orders.list({
      limit: 100,
    });

    for await (const page of result) {
      const orders = page.result.items || [];

      for (const order of orders) {
        if (order.status === OrderStatus.Paid && order.totalAmount > 0) {
          totalRevenue += order.totalAmount;
        }
      }
    }

    // Revenue is in cents so we convert to dollars (or your main currency unit)
    return totalRevenue / 100;
  },
};