1: <?php
 2: /**
 3:  * Magento
 4:  *
 5:  * NOTICE OF LICENSE
 6:  *
 7:  * This source file is subject to the Open Software License (OSL 3.0)
 8:  * that is bundled with this package in the file LICENSE.txt.
 9:  * It is also available through the world-wide-web at this URL:
10:  * http://opensource.org/licenses/osl-3.0.php
11:  * If you did not receive a copy of the license and are unable to
12:  * obtain it through the world-wide-web, please send an email
13:  * to license@magentocommerce.com so we can send you a copy immediately.
14:  *
15:  * DISCLAIMER
16:  *
17:  * Do not edit or add to this file if you wish to upgrade Magento to newer
18:  * versions in the future. If you wish to customize Magento for your
19:  * needs please refer to http://www.magentocommerce.com for more information.
20:  *
21:  * @category    Mage
22:  * @package     Mage_Sales
23:  * @copyright   Copyright (c) 2012 Magento Inc. (http://www.magentocommerce.com)
24:  * @license     http://opensource.org/licenses/osl-3.0.php  Open Software License (OSL 3.0)
25:  */
26: 
27: 
28: class Mage_Sales_Model_Order_Invoice_Total_Discount extends Mage_Sales_Model_Order_Invoice_Total_Abstract
29: {
30:     public function collect(Mage_Sales_Model_Order_Invoice $invoice)
31:     {
32:         $invoice->setDiscountAmount(0);
33:         $invoice->setBaseDiscountAmount(0);
34: 
35:         $totalDiscountAmount     = 0;
36:         $baseTotalDiscountAmount = 0;
37: 
38:         /**
39:          * Checking if shipping discount was added in previous invoices.
40:          * So basically if we have invoice with positive discount and it
41:          * was not canceled we don't add shipping discount to this one.
42:          */
43:         $addShippingDicount = true;
44:         foreach ($invoice->getOrder()->getInvoiceCollection() as $previusInvoice) {
45:             if ($previusInvoice->getDiscountAmount()) {
46:                 $addShippingDicount = false;
47:             }
48:         }
49: 
50:         if ($addShippingDicount) {
51:             $totalDiscountAmount     = $totalDiscountAmount + $invoice->getOrder()->getShippingDiscountAmount();
52:             $baseTotalDiscountAmount = $baseTotalDiscountAmount + $invoice->getOrder()->getBaseShippingDiscountAmount();
53:         }
54: 
55:         /** @var $item Mage_Sales_Model_Order_Invoice_Item */
56:         foreach ($invoice->getAllItems() as $item) {
57:             $orderItem = $item->getOrderItem();
58:             if ($orderItem->isDummy()) {
59:                  continue;
60:             }
61: 
62:             $orderItemDiscount      = (float) $orderItem->getDiscountAmount();
63:             $baseOrderItemDiscount  = (float) $orderItem->getBaseDiscountAmount();
64:             $orderItemQty       = $orderItem->getQtyOrdered();
65: 
66:             if ($orderItemDiscount && $orderItemQty) {
67:                 /**
68:                  * Resolve rounding problems
69:                  */
70:                 $discount = $orderItemDiscount - $orderItem->getDiscountInvoiced();
71:                 $baseDiscount = $baseOrderItemDiscount - $orderItem->getBaseDiscountInvoiced();
72:                 if (!$item->isLast()) {
73:                     $activeQty = $orderItemQty - $orderItem->getQtyInvoiced();
74:                     $discount = $invoice->roundPrice($discount / $activeQty * $item->getQty(), 'regular', true);
75:                     $baseDiscount = $invoice->roundPrice($baseDiscount / $activeQty * $item->getQty(), 'base', true);
76:                 }
77: 
78:                 $item->setDiscountAmount($discount);
79:                 $item->setBaseDiscountAmount($baseDiscount);
80: 
81:                 $totalDiscountAmount += $discount;
82:                 $baseTotalDiscountAmount += $baseDiscount;
83:             }
84:         }
85: 
86: 
87:         $invoice->setDiscountAmount($totalDiscountAmount);
88:         $invoice->setBaseDiscountAmount($baseTotalDiscountAmount);
89: 
90:         $invoice->setGrandTotal($invoice->getGrandTotal() - $totalDiscountAmount);
91:         $invoice->setBaseGrandTotal($invoice->getBaseGrandTotal() - $baseTotalDiscountAmount);
92:         return $this;
93:     }
94: }
95: