Screenity
- ().create({
transformer: superjson,
diff --git a/src/server/auth.ts b/src/server/auth.ts
index 9fd14eb..d98f7f0 100644
--- a/src/server/auth.ts
+++ b/src/server/auth.ts
@@ -20,15 +20,15 @@ declare module "next-auth" {
interface Session extends DefaultSession {
user: {
id: string;
+ stripeSubscriptionStatus: string;
// ...other properties
// role: UserRole;
} & DefaultSession["user"];
}
- // interface User {
- // // ...other properties
- // // role: UserRole;
- // }
+ interface User {
+ stripeSubscriptionStatus: string;
+ }
}
/**
@@ -42,6 +42,7 @@ export const authOptions: NextAuthOptions = {
...session,
user: {
...session.user,
+ stripeSubscriptionStatus: user.stripeSubscriptionStatus,
id: user.id,
},
}),
diff --git a/src/server/stripe-webhook-handlers.ts b/src/server/stripe-webhook-handlers.ts
new file mode 100644
index 0000000..aae52ba
--- /dev/null
+++ b/src/server/stripe-webhook-handlers.ts
@@ -0,0 +1,123 @@
+import type { PrismaClient } from "@prisma/client";
+import type Stripe from "stripe";
+
+// retrieves a Stripe customer id for a given user if it exists or creates a new one
+export const getOrCreateStripeCustomerIdForUser = async ({
+ stripe,
+ prisma,
+ userId,
+}: {
+ stripe: Stripe;
+ prisma: PrismaClient;
+ userId: string;
+}): Promise => {
+ const user = await prisma.user.findUnique({
+ where: {
+ id: userId,
+ },
+ });
+
+ if (!user) throw new Error("User not found");
+
+ if (user.stripeCustomerId) {
+ return user.stripeCustomerId;
+ }
+
+ // create a new customer
+ const customer = await stripe.customers.create({
+ email: user.email ?? undefined,
+ name: user.name ?? undefined,
+ // use metadata to link this Stripe customer to internal user id
+ metadata: {
+ userId,
+ },
+ });
+
+ // update with new customer id
+ const updatedUser = await prisma.user.update({
+ where: {
+ id: userId,
+ },
+ data: {
+ stripeCustomerId: customer.id,
+ },
+ });
+
+ if (updatedUser.stripeCustomerId) {
+ return updatedUser.stripeCustomerId;
+ }
+
+ return null;
+};
+
+export const handleInvoicePaid = async ({
+ event,
+ stripe,
+ prisma,
+}: {
+ event: Stripe.Event;
+ stripe: Stripe;
+ prisma: PrismaClient;
+}) => {
+ const invoice = event.data.object as Stripe.Invoice;
+ const subscriptionId = invoice.subscription;
+ const subscription = await stripe.subscriptions.retrieve(
+ subscriptionId as string
+ );
+ const userId = subscription.metadata.userId;
+
+ // update user with subscription data
+ await prisma.user.update({
+ where: {
+ id: userId,
+ },
+ data: {
+ stripeSubscriptionId: subscription.id,
+ stripeSubscriptionStatus: subscription.status,
+ },
+ });
+};
+
+export const handleSubscriptionCreatedOrUpdated = async ({
+ event,
+ prisma,
+}: {
+ event: Stripe.Event;
+ prisma: PrismaClient;
+}) => {
+ const subscription = event.data.object as Stripe.Subscription;
+ const userId = subscription.metadata.userId;
+
+ // update user with subscription data
+ await prisma.user.update({
+ where: {
+ id: userId,
+ },
+ data: {
+ stripeSubscriptionId: subscription.id,
+ stripeSubscriptionStatus: subscription.status,
+ },
+ });
+};
+
+export const handleSubscriptionCanceled = async ({
+ event,
+ prisma,
+}: {
+ event: Stripe.Event;
+ prisma: PrismaClient;
+}) => {
+ const subscription = event.data.object as Stripe.Subscription;
+ const userId = subscription.metadata.userId;
+
+ // remove subscription data from user
+ await prisma.user.update({
+ where: {
+ id: userId,
+ },
+ data: {
+ stripeSubscriptionId: null,
+ stripeSubscriptionStatus: null,
+ },
+ });
+};
diff --git a/src/server/stripe.ts b/src/server/stripe.ts
new file mode 100644
index 0000000..ad68bf7
--- /dev/null
+++ b/src/server/stripe.ts
@@ -0,0 +1,5 @@
+import Stripe from "stripe";
+import { env } from "~/env.mjs";
+export const stripe = new Stripe(env.STRIPE_SECRET_KEY, {
+ apiVersion: "2022-11-15",
+});
+
+
+ )}
-
diff --git a/src/server/api/root.ts b/src/server/api/root.ts
index 9cc2ab5..1f6bea9 100644
--- a/src/server/api/root.ts
+++ b/src/server/api/root.ts
@@ -1,6 +1,7 @@
import { createTRPCRouter } from "~/server/api/trpc";
import { exampleRouter } from "~/server/api/routers/example";
import { videoRouter } from "~/server/api/routers/video";
+import { stripeRouter } from "~/server/api/routers/stripe";
/**
* This is the primary router for your server.
@@ -10,6 +11,7 @@ import { videoRouter } from "~/server/api/routers/video";
export const appRouter = createTRPCRouter({
example: exampleRouter,
video: videoRouter,
+ stripe: stripeRouter,
});
// export type definition of API
diff --git a/src/server/api/routers/stripe.ts b/src/server/api/routers/stripe.ts
new file mode 100644
index 0000000..2caa590
--- /dev/null
+++ b/src/server/api/routers/stripe.ts
@@ -0,0 +1,80 @@
+import { env } from "~/env.mjs";
+import { getOrCreateStripeCustomerIdForUser } from "~/server/stripe-webhook-handlers";
+import { createTRPCRouter, protectedProcedure } from "~/server/api/trpc";
+
+export const stripeRouter = createTRPCRouter({
+ createCheckoutSession: protectedProcedure.mutation(async ({ ctx }) => {
+ const { stripe, session, prisma, req } = ctx;
+
+ const customerId = await getOrCreateStripeCustomerIdForUser({
+ prisma,
+ stripe,
+ userId: session.user?.id,
+ });
+
+ if (!customerId) {
+ throw new Error("Could not create customer");
+ }
+
+ const baseUrl =
+ env.NODE_ENV === "development"
+ ? `http://${req.headers.host ?? "localhost:3000"}`
+ : `https://${req.headers.host ?? env.NEXTAUTH_URL}`;
+
+ const checkoutSession = await stripe.checkout.sessions.create({
+ customer: customerId,
+ client_reference_id: session.user?.id,
+ payment_method_types: ["card"],
+ mode: "subscription",
+ line_items: [
+ {
+ price: env.STRIPE_PRICE_ID,
+ quantity: 1,
+ },
+ ],
+ success_url: `${baseUrl}/videos?checkoutSuccess=true`,
+ cancel_url: `${baseUrl}/videos?checkoutCanceled=true`,
+ subscription_data: {
+ metadata: {
+ userId: session.user?.id,
+ },
+ },
+ });
+
+ if (!checkoutSession) {
+ throw new Error("Could not create checkout session");
+ }
+
+ return { checkoutUrl: checkoutSession.url };
+ }),
+ createBillingPortalSession: protectedProcedure.mutation(async ({ ctx }) => {
+ const { stripe, session, prisma, req } = ctx;
+
+ const customerId = await getOrCreateStripeCustomerIdForUser({
+ prisma,
+ stripe,
+ userId: session.user?.id,
+ });
+
+ if (!customerId) {
+ throw new Error("Could not create customer");
+ }
+
+ const baseUrl =
+ env.NODE_ENV === "development"
+ ? `http://${req.headers.host ?? "localhost:3000"}`
+ : `https://${req.headers.host ?? env.NEXTAUTH_URL}`;
+
+ const stripeBillingPortalSession =
+ await stripe.billingPortal.sessions.create({
+ customer: customerId,
+ return_url: `${baseUrl}/videos`,
+ });
+
+ if (!stripeBillingPortalSession) {
+ throw new Error("Could not create billing portal session");
+ }
+
+ return { billingPortalUrl: stripeBillingPortalSession.url };
+ }),
+});
diff --git a/src/server/api/trpc.ts b/src/server/api/trpc.ts
index 7a5fc7c..9224b68 100644
--- a/src/server/api/trpc.ts
+++ b/src/server/api/trpc.ts
@@ -22,6 +22,8 @@ import { prisma } from "~/server/db";
type CreateContextOptions = {
session: Session | null;
+ req: NextApiRequest;
+ res: NextApiResponse;
};
/**
@@ -39,6 +41,9 @@ const createInnerTRPCContext = (opts: CreateContextOptions) => {
session: opts.session,
prisma,
s3,
+ stripe,
+ req: opts.req,
+ res: opts.res,
};
};
@@ -56,6 +61,8 @@ export const createTRPCContext = async (opts: CreateNextContextOptions) => {
return createInnerTRPCContext({
session,
+ req,
+ res,
});
};
@@ -70,6 +77,8 @@ import { initTRPC, TRPCError } from "@trpc/server";
import superjson from "superjson";
import { ZodError } from "zod";
import { s3 } from "~/server/aws/s3";
+import { stripe } from "~/server/stripe";
+import { type NextApiRequest, type NextApiResponse } from "next";
const t = initTRPC.context
- {videos &&
- videos.map(({ title, id, createdAt }) => (
-
+ )}
+ {videos &&
+ videos.map(({ title, id, createdAt }) => (
+
+ {videos && videos?.length <= 0 ? (
+
- You do not have any recordings.
-
- ) : null}
-
+ You do not have any recordings.
+
+ ) : null}
+