Migrate sales to earnings model and update related routes

dubinc · Feb 4, 2025 · fd10e45 · fd10e45
1 parent f3eaedd
commit fd10e45
Show file tree

Hide file tree

Showing 8 changed files with 25 additions and 14 deletions.
diff --git a/apps/web/app/api/programs/[programId]/metrics/route.ts b/apps/web/app/api/programs/[programId]/metrics/route.ts
@@ -33,15 +33,21 @@ export const GET = withWorkspace(
         },
       }),
 
-      prisma.sale.aggregate({
-        where,
+      prisma.earnings.aggregate({
+        where: {
+          ...where,
+          type: "sale",
+        },
         _sum: {
           amount: true,
         },
       }),
 
-      prisma.sale.count({
-        where,
+      prisma.earnings.count({
+        where: {
+          ...where,
+          type: "sale",
+        },
       }),
 
       prisma.programEnrollment.count({

diff --git a/apps/web/app/api/programs/[programId]/sales/amount/route.ts b/apps/web/app/api/programs/[programId]/sales/amount/route.ts
@@ -21,8 +21,9 @@ export const GET = withWorkspace(
 
     const { partnerId } = parsed;
 
-    const salesAmount = await prisma.sale.aggregate({
+    const salesAmount = await prisma.earnings.aggregate({
       where: {
+        type: "sale",
         programId,
         partnerId,
         createdAt: {

diff --git a/apps/web/app/api/programs/[programId]/sales/count/route.ts b/apps/web/app/api/programs/[programId]/sales/count/route.ts
@@ -21,9 +21,10 @@ export const GET = withWorkspace(
 
     const { startDate, endDate } = getStartEndDates(parsed);
 
-    const salesCount = await prisma.sale.groupBy({
+    const salesCount = await prisma.earnings.groupBy({
       by: ["status"],
       where: {
+        type: "sale",
         programId,
         status,
         partnerId,

diff --git a/apps/web/app/api/programs/[programId]/sales/route.ts b/apps/web/app/api/programs/[programId]/sales/route.ts
@@ -32,9 +32,10 @@ export const GET = withWorkspace(
       programId,
     });
 
-    const sales = await prisma.sale.findMany({
+    const sales = await prisma.earnings.findMany({
       where: {
         programId,
+        type: "sale",
         ...(status && { status }),
         ...(customerId && { customerId }),
         ...(payoutId && { payoutId }),

diff --git a/apps/web/lib/actions/partners/update-sale-status.ts b/apps/web/lib/actions/partners/update-sale-status.ts
@@ -18,7 +18,7 @@ export const updateSaleStatusAction = authActionClient
     const { workspace } = ctx;
     const { saleId, status } = parsedInput;
 
-    const sale = await prisma.sale.findUniqueOrThrow({
+    const sale = await prisma.earnings.findUniqueOrThrow({
       where: {
         id: saleId,
         program: {
@@ -50,7 +50,7 @@ export const updateSaleStatusAction = authActionClient
       });
     }
 
-    await prisma.sale.update({
+    await prisma.earnings.update({
       where: {
         id: sale.id,
       },

diff --git a/apps/web/scripts/migrate-sales.ts b/apps/web/scripts/migrate-sales.ts
@@ -1,10 +1,11 @@
-import "dotenv-flow/config";
 import { prisma } from "@dub/prisma";
 import { EventType } from "@dub/prisma/client";
+import "dotenv-flow/config";
 
 async function main() {
   const sales = await prisma.sale.findMany({
     select: {
+      id: true,
       programId: true,
       partnerId: true,
       linkId: true,
@@ -14,6 +15,7 @@ async function main() {
       eventId: true,
       amount: true,
       earnings: true,
+      currency: true,
       status: true,
       createdAt: true,
       updatedAt: true,

diff --git a/packages/prisma/schema/customer.prisma b/packages/prisma/schema/customer.prisma
@@ -16,8 +16,8 @@ model Customer {
   createdAt        DateTime @default(now())
   updatedAt        DateTime @updatedAt
 
-  project  Project   @relation(fields: [projectId], references: [id], onDelete: Cascade)
-  link     Link?     @relation(fields: [linkId], references: [id])
+  project  Project    @relation(fields: [projectId], references: [id], onDelete: Cascade)
+  link     Link?      @relation(fields: [linkId], references: [id])
   sales    Sale[]
   earnings Earnings[]
 

diff --git a/packages/prisma/schema/earnings.prisma b/packages/prisma/schema/earnings.prisma
@@ -26,8 +26,8 @@ model Earnings {
   type     EventType
   amount   Int?
   quantity Int
-  earnings Int?
-  // currency String
+  earnings Int            @default(0)
+  currency String?
   status   EarningsStatus @default(pending)
 
   createdAt DateTime @default(now())