woothemes/woocommerce

View on GitHub
includes/class-wc-shipping.php

Summary

Maintainability
A
3 hrs
Test Coverage
<?php
/**
 * WooCommerce Shipping
 *
 * Handles shipping and loads shipping methods via hooks.
 *
 * @version 2.6.0
 * @package WooCommerce\Classes\Shipping
 */

use Automattic\Jetpack\Constants;

if ( ! defined( 'ABSPATH' ) ) {
    exit;
}

/**
 * Shipping class.
 */
class WC_Shipping {

    /**
     * True if shipping is enabled.
     *
     * @var bool
     */
    public $enabled = false;

    /**
     * Stores methods loaded into woocommerce.
     *
     * @var array|null
     */
    public $shipping_methods = null;

    /**
     * Stores the shipping classes.
     *
     * @var array
     */
    public $shipping_classes = array();

    /**
     * Stores packages to ship and to get quotes for.
     *
     * @var array
     */
    public $packages = array();

    /**
     * The single instance of the class
     *
     * @var WC_Shipping
     * @since 2.1
     */
    protected static $_instance = null;

    /**
     * Main WC_Shipping Instance.
     *
     * Ensures only one instance of WC_Shipping is loaded or can be loaded.
     *
     * @since 2.1
     * @return WC_Shipping Main instance
     */
    public static function instance() {
        if ( is_null( self::$_instance ) ) {
            self::$_instance = new self();
        }
        return self::$_instance;
    }

    /**
     * Cloning is forbidden.
     *
     * @since 2.1
     */
    public function __clone() {
        wc_doing_it_wrong( __FUNCTION__, __( 'Cloning is forbidden.', 'woocommerce' ), '2.1' );
    }

    /**
     * Unserializing instances of this class is forbidden.
     *
     * @since 2.1
     */
    public function __wakeup() {
        wc_doing_it_wrong( __FUNCTION__, __( 'Unserializing instances of this class is forbidden.', 'woocommerce' ), '2.1' );
    }

    /**
     * Magic getter.
     *
     * @param string $name Property name.
     * @return mixed
     */
    public function __get( $name ) {
        // Grab from cart for backwards compatibility with versions prior to 3.2.
        if ( 'shipping_total' === $name ) {
            return WC()->cart->get_shipping_total();
        }
        if ( 'shipping_taxes' === $name ) {
            return WC()->cart->get_shipping_taxes();
        }
    }

    /**
     * Initialize shipping.
     */
    public function __construct() {
        $this->enabled = wc_shipping_enabled();

        if ( $this->enabled ) {
            $this->init();
        }
    }

    /**
     * Initialize shipping.
     */
    public function init() {
        do_action( 'woocommerce_shipping_init' );
    }

    /**
     * Shipping methods register themselves by returning their main class name through the woocommerce_shipping_methods filter.
     *
     * @return array
     */
    public function get_shipping_method_class_names() {
        // Unique Method ID => Method Class name.
        $shipping_methods = array(
            'flat_rate'     => 'WC_Shipping_Flat_Rate',
            'free_shipping' => 'WC_Shipping_Free_Shipping',
            'local_pickup'  => 'WC_Shipping_Local_Pickup',
        );

        // For backwards compatibility with 2.5.x we load any ENABLED legacy shipping methods here.
        $maybe_load_legacy_methods = array( 'flat_rate', 'free_shipping', 'international_delivery', 'local_delivery', 'local_pickup' );

        foreach ( $maybe_load_legacy_methods as $method ) {
            $options = get_option( 'woocommerce_' . $method . '_settings' );
            if ( $options && isset( $options['enabled'] ) && 'yes' === $options['enabled'] ) {
                $shipping_methods[ 'legacy_' . $method ] = 'WC_Shipping_Legacy_' . $method;
            }
        }

        return apply_filters( 'woocommerce_shipping_methods', $shipping_methods );
    }

