Merge pull request dubinc#2407 from dubinc/update-partners-import

Update partners import script

Author: steven-tey

apps/web/lib/api/partners/create-and-enroll-partner.ts

@@ -30,6 +30,7 @@ export const createAndEnrollPartner = async ({
   tenantId,
   status = "approved",
   skipEnrollmentCheck = false,
+  enrolledAt,
 }: {
   program: Pick<
     ProgramProps,
@@ -46,6 +47,7 @@ export const createAndEnrollPartner = async ({
   tenantId?: string;
   status?: ProgramEnrollmentStatus;
   skipEnrollmentCheck?: boolean;
+  enrolledAt?: Date;
 }) => {
   if (!skipEnrollmentCheck && partner.email) {
     const programEnrollment = await prisma.programEnrollment.findFirst({
@@ -107,6 +109,9 @@ export const createAndEnrollPartner = async ({
           discountId !== program.defaultDiscountId && {
             discountId,
           }),
+        ...(enrolledAt && {
+          createdAt: enrolledAt,
+        }),
       },
     },
   };

apps/web/scripts/partners/import-partners.ts

@@ -1,4 +1,6 @@
 import { prisma } from "@dub/prisma";
+import { nanoid } from "@dub/utils";
+import slugify from "@sindresorhus/slugify";
 import "dotenv-flow/config";
 import * as fs from "fs";
 import * as Papa from "papaparse";
@@ -7,19 +9,35 @@ import { createPartnerLink } from "../../lib/api/partners/create-partner-link";
 
 const programId = "xxx";
 const userId = "xxx";
-const partnersToImport: { email: string; slug: string }[] = [];
+const partnersToImport: { email: string; slug: string; enrolledAt: Date }[] =
+  [];
 
 async function main() {
   Papa.parse(fs.createReadStream("affiliates.csv", "utf-8"), {
     header: true,
     skipEmptyLines: true,
-    step: (result: { data: { email: string; slug: string } }) => {
+    step: (result: {
+      data: { payment_email?: string; date_registered: string };
+    }) => {
+      const email = result.data.payment_email;
+      if (!email) {
+        return;
+      }
+
+      let slug = slugify(email.split("@")[0]);
+      // check if slug is already used by another partner in partnersToImport
+      while (partnersToImport.some((partner) => partner.slug === slug)) {
+        slug = `${slug}-${nanoid(4).toLowerCase()}`;
+      }
+
       partnersToImport.push({
-        email: result.data.email,
-        slug: result.data.slug.slice(1),
+        email,
+        slug,
+        enrolledAt: new Date(result.data.date_registered),
       });
     },
     complete: async () => {
+      console.table(partnersToImport);
       const program = await prisma.program.findUniqueOrThrow({
         where: {
           id: programId,
@@ -57,9 +75,12 @@ async function main() {
           link: partnerLink,
           workspace: program.workspace,
           partner: partnerToCreate,
+          enrolledAt: partner.enrolledAt,
         });
 
-        console.log(enrolledPartner);
+        console.log(
+          `Created and enrolled partner ${enrolledPartner.email} with link ${enrolledPartner?.links?.[0]?.shortLink}`,
+        );
       }
     },
   });