🛂 Add setCustomPlan script

packages/scripts/setCustomPlan.ts

@@ -0,0 +1,83 @@
+import { Plan, PrismaClient } from 'db'
+import Stripe from 'stripe'
+
+const prisma = new PrismaClient()
+
+export const setCustomPlan = async () => {
+  if (
+    !process.env.STRIPE_SECRET_KEY ||
+    !process.env.STRIPE_PRODUCT_ID ||
+    !process.env.STRIPE_SUBSCRIPTION_ID ||
+    !process.env.WORKSPACE_ID
+  )
+    throw Error(
+      'STRIPE_SECRET_KEY or STRIPE_SUBSCRIPTION_ID or STRIPE_PRODUCT_ID or process.env.WORKSPACE_ID var is missing'
+    )
+  const stripe = new Stripe(process.env.STRIPE_SECRET_KEY, {
+    apiVersion: '2022-08-01',
+  })
+
+  const claimablePlan = await prisma.claimableCustomPlan.findFirst({
+    where: { workspaceId: process.env.WORKSPACE_ID, claimedAt: null },
+  })
+
+  if (!claimablePlan) throw Error('No claimable plan found')
+
+  console.log('Claimable plan found')
+
+  const { items: existingItems } = await stripe.subscriptions.retrieve(
+    process.env.STRIPE_SUBSCRIPTION_ID
+  )
+  if (existingItems.data.length === 0) return
+
+  const planItem = existingItems.data.find(
+    (item) => item.plan.product === process.env.STRIPE_PRODUCT_ID
+  )
+
+  if (!planItem) throw Error("Couldn't find plan item")
+
+  console.log('Updating subscription...')
+
+  await stripe.subscriptions.update(process.env.STRIPE_SUBSCRIPTION_ID, {
+    items: [
+      {
+        id: planItem.id,
+        price_data: {
+          currency: 'usd',
+          tax_behavior: 'exclusive',
+          recurring: { interval: 'month' },
+          product: process.env.STRIPE_PRODUCT_ID,
+          unit_amount: claimablePlan.price * 100,
+        },
+      },
+      ...existingItems.data
+        .filter((item) => item.plan.product !== process.env.STRIPE_PRODUCT_ID)
+        .map((item) => ({ id: item.id, deleted: true })),
+    ],
+  })
+
+  console.log('Subscription updated!')
+
+  console.log('Updating workspace...')
+
+  await prisma.workspace.update({
+    where: { id: process.env.WORKSPACE_ID },
+    data: {
+      plan: Plan.CUSTOM,
+      customChatsLimit: claimablePlan.chatsLimit,
+      customSeatsLimit: claimablePlan.seatsLimit,
+      customStorageLimit: claimablePlan.storageLimit,
+    },
+  })
+
+  console.log('Workspace updated!')
+
+  console.log('Updating claimable plan...')
+
+  await prisma.claimableCustomPlan.update({
+    where: { id: claimablePlan.id },
+    data: { claimedAt: new Date() },
+  })
+
+  console.log('Claimable plan updated!')
+}