Fix partners ban cron crash when link has no discount code

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: devkiran

apps/web/app/(ee)/api/cron/partners/ban/route.ts

@@ -131,14 +131,7 @@ export const POST = withCron(async ({ rawBody }) => {
     recordLink(links, { deleted: true }),
 
     // Queue discount code deletions
-    deleteDiscountCodes(
-      links
-        .map((link) => link.discountCode)
-        .filter(
-          (discountCode): discountCode is NonNullable<typeof discountCode> =>
-            discountCode !== null,
-        ),
-    ),
+    deleteDiscountCodes(links.map((link) => link.discountCode)),
   ]);
 
   const affectedProgramEnrollments = await prisma.programEnrollment.findMany({

apps/web/lib/api/discounts/delete-discount-code.ts

@@ -7,17 +7,22 @@ const queue = qstash.queue({
   queueName: "delete-discount-code",
 });
 
+type DiscountCodePayload = Pick<DiscountCode, "id" | "code" | "programId">;
+
 type DeleteDiscountCodesParams =
-  | Pick<DiscountCode, "id" | "code" | "programId">
-  | Pick<DiscountCode, "id" | "code" | "programId">[];
+  | DiscountCodePayload
+  | (DiscountCodePayload | null | undefined)[];
 
 // Triggered in the following cases:
 // 1. When a discount is deleted
 // 2. When a link is deleted that has a discount code associated with it
 // 3. When partners are banned / deactivated
 // 4. When a partner is moved to a different group
 export async function deleteDiscountCodes(input: DeleteDiscountCodesParams) {
-  const discountCodes = Array.isArray(input) ? input : [input];
+  const raw = Array.isArray(input) ? input : [input];
+  const discountCodes = raw.filter(
+    (dc): dc is NonNullable<typeof dc> => dc != null,
+  );
 
   if (discountCodes.length === 0) {
     return;