    /**
     * Loads all shipping methods which are hooked in.
     * If a $package is passed, some methods may add themselves conditionally and zones will be used.
     *
     * @param array $package Package information.
     * @return WC_Shipping_Method[]
     */
    public function load_shipping_methods( $package = array() ) {
        if ( ! empty( $package ) ) {
            $debug_mode             = 'yes' === get_option( 'woocommerce_shipping_debug_mode', 'no' );
            $shipping_zone          = WC_Shipping_Zones::get_zone_matching_package( $package );
            $this->shipping_methods = $shipping_zone->get_shipping_methods( true );

            // translators: %s: shipping zone name.
            $matched_zone_notice = sprintf( __( 'Customer matched zone "%s"', 'woocommerce' ), $shipping_zone->get_zone_name() );

            // Debug output.
            if ( $debug_mode && ! Constants::is_defined( 'WOOCOMMERCE_CHECKOUT' ) && ! Constants::is_defined( 'WC_DOING_AJAX' ) && ! wc_has_notice( $matched_zone_notice ) ) {
                wc_add_notice( $matched_zone_notice );
            }
        } else {
            $this->shipping_methods = array();
        }

        // For the settings in the backend, and for non-shipping zone methods, we still need to load any registered classes here.
        foreach ( $this->get_shipping_method_class_names() as $method_id => $method_class ) {
            $this->register_shipping_method( $method_class );
        }

        // Methods can register themselves manually through this hook if necessary.
        do_action( 'woocommerce_load_shipping_methods', $package );

        // Return loaded methods.
        return $this->get_shipping_methods();
    }

    /**
     * Register a shipping method.
     *
     * @param object|string $method Either the name of the method's class, or an instance of the method's class.
     *
     * @return bool|void
     */
    public function register_shipping_method( $method ) {
        if ( ! is_object( $method ) ) {
            if ( ! class_exists( $method ) ) {
                return false;
            }
            $method = new $method();
        }
        if ( is_null( $this->shipping_methods ) ) {
            $this->shipping_methods = array();
        }
        $this->shipping_methods[ $method->id ] = $method;
    }

    /**
     * Unregister shipping methods.
     */
    public function unregister_shipping_methods() {
        $this->shipping_methods = null;
    }

    /**
     * Returns all registered shipping methods for usage.
     *
     * @return WC_Shipping_Method[]
     */
    public function get_shipping_methods() {
        if ( is_null( $this->shipping_methods ) ) {
            $this->load_shipping_methods();
        }
        return $this->shipping_methods;
    }

    /**
     * Get an array of shipping classes.
     *
     * @return array
     */
    public function get_shipping_classes() {
        if ( empty( $this->shipping_classes ) ) {
            $classes                = get_terms(
                'product_shipping_class',
                array(
                    'hide_empty' => '0',
                    'orderby'    => 'name',
                )
            );
            $this->shipping_classes = ! is_wp_error( $classes ) ? $classes : array();
        }
        return apply_filters( 'woocommerce_get_shipping_classes', $this->shipping_classes );
    }

    /**
     * Calculate shipping for (multiple) packages of cart items.
     *
     * @param array $packages multi-dimensional array of cart items to calc shipping for.
     * @return array Array of calculated packages.
     */
    public function calculate_shipping( $packages = array() ) {
        $this->packages = array();

        if ( ! $this->enabled || empty( $packages ) ) {
            return array();
        }

        // Calculate costs for passed packages.
        foreach ( $packages as $package_key => $package ) {
            $this->packages[ $package_key ] = $this->calculate_shipping_for_package( $package, $package_key );
        }

        /**
         * Allow packages to be reorganized after calculating the shipping.
         *
         * This filter can be used to apply some extra manipulation after the shipping costs are calculated for the packages
         * but before WooCommerce does anything with them. A good example of usage is to merge the shipping methods for multiple
         * packages for marketplaces.
         *
         * @since 2.6.0
         *
         * @param array $packages The array of packages after shipping costs are calculated.
         */
        $this->packages = array_filter( (array) apply_filters( 'woocommerce_shipping_packages', $this->packages ) );

        return $this->packages;
    }

