📄 Add Commercial License for ee folder (#1532)

2024-05-23 10:42:23 +02:00
parent 5680829906
commit 0eacbebbbe
246 changed files with 1472 additions and 1588 deletions
--- a/ee/packages/billing/api/updateSubscription.ts
+++ b/ee/packages/billing/api/updateSubscription.ts
@@ -0,0 +1,153 @@
+import { TRPCError } from '@trpc/server'
+import { env } from '@typebot.io/env'
+import prisma from '@typebot.io/lib/prisma'
+import { Plan } from '@typebot.io/prisma'
+import Stripe from 'stripe'
+import { trackEvents } from '@typebot.io/telemetry/trackEvents'
+import { isAdminWriteWorkspaceForbidden } from '@typebot.io/db-rules/isAdminWriteWorkspaceForbidden'
+import { User } from '@typebot.io/schemas'
+import { createCheckoutSessionUrl } from '../helpers/createCheckoutSessionUrl'
+
+type Props = {
+  workspaceId: string
+  user: Pick<User, 'email' | 'id'>
+  plan: 'STARTER' | 'PRO'
+  returnUrl: string
+  currency: 'usd' | 'eur'
+}
+
+export const updateSubscription = async ({
+  workspaceId,
+  user,
+  plan,
+  returnUrl,
+  currency,
+}: Props) => {
+  if (!env.STRIPE_SECRET_KEY)
+    throw new TRPCError({
+      code: 'INTERNAL_SERVER_ERROR',
+      message: 'Stripe environment variables are missing',
+    })
+  const workspace = await prisma.workspace.findFirst({
+    where: {
+      id: workspaceId,
+    },
+    select: {
+      isPastDue: true,
+      stripeId: true,
+      members: {
+        select: {
+          userId: true,
+          role: true,
+        },
+      },
+    },
+  })
+  if (workspace?.isPastDue)
+    throw new TRPCError({
+      code: 'BAD_REQUEST',
+      message:
+        'You have unpaid invoices. Please head over your billing portal to pay it.',
+    })
+  if (!workspace?.stripeId || isAdminWriteWorkspaceForbidden(workspace, user))
+    throw new TRPCError({
+      code: 'NOT_FOUND',
+      message: 'Workspace not found',
+    })
+
+  const stripe = new Stripe(env.STRIPE_SECRET_KEY, {
+    apiVersion: '2022-11-15',
+  })
+  const { data } = await stripe.subscriptions.list({
+    customer: workspace.stripeId,
+    limit: 1,
+    status: 'active',
+  })
+  const subscription = data[0] as Stripe.Subscription | undefined
+  const currentPlanItemId = subscription?.items.data.find((item) =>
+    [env.STRIPE_STARTER_PRICE_ID, env.STRIPE_PRO_PRICE_ID].includes(
+      item.price.id
+    )
+  )?.id
+  const currentUsageItemId = subscription?.items.data.find(
+    (item) =>
+      item.price.id === env.STRIPE_STARTER_CHATS_PRICE_ID ||
+      item.price.id === env.STRIPE_PRO_CHATS_PRICE_ID
+  )?.id
+
+  const items = [
+    {
+      id: currentPlanItemId,
+      price:
+        plan === Plan.STARTER
+          ? env.STRIPE_STARTER_PRICE_ID
+          : env.STRIPE_PRO_PRICE_ID,
+      quantity: 1,
+    },
+    {
+      id: currentUsageItemId,
+      price:
+        plan === Plan.STARTER
+          ? env.STRIPE_STARTER_CHATS_PRICE_ID
+          : env.STRIPE_PRO_CHATS_PRICE_ID,
+    },
+  ]
+
+  if (subscription) {
+    if (plan === 'STARTER') {
+      const totalChatsUsed = await prisma.result.count({
+        where: {
+          typebot: { workspaceId },
+          hasStarted: true,
+          createdAt: {
+            gte: new Date(subscription.current_period_start * 1000),
+          },
+        },
+      })
+      if (totalChatsUsed >= 4000) {
+        throw new TRPCError({
+          code: 'BAD_REQUEST',
+          message:
+            "You have collected more than 4000 chats during this billing cycle. You can't downgrade to the Starter.",
+        })
+      }
+    }
+
+    await stripe.subscriptions.update(subscription.id, {
+      items,
+      proration_behavior: 'always_invoice',
+    })
+  } else {
+    const checkoutUrl = await createCheckoutSessionUrl(stripe)({
+      customerId: workspace.stripeId,
+      userId: user.id,
+      workspaceId,
+      currency,
+      plan,
+      returnUrl,
+    })
+
+    return { checkoutUrl }
+  }
+
+  const updatedWorkspace = await prisma.workspace.update({
+    where: { id: workspaceId },
+    data: {
+      plan,
+      isQuarantined: false,
+    },
+  })
+
+  await trackEvents([
+    {
+      name: 'Subscription updated',
+      workspaceId,
+      userId: user.id,
+      data: {
+        plan,
+      },
+    },
+  ])
+
+  return { workspace: updatedWorkspace }
+}