Allow targeting campaigns and bounties by partner tags

apps/web/app/(ee)/api/bounties/[bountyId]/route.ts

@@ -1,6 +1,7 @@
 import { recordAuditLog } from "@/lib/api/audit-logs/record-audit-log";
 import { DubApiError } from "@/lib/api/errors";
 import { throwIfInvalidGroupIds } from "@/lib/api/groups/throw-if-invalid-group-ids";
+import { throwIfInvalidPartnerTagIds } from "@/lib/api/partner-tags/throw-if-invalid-partner-tag-ids";
 import { getDefaultProgramIdOrThrow } from "@/lib/api/programs/get-default-program-id-or-throw";
 import { parseRequestBody } from "@/lib/api/utils";
 import { WorkflowCondition } from "@/lib/api/workflows/types";
@@ -11,6 +12,7 @@ import { generatePerformanceBountyName } from "@/lib/bounty/api/generate-perform
 import { getBountyOrThrow } from "@/lib/bounty/api/get-bounty-or-throw";
 import { getBountyWithDetails } from "@/lib/bounty/api/get-bounty-with-details";
 import { PERFORMANCE_BOUNTY_SCOPE_ATTRIBUTES } from "@/lib/bounty/api/performance-bounty-scope-attributes";
+import { transformBounty } from "@/lib/bounty/api/transform-bounty";
 import { shouldUpsertDraftSubmissionsOnReopen } from "@/lib/bounty/api/upsert-draft-bounty-submissions";
 import { validateBounty } from "@/lib/bounty/api/validate-bounty";
 import { qstash } from "@/lib/cron";
@@ -22,7 +24,12 @@ import {
   submissionRequirementsSchema,
   updateBountySchema,
 } from "@/lib/zod/schemas/bounties";
-import { APP_DOMAIN_WITH_NGROK, arrayEqual, deepEqual } from "@dub/utils";
+import {
+  APP_DOMAIN_WITH_NGROK,
+  arrayEqual,
+  deepEqual,
+  pluck,
+} from "@dub/utils";
 import { BountyStartMode, PartnerGroup, Prisma } from "@prisma/client";
 import { waitUntil } from "@vercel/functions";
 import { NextResponse } from "next/server";
