feat: add support for discountId in pricing preview queries (#3447)

Author: idoshamun

src/common/paddle/pricing.ts

@@ -213,18 +213,24 @@ export const getPricingMetadataByPriceIds = async (
   return Object.fromEntries(items.map(({ item }) => [item.idMap.paddle, item]));
 };
 
-export const getPlusPricePreview = async (ctx: AuthContext, ids: string[]) => {
+export const getPlusPricePreview = async (
+  ctx: AuthContext,
+  ids: string[],
+  discountId?: string,
+) => {
   const region = ctx.region;
   const sortedIds = ids.sort();
 
   const hmac = createHmac('sha1', StorageTopic.Paddle);
   hmac.update(sortedIds.toString());
   const pricesHash = hmac.digest().toString('hex');
 
+  // Include discountId in cache key when present
+  const cacheKeyParts = [pricesHash, region, discountId].filter(Boolean);
   const redisKey = generateStorageKey(
     StorageTopic.Paddle,
     StorageKey.PricingPreviewPlus,
-    [pricesHash, region].join(':'),
+    cacheKeyParts.join(':'),
   );
 
   const redisResult = await getRedisObject(redisKey);
@@ -239,6 +245,7 @@ export const getPlusPricePreview = async (ctx: AuthContext, ids: string[]) => {
       quantity: 1,
     })),
     address: region ? { countryCode: region as CountryCode } : undefined,
+    discountId,
   });
 
   await setRedisObjectWithExpiry(

src/schema/paddle.ts

@@ -134,7 +134,11 @@ export const typeDefs = /* GraphQL */ `
     pricePreviews: PricePreviews! @auth
     corePricePreviews: PricePreviews! @auth
     pricingMetadata(type: PricingType): [ProductPricingMetadata!]! @auth
-    pricingPreview(type: PricingType, locale: String): [ProductPricingPreview!]!
+    pricingPreview(
+      type: PricingType
+      locale: String
+      discountId: String
+    ): [ProductPricingPreview!]!
     pricingPreviewByIds(
       """
       The IDs of the prices to preview
@@ -323,6 +327,7 @@ export interface GQLCustomData {
 interface PaddlePricingPreviewArgs {
   type?: PurchaseType;
   locale?: string;
+  discountId?: string;
 }
 
 interface PaddlePricingPreviewByIdsArgs {
@@ -445,15 +450,19 @@ export const resolvers: IResolvers<unknown, AuthContext> = traceResolvers<
     ): Promise<BasePricingMetadata[]> => getPricingMetadata(ctx, type),
     pricingPreview: async (
       _,
-      { type = PurchaseType.Plus, locale }: PaddlePricingPreviewArgs,
+      {
+        type = PurchaseType.Plus,
+        locale,
+        discountId,
+      }: PaddlePricingPreviewArgs,
       ctx,
     ): Promise<BasePricingPreview[]> => {
       const metadata = await getPricingMetadata(ctx, type);
       const ids = metadata
         .map(({ idMap }) => idMap.paddle)
         .filter(Boolean) as string[];
 
-      const preview = await getPlusPricePreview(ctx, ids);
+      const preview = await getPlusPricePreview(ctx, ids, discountId);
 
       // consolidate the preview data and metadata
       const consolidated = metadata.map((meta) => {