woocommerce/classes/shipping/shipping.class.php

<?php    
/**
 * WooCommerce Shipping Class
 * 
 * Handles shipping and loads shipping methods via hooks.
 *
 * @class 		woocommerce_shipping
 * @package		WooCommerce
 * @category	Shipping
 * @author		WooThemes
 */  
class woocommerce_shipping {
	
	var $enabled			= false;
	var $shipping_methods 	= array();
	var $chosen_method		= null;
	var $shipping_total 	= 0;
	var $shipping_tax 		= 0;
	var $shipping_taxes		= array();
	var $shipping_label		= null;
	var $shipping_classes;
	
    function init() {

		if (get_option('woocommerce_calc_shipping')!='no') $this->enabled = true; 
		
		do_action('woocommerce_shipping_init');
		
		$load_methods = apply_filters('woocommerce_shipping_methods', array());
		
		// Get order option
		$ordering 	= (array) get_option('woocommerce_shipping_method_order');
		$order_end 	= 999;
		
		// Load gateways in order
		foreach ($load_methods as $method) :
			
			$load_method = &new $method();
			
			if (isset($ordering[$load_method->id]) && is_numeric($ordering[$load_method->id])) :
				// Add in position
				$this->shipping_methods[$ordering[$load_method->id]] = $load_method;
			else :
				// Add to end of the array
				$this->shipping_methods[$order_end] = $load_method;
				$order_end++;
			endif;
			
		endforeach;
		
		ksort($this->shipping_methods);
		
		add_action('woocommerce_update_options_shipping', array(&$this, 'process_admin_options'));
		
	}
	
	function get_shipping_classes() {
		
		if (!is_array($this->shipping_classes)) :
			
			$args = array(
				'hide_empty' => '0'
			);
			$classes = get_terms( 'product_shipping_class', $args );
			
			$this->shipping_classes = ($classes) ? $classes : array();
				
		endif;
		
		return (array) $this->shipping_classes;
	}

	function get_available_shipping_methods() {

		if ($this->enabled=='yes') :
		
			$_available_methods = array();
			
			foreach ( $this->shipping_methods as $shipping_method ) :
				
				if ($shipping_method->is_available()) :
					
					$shipping_method->calculate_shipping();
					
					// If available, put available methods/rates in the array
					if ($shipping_method->multiple_rates) :
							
							foreach ($shipping_method->rates as $rate) :
								
								$method = $rate;
								
								$_available_methods[$method->id] = $method;
								
							endforeach;
							
					else :
						
						$method = $shipping_method;
						
						$_available_methods[$method->id] = $method;

					endif;
					
				endif;
				
			endforeach;
			
			return $_available_methods;
			
		endif;
	}
	
	function reset_shipping_methods() {
		foreach ( $this->shipping_methods as $shipping_method ) :
			$shipping_method->shipping_total = 0;
			$shipping_method->shipping_tax = 0;
			$shipping_method->shipping_taxes = array();
			$shipping_method->rates = array();
		endforeach;
	}
	
	function calculate_shipping() {
		
		if ($this->enabled=='yes') :
		
			$this->shipping_total = 0;
			$this->shipping_tax = 0;
			$this->shipping_taxes = array();
			$this->shipping_label = null;
			$_cheapest_fee = '';
			$_cheapest_method = '';
			if (isset($_SESSION['_chosen_shipping_method'])) $chosen_method = $_SESSION['_chosen_shipping_method']; else $chosen_method = '';
			$calc_cheapest = false;
			
			if (!$chosen_method || empty($chosen_method)) $calc_cheapest = true;
			
			$this->reset_shipping_methods();
			
			$_available_methods = $this->get_available_shipping_methods();
			
			if (sizeof($_available_methods)>0) :
			
				foreach ($_available_methods as $method_id => $method) :
					
					$fee = $method->shipping_total;
					if ($fee < $_cheapest_fee || !is_numeric($_cheapest_fee)) :
						$_cheapest_fee 		= $fee;
						$_cheapest_method 	= $method_id;
					endif;

				endforeach;
				
				// Default to cheapest
				if ($calc_cheapest || !isset($_available_methods[$chosen_method])) :
					$chosen_method = $_cheapest_method;
				endif;
				
				if ($chosen_method) :
					
					$_SESSION['_chosen_shipping_method'] = $chosen_method;
					$this->shipping_total 	= $_available_methods[$chosen_method]->shipping_total;
					$this->shipping_tax 	= $_available_methods[$chosen_method]->shipping_tax;
					$this->shipping_taxes 	= $_available_methods[$chosen_method]->shipping_taxes;
					$this->shipping_label 	= $_available_methods[$chosen_method]->title;
					
				endif;
			endif;

		endif;
		
	}
	
	function reset_shipping() {
		unset($_SESSION['_chosen_shipping_method']);
		$this->shipping_total = 0;
		$this->shipping_tax = 0;
		$this->shipping_taxes = array();
		$this->shipping_label = null;
	}
	
	function process_admin_options() {
		$method_order = (isset($_POST['method_order'])) ? $_POST['method_order'] : '';
		
		$order = array();
		
		if (is_array($method_order) && sizeof($method_order)>0) :
			$loop = 0;
			foreach ($method_order as $method_id) :
				$order[$method_id] = $loop;
				$loop++;
			endforeach;
		endif;
		
		update_option( 'woocommerce_shipping_method_order', $order );
	}
	
}