dubinc · steven-tey · Jul 17, 2025 · Jul 13, 2025 · Jul 13, 2025 · Jul 14, 2025
diff --git a/apps/web/app/(ee)/api/cron/import/partnerstack/route.ts b/apps/web/app/(ee)/api/cron/import/partnerstack/route.ts
@@ -0,0 +1,48 @@
+import { handleAndReturnErrorResponse } from "@/lib/api/errors";
+import { verifyQstashSignature } from "@/lib/cron/verify-qstash";
+import { importCommissions } from "@/lib/partnerstack/import-commissions";
+import { importCustomers } from "@/lib/partnerstack/import-customers";
+import { importLinks } from "@/lib/partnerstack/import-links";
+import { importPartners } from "@/lib/partnerstack/import-partners";
+import { partnerStackImportPayloadSchema } from "@/lib/partnerstack/schemas";
+import { updateStripeCustomers } from "@/lib/partnerstack/update-stripe-customers";
+import { NextResponse } from "next/server";
+
+export const dynamic = "force-dynamic";
+
+export async function POST(req: Request) {
+  try {
+    const rawBody = await req.text();
+
+    await verifyQstashSignature({
+      req,
+      rawBody,
+    });
+
+    const payload = partnerStackImportPayloadSchema.parse(JSON.parse(rawBody));
+
+    switch (payload.action) {
+      case "import-partners":
+        await importPartners(payload);
+        break;
+      case "import-links":
+        await importLinks(payload);
+        break;
+      case "import-customers":
+        await importCustomers(payload);
+        break;
+      case "import-commissions":
+        await importCommissions(payload);
+        break;
+      case "update-stripe-customers":
+        await updateStripeCustomers(payload);
+        break;
+      default:
+        throw new Error(`Unknown action: ${payload.action}`);
+    }
+
+    return NextResponse.json("OK");
+  } catch (error) {
+    return handleAndReturnErrorResponse(error);
+  }
+}
diff --git a/apps/web/app/app.dub.co/(dashboard)/[slug]/(ee)/program/partners/import-export-buttons.tsx b/apps/web/app/app.dub.co/(dashboard)/[slug]/(ee)/program/partners/import-export-buttons.tsx
@@ -3,6 +3,7 @@
 import { PROGRAM_IMPORT_SOURCES } from "@/lib/partners/constants";
 import useWorkspace from "@/lib/swr/use-workspace";
 import { useExportPartnersModal } from "@/ui/modals/export-partners-modal";
+import { useImportPartnerStackModal } from "@/ui/modals/import-partnerstack-modal";
 import { useImportRewardfulModal } from "@/ui/modals/import-rewardful-modal";
 import { useImportToltModal } from "@/ui/modals/import-tolt-modal";
 import { Download, ThreeDots } from "@/ui/shared/icons";
