feat: storage and token limiting

lib/auth.ts

@@ -1,6 +1,7 @@
 import config from "@/lib/config"
 import { createUserDefaults } from "@/models/defaults"
 import { getSelfHostedUser, getUserByEmail } from "@/models/users"
+import { stripe } from "@better-auth/stripe"
 import { User } from "@prisma/client"
 import { betterAuth } from "better-auth"
 import { prismaAdapter } from "better-auth/adapters/prisma"
@@ -11,6 +12,7 @@ import { headers } from "next/headers"
 import { redirect } from "next/navigation"
 import { prisma } from "./db"
 import { resend, sendOTPCodeEmail } from "./email"
+import { isStripeEnabled, stripeClient } from "./stripe"
 
 export type UserProfile = {
   id: string
@@ -18,6 +20,7 @@ export type UserProfile = {
   email: string
   avatar?: string
   storageUsed?: number
+  tokenBalance?: number
 }
 
 export const auth = betterAuth({
@@ -65,6 +68,13 @@ export const auth = betterAuth({
         await sendOTPCodeEmail({ email, otp })
       },
     }),
+    isStripeEnabled(stripeClient)
+      ? stripe({
+          stripeClient: stripeClient!,
+          stripeWebhookSecret: config.stripe.webhookSecret,
+          createCustomerOnSignUp: true,
+        })
+      : { id: "stripe", endpoints: {} },
     nextCookies(), // make sure this is the last plugin in the array
   ],
 })