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,6 +21,18 @@ 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;

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

@@ -4,4 +4,7 @@ query DeliveryInput {
       id
     }
   }
+  discount {
+    discountClasses
+  }
 }

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

@@ -1,7 +1,10 @@
 {%- if flavor contains "vanilla-js" -%}
 // @ts-check
 
-import {  DeliveryDiscountSelectionStrategy} from '../generated/api';
+import {
+  DeliveryDiscountSelectionStrategy,
+  DiscountClass,
+} from "../generated/api";
 
 /**
  * @typedef {import("../generated/api").Input} RunInput
@@ -17,6 +20,15 @@ export function generateDeliveryRun(input) {
   if (!firstDeliveryGroup) {
       throw new Error('No delivery groups found');
   }
+
+  const hasShippingDiscountClass = input.discount.discountClasses.includes(
+    DiscountClass.Shipping
+  );
+
+  if (!hasShippingDiscountClass) {
+    return { operations: [] };
+  }
+
   return {
       operations: [
           {

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

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

discounts/rust/discount/default/src/generate_cart_run.rs

@@ -9,7 +9,7 @@ use cart_lines_discounts_generate_run::output::{
     ProductDiscountSelectionStrategy, ProductDiscountsAddOperation,
 };
 
-use cart_lines_discounts_generate_run::input::ResponseData;
+use cart_lines_discounts_generate_run::input::{ResponseData, DiscountClass};
 
 #[shopify_function_target(
     target = "cartLinesDiscountsGenerateRun",
@@ -30,38 +30,56 @@ fn generate_cart_run(input: ResponseData) -> Result<CartLinesDiscountsGenerateRu
         })
         .ok_or("No cart lines found")?;
 
+    let has_order_discount_class = input.discount.discount_classes.contains(&DiscountClass::ORDER);
+    let has_product_discount_class = input.discount.discount_classes.contains(&DiscountClass::PRODUCT);
+
+    if !has_order_discount_class && !has_product_discount_class {
+        return Ok(CartLinesDiscountsGenerateRunResult {
+            operations: vec![],
+        });
+    }
+
+    let mut operations = vec![];
+
+    // Check if the discount has the ORDER class
+    if has_order_discount_class {
+      operations.push(CartOperation::OrderDiscountsAdd(OrderDiscountsAddOperation {
+          selection_strategy: OrderDiscountSelectionStrategy::FIRST,
+          candidates: vec![OrderDiscountCandidate {
+              targets: vec![OrderDiscountCandidateTarget::OrderSubtotal(
+                  OrderSubtotalTarget {
+                      excluded_cart_line_ids: vec![],
+                  },
+              )],
+              message: Some("10% OFF ORDER".to_string()),
+              value: OrderDiscountCandidateValue::Percentage(Percentage {
+                  value: Decimal(10.0),
+              }),
+              conditions: None,
+              associated_discount_code: None,
+          }],
+      }));
+    }
+
+    // Check if the discount has the PRODUCT class
+    if has_product_discount_class {
+        operations.push(CartOperation::ProductDiscountsAdd(ProductDiscountsAddOperation {
+            selection_strategy: ProductDiscountSelectionStrategy::FIRST,
+            candidates: vec![ProductDiscountCandidate {
+                targets: vec![ProductDiscountCandidateTarget::CartLine(CartLineTarget {
+                    id: max_cart_line.id.clone(),
+                    quantity: None,
+                })],
+                message: Some("20% OFF PRODUCT".to_string()),
+                value: ProductDiscountCandidateValue::Percentage(Percentage {
+                    value: Decimal(20.0),
+                }),
+                associated_discount_code: None,
+            }],
+        }));
+    }
+
     Ok(CartLinesDiscountsGenerateRunResult {
-        operations: vec![
-            CartOperation::OrderDiscountsAdd(OrderDiscountsAddOperation {
-                selection_strategy: OrderDiscountSelectionStrategy::FIRST,
-                candidates: vec![OrderDiscountCandidate {
-                    targets: vec![OrderDiscountCandidateTarget::OrderSubtotal(
-                        OrderSubtotalTarget {
-                            excluded_cart_line_ids: vec![],
-                        },
-                    )],
-                    message: Some("10% OFF ORDER".to_string()),
-                    value: OrderDiscountCandidateValue::Percentage(Percentage {
-                        value: Decimal(10.0),
-                    }),
-                    conditions: None,
-                    associated_discount_code: None,
-                }],
-            }),
-            CartOperation::ProductDiscountsAdd(ProductDiscountsAddOperation {
-                selection_strategy: ProductDiscountSelectionStrategy::FIRST,
-                candidates: vec![ProductDiscountCandidate {
-                    targets: vec![ProductDiscountCandidateTarget::CartLine(CartLineTarget {
-                        id: max_cart_line.id.clone(),
-                        quantity: None,
-                    })],
-                    message: Some("20% OFF PRODUCT".to_string()),
-                    value: ProductDiscountCandidateValue::Percentage(Percentage {
-                        value: Decimal(20.0),
-                    }),
-                    associated_discount_code: None,
-                }],
-            }),
-        ],
+      operations,
     })
 }

discounts/rust/discount/default/src/generate_delivery_run.graphql.liquid

@@ -4,4 +4,7 @@ query Input {
       id
     }
   }
+  discount {
+    discountClasses
+  }
 }

discounts/rust/discount/default/src/generate_delivery_run.rs

@@ -8,7 +8,7 @@ use cart_delivery_options_discounts_generate_run::output::{
     DeliveryOperation, Percentage,
 };
 
-use cart_delivery_options_discounts_generate_run::input::ResponseData;
+use cart_delivery_options_discounts_generate_run::input::{ResponseData, DiscountClass};
 
 #[shopify_function_target(
     target = "cartDeliveryOptionsDiscountsGenerateRun",
@@ -18,29 +18,36 @@ use cart_delivery_options_discounts_generate_run::input::ResponseData;
 fn generate_delivery_run(
     input: ResponseData,
 ) -> Result<CartDeliveryOptionsDiscountsGenerateRunResult> {
+    let has_shipping_discount_class = input.discount.discount_classes.contains(&DiscountClass::SHIPPING);
+      if !has_shipping_discount_class {
+          return Ok(CartDeliveryOptionsDiscountsGenerateRunResult {
+              operations: vec![],
+          });
+      }
+
     let first_delivery_group = input
         .cart
         .delivery_groups
         .first()
         .ok_or("No delivery groups found")?;
 
     Ok(CartDeliveryOptionsDiscountsGenerateRunResult {
-        operations: vec![DeliveryOperation::DeliveryDiscountsAdd(
-            DeliveryDiscountsAddOperation {
-                selection_strategy: DeliveryDiscountSelectionStrategy::ALL,
-                candidates: vec![DeliveryDiscountCandidate {
-                    targets: vec![DeliveryDiscountCandidateTarget::DeliveryGroup(
-                        DeliveryGroupTarget {
-                            id: first_delivery_group.id.clone(),
-                        },
-                    )],
-                    value: DeliveryDiscountCandidateValue::Percentage(Percentage {
-                        value: Decimal(100.0),
-                    }),
-                    message: Some("FREE DELIVERY".to_string()),
-                    associated_discount_code: None,
-                }],
-            },
-        )],
-    })
+      operations: vec![
+          DeliveryOperation::DeliveryDiscountsAdd(DeliveryDiscountsAddOperation {
+              selection_strategy: DeliveryDiscountSelectionStrategy::ALL,
+              candidates: vec![DeliveryDiscountCandidate {
+                  targets: vec![DeliveryDiscountCandidateTarget::DeliveryGroup(
+                      DeliveryGroupTarget {
+                          id: first_delivery_group.id.clone(),
+                      },
+                  )],
+                  value: DeliveryDiscountCandidateValue::Percentage(Percentage {
+                      value: Decimal(100.0),
+                  }),
+                  message: Some("FREE DELIVERY".to_string()),
+                  associated_discount_code: None,
+              }],
+          })
+      ],
+  })
 }