Customer subscription duration reward modifier

apps/web/app/(ee)/api/stripe/integration/webhook/checkout-session-completed.ts

@@ -463,6 +463,7 @@ export async function checkoutSessionCompleted(
       context: {
         customer: {
           country: customer.country,
+          firstSaleAt: customer.firstSaleAt,
         },
         sale: {
           productId,

apps/web/app/(ee)/api/stripe/integration/webhook/invoice-paid.ts

@@ -232,6 +232,7 @@ export async function invoicePaid(event: Stripe.Event, mode: StripeMode) {
       context: {
         customer: {
           country: customer.country,
+          firstSaleAt: customer.firstSaleAt,
         },
         sale: {
           productId: invoice.lines.data[0]?.pricing?.price_details?.product,

apps/web/lib/actions/partners/create-manual-commission.ts

@@ -292,8 +292,13 @@ export const createManualCommissionAction = authActionClient
               createdAt: new Date(saleEventData.timestamp + "Z"), // add the "Z" to the timestamp to make it UTC
               user,
               context: {
-                customer: { country: customer.country },
-                sale: { productId },
+                customer: {
+                  country: customer.country,
+                  firstSaleAt: customer.firstSaleAt,
+                },
+                sale: {
+                  productId,
+                },
               },
             })),
           );

apps/web/lib/api/conversions/track-sale.ts

@@ -552,10 +552,11 @@ const _trackSale = async ({
           context: {
             customer: {
               country: customer.country,
-              source: source!,
+              source,
+              firstSaleAt: customer.firstSaleAt,
             },
             sale: {
-              productId: metadata?.productId as string,
+              productId: metadata?.productId,
               amount: saleData.amount,
             },
           },

apps/web/lib/integrations/shopify/create-sale.ts

@@ -150,6 +150,7 @@ export async function createShopifySale({
       context: {
         customer: {
           country: customer.country,
+          firstSaleAt: customer.firstSaleAt,
         },
         sale: {
           amount: saleData.amount,

apps/web/lib/partners/determine-partner-reward.ts

@@ -1,4 +1,5 @@
 import { EventType, Link, Prisma, Reward } from "@dub/prisma/client";
+import { differenceInMonths } from "date-fns";
 import { serializeReward } from "../api/partners/serialize-reward";
 import { RewardContext } from "../types";
 import {
@@ -51,6 +52,13 @@ export const determinePartnerReward = ({
       totalCommissions: programEnrollment.totalCommissions,
       country: programEnrollment.partner?.country,
     },
+    customer: {
+      ...context?.customer,
+      subscriptionDuration: differenceInMonths(
+        new Date(),
+        context?.customer?.firstSaleAt ?? new Date(),
+      ),
+    },
   };
 
   if (partnerReward.modifiers && context) {

apps/web/lib/zod/schemas/rewards.ts

@@ -147,6 +147,11 @@ export const REWARD_CONDITIONS: Record<
               },
             ],
           },
+          {
+            id: "subscriptionDuration",
+            label: "Subscription duration",
+            type: "number",
+          },
         ],
       },
       PARTNER_ENTITY,
@@ -347,6 +352,8 @@ export const rewardContextSchema = z.object({
     .object({
       country: z.string().nullish(),
       source: z.enum(CUSTOMER_SOURCES).default("tracked").nullish(),
+      firstSaleAt: z.date().nullish(),
+      subscriptionDuration: z.number().nullish(),
     })
     .optional(),
 

apps/web/tests/rewards/reward-conditions.test.ts

@@ -1660,4 +1660,70 @@ describe("evaluateRewardConditions", () => {
       expect(result).toBeNull(); // Should not match when country is null
     });
   });
+
+  describe("subscription duration conditions", () => {
+    describe("customer.subscriptionDuration", () => {
+      test("should match when subscription duration meets greater_than_or_equal condition", () => {
+        const conditions = [
+          {
+            operator: "AND" as const,
+            type: "flat" as const,
+            amountInCents: 5000,
+            conditions: [
+              {
+                entity: "customer" as const,
+                attribute: "subscriptionDuration" as const,
+                operator: "greater_than_or_equal" as const,
+                value: 12,
+              },
+            ],
+          },
+        ];
+
+        const context: RewardContext = {
+          customer: {
+            subscriptionDuration: 12,
+          },
+        };
+
+        const result = evaluateRewardConditions({
+          conditions,
+          context,
+        });
+
+        expect(result).toEqual(conditions[0]);
+      });
+
+      test("should not match when subscription duration is less than condition value", () => {
+        const conditions = [
+          {
+            operator: "AND" as const,
+            type: "flat" as const,
+            amountInCents: 5000,
+            conditions: [
+              {
+                entity: "customer" as const,
+                attribute: "subscriptionDuration" as const,
+                operator: "greater_than_or_equal" as const,
+                value: 12,
+              },
+            ],
+          },
+        ];
+
+        const context: RewardContext = {
+          customer: {
+            subscriptionDuration: 6,
+          },
+        };
+
+        const result = evaluateRewardConditions({
+          conditions,
+          context,
+        });
+
+        expect(result).toBe(null);
+      });
+    });
+  });
 });

apps/web/ui/partners/program-reward-modifiers-tooltip.tsx

@@ -17,6 +17,7 @@ import {
   currencyFormatter,
   pluralize,
 } from "@dub/utils";
+import { formatDuration } from "date-fns";
 import { useRef } from "react";
 import * as z from "zod/v4";
 
@@ -118,6 +119,8 @@ export function ProgramRewardModifiersTooltipContent({
   );
 }
 
+// TODO:
+// This became a bit of a mess, let's clean it up a bit.
 const RewardItem = ({
   reward,
   conditions,
@@ -169,7 +172,12 @@ const RewardItem = ({
                 <span className="min-w-0">
                   {idx === 0 ? "If" : capitalize(operator.toLowerCase())}{" "}
                   {condition.entity}{" "}
-                  {(attribute?.label ?? condition.attribute)?.toLowerCase()}{" "}
+                  {(
+                    attribute?.label ??
+                    (condition.attribute === "subscriptionDuration"
+                      ? "Subscription duration"
+                      : condition.attribute)
+                  )?.toLowerCase()}{" "}
                   {CONDITION_OPERATOR_LABELS[condition.operator]}{" "}
                   {condition.value &&
                     (condition.attribute === "country"
@@ -180,29 +188,36 @@ const RewardItem = ({
                             .join(", ")
                         : COUNTRIES[condition.value?.toString()] ??
                           condition.value
-                      : // Non-country value(s)
-                        Array.isArray(condition.value)
-                        ? // Basic array
-                          (attribute?.options
-                            ? (condition.value as string[] | number[]).map(
-                                (v) =>
-                                  attribute.options?.find((o) => o.id === v)
-                                    ?.label ?? v,
-                              )
-                            : condition.value
-                          ).join(", ")
-                        : condition.attribute === "productId" && condition.label
-                          ? // Product label
-                            condition.label
-                          : attribute?.type === "currency"
-                            ? // Currency value
-                              currencyFormatter(Number(condition.value))
-                            : // Everything else
-                              attribute?.options
-                              ? attribute.options.find(
-                                  (o) => o.id === condition.value,
-                                )?.label ?? condition.value.toString()
-                              : condition.value.toString())}
+                      : condition.attribute === "subscriptionDuration"
+                        ? Array.isArray(condition.value)
+                          ? (condition.value as number[])
+                              .map(formatSubscriptionDuration)
+                              .join(", ")
+                          : formatSubscriptionDuration(Number(condition.value))
+                        : // Non-country value(s)
+                          Array.isArray(condition.value)
+                          ? // Basic array
+                            (attribute?.options
+                              ? (condition.value as string[] | number[]).map(
+                                  (v) =>
+                                    attribute.options?.find((o) => o.id === v)
+                                      ?.label ?? v,
+                                )
+                              : condition.value
+                            ).join(", ")
+                          : condition.attribute === "productId" &&
+                              condition.label
+                            ? // Product label
+                              condition.label
+                            : attribute?.type === "currency"
+                              ? // Currency value
+                                currencyFormatter(Number(condition.value))
+                              : // Everything else
+                                attribute?.options
+                                ? attribute.options.find(
+                                    (o) => o.id === condition.value,
+                                  )?.label ?? condition.value.toString()
+                                : condition.value.toString())}
                 </span>
               </li>
             );
@@ -212,3 +227,9 @@ const RewardItem = ({
     </div>
   );
 };
+
+function formatSubscriptionDuration(v: number): string {
+  return formatDuration(
+    v >= 12 ? { years: Math.floor(v / 12), months: v % 12 } : { months: v },
+  );
+}