stripe downgrade

Author: thibaultleouay

packages/api/src/router/stripe/webhook.ts

@@ -3,10 +3,15 @@ import type Stripe from "stripe";
 import { z } from "zod";
 
 import { Events, setupAnalytics } from "@openstatus/analytics";
-import { eq } from "@openstatus/db";
+import { and, asc, eq, isNull } from "@openstatus/db";
 import {
+  invitation,
+  monitor,
+  notification,
+  page,
   selectWorkspaceSchema,
   user,
+  usersToWorkspaces,
   workspace,
 } from "@openstatus/db/src/schema";
 
@@ -213,19 +218,110 @@ export const webhookRouter = createTRPCRouter({
         subscriptionId: null,
         plan: "free",
         paidUntil: null,
+        limits: JSON.stringify(getLimits("free")),
       })
       .where(eq(workspace.stripeId, customerId))
       .returning();
 
-    const customer = await stripe.customers.retrieve(customerId);
-
-    if (!_workspace) {
+    if (!_workspace.length) {
       throw new TRPCError({
         code: "BAD_REQUEST",
         message: "Workspace not found",
       });
     }
 
+    const workspaceId = _workspace[0].id;
+
+    const activeMonitors = await opts.ctx.db
+      .select({ id: monitor.id })
+      .from(monitor)
+      .where(
+        and(
+          eq(monitor.workspaceId, workspaceId),
+          eq(monitor.active, true),
+          isNull(monitor.deletedAt),
+        ),
+      )
+      .orderBy(asc(monitor.createdAt));
+
+    for (const m of activeMonitors.slice(1)) {
+      await opts.ctx.db
+        .update(monitor)
+        .set({ active: false })
+        .where(eq(monitor.id, m.id))
+        .run();
+    }
+
+    const statusPages = await opts.ctx.db
+      .select({ id: page.id })
+      .from(page)
+      .where(eq(page.workspaceId, workspaceId))
+      .orderBy(asc(page.createdAt));
+
+    for (const p of statusPages.slice(1)) {
+      await opts.ctx.db.delete(page).where(eq(page.id, p.id)).run();
+    }
+
+    if (statusPages.length > 0) {
+      await opts.ctx.db
+        .update(page)
+        .set({
+          customDomain: "",
+          password: null,
+          accessType: "public",
+          authEmailDomains: null,
+        })
+        .where(eq(page.id, statusPages[0].id))
+        .run();
+    }
+
+    const notifications = await opts.ctx.db
+      .select({ id: notification.id, provider: notification.provider })
+      .from(notification)
+      .where(eq(notification.workspaceId, workspaceId))
+      .orderBy(asc(notification.createdAt));
+
+    const keepNotification =
+      notifications.find((n) => n.provider === "email") ?? notifications[0];
+
+    for (const n of notifications.filter(
+      (n) => n.id !== keepNotification?.id,
+    )) {
+      await opts.ctx.db
+        .delete(notification)
+        .where(eq(notification.id, n.id))
+        .run();
+    }
+
+    // Remove all non-owner members from the workspace
+    await opts.ctx.db
+      .delete(usersToWorkspaces)
+      .where(
+        and(
+          eq(usersToWorkspaces.workspaceId, workspaceId),
+          eq(usersToWorkspaces.role, "member"),
+        ),
+      )
+      .run();
+
+    await opts.ctx.db
+      .delete(usersToWorkspaces)
+      .where(
+        and(
+          eq(usersToWorkspaces.workspaceId, workspaceId),
+          eq(usersToWorkspaces.role, "admin"),
+        ),
+      )
+      .run();
+
+    // Remove all pending invitations for the workspace
+    await opts.ctx.db
+      .delete(invitation)
+      .where(eq(invitation.workspaceId, workspaceId))
+      .run();
+
+    const customer = await stripe.customers.retrieve(customerId);
+
     if (!customer.deleted && customer.email) {
       const userResult = await opts.ctx.db
         .select()
@@ -237,7 +333,7 @@ export const webhookRouter = createTRPCRouter({
       const analytics = await setupAnalytics({
         userId: `usr_${userResult.id}`,
         email: customer.email || undefined,
-        workspaceId: String(_workspace[0].id),
+        workspaceId: String(workspaceId),
         plan: "free",
       });
       await analytics.track(Events.DowngradeWorkspace);