Improve markApplicationEventSubmitted

Author: steven-tey

apps/web/app/app.dub.co/(dashboard)/[slug]/(ee)/program/partners/applications/page-client.tsx

@@ -13,6 +13,7 @@ import { useBulkRejectPartnersModal } from "@/ui/modals/bulk-reject-partners-mod
 import { useRejectPartnerApplicationModal } from "@/ui/modals/reject-partner-application-modal";
 import { GroupColorCircle } from "@/ui/partners/groups/group-color-circle";
 import { PartnerApplicationSheet } from "@/ui/partners/partner-application-sheet";
+import { PartnerApplicationSource } from "@/ui/partners/partner-application-source";
 import { PartnerRowItem } from "@/ui/partners/partner-row-item";
 import { PartnerSocialColumn } from "@/ui/partners/partner-social-column";
 import { AnimatedEmptyState } from "@/ui/shared/animated-empty-state";
@@ -196,17 +197,16 @@ export function ProgramPartnersApplicationsPageClient() {
         header: "Applied",
         accessorFn: (d) => formatDate(d.createdAt, { month: "short" }),
       },
-      // TODO: add source column back once we fix application source display
-      // {
-      //   id: "source",
-      //   header: "Source",
-      //   minSize: 170,
-      //   cell: ({ row }) => (
-      //     <PartnerApplicationSource
-      //       referralSource={row.original.applicationEvent?.referralSource}
-      //     />
-      //   ),
-      // },
+      {
+        id: "source",
+        header: "Source",
+        minSize: 170,
+        cell: ({ row }) => (
+          <PartnerApplicationSource
+            referralSource={row.original.applicationEvent?.referralSource}
+          />
+        ),
+      },
       {
         id: "group",
         header: "Group",

apps/web/lib/actions/partners/create-program-application.ts

@@ -381,7 +381,9 @@ async function createApplicationAndEnrollment({
           },
         }),
 
-        markApplicationEventSubmitted(programEnrollment),
+        markApplicationEventSubmitted(programEnrollment, {
+          partnerNetworkStatus: partner.networkStatus,
+        }),
       ]);
     })(),
   );

apps/web/lib/application-events/update-application-event.ts

@@ -1,20 +1,20 @@
 import { prisma } from "@dub/prisma";
-import { Prisma } from "@dub/prisma/client";
+import { PartnerNetworkStatus, Prisma } from "@dub/prisma/client";
 import { cookies } from "next/headers";
 import { getApplicationEventCookieName } from "./utils";
 
 // Application events visited as a guest have partnerId: null — look up the
 // event via the browser cookie and backfill partnerId + submittedAt.
 // Fallback to the (programId, partnerId) lookup if no cookie is present (e.g. the
 // partner visited while already logged in on a different browser).
-export async function markApplicationEventSubmitted({
-  programId,
-  partnerId,
-  applicationId,
-}: Pick<
-  Prisma.ProgramEnrollmentCreateManyInput,
-  "programId" | "partnerId" | "applicationId"
->) {
+export async function markApplicationEventSubmitted(
+  programEnrollment: Pick<
+    Prisma.ProgramEnrollmentCreateManyInput,
+    "programId" | "partnerId" | "applicationId"
+  >,
+  { partnerNetworkStatus }: { partnerNetworkStatus: PartnerNetworkStatus },
+) {
+  const { programId, partnerId, applicationId } = programEnrollment;
   const cookieStore = await cookies();
   const cookieName = getApplicationEventCookieName(programId);
   const applicationEventId = cookieStore.get(cookieName)?.value;
@@ -25,21 +25,44 @@ export async function markApplicationEventSubmitted({
     return;
   }
 
+  const applicationEvent = await prisma.programApplicationEvent.findUnique({
+    where: {
+      ...(applicationEventId
+        ? { id: applicationEventId }
+        : { programId_partnerId: { programId, partnerId } }),
+    },
+  });
+
+  if (!applicationEvent) {
+    console.error(
+      "[markApplicationEventSubmitted]: No application event found, skipping...",
+    );
+    return;
+  }
+
   try {
-    await prisma.programApplicationEvent.updateMany({
+    await prisma.programApplicationEvent.update({
       where: {
-        ...(applicationEventId
-          ? { id: applicationEventId }
-          : { programId, partnerId }),
-        submittedAt: null,
+        id: applicationEvent.id,
       },
       data: {
-        partnerId,
         submittedAt: new Date(),
+        partnerId,
         programApplicationId: applicationId,
+        ...(applicationEvent.referralSource === "marketplace" &&
+        !["approved", "trusted"].includes(partnerNetworkStatus)
+          ? {
+              referralSource: "direct",
+            }
+          : {}),
       },
     });
-  } catch {}
+  } catch (error) {
+    console.error(
+      "[markApplicationEventSubmitted]: Error updating application event:",
+      error,
+    );
+  }
 }
 
 export async function trackApplicationEvents({

apps/web/lib/partners/complete-program-applications.ts

@@ -260,7 +260,9 @@ export async function completeProgramApplications(userEmail: string) {
 
     await Promise.allSettled(
       programEnrollments.map((programEnrollment) =>
-        markApplicationEventSubmitted(programEnrollment),
+        markApplicationEventSubmitted(programEnrollment, {
+          partnerNetworkStatus: partner.networkStatus,
+        }),
       ),
     );
   } catch (error) {

apps/web/ui/partners/partner-info-cards.tsx

@@ -45,6 +45,7 @@ import {
   PartnerFraudBanner,
 } from "./fraud-risks/partner-fraud-banner";
 import { PartnerFraudIndicator } from "./fraud-risks/partner-fraud-indicator";
+import { PartnerApplicationSource } from "./partner-application-source";
 import { PartnerAvatar } from "./partner-avatar";
 import { PartnerInfoGroup } from "./partner-info-group";
 import { PartnerStarButton } from "./partner-star-button";
@@ -175,16 +176,15 @@ export function PartnerInfoCards({
             >
               <span>Applied {formatDate(partner.createdAt)}</span>
             </TimestampTooltip>
-            {/* TODO: add source column back once we fix application source display */}
-            {/* {applicationReferralSource && (
+            {applicationReferralSource && (
               <>
                 <span>via</span>
                 <PartnerApplicationSource
                   referralSource={applicationReferralSource}
                   variant="inline"
                 />
               </>
-            )} */}
+            )}
           </span>
         ) : (
           `${isPendingApplication ? "Applied" : "Partner since"} ${formatDate(partner.createdAt)}`