Merge pull request #2810 from dubinc/add-missing-workflows

Add executeWorkflows to a few spots that were missed earlier

Author: steven-tey

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

@@ -341,7 +341,7 @@ export async function checkoutSessionCompleted(event: Stripe.Event) {
       livemode: event.livemode,
     });
 
-    const commission = await createPartnerCommission({
+    await createPartnerCommission({
       event: "sale",
       programId: link.programId,
       partnerId: link.partnerId,
@@ -362,15 +362,25 @@ export async function checkoutSessionCompleted(event: Stripe.Event) {
       },
     });
 
-    if (commission) {
-      waitUntil(
+    waitUntil(
+      Promise.allSettled([
         executeWorkflows({
           trigger: WorkflowTrigger.saleRecorded,
           programId: link.programId,
           partnerId: link.partnerId,
         }),
-      );
-    }
+        // same logic as lead.created webhook below:
+        // if the clickEvent variable exists and there was no existing customer before,
+        // we need to trigger the leadRecorded workflow
+        clickEvent &&
+          !existingCustomer &&
+          executeWorkflows({
+            trigger: WorkflowTrigger.leadRecorded,
+            programId: link.programId,
+            partnerId: link.partnerId,
+          }),
+      ]),
+    );
   }
 
   waitUntil(

apps/web/app/(ee)/api/stripe/integration/webhook/invoice-paid.ts

@@ -209,15 +209,13 @@ export async function invoicePaid(event: Stripe.Event) {
       },
     });
 
-    if (commission) {
-      waitUntil(
-        executeWorkflows({
-          trigger: WorkflowTrigger.saleRecorded,
-          programId: link.programId,
-          partnerId: link.partnerId,
-        }),
-      );
-    }
+    waitUntil(
+      executeWorkflows({
+        trigger: WorkflowTrigger.saleRecorded,
+        programId: link.programId,
+        partnerId: link.partnerId,
+      }),
+    );
   }
 
   // send workspace webhook

apps/web/lib/actions/partners/create-commission.ts

@@ -3,6 +3,7 @@
 import { isFirstConversion } from "@/lib/analytics/is-first-conversion";
 import { getDefaultProgramIdOrThrow } from "@/lib/api/programs/get-default-program-id-or-throw";
 import { getProgramEnrollmentOrThrow } from "@/lib/api/programs/get-program-enrollment-or-throw";
+import { executeWorkflows } from "@/lib/api/workflows/execute-workflows";
 import { createPartnerCommission } from "@/lib/partners/create-partner-commission";
 import { getLeadEvent } from "@/lib/tinybird";
 import { recordClick } from "@/lib/tinybird/record-click";
@@ -15,6 +16,7 @@ import { leadEventSchemaTB } from "@/lib/zod/schemas/leads";
 import { prisma } from "@dub/prisma";
 import { nanoid } from "@dub/utils";
 import { COUNTRIES_TO_CONTINENTS } from "@dub/utils/src";
+import { WorkflowTrigger } from "@prisma/client";
 import { waitUntil } from "@vercel/functions";
 import { authActionClient } from "../safe-action";
 
@@ -200,6 +202,12 @@ export const createCommissionAction = authActionClient
             },
           },
         }),
+
+        executeWorkflows({
+          trigger: WorkflowTrigger.leadRecorded,
+          programId,
+          partnerId,
+        }),
       ]);
     }
 
@@ -238,6 +246,12 @@ export const createCommissionAction = authActionClient
             },
           },
         }),
+
+        executeWorkflows({
+          trigger: WorkflowTrigger.saleRecorded,
+          programId,
+          partnerId,
+        }),
       ]);
     }
 

apps/web/lib/integrations/shopify/create-sale.ts

@@ -1,5 +1,6 @@
 import { isFirstConversion } from "@/lib/analytics/is-first-conversion";
 import { includeTags } from "@/lib/api/links/include-tags";
+import { executeWorkflows } from "@/lib/api/workflows/execute-workflows";
 import { createPartnerCommission } from "@/lib/partners/create-partner-commission";
 import { recordSale } from "@/lib/tinybird";
 import { LeadEventTB } from "@/lib/types";
@@ -8,6 +9,7 @@ import { sendWorkspaceWebhook } from "@/lib/webhook/publish";
 import { transformSaleEventData } from "@/lib/webhook/transform";
 import { prisma } from "@dub/prisma";
 import { nanoid } from "@dub/utils";
+import { WorkflowTrigger } from "@prisma/client";
 import { waitUntil } from "@vercel/functions";
 import { orderSchema } from "./schema";
 
@@ -154,5 +156,13 @@ export async function createShopifySale({
         },
       },
     });
+
+    waitUntil(
+      executeWorkflows({
+        trigger: WorkflowTrigger.saleRecorded,
+        programId: link.programId,
+        partnerId: link.partnerId,
+      }),
+    );
   }
 }

apps/web/lib/partners/create-partner-commission.ts

@@ -5,7 +5,7 @@ import {
   EventType,
   WorkflowTrigger,
 } from "@dub/prisma/client";
-import { log } from "@dub/utils";
+import { currencyFormatter, log } from "@dub/utils";
 import { waitUntil } from "@vercel/functions";
 import { differenceInMonths } from "date-fns";
 import { recordAuditLog } from "../api/audit-logs/record-audit-log";
@@ -238,6 +238,10 @@ export const createPartnerCommission = async ({
       },
     });
 
+    console.log(
+      `Created a ${event} commission ${commission.id} (${currencyFormatter(commission.earnings)}) for ${partnerId}: ${JSON.stringify(commission)}`,
+    );
+
     waitUntil(
       (async () => {
         const program = await prisma.program.findUniqueOrThrow({