@@ -69,6 +76,7 @@ export const PATCH = withWorkspace(
       submissionRequirements,
       performanceCondition,
       groupIds,
+      partnerTagIds,
     } = updateBountySchema.parse(await parseRequestBody(req));
 
     const bounty = await getBountyOrThrow({
@@ -90,13 +98,16 @@ export const PATCH = withWorkspace(
 
     // Absolute end dates are cleared when switching to relative (unless the
     // client explicitly sends endsAt) or when setting endsAfterDays.
+    // Do not clear a relative bounty's fixed endsAt on unrelated PATCHes —
+    // relative + calendar endsAt is a supported shape (custom end).
     let endsAtUpdate: { endsAt?: Date | null } = {};
 
     if (endsAt !== undefined) {
       endsAtUpdate = { endsAt };
     } else if (endsAfterDays != null) {
       endsAtUpdate = { endsAt: null };
     } else if (
+      bounty.startMode === BountyStartMode.absolute &&
       nextStartMode === BountyStartMode.relative &&
       bounty.endsAt != null
     ) {
@@ -159,9 +170,11 @@ export const PATCH = withWorkspace(
     // if groupIds is provided and is different from the current groupIds, update the groups
     let updatedPartnerGroups: PartnerGroup[] | undefined = undefined;
     let shouldUpdatePartnerGroups = false;
+    let updatedPartnerTags: { id: string }[] | undefined = undefined;
+    let shouldUpdatePartnerTags = false;
 
     if (groupIds !== undefined) {
-      const currentGroupIds = bounty.groups.map((group) => group.groupId);
+      const currentGroupIds = pluck(bounty.groups, "groupId");
       const newGroupIds = groupIds || [];
 
       if (!arrayEqual(currentGroupIds, newGroupIds)) {
@@ -176,6 +189,22 @@ export const PATCH = withWorkspace(
       }
     }
 
+    if (partnerTagIds !== undefined) {
+      const currentPartnerTagIds = pluck(bounty.partnerTags, "partnerTagId");
+      const newPartnerTagIds = partnerTagIds || [];
+
+      if (!arrayEqual(currentPartnerTagIds, newPartnerTagIds)) {
+        if (newPartnerTagIds.length > 0) {
+          updatedPartnerTags = await throwIfInvalidPartnerTagIds({
+            programId,
+            partnerTagIds: newPartnerTagIds,
+          });
+        }
+
+        shouldUpdatePartnerTags = true;
+      }
+    }
+
     // Prevent updates if `performanceCondition.attribute` differs from the current value if there are existing submissions
     if (performanceCondition && bounty.workflow) {
       const submissionCount = bounty._count.submissions;
@@ -285,6 +314,17 @@ export const PATCH = withWorkspace(
                 }),
             },
           }),
+          ...(shouldUpdatePartnerTags && {
+            partnerTags: {
+              deleteMany: {},
+              ...(updatedPartnerTags &&
+                updatedPartnerTags.length > 0 && {
+                  create: updatedPartnerTags.map((tag) => ({
+                    partnerTagId: tag.id,
+                  })),
+                }),
+            },
+          }),
         },
         include: {
           workflow: true,
@@ -309,11 +349,7 @@ export const PATCH = withWorkspace(
       };
     });
 
-    const updatedBounty = BountySchema.parse({
-      ...data,
-      groups: data.groups.map(({ groupId }) => ({ id: groupId })),
-      performanceCondition: data.workflow?.triggerConditions?.[0],
-    });
+    const updatedBounty = BountySchema.parse(transformBounty(data));
 
     const shouldUpsertDraftSubmissions = shouldUpsertDraftSubmissionsOnReopen({
       type: bounty.type,
@@ -412,11 +448,7 @@ export const DELETE = withWorkspace(
       }
     });
 
-    const deletedBounty = BountySchema.parse({
-      ...bounty,
-      groups: bounty.groups.map(({ groupId }) => ({ id: groupId })),
-      performanceCondition: bounty.workflow?.triggerConditions?.[0],
-    });
+    const deletedBounty = BountySchema.parse(transformBounty(bounty));
 
     waitUntil(
       recordAuditLog({

apps/web/app/(ee)/api/bounties/route.ts

@@ -2,6 +2,7 @@ import { recordAuditLog } from "@/lib/api/audit-logs/record-audit-log";
 import { createId } from "@/lib/api/create-id";
 import { DubApiError } from "@/lib/api/errors";
 import { throwIfInvalidGroupIds } from "@/lib/api/groups/throw-if-invalid-group-ids";
+import { throwIfInvalidPartnerTagIds } from "@/lib/api/partner-tags/throw-if-invalid-partner-tag-ids";
 import { getDefaultProgramIdOrThrow } from "@/lib/api/programs/get-default-program-id-or-throw";
 import { getProgramEnrollmentOrThrow } from "@/lib/api/programs/get-program-enrollment-or-throw";
 import { parseRequestBody } from "@/lib/api/utils";
@@ -15,6 +16,7 @@ import {
   isPartnerEligibleForBounty,
 } from "@/lib/bounty/api/bounty-availability";
 import { generatePerformanceBountyName } from "@/lib/bounty/api/generate-performance-bounty-name";
+import { transformBounty } from "@/lib/bounty/api/transform-bounty";
 import { validateBounty } from "@/lib/bounty/api/validate-bounty";
 import { qstash } from "@/lib/cron";
 import { getPlanCapabilities } from "@/lib/plan-capabilities";
@@ -26,11 +28,8 @@ import {
   createBountySchema,
   getBountiesQuerySchema,
 } from "@/lib/zod/schemas/bounties";
-import {
-  WORKFLOW_ACTION_TYPES,
-  WORKFLOW_ATTRIBUTE_TRIGGER,
-} from "@/lib/zod/schemas/workflows";
-import { APP_DOMAIN_WITH_NGROK } from "@dub/utils";
+import { WORKFLOW_ACTION_TYPES } from "@/lib/zod/schemas/workflows";
+import { APP_DOMAIN_WITH_NGROK, pluck } from "@dub/utils";
 import { BountyStartMode, Workflow } from "@prisma/client";
 import { waitUntil } from "@vercel/functions";
 import { NextResponse } from "next/server";
@@ -53,18 +52,30 @@ export const GET = withWorkspace(
                 defaultGroupId: true,
               },
             },
+            programPartnerTags: {
+              select: {
+                partnerTagId: true,
+              },
+            },
           },
         })
       : null;
 
     const partnerGroupId =
       programEnrollment?.groupId || programEnrollment?.program.defaultGroupId;
