From ed0cfb8875c764af2efcf4d1a29c50ac6209c311 Mon Sep 17 00:00:00 2001
From: Tim Wilson <tim@twilson.net>
Date: Fri, 15 Nov 2024 12:43:19 -0500
Subject: [PATCH] Save applications and create enrollments when logged in

---
 .../apply/[programSlug]/application/form.tsx  |  10 +-
 .../application/success/cta-button.tsx        |  23 --
 .../application/success/cta-buttons.tsx       |  35 +++
 .../application/success/page.tsx              |  52 ++--
 .../partners/create-program-application.ts    | 227 ++++++++++--------
 .../lib/actions/partners/onboard-partner.ts   |  13 +-
 apps/web/lib/auth/options.ts                  |   4 +
 .../partners/complete-program-applications.ts | 102 ++++++++
 apps/web/prisma/schema/partner.prisma         |   1 -
 apps/web/prisma/schema/program.prisma         |  25 +-
 10 files changed, 322 insertions(+), 170 deletions(-)
 delete mode 100644 apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/success/cta-button.tsx
 create mode 100644 apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/success/cta-buttons.tsx
 create mode 100644 apps/web/lib/partners/complete-program-applications.ts

diff --git a/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/form.tsx b/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/form.tsx
index e94fc33abd..d1b661641c 100644
--- a/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/form.tsx
+++ b/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/form.tsx
@@ -50,7 +50,8 @@ export function ProgramApplicationForm({
 
         if (!response?.data?.ok) {
           toast.error(
-            response?.data?.message ?? "Failed to submit application",
+            (response?.data as { message?: string }).message ??
+              "Failed to submit application",
           );
           setError("root.serverError", {
             message: "Failed to submit application. Please try again.",
@@ -60,7 +61,12 @@ export function ProgramApplicationForm({
 
         toast.success("Application submitted successfully");
         router.push(
-          `/apply/${program.slug}/application/success?applicationId=${response.data.programApplicationId}`,
+          `/apply/${program.slug}/application/success?${new URLSearchParams({
+            applicationId: response.data.programApplicationId,
+            ...("programEnrollmentId" in response.data && {
+              enrollmentId: response.data.programEnrollmentId,
+            }),
+          }).toString()}`,
         );
       })}
       className="flex flex-col gap-6"
diff --git a/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/success/cta-button.tsx b/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/success/cta-button.tsx
deleted file mode 100644
index 21021f7c47..0000000000
--- a/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/success/cta-button.tsx
+++ /dev/null
@@ -1,23 +0,0 @@
-"use client";
-
-import { Button } from "@dub/ui";
-import { useSession } from "next-auth/react";
-import { useRouter } from "next/navigation";
-
-export function CTAButton() {
-  const router = useRouter();
-  const { data: session } = useSession();
-
-  return (
-    <Button
-      type="button"
-      text={
-        session?.user
-          ? "Continue to dashboard"
-          : "Create your Dub Partner account"
-      }
-      className="border-[var(--brand)] bg-[var(--brand)] hover:bg-[var(--brand)] hover:ring-[var(--brand-ring)]"
-      onClick={() => router.push("/register")}
-    />
-  );
-}
diff --git a/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/success/cta-buttons.tsx b/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/success/cta-buttons.tsx
new file mode 100644
index 0000000000..2b3b5b099c
--- /dev/null
+++ b/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/success/cta-buttons.tsx
@@ -0,0 +1,35 @@
+"use client";
+
+import { Button } from "@dub/ui";
+import { useSession } from "next-auth/react";
+import { useRouter } from "next/navigation";
+
+export function CTAButtons() {
+  const router = useRouter();
+  const { data: session, status } = useSession();
+
+  return (
+    <>
+      <Button
+        type="button"
+        text={
+          status === "authenticated"
+            ? "Continue to Dub Partners"
+            : "Create your Dub Partners account"
+        }
+        className="border-[var(--brand)] bg-[var(--brand)] hover:bg-[var(--brand)] hover:ring-[var(--brand-ring)]"
+        onClick={() =>
+          router.push(status === "authenticated" ? "/" : "/register")
+        }
+      />
+      {status === "unauthenticated" && (
+        <Button
+          type="button"
+          variant="secondary"
+          text="Log in to Dub Partners"
+          onClick={() => router.push("/login")}
+        />
+      )}
+    </>
+  );
+}
diff --git a/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/success/page.tsx b/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/success/page.tsx
index f86a94a913..0e57301676 100644
--- a/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/success/page.tsx
+++ b/apps/web/app/partners.dub.co/(apply)/apply/[programSlug]/application/success/page.tsx
@@ -7,7 +7,7 @@ import { Store } from "lucide-react";
 import { notFound } from "next/navigation";
 import { CSSProperties } from "react";
 import { Header } from "../../header";
-import { CTAButton } from "./cta-button";
+import { CTAButtons } from "./cta-buttons";
 import { Screenshot } from "./screenshot";
 
 const FEATURES = [
@@ -42,7 +42,7 @@ export default async function SuccessPage({
   searchParams,
 }: {
   params: { programSlug: string };
-  searchParams: { applicationId?: string };
+  searchParams: { applicationId?: string; enrollmentId?: string };
 }) {
   const { programSlug } = params;
   const program = await prisma.program.findUnique({
@@ -64,6 +64,8 @@ export default async function SuccessPage({
 
   if (!program) notFound();
 
+  const isEnrolled = !!searchParams.enrollmentId;
+
   const application = searchParams.applicationId
     ? await prisma.programApplication.findFirst({
         where: {
@@ -97,24 +99,32 @@ export default async function SuccessPage({
       />
       <div className="p-6">
         <div className="grid grid-cols-1 gap-5 sm:pt-20">
-          <h1 className="text-4xl font-semibold">Application submitted</h1>
+          <h1 className="text-4xl font-semibold">
+            Application {isEnrolled ? "submitted" : "saved"}
+          </h1>
           <div className="flex flex-col gap-4 text-base text-neutral-700">
-            <p>
-              Your application has been submitted for review.
-              {application && (
-                <>
-                  {" "}
-                  You'll receive an update at{" "}
-                  <strong className="font-semibold">{application.email}</strong>
-                  .
-                </>
-              )}
-            </p>
-            <p>
-              While you're waiting, complete your account setup on{" "}
-              <strong className="font-semibold">Dub Partners</strong>, which
-              powers this program.
-            </p>
+            {isEnrolled && (
+              <p>
+                Your application has been submitted for review.
+                {application && (
+                  <>
+                    {" "}
+                    You'll receive an update at{" "}
+                    <strong className="font-semibold">
+                      {application.email}
+                    </strong>
+                    .
+                  </>
+                )}
+              </p>
+            )}
+            {!isEnrolled && (
+              <p>
+                Complete your account setup on{" "}
+                <strong className="font-semibold">Dub Partners</strong> to
+                finish submitting your application.
+              </p>
+            )}
           </div>
         </div>
 
@@ -149,8 +159,8 @@ export default async function SuccessPage({
           ))}
         </div>
 
-        <div className="mt-12">
-          <CTAButton />
+        <div className="mt-12 flex flex-col gap-3">
+          <CTAButtons />
         </div>
       </div>
     </div>
diff --git a/apps/web/lib/actions/partners/create-program-application.ts b/apps/web/lib/actions/partners/create-program-application.ts
index 2fdd4cfaa9..44313a1fc1 100644
--- a/apps/web/lib/actions/partners/create-program-application.ts
+++ b/apps/web/lib/actions/partners/create-program-application.ts
@@ -4,7 +4,9 @@ import { createId, getIP } from "@/lib/api/utils";
 import { getSession } from "@/lib/auth";
 import { prisma } from "@/lib/prisma";
 import { ratelimit } from "@/lib/upstash";
-import { Partner, ProgramApplicationStatus } from "@prisma/client";
+import { Partner, Program, ProgramEnrollment } from "@prisma/client";
+import { addDays } from "date-fns";
+import { cookies } from "next/headers";
 import z from "../../zod";
 import { actionClient } from "../safe-action";
 
@@ -17,115 +19,132 @@ const createProgramApplicationSchema = z.object({
   comments: z.string().optional(),
 });
 
-// Onboard a new partner
+// Create a program application (or enrollment if a partner is already logged in)
 export const createProgramApplicationAction = actionClient
   .schema(createProgramApplicationSchema)
-  .action(async ({ parsedInput }) => {
-    const { programId, ...data } = parsedInput;
-
-    // Limit to 3 requests per minute per program per IP
-    const { success } = await ratelimit(3, "1 m").limit(
-      `create-program-application:${programId}:${getIP()}`,
-    );
-
-    if (!success)
-      return {
-        ok: false,
-        message: "Too many requests. Please try again later.",
-      };
-
-    try {
-      const program = await prisma.program.findUnique({
-        where: { id: programId },
-      });
-
-      if (!program) return { ok: false, message: "Program not found." };
-
-      // Find existing partner to automatically associate
-      const existingPartner = await getExistingPartner({
-        email: data.email,
-        programId: program.id,
-      });
-
-      const existingApplication = await prisma.programApplication.findFirst({
-        where: {
-          programId: program.id,
-          ...(existingPartner
-            ? { partnerId: existingPartner.id }
-            : {
-                email: data.email,
-              }),
-          status: {
-            not: ProgramApplicationStatus.rejected,
-          },
-        },
-      });
-
-      if (existingApplication)
+  .action(
+    async ({
+      parsedInput,
+    }): Promise<
+      | { ok: false; message?: string }
+      | { ok: true; programApplicationId: string }
+      | { ok: true; programApplicationId: string; programEnrollmentId: string }
+    > => {
+      const { programId, ...data } = parsedInput;
+
+      // Limit to 3 requests per minute per program per IP
+      const { success } = await ratelimit(3, "1 m").limit(
+        `create-program-application:${programId}:${getIP()}`,
+      );
+
+      if (!success)
         return {
           ok: false,
-          message: "You have already applied to this program.",
+          message: "Too many requests. Please try again later.",
         };
 
-      const application = await prisma.programApplication.create({
-        data: {
-          ...parsedInput,
-          id: createId({ prefix: "pga_" }),
-          programId: program.id,
-          partnerId: existingPartner?.id,
-        },
-      });
-
-      return { ok: true, programApplicationId: application.id };
-    } catch (e) {
-      console.error(e);
-      return {
-        ok: false,
-      };
-    }
-  });
+      try {
+        const program = await prisma.program.findUnique({
+          where: { id: programId },
+        });
+
+        if (!program) return { ok: false, message: "Program not found." };
 
-async function getExistingPartner({
-  email,
-  programId,
-}: {
-  email: string;
-  programId: string;
-}) {
-  const session = await getSession();
-
-  let partner: Partner | null = null;
-
-  // First try any currently logged in user
-  if (session?.user.id)
-    partner = await prisma.partner.findFirst({
-      where: {
-        users: { some: { userId: session.user.id } },
-        programs: { none: { programId } },
-      },
-    });
-
-  // Then try to find a matching partner by user email
-  if (!partner)
-    partner = await prisma.partner.findFirst({
-      where: {
-        users: {
-          some: {
-            user: {
-              email,
-              emailVerified: {
-                not: null,
+        const session = await getSession();
+
+        // Get currently logged in partner
+        const existingPartner = session?.user.id
+          ? await prisma.partner.findFirst({
+              where: {
+                users: { some: { userId: session.user.id } },
+              },
+              include: {
+                programs: true,
               },
-            },
-          },
-        },
-        programs: {
-          none: {
-            programId,
-          },
-        },
-      },
-    });
-
-  return partner;
+            })
+          : null;
+
+        if (existingPartner)
+          return createApplicationAndEnrollment(
+            existingPartner,
+            program,
+            parsedInput,
+          );
+        else return createApplication(program, parsedInput);
+      } catch (e) {
+        console.error(e);
+        return {
+          ok: false,
+        };
+      }
+    },
+  );
+
+async function createApplicationAndEnrollment(
+  partner: Partner & { programs: ProgramEnrollment[] },
+  program: Program,
+  data: z.infer<typeof createProgramApplicationSchema>,
+): Promise<
+  | { ok: false; message?: string }
+  | { ok: true; programApplicationId: string; programEnrollmentId: string }
+> {
+  // Check if ProgramEnrollment already exists
+  if (partner.programs.some((p) => p.id === program.id))
+    return {
+      ok: false,
+      message: "You have already applied to this program.",
+    };
+
+  const application = await prisma.programApplication.create({
+    data: {
+      ...data,
+      id: createId({ prefix: "pga_" }),
+      programId: program.id,
+    },
+  });
+
+  const enrollment = await prisma.programEnrollment.create({
+    data: {
+      id: createId({ prefix: "pge_" }),
+      partnerId: partner.id,
+      programId: program.id,
+      status: "pending",
+      applicationId: application.id,
+    },
+  });
+
+  return {
+    ok: true,
+    programApplicationId: application.id,
+    programEnrollmentId: enrollment.id,
+  };
+}
+
+async function createApplication(
+  program: Program,
+  data: z.infer<typeof createProgramApplicationSchema>,
+): Promise<
+  { ok: false; message?: string } | { ok: true; programApplicationId: string }
+> {
+  const application = await prisma.programApplication.create({
+    data: {
+      ...data,
+      id: createId({ prefix: "pga_" }),
+      programId: program.id,
+    },
+  });
+
+  // Add application ID to cookie
+  const cookieStore = await cookies();
+  const existingApplicationIds =
+    cookieStore.get("programApplicationIds")?.value?.split(",") || [];
+  cookieStore.set(
+    "programApplicationIds",
+    [...existingApplicationIds, application.id].join(","),
+    {
+      httpOnly: true,
+      expires: addDays(new Date(), 1),
+    },
+  );
+  return { ok: true, programApplicationId: application.id };
 }
diff --git a/apps/web/lib/actions/partners/onboard-partner.ts b/apps/web/lib/actions/partners/onboard-partner.ts
index d62fa32e9f..d0577730f6 100644
--- a/apps/web/lib/actions/partners/onboard-partner.ts
+++ b/apps/web/lib/actions/partners/onboard-partner.ts
@@ -3,6 +3,7 @@
 import { createId } from "@/lib/api/utils";
 import { createDotsUser } from "@/lib/dots/create-dots-user";
 import { userIsInBeta } from "@/lib/edge-config";
+import { completeProgramApplications } from "@/lib/partners/complete-program-applications";
 import { prisma } from "@/lib/prisma";
 import { storage } from "@/lib/storage";
 import { onboardPartnerSchema } from "@/lib/zod/schemas/partners";
@@ -66,6 +67,7 @@ export const onboardPartnerAction = authUserActionClient
       if (programInvite) {
         await prisma.programEnrollment.create({
           data: {
+            id: createId({ prefix: "pge_" }),
             programId: programInvite.programId,
             linkId: programInvite.linkId,
             partnerId: partner.id,
@@ -78,15 +80,8 @@ export const onboardPartnerAction = authUserActionClient
         });
       }
 
-      // Associate any matching program applications with the partner
-      await prisma.programApplication.updateMany({
-        where: {
-          email: user.email,
-          partnerId: null,
-          programId: { not: programInvite?.id },
-        },
-        data: { partnerId: partner.id },
-      });
+      // Complete any outstanding program applications
+      await completeProgramApplications(user.id);
 
       // Create the Dots user with DOTS_DEFAULT_APP_ID
       const [firstName, lastName] = name.split(" ");
diff --git a/apps/web/lib/auth/options.ts b/apps/web/lib/auth/options.ts
index 765af3ceac..c3a742384b 100644
--- a/apps/web/lib/auth/options.ts
+++ b/apps/web/lib/auth/options.ts
@@ -17,6 +17,7 @@ import CredentialsProvider from "next-auth/providers/credentials";
 import EmailProvider from "next-auth/providers/email";
 import GithubProvider from "next-auth/providers/github";
 import GoogleProvider from "next-auth/providers/google";
+import { completeProgramApplications } from "../partners/complete-program-applications";
 import {
   exceededLoginAttemptsThreshold,
   incrementLoginAttempts,
@@ -479,6 +480,9 @@ export const authOptions: NextAuthOptions = {
           })(),
         );
       }
+
+      // Complete any outstanding program applications
+      await completeProgramApplications(message.user.id);
     },
   },
 };
diff --git a/apps/web/lib/partners/complete-program-applications.ts b/apps/web/lib/partners/complete-program-applications.ts
new file mode 100644
index 0000000000..ca704cf1da
--- /dev/null
+++ b/apps/web/lib/partners/complete-program-applications.ts
@@ -0,0 +1,102 @@
+import { ProgramApplicationStatus } from "@prisma/client";
+import { cookies } from "next/headers";
+import { createId } from "../api/utils";
+import { prisma } from "../prisma";
+import { ratelimit } from "../upstash";
+
+/**
+ * Completes any outstanding program applications for a user
+ * by creating a program enrollment for each
+ */
+export async function completeProgramApplications(userId: string) {
+  try {
+    const cookieStore = await cookies();
+    const programApplicationIds = cookieStore
+      .get("programApplicationIds")
+      ?.value?.split(",");
+    if (!programApplicationIds?.length) return;
+
+    // Prevent brute forcing
+    const { success } = await ratelimit(3, "1 m").limit(
+      `complete-program-applications:${userId}`,
+    );
+
+    if (!success) {
+      console.warn("Not completing program applications due to rate limiting", {
+        userId,
+      });
+      return;
+    }
+
+    const user = await prisma.user.findUniqueOrThrow({
+      where: { id: userId },
+      select: {
+        partners: {
+          select: {
+            partnerId: true,
+            partner: {
+              select: {
+                programs: {
+                  select: {
+                    programId: true,
+                  },
+                },
+              },
+            },
+          },
+        },
+      },
+    });
+
+    if (!user.partners.length) return;
+
+    let programApplications = await prisma.programApplication.findMany({
+      where: {
+        id: { in: programApplicationIds.filter(Boolean) },
+
+        // Exclude any applications for programs the user is already enrolled in
+        programId: {
+          notIn: user.partners
+            .map((p) => p.partner.programs.map((pp) => pp.programId))
+            .flat(),
+        },
+      },
+      orderBy: {
+        createdAt: "desc",
+      },
+    });
+
+    if (!programApplications.length) return;
+
+    // Filter out duplicate program applications
+    let seenIds = new Set<string>();
+    programApplications = programApplications.filter(({ programId }) => {
+      if (seenIds.has(programId)) return false;
+      seenIds.add(programId);
+      return true;
+    });
+
+    console.log("Completing program applications", {
+      userId,
+      partnerId: user.partners[0].partnerId,
+      ids: programApplications.map(({ id }) => id),
+    });
+
+    await prisma.programEnrollment.createMany({
+      data: programApplications.map((programApplication) => ({
+        id: createId({ prefix: "pge_" }),
+        programId: programApplication.programId,
+        partnerId: user.partners[0].partnerId,
+        applicationId: programApplication.id,
+        status: ProgramApplicationStatus.pending,
+      })),
+    });
+
+    cookieStore.delete("programApplicationIds");
+  } catch (error) {
+    console.error(
+      "Failed to complete program applications from cookies",
+      error,
+    );
+  }
+}
diff --git a/apps/web/prisma/schema/partner.prisma b/apps/web/prisma/schema/partner.prisma
index 5b5c9c3500..818004fd1d 100644
--- a/apps/web/prisma/schema/partner.prisma
+++ b/apps/web/prisma/schema/partner.prisma
@@ -26,7 +26,6 @@ model Partner {
   invites  PartnerInvite[]
   payouts  Payout[]
   sales    Sale[]
-  applications ProgramApplication[]
 }
 
 model PartnerUser {
diff --git a/apps/web/prisma/schema/program.prisma b/apps/web/prisma/schema/program.prisma
index 19f2ae1547..6af318d126 100644
--- a/apps/web/prisma/schema/program.prisma
+++ b/apps/web/prisma/schema/program.prisma
@@ -58,12 +58,16 @@ model Program {
 }
 
 model ProgramEnrollment {
-  id        String                  @id @default(cuid())
-  partnerId String
-  programId String
-  linkId    String?                 @unique
-  dotsUserId String?                @unique
-  status    ProgramEnrollmentStatus @default(pending)
+  id            String  @id @default(cuid())
+  partnerId     String
+  programId     String
+
+  applicationId String? @unique
+  linkId        String? @unique
+  dotsUserId    String? @unique
+
+  status ProgramEnrollmentStatus @default(pending)
+
   createdAt DateTime                @default(now())
   updatedAt DateTime                @updatedAt
 
@@ -71,9 +75,12 @@ model ProgramEnrollment {
   program Program @relation(fields: [programId], references: [id])
   link    Link?   @relation(fields: [linkId], references: [id])
 
+  application ProgramApplication? @relation(fields: [applicationId], references: [id])
+
   @@unique([partnerId, programId])
   @@index([programId])
   @@index([linkId])
+  @@index([applicationId])
 }
 
 enum ProgramApplicationStatus {
@@ -85,7 +92,6 @@ enum ProgramApplicationStatus {
 model ProgramApplication {
   id String @id @default(cuid())
   programId String
-  partnerId String?
 
   name     String
   email    String
@@ -97,11 +103,10 @@ model ProgramApplication {
   createdAt DateTime                 @default(now())
   updatedAt DateTime                 @updatedAt
 
-  program Program @relation(fields: [programId], references: [id])
-  partner Partner? @relation(fields: [partnerId], references: [id])
+  program    Program @relation(fields: [programId], references: [id])
+  enrollment ProgramEnrollment?
 
   @@index([programId])
-  @@index([partnerId])
 }
 
 // TODO: Haven't finalized the resource table yet.