Merge pull request #1829 from dubinc/dub-customer

Set referral discount via Dub Customer object

apps/web/app/api/analytics/client/route.ts

@@ -1,5 +1,5 @@
 import { getAnalytics } from "@/lib/analytics/get-analytics";
-import { calculateEarnings } from "@/lib/api/sales/commission";
+import { calculateEarnings } from "@/lib/api/sales/calculate-earnings";
 import { withEmbedToken } from "@/lib/embed/auth";
 import { analyticsQuerySchema } from "@/lib/zod/schemas/analytics";
 import { NextResponse } from "next/server";

apps/web/app/api/customers/[id]/route.ts

@@ -1,4 +1,5 @@
 import { getCustomerOrThrow } from "@/lib/api/customers/get-customer-or-throw";
+import { transformCustomer } from "@/lib/api/customers/transform-customer";
 import { DubApiError } from "@/lib/api/errors";
 import { parseRequestBody } from "@/lib/api/utils";
 import { withWorkspace } from "@/lib/auth";
@@ -24,7 +25,7 @@ export const GET = withWorkspace(
       },
     );
 
-    return NextResponse.json(CustomerSchema.parse(customer));
+    return NextResponse.json(CustomerSchema.parse(transformCustomer(customer)));
   },
   {
     requiredAddOn: "conversion",
@@ -52,11 +53,22 @@ export const PATCH = withWorkspace(
         },
         data: { name, email, avatar, externalId },
         include: {
-          link: true,
+          link: {
+            include: {
+              programEnrollment: {
+                include: {
+                  partner: true,
+                  discount: true,
+                },
+              },
+            },
+          },
         },
       });
 
-      return NextResponse.json(CustomerSchema.parse(updatedCustomer));
+      return NextResponse.json(
+        CustomerSchema.parse(transformCustomer(updatedCustomer)),
+      );
     } catch (error) {
       if (error.code === "P2002") {
         throw new DubApiError({

apps/web/app/api/customers/route.ts

@@ -1,3 +1,4 @@
+import { transformCustomer } from "@/lib/api/customers/transform-customer";
 import { DubApiError } from "@/lib/api/errors";
 import { parseRequestBody } from "@/lib/api/utils";
 import { withWorkspace } from "@/lib/auth";
@@ -22,7 +23,9 @@ export const GET = withWorkspace(
       },
     });
 
-    return NextResponse.json(CustomerSchema.array().parse(customers));
+    return NextResponse.json(
+      CustomerSchema.array().parse(customers.map(transformCustomer)),
+    );
   },
   {
     requiredAddOn: "conversion",
@@ -48,14 +51,14 @@ export const POST = withWorkspace(
           projectId: workspace.id,
           projectConnectId: workspace.stripeConnectId,
         },
-        include: {
-          link: true,
-        },
       });
 
