Add reactivateProgram (dubinc#3764)

Author: steven-tey

…e)/api/cron/partners/reactivate/route.ts …e)/api/cron/programs/reactivate/route.tsapps/web/app/(ee)/api/cron/partners/reactivate/route.ts renamed to apps/web/app/(ee)/api/cron/programs/reactivate/route.ts apps/web/app/(ee)/api/cron/partners/reactivate/route.ts renamed to apps/web/app/(ee)/api/cron/programs/reactivate/route.ts

@@ -11,7 +11,7 @@ const inputSchema = z.object({
   programId: z.string(),
 });
 
-// POST /api/cron/partners/reactivate - reactivate partners in a program
+// POST /api/cron/programs/reactivate - reactivate all partners in a program
 export const POST = withCron(async ({ rawBody }) => {
   const { programId } = inputSchema.parse(JSON.parse(rawBody));
 
@@ -69,7 +69,7 @@ export const POST = withCron(async ({ rawBody }) => {
     // Self-queue the next batch if there are more partners to process
     if (programEnrollments.length === CRON_BATCH_SIZE) {
       const response = await qstash.publishJSON({
-        url: `${APP_DOMAIN_WITH_NGROK}/api/cron/partners/reactivate`,
+        url: `${APP_DOMAIN_WITH_NGROK}/api/cron/programs/reactivate`,
         body: {
           programId,
         },

apps/web/app/(ee)/api/stripe/webhook/checkout-session-completed.ts

@@ -1,7 +1,9 @@
 import { createProgram } from "@/lib/actions/partners/create-program";
 import { claimDotLinkDomain } from "@/lib/api/domains/claim-dot-link-domain";
+import { reactivateProgram } from "@/lib/api/programs/reactivate-program";
 import { onboardingStepCache } from "@/lib/api/workspaces/onboarding-step-cache";
 import { tokenCache } from "@/lib/auth/token-cache";
+import { wouldGainPartnerAccess } from "@/lib/plans/has-partner-access";
 import { stripe } from "@/lib/stripe";
 import { WorkspaceProps } from "@/lib/types";
 import { redis } from "@/lib/upstash";
@@ -54,7 +56,7 @@ export async function checkoutSessionCompleted(
   // in the database for easy identification in future webhook events
   // also update the billingCycleStart to today's date
 
-  const workspace = await prisma.project.update({
+  const updatedWorkspace = await prisma.project.update({
     where: {
       id: workspaceId,
     },
@@ -102,14 +104,21 @@ export async function checkoutSessionCompleted(
     },
   });
 
-  const users = workspace.users.map(({ user }) => ({
+  const users = updatedWorkspace.users.map(({ user }) => ({
     id: user.id,
     name: user.name,
     email: user.email,
   }));
 
   await Promise.allSettled([
     completeOnboarding({ users, workspaceId }),
+    // if workspace had a program from before and is upgrading to an eligible plan, reactivate it
+    updatedWorkspace.defaultProgramId &&
+      wouldGainPartnerAccess({
+        currentPlan: "free",
+        newPlan: updatedWorkspace.plan,
+      }) &&
+      reactivateProgram(updatedWorkspace.defaultProgramId),
     sendBatchEmail(
       users.map((user) => ({
         to: user.email as string,
@@ -135,7 +144,9 @@ export async function checkoutSessionCompleted(
     }),
     // expire tokens cache
     tokenCache.expireMany({
-      hashedKeys: workspace.restrictedTokens.map(({ hashedKey }) => hashedKey),
+      hashedKeys: updatedWorkspace.restrictedTokens.map(
+        ({ hashedKey }) => hashedKey,
+      ),
     }),
   ]);
 

apps/web/app/(ee)/api/stripe/webhook/utils/update-workspace-plan.ts

@@ -1,7 +1,7 @@
 import { deleteWorkspaceFolders } from "@/lib/api/folders/delete-workspace-folders";
 import { deactivateProgram } from "@/lib/api/programs/deactivate-program";
+import { reactivateProgram } from "@/lib/api/programs/reactivate-program";
 import { tokenCache } from "@/lib/auth/token-cache";
-import { qstash } from "@/lib/cron";
 import { syncUserPlanToPlain } from "@/lib/plain/sync-user-plan";
 import { getPlanCapabilities } from "@/lib/plan-capabilities";
 import {
@@ -11,7 +11,7 @@ import {
 import { WorkspaceProps } from "@/lib/types";
 import { webhookCache } from "@/lib/webhook/cache";
 import { prisma } from "@dub/prisma";
-import { APP_DOMAIN_WITH_NGROK, getPlanAndTierFromPriceId } from "@dub/utils";
+import { getPlanAndTierFromPriceId } from "@dub/utils";
 import { NEW_BUSINESS_PRICE_IDS } from "@dub/utils/src";
 import { waitUntil } from "@vercel/functions";
 
@@ -187,17 +187,7 @@ export async function updateWorkspacePlan({
       }) &&
       workspace.defaultProgramId
     ) {
-      const response = await qstash.publishJSON({
-        url: `${APP_DOMAIN_WITH_NGROK}/api/cron/partners/reactivate`,
-        body: {
-          programId: workspace.defaultProgramId,
-        },
-        deduplicationId: `reactivate-program-${workspace.defaultProgramId}`,
-      });
-
-      console.log("Reactivation job enqueued.", {
-        response,
-      });
+      await reactivateProgram(workspace.defaultProgramId);
     }
 
     if (

apps/web/lib/api/programs/reactivate-program.ts

@@ -0,0 +1,47 @@
+import { qstash } from "@/lib/cron";
+import { DEFAULT_PARTNER_GROUP } from "@/lib/zod/schemas/groups";
+import { prisma } from "@dub/prisma";
+import { APP_DOMAIN_WITH_NGROK } from "@dub/utils";
+
+export async function reactivateProgram(programId: string) {
+  if (!programId) {
+    throw new Error("[reactivateProgram] programId is required");
+  }
+
+  await prisma.$transaction([
+    prisma.program.update({
+      where: {
+        id: programId,
+      },
+      data: {
+        deactivatedAt: null,
+      },
+    }),
+
+    // republish the default group
+    prisma.partnerGroup.update({
+      where: {
+        programId_slug: {
+          programId,
+          slug: DEFAULT_PARTNER_GROUP.slug,
+        },
+      },
+      data: {
+        applicationFormPublishedAt: new Date(),
+        landerPublishedAt: new Date(),
+      },
+    }),
+  ]);
+
+  const response = await qstash.publishJSON({
+    url: `${APP_DOMAIN_WITH_NGROK}/api/cron/programs/reactivate`,
+    body: {
+      programId,
+    },
+    deduplicationId: `reactivate-program-${programId}`,
+  });
+
+  console.log("[reactivateProgram] Reactivation job enqueued.", { response });
+
+  return response;
+}