    /**
     * See if package is shippable.
     *
     * Packages are shippable until proven otherwise e.g. after getting a shipping country.
     *
     * @param  array $package Package of cart items.
     * @return bool
     */
    public function is_package_shippable( $package ) {
        // Packages are shippable until proven otherwise.
        if ( empty( $package['destination']['country'] ) ) {
            return true;
        }

        $allowed = array_keys( WC()->countries->get_shipping_countries() );
        return in_array( $package['destination']['country'], $allowed, true );
    }

    /**
     * Calculate shipping rates for a package,
     *
     * Calculates each shipping methods cost. Rates are stored in the session based on the package hash to avoid re-calculation every page load.
     *
     * @param array $package Package of cart items.
     * @param int   $package_key Index of the package being calculated. Used to cache multiple package rates.
     *
     * @return array|bool
     */
    public function calculate_shipping_for_package( $package = array(), $package_key = 0 ) {
        // If shipping is disabled or the package is invalid, return false.
        if ( ! $this->enabled || empty( $package ) ) {
            return false;
        }

        $package['rates'] = array();

        // If the package is not shippable, e.g. trying to ship to an invalid country, do not calculate rates.
        if ( ! $this->is_package_shippable( $package ) ) {
            return $package;
        }

        // Check if we need to recalculate shipping for this package.
        $package_to_hash = $package;

        // Remove data objects so hashes are consistent.
        foreach ( $package_to_hash['contents'] as $item_id => $item ) {
            unset( $package_to_hash['contents'][ $item_id ]['data'] );
        }

        // Get rates stored in the WC session data for this package.
        $wc_session_key = 'shipping_for_package_' . $package_key;
        $stored_rates   = WC()->session->get( $wc_session_key );

        // Calculate the hash for this package so we can tell if it's changed since last calculation.
        $package_hash = 'wc_ship_' . md5( wp_json_encode( $package_to_hash ) . WC_Cache_Helper::get_transient_version( 'shipping' ) );

        if ( ! is_array( $stored_rates ) || $package_hash !== $stored_rates['package_hash'] || 'yes' === get_option( 'woocommerce_shipping_debug_mode', 'no' ) ) {
            foreach ( $this->load_shipping_methods( $package ) as $shipping_method ) {
                if ( ! $shipping_method->supports( 'shipping-zones' ) || $shipping_method->get_instance_id() ) {
                    /**
                     * Fires before getting shipping rates for a package.
                     *
                     * @since 4.3.0
                     * @param array $package Package of cart items.
                     * @param WC_Shipping_Method $shipping_method Shipping method instance.
                     */
                    do_action( 'woocommerce_before_get_rates_for_package', $package, $shipping_method );

                    // Use + instead of array_merge to maintain numeric keys.
                    $package['rates'] = $package['rates'] + $shipping_method->get_rates_for_package( $package );

                    /**
                     * Fires after getting shipping rates for a package.
                     *
                     * @since 4.3.0
                     * @param array $package Package of cart items.
                     * @param WC_Shipping_Method $shipping_method Shipping method instance.
                     */
                    do_action( 'woocommerce_after_get_rates_for_package', $package, $shipping_method );
                }
            }

            // Filter the calculated rates.
            $package['rates'] = apply_filters( 'woocommerce_package_rates', $package['rates'], $package );

            // Store in session to avoid recalculation.
            WC()->session->set(
                $wc_session_key,
                array(
                    'package_hash' => $package_hash,
                    'rates'        => $package['rates'],
                )
            );
        } else {
            $package['rates'] = $stored_rates['rates'];
        }

        return $package;
    }

    /**
     * Get packages.
     *
     * @return array
     */
    public function get_packages() {
        return $this->packages;
    }

    /**
     * Reset shipping.
     *
     * Reset the totals for shipping as a whole.
     */
    public function reset_shipping() {
        unset( WC()->session->chosen_shipping_methods );
        $this->packages = array();
    }

    /**
     * Deprecated
     *
     * @deprecated 2.6.0 Was previously used to determine sort order of methods, but this is now controlled by zones and thus unused.
     */
    public function sort_shipping_methods() {
        wc_deprecated_function( 'sort_shipping_methods', '2.6' );
        return $this->shipping_methods;
    }
}