import { z } from 'zod';

import { PlanSchema, ProductSchema } from '@kit/billing';
import { resolveProductPlan } from '@kit/billing-gateway';
import {
  BillingPortalCard,
  CurrentLifetimeOrderCard,
  CurrentSubscriptionCard,
} from '@kit/billing-gateway/components';
import { AppBreadcrumbs } from '@kit/ui/app-breadcrumbs';
import { If } from '@kit/ui/if';
import { PageBody } from '@kit/ui/page';
import { Trans } from '@kit/ui/trans';

import billingConfig from '~/config/billing.config';
import { createI18nServerInstance } from '~/lib/i18n/i18n.server';
import { withI18n } from '~/lib/i18n/with-i18n';
import { requireUserInServerComponent } from '~/lib/server/require-user-in-server-component';

// local imports
import { HomeLayoutPageHeader } from '../_components/home-page-header';
import { createPersonalAccountBillingPortalSession } from '../billing/_lib/server/server-actions';
import { PersonalAccountCheckoutForm } from './_components/personal-account-checkout-form';
import { loadPersonalAccountBillingPageData } from './_lib/server/personal-account-billing-page.loader';

export const generateMetadata = async () => {
  const i18n = await createI18nServerInstance();
  const title = i18n.t('account:billingTab');

  return {
    title,
  };
};

async function PersonalAccountBillingPage() {
  const user = await requireUserInServerComponent();

  const [subscription, order, customerId] =
    await loadPersonalAccountBillingPageData(user.id);

  const subscriptionProductPlan = subscription
    ? await getProductPlan(
        subscription.items[0]?.variant_id,
        subscription.currency,
      )
    : undefined;

  const orderProductPlan = order
    ? await getProductPlan(order.items[0]?.variant_id, order.currency)
    : undefined;

  const hasBillingData = subscription || order;

  return (
    <>
      <HomeLayoutPageHeader
        title={<Trans i18nKey={'common:routes.billing'} />}
        description={<AppBreadcrumbs />}
      />

      <PageBody>
        <div className={'flex flex-col space-y-4'}>
          <If condition={!hasBillingData}>
            <PersonalAccountCheckoutForm customerId={customerId} />

            <If condition={customerId}>
              <CustomerBillingPortalForm />
            </If>
          </If>

          <If condition={hasBillingData}>
            <div className={'flex w-full max-w-2xl flex-col space-y-6'}>
              <If condition={subscription}>
                {(subscription) => {
                  return (
                    <CurrentSubscriptionCard
                      subscription={subscription}
                      product={subscriptionProductPlan!.product}
                      plan={subscriptionProductPlan!.plan}
                    />
                  );
                }}
              </If>

              <If condition={order}>
                {(order) => {
                  return (
                    <CurrentLifetimeOrderCard
                      order={order}
                      product={orderProductPlan!.product}
                      plan={orderProductPlan!.plan}
                    />
                  );
                }}
              </If>
            </div>
          </If>

          <If condition={customerId}>
            <CustomerBillingPortalForm />
          </If>
        </div>
      </PageBody>
    </>
  );
}

export default withI18n(PersonalAccountBillingPage);

function CustomerBillingPortalForm() {
  return (
    <form action={createPersonalAccountBillingPortalSession}>
      <BillingPortalCard />
    </form>
  );
}

async function getProductPlan(
  variantId: string | undefined,
  currency: string,
): Promise<
  | {
      product: ProductSchema;
      plan: z.infer<typeof PlanSchema>;
    }
  | undefined
> {
  if (!variantId) {
    return undefined;
  }

  return resolveProductPlan(billingConfig, variantId, currency);
}
