feat: update current & surprise queries to support new endpoint

server/routers/contentstack/reward/output.ts

@@ -91,22 +91,6 @@ export const validateApiTierRewardsSchema = z.record(
   )
 )
 
-export const validateApiAllTiersSchema = z.record(
-  z.nativeEnum(TierKey).transform((data) => {
-    return TierKey[data as unknown as Key]
-  }),
-  z.array(
-    z.object({
-      id: z.string().optional(),
-      status: z.string().optional(),
-      rewardId: z.string().optional(),
-      rewardTierLevel: z.string().optional(),
-      rewardType: z.string().optional(),
-      title: z.string().optional(),
-    })
-  )
-)
-
 export const validateCmsRewardsSchema = z
   .object({
     data: z.object({
@@ -138,3 +122,61 @@ export type SurpriseReward = z.output<typeof SurpriseReward>
 export type CmsRewardsResponse = z.input<typeof validateCmsRewardsSchema>
 
 export type Reward = z.output<typeof validateCmsRewardsSchema>[0]
+
+// New endpoint related types and schemas.
+
+const BenefitReward = z.object({
+  title: z.string().optional(),
+  id: z.string().optional(),
+  status: z.string().optional(),
+  rewardId: z.string().optional(),
+  rewardType: z.string().optional(),
+  rewardTierLevel: z.string().optional(),
+})
+
+const CouponState = z.enum(["claimed", "redeemed", "viewed"])
+const CouponData = z.object({
+  couponCode: z.string().optional(),
+  unwrapped: z.boolean().default(false),
+  state: CouponState,
+  expiresAt: z.string().datetime({ offset: true }).optional(),
+})
+
+const CouponReward = z.object({
+  title: z.string().optional(),
+  id: z.string().optional(),
+  rewardId: z.string().optional(),
+  rewardType: z.string().optional(),
+  status: z.string().optional(),
+  coupon: z.array(CouponData).optional(),
+})
+
+/**
+ * Schema for the new /profile/v1/Reward endpoint.
+ *
+ * TODO: Once we fully migrate to the new endpoint:
+ * 1. Remove the data transform and use the categorized structure directly.
+ * 2. Simplify surprise filtering in the query.
+ */
+export const validateCategorizedRewardsSchema = z
+  .object({
+    benefits: z.array(BenefitReward),
+    coupons: z.array(CouponReward),
+  })
+  .transform((data) => [
+    ...data.benefits.map((benefit) => ({
+      ...benefit,
+      type: "custom" as const, // Added for legacy compatibility.
+    })),
+    ...data.coupons.map((coupon) => ({
+      ...coupon,
+      type: "coupon" as const, // Added for legacy compatibility.
+    })),
+  ])
+
+export const validateApiAllTiersSchema = z.record(
+  z.nativeEnum(TierKey).transform((data) => {
+    return TierKey[data as unknown as Key]
+  }),
+  z.array(BenefitReward)
+)