feat: add stripe portal

Author: G3root

src/trpc/routers/billing-router/procedures/stripe-portal.ts

@@ -0,0 +1,39 @@
+import { env } from "@/env";
+import { invariant } from "@/lib/error";
+import { checkMembership } from "@/server/auth";
+import { createOrRetrieveCustomer, stripe } from "@/server/stripe";
+import { withAuth } from "@/trpc/api/trpc";
+
+export const stripePortalProcedure = withAuth.mutation(async ({ ctx }) => {
+  const { db, session } = ctx;
+
+  const { url } = await db.$transaction(async (tx) => {
+    const { companyId } = await checkMembership({ session, tx });
+
+    let customer: string;
+    try {
+      customer = await createOrRetrieveCustomer({
+        tx,
+        companyId,
+        email: "",
+      });
+    } catch (_error) {
+      throw new Error("Unable to access customer record.");
+    }
+
+    invariant(customer, "Could not get customer.");
+
+    try {
+      const { url } = await stripe.billingPortal.sessions.create({
+        customer,
+        return_url: `${env.NEXT_PUBLIC_BASE_URL}/${session.user.companyPublicId}/settings/billing`,
+      });
+
+      return { url };
+    } catch (_error) {
+      throw new Error("Could not create billing portal.");
+    }
+  });
+
+  return { url };
+});

src/trpc/routers/billing-router/router.ts

@@ -2,9 +2,11 @@ import { createTRPCRouter } from "@/trpc/api/trpc";
 import { checkoutProcedure } from "./procedures/checkout";
 import { getProductsProcedure } from "./procedures/get-products";
 import { getSubscriptionProcedure } from "./procedures/get-subscription";
+import { stripePortalProcedure } from "./procedures/stripe-portal";
 
 export const billingRouter = createTRPCRouter({
   getProducts: getProductsProcedure,
   getSubscription: getSubscriptionProcedure,
   checkout: checkoutProcedure,
+  stripePortal: stripePortalProcedure,
 });