Merge pull request dubinc#2409 from dubinc/default-program-id

Add workspace.defaultProgramId

Author: steven-tey

apps/web/app/api/commissions/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-or-throw";
+import { DubApiError } from "@/lib/api/errors";
 import { withWorkspace } from "@/lib/auth";
 import { getCommissionsCountQuerySchema } from "@/lib/zod/schemas/commissions";
 import { prisma } from "@dub/prisma";
@@ -10,10 +10,12 @@ import { NextResponse } from "next/server";
 export const GET = withWorkspace(async ({ workspace, searchParams }) => {
   const { programId } = searchParams;
 
-  await getProgramOrThrow({
-    workspaceId: workspace.id,
-    programId,
-  });
+  if (programId !== workspace.defaultProgramId) {
+    throw new DubApiError({
+      code: "not_found",
+      message: "Program not found",
+    });
+  }
 
   const {
     status,

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

@@ -1,5 +1,5 @@
 import { getStartEndDates } from "@/lib/analytics/utils/get-start-end-dates";
-import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
+import { DubApiError } from "@/lib/api/errors";
 import { withWorkspace } from "@/lib/auth";
 import {
   CommissionResponseSchema,
@@ -13,10 +13,12 @@ import { z } from "zod";
 export const GET = withWorkspace(async ({ workspace, searchParams }) => {
   const { programId } = searchParams;
 
-  await getProgramOrThrow({
-    workspaceId: workspace.id,
-    programId,
-  });
+  if (programId !== workspace.defaultProgramId) {
+    throw new DubApiError({
+      code: "not_found",
+      message: "Program not found",
+    });
+  }
 
   const {
     status,

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

@@ -1,6 +1,5 @@
 import { DubApiError } from "@/lib/api/errors";
 import { getPartnerForProgram } from "@/lib/api/partners/get-partner-for-program";
-import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import { EnrolledPartnerSchemaWithExpandedFields } from "@/lib/zod/schemas/partners";
 import { NextResponse } from "next/server";
@@ -19,16 +18,17 @@ export const GET = withWorkspace(
       });
     }
 
-    const [_program, partner] = await Promise.all([
-      getProgramOrThrow({
-        workspaceId: workspace.id,
-        programId,
-      }),
-      getPartnerForProgram({
-        programId,
-        partnerId,
-      }),
-    ]);
+    if (programId !== workspace.defaultProgramId) {
+      throw new DubApiError({
+        code: "not_found",
+        message: "Program not found",
+      });
+    }
+
+    const partner = await getPartnerForProgram({
+      programId,
+      partnerId,
+    });
 
     if (!partner)
       throw new DubApiError({

apps/web/app/api/partners/export/route.ts

@@ -1,4 +1,5 @@
 import { convertToCSV } from "@/lib/analytics/utils/convert-to-csv";
+import { DubApiError } from "@/lib/api/errors";
 import { getPartners } from "@/lib/api/partners/get-partners";
 import { withWorkspace } from "@/lib/auth";
 import {
@@ -17,6 +18,13 @@ export const GET = withWorkspace(
   async ({ searchParams, workspace }) => {
     const { programId } = searchParams;
 
+    if (programId !== workspace.defaultProgramId) {
+      throw new DubApiError({
+        code: "not_found",
+        message: "Program not found",
+      });
+    }
+
     let { columns, ...filters } = partnersExportQuerySchema.parse(searchParams);
 
     const partners = await getPartners({

apps/web/app/api/partners/links/route.ts

@@ -22,10 +22,12 @@ export const GET = withWorkspace(
     const { programId, partnerId, tenantId } =
       retrievePartnerLinksSchema.parse(searchParams);
 
-    await getProgramOrThrow({
-      programId,
-      workspaceId: workspace.id,
-    });
+    if (programId !== workspace.defaultProgramId) {
+      throw new DubApiError({
+        code: "not_found",
+        message: "Program not found",
+      });
+    }
 
     const programEnrollment = await prisma.programEnrollment.findUnique({
       where: partnerId

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

@@ -27,6 +27,13 @@ export const GET = withWorkspace(
       });
     }
 
+    if (programId !== workspace.defaultProgramId) {
+      throw new DubApiError({
+        code: "not_found",
+        message: "Program not found",
+      });
+    }
+
     const partners = await getPartners({
       ...partnersQuerySchema.parse(searchParams),
       workspaceId: workspace.id,

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

@@ -1,5 +1,4 @@
 import { DubApiError } from "@/lib/api/errors";
-import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { calculateSaleEarnings } from "@/lib/api/sales/calculate-sale-earnings";
 import { parseRequestBody } from "@/lib/api/utils";
 import { withWorkspace } from "@/lib/auth/workspace";
@@ -17,15 +16,17 @@ export const PATCH = withWorkspace(
     let { programId, invoiceId, amount, modifyAmount, currency } =
       updatePartnerSaleSchema.parse(await parseRequestBody(req));
 
-    const program = await getProgramOrThrow({
-      workspaceId: workspace.id,
-      programId,
-    });
+    if (programId !== workspace.defaultProgramId) {
+      throw new DubApiError({
+        code: "not_found",
+        message: "Program not found",
+      });
+    }
 
     const commission = await prisma.commission.findUnique({
       where: {
         programId_invoiceId: {
-          programId: program.id,
+          programId,
           invoiceId,
         },
       },
@@ -72,13 +73,13 @@ export const PATCH = withWorkspace(
     const reward = await determinePartnerReward({
       event: "sale",
       partnerId: partner.id,
-      programId: program.id,
+      programId: programId,
     });
 
     if (!reward) {
       throw new DubApiError({
         code: "not_found",
-        message: `No reward found for partner ${partner.id} in program ${program.id}.`,
+        message: `No reward found for partner ${partner.id} in program ${programId}.`,
       });
     }
 

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

@@ -1,13 +1,15 @@
-import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
+import { DubApiError } from "@/lib/api/errors";
 import { withWorkspace } from "@/lib/auth";
 import { prisma } from "@dub/prisma";
 import { NextResponse } from "next/server";
 
 export const GET = withWorkspace(async ({ workspace, params }) => {
-  await getProgramOrThrow({
-    workspaceId: workspace.id,
-    programId: params.programId,
-  });
+  if (params.programId !== workspace.defaultProgramId) {
+    throw new DubApiError({
+      code: "not_found",
+      message: "Program not found",
+    });
+  }
 
   const application = await prisma.programApplication.findUnique({
     where: { id: params.applicationId },

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

@@ -1,5 +1,5 @@
+import { DubApiError } from "@/lib/api/errors";
 import { getDiscountOrThrow } from "@/lib/api/partners/get-discount-or-throw";
-import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
 import { withWorkspace } from "@/lib/auth";
 import { discountPartnersQuerySchema } from "@/lib/zod/schemas/discount";
 import { prisma } from "@dub/prisma";
@@ -9,19 +9,19 @@ import { NextResponse } from "next/server";
 export const GET = withWorkspace(
   async ({ workspace, params, searchParams }) => {
     const { programId } = params;
-    const { discountId } = discountPartnersQuerySchema.parse(searchParams);
+    if (programId !== workspace.defaultProgramId) {
+      throw new DubApiError({
+        code: "not_found",
+        message: "Program not found",
+      });
+    }
 
-    await Promise.all([
-      getProgramOrThrow({
-        workspaceId: workspace.id,
-        programId,
-      }),
+    const { discountId } = discountPartnersQuerySchema.parse(searchParams);
 
-      getDiscountOrThrow({
-        programId,
-        discountId,
-      }),
-    ]);
+    await getDiscountOrThrow({
+      programId,
+      discountId,
+    });
 
     const partners = await prisma.programEnrollment.findMany({
       where: {

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

@@ -1,4 +1,4 @@
-import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
+import { DubApiError } from "@/lib/api/errors";
 import { withWorkspace } from "@/lib/auth";
 import { DiscountSchema } from "@/lib/zod/schemas/discount";
 import { prisma } from "@dub/prisma";
@@ -9,10 +9,12 @@ import { z } from "zod";
 export const GET = withWorkspace(async ({ workspace, params }) => {
   const { programId } = params;
 
-  await getProgramOrThrow({
-    workspaceId: workspace.id,
-    programId,
-  });
+  if (programId !== workspace.defaultProgramId) {
+    throw new DubApiError({
+      code: "not_found",
+      message: "Program not found",
+    });
+  }
 
   const discounts = await prisma.discount.findMany({
     where: {