use partner commissionAmount

dubinc · Dec 24, 2024 · 99cf016 · 99cf016
1 parent 7d48eec
commit 99cf016
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 78 deletions.
diff --git a/apps/web/lib/api/sales/commission.ts b/apps/web/lib/api/sales/commission.ts
@@ -1,27 +1,34 @@
-import { Program } from "@dub/prisma/client";
+import { Program, ProgramEnrollment } from "@dub/prisma/client";
 
 /* 
   Calculate the commission earned for a sale
 */
 export const calculateEarnings = ({
   program,
+  partner,
   sales,
   saleAmount,
 }: {
   program: Pick<Program, "commissionAmount" | "commissionType">;
+  partner?: Pick<ProgramEnrollment, "commissionAmount">;
   sales: number;
   saleAmount: number;
 }) => {
-  if (program.commissionAmount === 0) {
+  const commissionAmount =
+    partner?.commissionAmount !== null
+      ? partner?.commissionAmount
+      : program.commissionAmount;
+
+  if (!commissionAmount) {
     return 0;
   }
 
   if (program.commissionType === "percentage") {
-    return saleAmount * (program.commissionAmount / 100);
+    return saleAmount * (commissionAmount / 100);
   }
 
   if (program.commissionType === "flat") {
-    return sales * program.commissionAmount;
+    return sales * commissionAmount;
   }
 
   throw new Error("Invalid commissionType");

diff --git a/apps/web/lib/api/sales/recalculate-sales-earnings.ts b/apps/web/lib/api/sales/recalculate-sales-earnings.ts
diff --git a/apps/web/lib/api/sales/sale.ts b/apps/web/lib/api/sales/sale.ts
@@ -32,6 +32,7 @@ export const createSaleData = ({
     program,
     sales: 1,
     saleAmount: sale.amount,
+    partner,
   });
 
   const commissionAmount =