Save applications and create enrollments when logged in

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"

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")}
+        />
+      )}
+    </>
+  );
+}

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>