small updates

Author: steven-tey

Diff for: apps/web/lib/actions/partners/accept-program-invite.ts

@@ -35,17 +35,12 @@ export const acceptProgramInviteAction = authPartnerActionClient
       prisma.programInvite.delete({
         where: { id: programInvite.id },
       }),
-      prisma.link.update({
-        where: { id: programInvite.linkId },
-        data: {
-          partnerId: partner.id,
-        },
-      }),
     ]);
 
     waitUntil(
       backfillLinkData({
-        programEnrollmentId: programEnrollment.id,
+        programId: programInvite.programId,
+        partnerId: partner.id,
         linkId: programInvite.linkId,
       }),
     );

Diff for: apps/web/lib/actions/partners/backfill-link-data.ts

@@ -1,29 +1,43 @@
 import { getEvents } from "@/lib/analytics/get-events";
+import { includeTags } from "@/lib/api/links/include-tags";
 import { createSaleData } from "@/lib/api/sales/create-sale-data";
+import { recordLink } from "@/lib/tinybird";
 import { SaleEvent } from "@/lib/types";
 import { prisma } from "@dub/prisma";
 
 export const backfillLinkData = async ({
-  programEnrollmentId,
+  programId,
+  partnerId,
   linkId,
 }: {
-  programEnrollmentId: string;
+  programId: string;
+  partnerId: string;
   linkId: string;
 }) => {
-  const link = await prisma.link.findUniqueOrThrow({
+  const link = await prisma.link.update({
     where: {
       id: linkId,
     },
+    data: {
+      partnerId,
+    },
+    include: includeTags,
   });
 
+  // update in tinybird
+  await recordLink(link);
+
   if (link.sales === 0) {
     console.log(`Link ${linkId} has no sales, skipping backfill`);
     return;
   }
 
   const programEnrollment = await prisma.programEnrollment.findUniqueOrThrow({
     where: {
-      id: programEnrollmentId,
+      partnerId_programId: {
+        partnerId,
+        programId,
+      },
     },
     include: {
       program: {

Diff for: apps/web/lib/api/partners/enroll-partner.ts

@@ -7,6 +7,7 @@ import { prisma } from "@dub/prisma";
 import { Prisma } from "@dub/prisma/client";
 import { waitUntil } from "@vercel/functions";
 import { DubApiError } from "../errors";
+import { includeTags } from "../links/include-tags";
 
 export const enrollPartner = async ({
   programId,
@@ -101,13 +102,7 @@ export const enrollPartner = async ({
           data: {
             partnerId: upsertedPartner.id,
           },
-          include: {
-            tags: {
-              select: {
-                tag: true,
-              },
-            },
-          },
+          include: includeTags,
         })
         .then((link) => recordLink(link)),
       // TODO: Remove this once we open up partners.dub.co to everyone

Diff for: packages/ui/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@dub/ui",
   "description": "UI components for Dub.co",
-  "version": "0.2.20",
+  "version": "0.2.21",
   "sideEffects": false,
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",