isCustomerSourceCondition

Author: steven-tey

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

@@ -113,6 +113,10 @@ export const REWARD_CONDITIONS: Record<
                 id: "submitted",
                 label: "qualified lead",
               },
+              {
+                id: "trial",
+                label: "free trial",
+              },
             ],
           },
         ],
@@ -206,9 +210,6 @@ export const NUMBER_CONDITION_OPERATORS: (typeof CONDITION_OPERATORS)[number][]
     "less_than_or_equal",
   ];
 
-export const SOURCE_CONDITION_OPERATORS: (typeof CONDITION_OPERATORS)[number][] =
-  ["equals_to", "not_equals"];
-
 export const CONDITION_OPERATOR_LABELS = {
   equals_to: "is",
   not_equals: "is not",
@@ -337,7 +338,7 @@ export const REWARD_EVENT_COLUMN_MAPPING = Object.freeze({
   sale: "saleRewardId",
 });
 
-export const CUSTOMER_SOURCES = ["tracked", "submitted"] as const;
+export const CUSTOMER_SOURCES = ["tracked", "submitted", "trial"] as const;
 
 export const rewardContextSchema = z.object({
   customer: z

apps/web/ui/partners/rewards/rewards-logic.tsx

@@ -11,7 +11,6 @@ import {
   NUMBER_CONDITION_OPERATORS,
   REWARD_CONDITIONS,
   RewardConditionEntityAttribute,
-  SOURCE_CONDITION_OPERATORS,
   STRING_CONDITION_OPERATORS,
 } from "@/lib/zod/schemas/rewards";
 import { X } from "@/ui/shared/icons";
@@ -303,6 +302,31 @@ function ConditionLogic({
 
   const [displayProductLabel, setDisplayProductLabel] = useState(false);
 
+  // Auto-set operator to "equals_to" for customer.source
+  const isCustomerSourceCondition =
+    condition.entity === "customer" && condition.attribute === "source";
+
+  useEffect(() => {
+    if (isCustomerSourceCondition && condition.operator !== "equals_to") {
+      setValue(
+        conditionKey,
+        {
+          ...condition,
+          operator: "equals_to",
+        },
+        {
+          shouldDirty: true,
+        },
+      );
+    }
+  }, [
+    isCustomerSourceCondition,
+    condition.operator,
+    condition,
+    conditionKey,
+    setValue,
+  ]);
+
   return (
     <div className="flex w-full flex-col">
       <div className="flex items-center justify-between p-2.5">
@@ -369,53 +393,54 @@ function ConditionLogic({
                     }))}
                   />
                 </InlineBadgePopover>{" "}
-                <InlineBadgePopover
-                  text={
-                    condition.operator
-                      ? CONDITION_OPERATOR_LABELS[condition.operator]
-                      : "Condition"
-                  }
-                  invalid={!condition.operator}
-                >
-                  <InlineBadgePopoverMenu
-                    selectedValue={condition.operator}
-                    onSelect={(value) =>
-                      setValue(
-                        conditionKey,
-                        {
-                          ...condition,
-                          operator:
-                            value as (typeof CONDITION_OPERATORS)[number],
-                          // Update value to array / string / number if needed
-                          ...(["in", "not_in"].includes(value)
-                            ? !Array.isArray(condition.value)
-                              ? { value: [] }
-                              : null
-                            : ["number", "currency"].includes(attributeType)
-                              ? typeof condition.value !== "number"
-                                ? { value: "" }
-                                : null
-                              : typeof condition.value !== "string"
-                                ? { value: "" }
-                                : null),
-                        },
-                        {
-                          shouldDirty: true,
-                        },
-                      )
+                {isCustomerSourceCondition ? (
+                  <span className="text-content-emphasis font-medium">is </span>
+                ) : (
+                  <InlineBadgePopover
+                    text={
+                      condition.operator
+                        ? CONDITION_OPERATOR_LABELS[condition.operator]
+                        : "Condition"
                     }
-                    items={(condition.entity === "customer" &&
-                    condition.attribute === "source"
-                      ? SOURCE_CONDITION_OPERATORS
-                      : ["number", "currency"].includes(attributeType)
+                    invalid={!condition.operator}
+                  >
+                    <InlineBadgePopoverMenu
+                      selectedValue={condition.operator}
+                      onSelect={(value) =>
+                        setValue(
+                          conditionKey,
+                          {
+                            ...condition,
+                            operator:
+                              value as (typeof CONDITION_OPERATORS)[number],
+                            // Update value to array / string / number if needed
+                            ...(["in", "not_in"].includes(value)
+                              ? !Array.isArray(condition.value)
+                                ? { value: [] }
+                                : null
+                              : ["number", "currency"].includes(attributeType)
+                                ? typeof condition.value !== "number"
+                                  ? { value: "" }
+                                  : null
+                                : typeof condition.value !== "string"
+                                  ? { value: "" }
+                                  : null),
+                          },
+                          {
+                            shouldDirty: true,
+                          },
+                        )
+                      }
+                      items={(["number", "currency"].includes(attributeType)
                         ? NUMBER_CONDITION_OPERATORS
                         : STRING_CONDITION_OPERATORS
-                    ).map((operator) => ({
-                      text: CONDITION_OPERATOR_LABELS[operator],
-                      value: operator,
-                    }))}
-                  />
-                </InlineBadgePopover>{" "}
+                      ).map((operator) => ({
+                        text: CONDITION_OPERATOR_LABELS[operator],
+                        value: operator,
+                      }))}
+                    />
+                  </InlineBadgePopover>
+                )}{" "}
                 {condition.operator && (
                   <>
                     <InlineBadgePopover