+    const partnerTagIds = programEnrollment
+      ? pluck(programEnrollment.programPartnerTags, "partnerTagId")
+      : [];
 
     const [bounties, allBountiesSubmissionsCount] = await Promise.all([
       prisma.bounty.findMany({
         where: {
           programId,
-          ...(programEnrollment && buildBountyEligibilityWhere(partnerGroupId)),
+          ...(programEnrollment &&
+            buildBountyEligibilityWhere({
+              groupId: partnerGroupId,
+              partnerTagIds,
+            })),
         },
         include: {
           ...bountyEligibilityIncludes,
@@ -138,11 +149,10 @@ export const GET = withWorkspace(
 
       return [
         BountyListSchema.parse({
-          ...bounty,
+          ...transformBounty(bounty),
           ...(allBountiesSubmissionsCount && {
             submissionsCountData: aggregateSubmissionsCountForBounty(bounty.id),
           }),
-          groups: bounty.groups.map(({ groupId }) => ({ id: groupId })),
         }),
       ];
     });
@@ -174,6 +184,7 @@ export const POST = withWorkspace(
       maxSubmissions,
       submissionRequirements,
       groupIds,
+      partnerTagIds,
       performanceCondition,
       performanceScope,
       sendNotificationEmails,
@@ -205,6 +216,11 @@ export const POST = withWorkspace(
       groupIds,
     });
 
+    const partnerTags = await throwIfInvalidPartnerTagIds({
+      programId,
+      partnerTagIds,
+    });
+
     // Bounty name
     let bountyName = name;
 
@@ -244,7 +260,6 @@ export const POST = withWorkspace(
           data: {
             id: createId({ prefix: "wf_" }),
             programId,
-            trigger: WORKFLOW_ATTRIBUTE_TRIGGER[performanceCondition.attribute],
             triggerConditions: [performanceCondition],
             actions: [action],
           },
@@ -284,6 +299,15 @@ export const POST = withWorkspace(
               },
             },
           }),
+          ...(partnerTags.length && {
+            partnerTags: {
+              createMany: {
+                data: partnerTags.map(({ id }) => ({
+                  partnerTagId: id,
+                })),
+              },
+            },
+          }),
         },
         include: {
           workflow: true,
@@ -292,11 +316,7 @@ export const POST = withWorkspace(
       });
     });
 
-    const createdBounty = BountySchema.parse({
-      ...bounty,
-      groups: bounty.groups.map(({ groupId }) => ({ id: groupId })),
-      performanceCondition: bounty.workflow?.triggerConditions?.[0],
-    });
+    const createdBounty = BountySchema.parse(transformBounty(bounty));
 
     const shouldScheduleDraftSubmissions =
       bounty.type === "performance" &&

apps/web/app/(ee)/api/campaigns/[campaignId]/duplicate/route.ts

@@ -1,5 +1,6 @@
 import { DEFAULT_CAMPAIGN_BODY } from "@/lib/api/campaigns/constants";
 import { getCampaignOrThrow } from "@/lib/api/campaigns/get-campaign-or-throw";
+import { campaignEligibilityIncludes } from "@/lib/api/campaigns/transform-campaign";
 import { createId } from "@/lib/api/create-id";
 import { getDefaultProgramIdOrThrow } from "@/lib/api/programs/get-default-program-id-or-throw";
 import { parseWorkflowConfig } from "@/lib/api/workflows/parse-workflow-config";
@@ -17,24 +18,33 @@ export const POST = withWorkspace(
     const campaign = await getCampaignOrThrow({
       programId,
       campaignId,
-      includeWorkflow: true,
-      includeGroups: true,
+      include: {
+        ...campaignEligibilityIncludes,
+        workflow: {
+          select: {
+            id: true,
+            actions: true,
+            triggerConditions: true,
+          },
+        },
+      },
     });
 
     const duplicatedCampaign = await prisma.$transaction(async (tx) => {
       let workflowId: string | null = null;
       const campaignId = createId({ prefix: "cmp_" });
 
       if (campaign.workflow) {
-        const { action, condition } = parseWorkflowConfig(campaign.workflow);
+        const { action, conditions: triggerConditions } = parseWorkflowConfig(
+          campaign.workflow,
+        );
 
         const newWorkflow = await tx.workflow.create({
           data: {
             id: createId({ prefix: "wf_" }),
             programId,
             name: campaign.name,
-            trigger: campaign.workflow.trigger,
-            triggerConditions: [condition],
+            triggerConditions,
             actions: [
               {
                 ...action,
@@ -70,6 +80,13 @@ export const POST = withWorkspace(
               })),
             },
           },
+          partnerTags: {
+            createMany: {
+              data: campaign.partnerTags.map(({ partnerTagId }) => ({
+                partnerTagId,
+              })),
+            },
+          },
         },
       });
     });