From 4b6f8b067d95e0aba33e8d062d6116acb440b029 Mon Sep 17 00:00:00 2001
From: lucasstark <lucas.stark@live.com>
Date: Fri, 23 Dec 2011 14:20:27 -0500
Subject: [PATCH] Added filter to coupon loading

---
 classes/coupons.class.php | 53 ++++++++++++++++++++++++---------------
 1 file changed, 33 insertions(+), 20 deletions(-)

diff --git a/classes/coupons.class.php b/classes/coupons.class.php
index 64950d18f40..18b44a51367 100644
--- a/classes/coupons.class.php
+++ b/classes/coupons.class.php
@@ -28,27 +28,40 @@ class woocommerce_coupon {
 		
 		$this->code = $code;
 		
-		$coupon = get_page_by_title( $this->code, 'OBJECT', 'shop_coupon' );
+		$coupon_data = apply_filters('woocommerce_get_shop_coupon_data', false, $code);
 		
-		if ($coupon && $coupon->post_status == 'publish') :
-			
-			$this->id					= $coupon->ID;
-			$this->type 				= get_post_meta($coupon->ID, 'discount_type', true);
-			$this->amount 				= get_post_meta($coupon->ID, 'coupon_amount', true);
-			$this->individual_use 		= get_post_meta($coupon->ID, 'individual_use', true);
-			$this->product_ids 			= array_filter(array_map('trim', explode(',', get_post_meta($coupon->ID, 'product_ids', true))));
-			$this->exclude_product_ids	= array_filter(array_map('trim', explode(',', get_post_meta($coupon->ID, 'exclude_product_ids', true))));
-			$this->usage_limit 			= get_post_meta($coupon->ID, 'usage_limit', true);
-			$this->usage_count 			= (int) get_post_meta($coupon->ID, 'usage_count', true);
-			$this->expiry_date 			= ($expires = get_post_meta($coupon->ID, 'expiry_date', true)) ? strtotime($expires) : '';
-			$this->apply_before_tax 	= get_post_meta($coupon->ID, 'apply_before_tax', true);
-			$this->free_shipping 		= get_post_meta($coupon->ID, 'free_shipping', true);
-			
-			return true;
-			
-		endif;
-		
-		return false;
+                if ($coupon_data) :
+                    $this->id = $coupon_data['id'];
+                    $this->type = $coupon_data['type'];
+                    $this->amount = $coupon_data['amount'];
+                    $this->individual_use = $coupon_data['individual_use'];
+                    $this->product_ids = $coupon_data['product_ids'];
+                    $this->exclude_product_ids = $coupon_data['exclude_product_ids'];
+                    $this->usage_limit = $coupon_data['usage_limit'];
+                    $this->usage_count = $coupon_data['usage_count'];
+                    $this->expiry_date = $coupon_data['expiry_date'];
+                    $this->apply_before_tax = $coupon_data['apply_before_tax'];
+                    $this->free_shipping = $coupon_data['free_shipping'];
+                    return true;
+                else:
+                    $coupon = get_page_by_title($this->code, 'OBJECT', 'shop_coupon');
+                    if ($coupon && $coupon->post_status == 'publish') :
+                        $this->id = $coupon->ID;
+                        $this->type = get_post_meta($coupon->ID, 'discount_type', true);
+                        $this->amount = get_post_meta($coupon->ID, 'coupon_amount', true);
+                        $this->individual_use = get_post_meta($coupon->ID, 'individual_use', true);
+                        $this->product_ids = array_filter(array_map('trim', explode(',', get_post_meta($coupon->ID, 'product_ids', true))));
+                        $this->exclude_product_ids = array_filter(array_map('trim', explode(',', get_post_meta($coupon->ID, 'exclude_product_ids', true))));
+                        $this->usage_limit = get_post_meta($coupon->ID, 'usage_limit', true);
+                        $this->usage_count = (int) get_post_meta($coupon->ID, 'usage_count', true);
+                        $this->expiry_date = ($expires = get_post_meta($coupon->ID, 'expiry_date', true)) ? strtotime($expires) : '';
+                        $this->apply_before_tax = get_post_meta($coupon->ID, 'apply_before_tax', true);
+                        $this->free_shipping = get_post_meta($coupon->ID, 'free_shipping', true);
+                        return true;
+                    endif;
+                endif;
+
+                return false;
 	}
 	
 	/** Check if coupon needs applying before tax **/