Merge pull request #615 from Shopify/04-30-update_to_include_discount_class_filtering

Update to include discount class filtering

Author: munarahmanatshop

discounts/javascript/discount/default/src/generate_cart_run.graphql.liquid

@@ -9,4 +9,7 @@ query CartInput {
       }
     }
   }
+  discount {
+    discountClasses
+  }
 }

discounts/javascript/discount/default/src/generate_cart_run.liquid

@@ -2,6 +2,7 @@
 // @ts-check
 
 import {
+  DiscountClass,
   OrderDiscountSelectionStrategy,
   ProductDiscountSelectionStrategy,
 } from '../generated/api';
@@ -20,63 +21,85 @@ export function generateCartRun(input) {
     if (!input.cart.lines.length) {
         throw new Error('No cart lines found');
     }
+
+    const hasOrderDiscountClass = input.discount.discountClasses.includes(
+        DiscountClass.Order
+    );
+
+    const hasProductDiscountClass = input.discount.discountClasses.includes(
+        DiscountClass.Product
+    );
+
+    if (!hasOrderDiscountClass && !hasProductDiscountClass) {
+        return { operations: [] };
+    }
+
     const maxCartLine = input.cart.lines.reduce((maxLine, line) => {
         if (line.cost.subtotalAmount.amount > maxLine.cost.subtotalAmount.amount) {
             return line;
         }
         return maxLine;
     }, input.cart.lines[0]);
-    return {
-        operations: [
-            {
-                orderDiscountsAdd: {
-                    candidates: [
-                        {
-                            message: '10% OFF ORDER',
-                            targets: [
-                                {
-                                    orderSubtotal: {
-                                        excludedCartLineIds: [],
-                                    },
-                                },
-                            ],
-                            value: {
-                                percentage: {
-                                    value: 10,
+
+    const operations = [];
+
+    if(hasOrderDiscountClass) {
+        operations.push({
+            orderDiscountsAdd: {
+                candidates: [
+                    {
+                        message: '10% OFF ORDER',
+                        targets: [
+                            {
+                                orderSubtotal: {
+                                    excludedCartLineIds: [],
                                 },
                             },
+                        ],
+                        value: {
+                            percentage: {
+                                value: 10,
+                            },
                         },
-                    ],
-                    selectionStrategy: OrderDiscountSelectionStrategy.First,
-                },
+                    },
+                ],
+                selectionStrategy: OrderDiscountSelectionStrategy.First,
             },
-            {
-                productDiscountsAdd: {
-                    candidates: [
-                        {
-                            message: '20% OFF PRODUCT',
-                            targets: [
-                                {
-                                    cartLine: {
-                                        id: maxCartLine.id,
-                                    },
-                                },
-                            ],
-                            value: {
-                                percentage: {
-                                    value: 20,
+        });
+    }
+
+    if (hasProductDiscountClass) {
+        operations.push({
+            productDiscountsAdd: {
+                candidates: [
+                    {
+                        message: '20% OFF PRODUCT',
+                        targets: [
+                            {
+                                cartLine: {
+                                    id: maxCartLine.id,
                                 },
                             },
+                        ],
+                        value: {
+                            percentage: {
+                                value: 20,
+                            },
                         },
-                    ],
-                    selectionStrategy: ProductDiscountSelectionStrategy.First,
-                },
+                    },
+                ],
+                selectionStrategy: ProductDiscountSelectionStrategy.First,
             },
-        ],
+        });
+    }
+
+    return {
+        operations,
     };
 }
 {%- elsif flavor contains "typescript" -%}
 import {
+  DiscountClass,
   OrderDiscountSelectionStrategy,
   ProductDiscountSelectionStrategy,
   CartInput,
@@ -88,60 +111,80 @@ export function generateCartRun(input: CartInput): CartLinesDiscountsGenerateRun
     throw new Error('No cart lines found');
   }
 
+  const hasOrderDiscountClass = input.discount.discountClasses.includes(
+    DiscountClass.Order
+  );
+
+  const hasProductDiscountClass = input.discount.discountClasses.includes(
+    DiscountClass.Product
+  );
+
+  if (!hasOrderDiscountClass && !hasProductDiscountClass) {
+    return { operations: [] };
+  }
+
   const maxCartLine = input.cart.lines.reduce((maxLine, line) => {
     if (line.cost.subtotalAmount.amount > maxLine.cost.subtotalAmount.amount) {
       return line;
     }
     return maxLine;
   }, input.cart.lines[0]);
 
-  return {
-    operations: [
-      {
-        orderDiscountsAdd: {
-          candidates: [
-            {
-              message: '10% OFF ORDER',
-              targets: [
-                {
-                  orderSubtotal: {
-                    excludedCartLineIds: [],
-                  },
-                },
-              ],
-              value: {
-                percentage: {
-                  value: 10,
+  const operations: CartLinesDiscountsGenerateRunResult['operations'] = [];
+
+  if (hasOrderDiscountClass) {
+    operations.push({
+      orderDiscountsAdd: {
+        candidates: [
+          {
+            message: '10% OFF ORDER',
+            targets: [
+              {
+                orderSubtotal: {
+                  excludedCartLineIds: [],
                 },
               },
+            ],
+            value: {
+              percentage: {
+                value: 10,
+              },
             },
-          ],
-          selectionStrategy: OrderDiscountSelectionStrategy.First,
-        },
-      },
-      {
-        productDiscountsAdd: {
-          candidates: [
-            {
-              message: '20% OFF PRODUCT',
-              targets: [
-                {
-                  cartLine: {
-                    id: maxCartLine.id,
-                  },
-                },
-              ],
-              value: {
-                percentage: {
-                  value: 20,
+          },
+        ],
+        selectionStrategy: OrderDiscountSelectionStrategy.First,
+      }
+    });
+  }
+
+  if (hasProductDiscountClass) {
+    operations.push({
+      productDiscountsAdd: {
+        candidates: [
+          {
+            message: '20% OFF PRODUCT',
+            targets: [
+              {
+                cartLine: {
+                  id: maxCartLine.id,
                 },
               },
+            ],
+            value: {
+              percentage: {
+                value: 20,
+              },
             },
-          ],
-          selectionStrategy: ProductDiscountSelectionStrategy.First,
-        },
+          },
+        ],
+        selectionStrategy: ProductDiscountSelectionStrategy.First,
       },
-    ],
+    });
+  };
+
+  return {
+    operations,
   };
 }
+
 {%- endif -%}