@@ -17,6 +18,7 @@ export function ImportExportButtons() {
 
   const { ImportToltModal } = useImportToltModal();
   const { ImportRewardfulModal } = useImportRewardfulModal();
+  const { ImportPartnerStackModal } = useImportPartnerStackModal();
 
   const { ExportPartnersModal, setShowExportPartnersModal } =
     useExportPartnersModal();
@@ -25,10 +27,11 @@ export function ImportExportButtons() {
     <>
       <ImportToltModal />
       <ImportRewardfulModal />
+      <ImportPartnerStackModal />
       <ExportPartnersModal />
       <Popover
         content={
-          <div className="w-full md:w-52">
+          <div className="w-full md:w-56">
             <div className="grid gap-px p-2">
               <p className="mb-1.5 mt-1 flex items-center gap-2 px-1 text-xs font-medium text-neutral-500">
                 Import Partners

diff --git a/apps/web/lib/actions/partners/start-partnerstack-import.ts b/apps/web/lib/actions/partners/start-partnerstack-import.ts
@@ -0,0 +1,54 @@
+"use server";
+
+import { getDefaultProgramIdOrThrow } from "@/lib/api/programs/get-default-program-id-or-throw";
+import { getProgramOrThrow } from "@/lib/api/programs/get-program-or-throw";
+import { PartnerStackApi } from "@/lib/partnerstack/api";
+import { partnerStackImporter } from "@/lib/partnerstack/importer";
+import { z } from "zod";
+import { authActionClient } from "../safe-action";
+
+const schema = z.object({
+  workspaceId: z.string(),
+  publicKey: z.string().min(1),
+  secretKey: z.string().min(1),
+});
+
+export const startPartnerStackImportAction = authActionClient
+  .schema(schema)
+  .action(async ({ ctx, parsedInput }) => {
+    const { workspace, user } = ctx;
+    const { publicKey, secretKey } = parsedInput;
+
+    const programId = getDefaultProgramIdOrThrow(workspace);
+
+    const program = await getProgramOrThrow({
+      workspaceId: workspace.id,
+      programId,
+    });
+
+    if (!program.domain) {
+      throw new Error("Program domain is not set.");
+    }
+
+    if (!program.url) {
+      throw new Error("Program URL is not set.");
+    }
+
+    const partnerStackApi = new PartnerStackApi({
+      publicKey,
+      secretKey,
+    });
+
+    await partnerStackApi.testConnection();
+
+    await partnerStackImporter.setCredentials(workspace.id, {
+      publicKey,
+      secretKey,
+    });
+
+    await partnerStackImporter.queue({
+      programId,
+      userId: user.id,
+      action: "import-partners",
+    });
+  });
diff --git a/apps/web/lib/partners/constants.ts b/apps/web/lib/partners/constants.ts
@@ -78,4 +78,10 @@ export const PROGRAM_IMPORT_SOURCES = [
     image: "https://assets.dub.co/misc/icons/tolt.svg",
     helpUrl: "https://dub.co/help/article/migrating-from-tolt",
   },
+  {
+    id: "partnerstack",
+    value: "PartnerStack",
+    image: "https://assets.dub.co/misc/icons/partnerstack.svg",
+    helpUrl: "https://dub.co/help/article/migrating-from-partnerstack",
+  },
 ] as const;
diff --git a/apps/web/lib/partnerstack/api.ts b/apps/web/lib/partnerstack/api.ts
@@ -0,0 +1,125 @@
+import {
+  partnerStackCommission,
+  partnerStackCustomer,
+  partnerStackLink,
+  partnerStackPartner,
+} from "./schemas";
+import {
+  PartnerStackCommission,
+  PartnerStackCustomer,
+  PartnerStackLink,
+  PartnerStackListResponse,
+  PartnerStackPartner,
+} from "./types";
+
+const PAGE_LIMIT = 100;
+
+export class PartnerStackApi {
+  private readonly baseUrl = "https://api.partnerstack.com/api/v2";
+  private readonly publicKey: string;
+  private readonly secretKey: string;
+
+  constructor({
+    publicKey,
+    secretKey,
+  }: {
+    publicKey: string;
+    secretKey: string;
+  }) {
+    this.publicKey = publicKey;
+    this.secretKey = secretKey;
+  }
+
+  private async fetch<T>(path: string): Promise<T> {
+    const token = Buffer.from(`${this.publicKey}:${this.secretKey}`).toString(
+      "base64",
+    );
+
+    const response = await fetch(`${this.baseUrl}${path}`, {
+      headers: {
+        Authorization: `Basic ${token}`,
+      },
+    });
+
+    if (!response.ok) {
+      const error = await response.json();
+
+      console.error("PartnerStack API Error:", error);
+
+      throw new Error(error.message || "Unknown error from PartnerStack API.");
+    }
+
+    return await response.json();
+  }
+
+  async testConnection() {
+    try {
+      await this.fetch("/customers?limit=1");
+      return true;
+    } catch (error) {
+      throw new Error("Invalid PartnerStack API token.");
+    }
+  }
+
+  async listPartners({ startingAfter }: { startingAfter?: string }) {
+    const searchParams = new URLSearchParams();
+    searchParams.append("approved_status", "approved");
+    searchParams.append("limit", PAGE_LIMIT.toString());
+
+    if (startingAfter) {
+      searchParams.append("starting_after", startingAfter);
+    }
+
+    const {
+      data: { items },
+    } = await this.fetch<PartnerStackListResponse<PartnerStackPartner>>(
+      `/partnerships?${searchParams.toString()}`,
+    );
+
+    return partnerStackPartner.array().parse(items);
+  }
+
+  async listLinks({ identifier }: { identifier: string }) {
+    const {
+      data: { items },
+    } = await this.fetch<PartnerStackListResponse<PartnerStackLink>>(
+      `/links/partnership/${identifier}`,
+    );
+
+    return partnerStackLink.array().parse(items);
+  }
+
+  async listCustomers({ startingAfter }: { startingAfter?: string }) {
+    const searchParams = new URLSearchParams();
+    searchParams.append("limit", PAGE_LIMIT.toString());
+
+    if (startingAfter) {
+      searchParams.append("starting_after", startingAfter);
+    }
+
+    const {
+      data: { items },
+    } = await this.fetch<PartnerStackListResponse<PartnerStackCustomer>>(
+      `/customers?${searchParams.toString()}`,
+    );
+
+    return partnerStackCustomer.array().parse(items);
+  }
+
+  async listCommissions({ startingAfter }: { startingAfter?: string }) {
+    const searchParams = new URLSearchParams();
+    searchParams.append("limit", PAGE_LIMIT.toString());
+
+    if (startingAfter) {
+      searchParams.append("starting_after", startingAfter);
+    }
+
+    const {
+      data: { items },
+    } = await this.fetch<PartnerStackListResponse<PartnerStackCommission>>(
+      `/rewards?${searchParams.toString()}`,
+    );
+
+    return partnerStackCommission.array().parse(items);
+  }
+}