tenantpilot/app/api/webhooks/stripe/route.ts

import { db } from "@/lib/db/index";
import { stripe } from "@/lib/stripe/index";
import { headers } from "next/headers";
import type Stripe from "stripe";
import { subscriptions } from "@/lib/db/schema/subscriptions";
import { eq } from "drizzle-orm";

export async function POST(request: Request) {
  const body = await request.text();
  const headersList = await headers();
  const signature = headersList.get("Stripe-Signature") ?? "";

  let event: Stripe.Event;

  try {
    event = stripe.webhooks.constructEvent(
      body,
      signature,
      process.env.STRIPE_WEBHOOK_SECRET || ""
    );
    console.log(event.type);
  } catch (err) {
    return new Response(
      `Webhook Error: ${err instanceof Error ? err.message : "Unknown Error"}`,
      { status: 400 }
    );
  }

  const session = event.data.object as Stripe.Checkout.Session;
  // console.log("this is the session metadata -> ", session);

  if (!session?.metadata?.userId && session.customer == null) {
    console.error("session customer", session.customer);
    console.error("no metadata for userid");
    return new Response(null, {
      status: 200,
    });
  }

  if (event.type === "checkout.session.completed") {
    const subscription = await stripe.subscriptions.retrieve(
      session.subscription as string
    ) as Stripe.Subscription;
    
    const subscriptionItem = subscription.items.data[0];
    const updatedData = {
      stripeSubscriptionId: subscription.id,
      stripeCustomerId: subscription.customer as string,
      stripePriceId: subscriptionItem.price.id,
      stripeCurrentPeriodEnd: new Date(subscriptionItem.current_period_end * 1000),
    };

    if (session?.metadata?.userId != null) {
      const [sub] = await db
        .select()
        .from(subscriptions)
        .where(eq(subscriptions.userId, session.metadata.userId));
      if (sub != undefined) {
        await db
          .update(subscriptions)
          .set(updatedData)
          .where(eq(subscriptions.userId, sub.userId!));
      } else {
        await db
          .insert(subscriptions)
          .values({ ...updatedData, userId: session.metadata.userId });
      }

    } else if (
      typeof session.customer === "string" &&
      session.customer != null
    ) {
      await db
        .update(subscriptions)
        .set(updatedData)
        .where(eq(subscriptions.stripeCustomerId, session.customer));

    }
  }

  if (event.type === "invoice.payment_succeeded") {
    // Retrieve the subscription details from Stripe.
    const subscription = await stripe.subscriptions.retrieve(
      session.subscription as string
    ) as Stripe.Subscription;

    // Update the price id and set the new period end.
    const subscriptionItem = subscription.items.data[0];
    await db
      .update(subscriptions)
      .set({
        stripePriceId: subscriptionItem.price.id,
        stripeCurrentPeriodEnd: new Date(
          subscriptionItem.current_period_end * 1000
        ),
      })
      .where(eq(subscriptions.stripeSubscriptionId, subscription.id));

  }

  return new Response(null, { status: 200 });
}