-      return NextResponse.json(CustomerSchema.parse(customer), {
-        status: 201,
-      });
+      return NextResponse.json(
+        CustomerSchema.parse(transformCustomer(customer)),
+        {
+          status: 201,
+        },
+      );
     } catch (error) {
       if (error.code === "P2002") {
         throw new DubApiError({

apps/web/app/api/embed/analytics/route.ts

@@ -0,0 +1,15 @@
+import { getAnalytics } from "@/lib/analytics/get-analytics";
+import { withEmbedToken } from "@/lib/embed/auth";
+import { NextResponse } from "next/server";
+
+// GET /api/embed/analytics – get timeseries analytics for a link from an embed token
+export const GET = withEmbedToken(async ({ link }) => {
+  const analytics = await getAnalytics({
+    event: "composite",
+    groupBy: "timeseries",
+    linkId: link.id,
+    interval: "1y",
+  });
+
+  return NextResponse.json(analytics);
+});

apps/web/app/api/embed/leaderboard/route.ts

@@ -0,0 +1,40 @@
+import { withEmbedToken } from "@/lib/embed/auth";
+import { LeaderboardPartnerSchema } from "@/lib/zod/schemas/partners";
+import { prisma } from "@dub/prisma";
+import { NextResponse } from "next/server";
+import z from "node_modules/zod/lib";
+
+// GET /api/embed/sales – get sales for a link from an embed token
+export const GET = withEmbedToken(async ({ program, searchParams }) => {
+  const programEnrollments = await prisma.programEnrollment.findMany({
+    where: {
+      programId: program.id,
+    },
+    orderBy: [
+      {
+        link: {
+          saleAmount: "desc",
+        },
+      },
+      {
+        link: {
+          leads: "desc",
+        },
+      },
+      {
+        link: {
+          clicks: "desc",
+        },
+      },
+    ],
+    select: {
+      partner: true,
+      link: true,
+    },
+    take: 20,
+  });
+
+  return NextResponse.json(
+    z.array(LeaderboardPartnerSchema).parse(programEnrollments),
+  );
+});

apps/web/app/api/embed/sales/route.ts

@@ -1,11 +1,16 @@
 import { withEmbedToken } from "@/lib/embed/auth";
+import { SALES_PAGE_SIZE } from "@/lib/partners/constants";
 import z from "@/lib/zod";
 import { PartnerSaleResponseSchema } from "@/lib/zod/schemas/partners";
 import { prisma } from "@dub/prisma";
 import { NextResponse } from "next/server";
 
 // GET /api/embed/sales – get sales for a link from an embed token
-export const GET = withEmbedToken(async ({ link }) => {
+export const GET = withEmbedToken(async ({ link, searchParams }) => {
+  const { page } = z
+    .object({ page: z.coerce.number().optional().default(1) })
+    .parse(searchParams);
+
   const sales = await prisma.sale.findMany({
     where: {
       linkId: link.id,
@@ -25,7 +30,8 @@ export const GET = withEmbedToken(async ({ link }) => {
         },
       },
     },
-    take: 3,
+    take: SALES_PAGE_SIZE,
+    skip: (page - 1) * SALES_PAGE_SIZE,
     orderBy: {
       createdAt: "desc",
     },

apps/web/app/api/events/client/route.ts

@@ -1,5 +1,5 @@
 import { getEvents } from "@/lib/analytics/get-events";
-import { calculateEarnings } from "@/lib/api/sales/commission";
+import { calculateEarnings } from "@/lib/api/sales/calculate-earnings";
 import { withEmbedToken } from "@/lib/embed/auth";
 import { eventsQuerySchema } from "@/lib/zod/schemas/analytics";
 import { NextResponse } from "next/server";

apps/web/app/api/partners/[partnerId]/programs/[programId]/analytics/route.ts

@@ -1,6 +1,6 @@
 import { getAnalytics } from "@/lib/analytics/get-analytics";
 import { getProgramEnrollmentOrThrow } from "@/lib/api/programs/get-program-enrollment-or-throw";
-import { calculateEarnings } from "@/lib/api/sales/commission";
+import { calculateEarnings } from "@/lib/api/sales/calculate-earnings";
 import { withPartner } from "@/lib/auth/partner";
 import { analyticsQuerySchema } from "@/lib/zod/schemas/analytics";
 import { NextResponse } from "next/server";

apps/web/app/api/programs/[programId]/analytics/route.ts

@@ -1,5 +1,5 @@
 import { getAnalytics } from "@/lib/analytics/get-analytics";
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import { analyticsQuerySchema } from "@/lib/zod/schemas/analytics";
 import { NextResponse } from "next/server";

apps/web/app/api/programs/[programId]/customers/count/route.ts

@@ -1,4 +1,4 @@
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import { prisma } from "@dub/prisma";
 import { NextResponse } from "next/server";

apps/web/app/api/programs/[programId]/customers/route.ts

@@ -1,4 +1,4 @@
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import {
   CustomerSchema,

apps/web/app/api/programs/[programId]/invites/count/route.ts

@@ -1,4 +1,4 @@
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import { prisma } from "@dub/prisma";
 import { NextResponse } from "next/server";

apps/web/app/api/programs/[programId]/invites/route.ts

@@ -1,4 +1,4 @@
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import { partnerInvitesQuerySchema } from "@/lib/zod/schemas/partners";
 import { ProgramInviteSchema } from "@/lib/zod/schemas/programs";

apps/web/app/api/programs/[programId]/metrics/route.ts

@@ -1,5 +1,5 @@
 import { getStartEndDates } from "@/lib/analytics/utils/get-start-end-dates";
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import { getProgramMetricsQuerySchema } from "@/lib/zod/schemas/programs";
 import { prisma } from "@dub/prisma";

apps/web/app/api/programs/[programId]/partners/[partnerId]/route.ts

@@ -1,4 +1,4 @@
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import { PartnerSchema } from "@/lib/zod/schemas/partners";
 import { prisma } from "@dub/prisma";

apps/web/app/api/programs/[programId]/partners/count/route.ts

@@ -1,4 +1,4 @@
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import { partnersCountQuerySchema } from "@/lib/zod/schemas/partners";
 import { prisma } from "@dub/prisma";

apps/web/app/api/programs/[programId]/partners/route.ts

@@ -1,4 +1,4 @@
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import {
   EnrolledPartnerSchema,

apps/web/app/api/programs/[programId]/payouts/count/route.ts

@@ -1,5 +1,5 @@
 import { getStartEndDates } from "@/lib/analytics/utils/get-start-end-dates";
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import { MIN_PAYOUT_AMOUNT } from "@/lib/partners/constants";
 import { payoutsCountQuerySchema } from "@/lib/zod/schemas/payouts";

apps/web/app/api/programs/[programId]/payouts/route.ts

@@ -1,5 +1,5 @@
 import { getStartEndDates } from "@/lib/analytics/utils/get-start-end-dates";
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import {
   PayoutResponseSchema,

apps/web/app/api/programs/[programId]/route.ts

@@ -1,4 +1,4 @@
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import { NextResponse } from "next/server";
 

apps/web/app/api/programs/[programId]/sales/amount/route.ts

@@ -1,5 +1,5 @@
 import { getStartEndDates } from "@/lib/analytics/utils/get-start-end-dates";
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import { getSalesAmountQuerySchema } from "@/lib/zod/schemas/partners";
 import { prisma } from "@dub/prisma";

apps/web/app/api/programs/[programId]/sales/count/route.ts

@@ -1,5 +1,5 @@
 import { getStartEndDates } from "@/lib/analytics/utils/get-start-end-dates";
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import { getSalesCountQuerySchema } from "@/lib/zod/schemas/partners";
 import { prisma } from "@dub/prisma";

apps/web/app/api/programs/[programId]/sales/route.ts

@@ -1,5 +1,5 @@
 import { getStartEndDates } from "@/lib/analytics/utils/get-start-end-dates";
-import { getProgramOrThrow } from "@/lib/api/programs/get-program";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import {
   getSalesQuerySchema,

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

@@ -1,5 +1,5 @@
 import { notifyPartnerSale } from "@/lib/api/partners/notify-partner-sale";
-import { createSaleData } from "@/lib/api/sales/sale";
+import { createSaleData } from "@/lib/api/sales/create-sale-data";
 import { getLeadEvent, recordSale } from "@/lib/tinybird";
 import { redis } from "@/lib/upstash";
 import { sendWorkspaceWebhook } from "@/lib/webhook/publish";
@@ -129,32 +129,36 @@ export async function checkoutSessionCompleted(event: Stripe.Event) {
 
   // for program links
   if (link.programId) {
-    const { program, partner } =
+    const { program, partnerId, commissionAmount } =
       await prisma.programEnrollment.findUniqueOrThrow({
         where: {
           linkId: link.id,
         },
         select: {
           program: true,
-          partner: {
-            select: {
-              id: true,
-            },
-          },
+          partnerId: true,
+          commissionAmount: true,
         },
       });
 
     const saleRecord = createSaleData({
-      customerId: saleData.customer_id,
-      linkId: saleData.link_id,
-      clickId: saleData.click_id,
-      invoiceId: saleData.invoice_id,
-      eventId: saleData.event_id,
-      paymentProcessor: saleData.payment_processor,
-      amount: saleData.amount,
-      currency: saleData.currency,
-      partnerId: partner.id,
       program,
+      partner: {
+        id: partnerId,
+        commissionAmount,
+      },
+      customer: {
+        id: saleData.customer_id,
+        linkId: saleData.link_id,
+        clickId: saleData.click_id,
+      },
+      sale: {
+        amount: saleData.amount,
+        currency: saleData.currency,
+        invoiceId: saleData.invoice_id,
+        eventId: saleData.event_id,
+        paymentProcessor: saleData.payment_processor,
+      },
       metadata: {
         ...leadEvent.data[0],
         stripeMetadata: charge,
@@ -168,7 +172,7 @@ export async function checkoutSessionCompleted(event: Stripe.Event) {
     waitUntil(
       notifyPartnerSale({
         partner: {
-          id: partner.id,
+          id: partnerId,
           referralLink: link.shortLink,
         },
         program,