1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

<?php
/**
 * WC Admin Order Trait
 *
 * WC Admin Order Trait class that houses shared functionality across order and refund classes.
 */

namespace Automattic\WooCommerce\Admin\Overrides;

defined( 'ABSPATH' ) || exit;

/**
 * OrderTraits class.
 */
trait OrderTraits {
    /**
     * Calculate shipping amount for line item/product as a total shipping amount ratio based on quantity.
     *
     * @param WC_Order_Item $item Line item from order.
     *
     * @return float|int
     */
    public function get_item_shipping_amount( $item ) {
        // Shipping amount loosely based on woocommerce code in includes/admin/meta-boxes/views/html-order-item(s).php
        // distributed simply based on number of line items.
        $product_qty = $item->get_quantity( 'edit' );
        $order_items = $this->get_item_count();
        if ( 0 === $order_items ) {
            return 0;
        }

        $total_shipping_amount = (float) $this->get_shipping_total();

        return $total_shipping_amount / $order_items * $product_qty;
    }

    /**
     * Calculate shipping tax amount for line item/product as a total shipping tax amount ratio based on quantity.
     *
     * Loosely based on code in includes/admin/meta-boxes/views/html-order-item(s).php.
     *
     * @todo If WC is currently not tax enabled, but it was before (or vice versa), would this work correctly?
     *
     * @param WC_Order_Item $item Line item from order.
     *
     * @return float|int
     */
    public function get_item_shipping_tax_amount( $item ) {
        $order_items = $this->get_item_count();
        if ( 0 === $order_items ) {
            return 0;
        }

        $product_qty               = $item->get_quantity( 'edit' );
        $order_taxes               = $this->get_taxes();
        $line_items_shipping       = $this->get_items( 'shipping' );
        $total_shipping_tax_amount = 0;
        foreach ( $line_items_shipping as $item_id => $shipping_item ) {
            $tax_data = $shipping_item->get_taxes();
            if ( $tax_data ) {
                foreach ( $order_taxes as $tax_item ) {
                    $tax_item_id                = $tax_item->get_rate_id();
                    $tax_item_total             = isset( $tax_data['total'][ $tax_item_id ] ) ? (float) $tax_data['total'][ $tax_item_id ] : 0;
                    $total_shipping_tax_amount += $tax_item_total;
                }
            }
        }
        return $total_shipping_tax_amount / $order_items * $product_qty;
    }

    /**
     * Calculates coupon amount for specified line item/product.
     *
     * Coupon calculation based on woocommerce code in includes/admin/meta-boxes/views/html-order-item.php.
     *
     * @param WC_Order_Item $item Line item from order.
     *
     * @return float
     */
    public function get_item_coupon_amount( $item ) {
        return floatval( $item->get_subtotal( 'edit' ) - $item->get_total( 'edit' ) );
    }
}