Create referral commissions in workflows

apps/web/app/(ee)/api/admin/payouts/get-payouts-timeseries.ts

@@ -1,5 +1,5 @@
-import { sqlGranularityMap } from "@/lib/planetscale/granularity";
 import { conn } from "@/lib/planetscale/connection";
+import { sqlGranularityMap } from "@/lib/planetscale/granularity";
 import { InvoiceStatus } from "@dub/prisma/client";
 import { ACME_PROGRAM_ID } from "@dub/utils";
 import { format } from "date-fns";

apps/web/app/(ee)/api/cron/commissions/referrals/create/route.ts

@@ -1,5 +1,8 @@
 import { withCron } from "@/lib/cron/with-cron";
-import { createReferralCommission } from "@/lib/partner-referrals/create-referral-commission";
+import {
+  createReferralCommission,
+  CreateReferralCommissionArgs,
+} from "@/lib/partner-referrals/create-referral-commission";
 import * as z from "zod/v4";
 import { logAndRespond } from "../../../utils";
 
@@ -14,13 +17,28 @@ const inputSchema = z.union([
 export const POST = withCron(async ({ rawBody }) => {
   const inputParsed = inputSchema.parse(JSON.parse(rawBody));
 
-  const referralCommission = await createReferralCommission(inputParsed);
+  let args: CreateReferralCommissionArgs;
 
-  if (referralCommission === null) {
-    return logAndRespond("Referral commission creation skipped.");
+  if ("sourceCommissionId" in inputParsed) {
+    args = {
+      source: "commission",
+      sourceCommissionId: inputParsed.sourceCommissionId,
+    };
+  } else {
+    args = {
+      source: "partner",
+      programId: inputParsed.programId,
+      partnerId: inputParsed.partnerId,
+    };
   }
 
-  return logAndRespond(
-    `Referral commission ${referralCommission.id} created successfully.`,
-  );
+  const { commission, reason } = await createReferralCommission(args);
+
+  if (commission) {
+    return logAndRespond(
+      `Referral commission ${commission.id} created successfully.`,
+    );
+  }
+
+  return logAndRespond(reason ?? "Referral commission creation skipped.");
 });

apps/web/app/(ee)/api/cron/commissions/referrals/network/route.ts

@@ -0,0 +1,56 @@
+import { withCron } from "@/lib/cron/with-cron";
+import { createNetworkReferralCommission } from "@/lib/partner-referrals/create-network-referral-commission";
+import { prisma } from "@dub/prisma";
+import * as z from "zod/v4";
+import { logAndRespond } from "../../../utils";
+
+export const dynamic = "force-dynamic";
+
+const inputSchema = z.object({
+  payoutId: z.string(),
+});
+
+// POST /api/cron/commissions/referrals/network
+// Creates a network referral commission for the referrer when a referred partner's payout is sent or completed.
+export const POST = withCron(async ({ rawBody }) => {
+  const { payoutId } = inputSchema.parse(JSON.parse(rawBody));
+
+  const payout = await prisma.payout.findUnique({
+    where: {
+      id: payoutId,
+    },
+    include: {
+      partner: {
+        select: {
+          id: true,
+          referredByPartnerId: true,
+        },
+      },
+    },
+  });
+
+  if (!payout) {
+    return logAndRespond(`Payout ${payoutId} not found.`);
+  }
+
+  if (!["sent", "completed"].includes(payout.status)) {
+    return logAndRespond(
+      `Payout ${payoutId} is not in a valid status to create referral commissions.`,
+    );
+  }
+
+  const commission = await createNetworkReferralCommission({
+    partner: payout.partner,
+    payout,
+  });
+
+  if (commission) {
+    return logAndRespond(
+      `Created network referral commission for payout ${payout.id}.`,
+    );
+  }
+
+  return logAndRespond(
+    `No referral commission created for payout ${payout.id}.`,
+  );
+});

apps/web/app/(ee)/api/cron/commissions/referrals/void/route.ts

@@ -0,0 +1,39 @@
+import {
+  cancelReferralCommissionsBelowThreshold,
+  voidReferralCommissions,
+  voidReferralCommissionsSchema,
+} from "@/lib/api/commissions/void-referral-commissions";
+import { withCron } from "@/lib/cron/with-cron";
+import { logAndRespond } from "../../../utils";
+
+export const dynamic = "force-dynamic";
+
+// POST /api/cron/commissions/referrals/void
+export const POST = withCron(async ({ rawBody }) => {
+  const {
+    workspaceId,
+    programId,
+    userId,
+    sourceCommissionIds,
+    sourceCommissionStatus,
+  } = voidReferralCommissionsSchema.parse(JSON.parse(rawBody));
+
+  await voidReferralCommissions({
+    workspaceId,
+    programId,
+    userId,
+    sourceCommissionIds,
+    sourceCommissionStatus,
+  });
+
+  await cancelReferralCommissionsBelowThreshold({
+    workspaceId,
+    userId,
+    sourceCommissionIds,
+    programId,
+  });
+
+  return logAndRespond(
+    `Voided referral commissions for ${sourceCommissionIds.length} source commission(s).`,
+  );
+});

apps/web/app/(ee)/api/cron/payouts/charge-succeeded/send-paypal-payouts.ts

@@ -83,7 +83,7 @@ export async function sendPaypalPayouts(invoice: Pick<Invoice, "id">) {
       enqueueBatchJobs(
         payouts.map((payout) => ({
           queueName: "create-referral-commissions",
-          url: `${APP_DOMAIN_WITH_NGROK}/api/cron/commissions/referrals/queue`,
+          url: `${APP_DOMAIN_WITH_NGROK}/api/cron/commissions/referrals/network`,
           body: {
             payoutId: payout.id,
           },

apps/web/app/(ee)/api/stripe/integration/webhook/charge-refunded.ts

@@ -1,8 +1,10 @@
 import { trackCommissionStatusUpdate } from "@/lib/api/commissions/track-commission-update-activity-log";
+import { queueVoidReferralCommissions } from "@/lib/api/commissions/void-referral-commissions";
 import { syncTotalCommissions } from "@/lib/api/partners/sync-total-commissions";
 import { stripeAppClient } from "@/lib/stripe";
 import { StripeMode } from "@/lib/types";
 import { prisma } from "@dub/prisma";
+import { CommissionStatus } from "@dub/prisma/client";
 import type Stripe from "stripe";
 
 // Handle event "charge.refunded"
@@ -140,6 +142,13 @@ export async function chargeRefunded(
     newStatus: "refunded",
   });
 
+  await queueVoidReferralCommissions({
+    workspaceId: workspace.id,
+    programId: commission.programId,
+    sourceCommissionIds: [commission.id],
+    sourceCommissionStatus: CommissionStatus.refunded,
+  });
+
   return {
     response: `Commission ${commission.id} updated to status "refunded"`,
     workspaceId,

apps/web/app/(ee)/api/workflows/create-partner-commission/route.ts

@@ -7,6 +7,7 @@ import { calculateSaleEarnings } from "@/lib/api/sales/calculate-sale-earnings";
 import { executeWorkflows } from "@/lib/api/workflows/execute-workflows";
 import { logger } from "@/lib/axiom/server";
 import { getWorkflowConfig } from "@/lib/cron/qstash-workflow";
+import { createReferralCommission } from "@/lib/partner-referrals/create-referral-commission";
 import { constructWebhookPartner } from "@/lib/partners/constuct-webhook-partner";
 import { determinePartnerReward } from "@/lib/partners/determine-partner-reward";
 import { getRewardAmount } from "@/lib/partners/get-reward-amount";
@@ -21,6 +22,7 @@ import { DEFAULT_PARTNER_GROUP } from "@/lib/zod/schemas/groups";
 import { prisma } from "@dub/prisma";
 import {
   Commission,
+  CommissionType,
   Link,
   Partner,
   PartnerGroup,
@@ -51,7 +53,7 @@ type StepFunctionInput = Input & {
 };
 
 type StepCreateCommissionOutput = {
-  commission: Pick<Commission, "id"> | null;
+  commission: Pick<Commission, "id" | "type"> | null;
   outputLog: string;
   isFirstCommission?: boolean;
 };
@@ -121,6 +123,16 @@ export const { POST } = serve<Input>(
         }
       });
     }
+
+    // Step 4: Create referral commission
+    if (commission && commission.type === CommissionType.sale) {
+      await context.run("create-referral-commission", async () => {
+        return await createReferralCommission({
+          source: "commission",
+          sourceCommissionId: commission.id,
+        });
+      });
+    }
   },
   {
     initialPayloadParser: (requestPayload) => {