/**
* Collects the taxes
* @param TinyBrick_OrderEdit_Model_Order_Address $address
* @return TinyBrick_OrderEdit_Model_Order_Address_Total_Tax
*/
public function collect(TinyBrick_OrderEdit_Model_Order_Address $address)
{
$store = $address->getOrder()->getStore();
$address->getOrder()->setTaxAmount(0);
$address->getOrder()->setBaseTaxAmount(0);
$address->setShippingTaxAmount(0);
$address->setBaseShippingTaxAmount(0);
$address->setAppliedTaxes(array());
$order = $address->getOrder();
$items = $order->getAllItems();
if (!count($items)) {
return $this;
}
$custTaxClassId = $address->getOrder()->getCustomerTaxClassId();
$taxCalculationModel = Mage::getSingleton('tax/calculation');
/**
* @var $taxCalculationModel Mage_Tax_Model_Calculation
*/
$request = $taxCalculationModel->getRateRequest($address->getOrder()->getShippingAddress(), $address->getOrder()->getBillingAddress(), $custTaxClassId, $store);
foreach ($items as $item) {
/**
* Child item's tax we calculate for parent
*/
if ($item->getParentItemId()) {
continue;
}
/**
* We calculate parent tax amount as sum of children's tax amounts
*/
if ($item->getHasChildren() && $item->isChildrenCalculated()) {
foreach ($item->getChildren() as $child) {
$discountBefore = $item->getDiscountAmount();
$baseDiscountBefore = $item->getBaseDiscountAmount();
$rate = $taxCalculationModel->getRate($request->setProductClassId($child->getProduct()->getTaxClassId()));
$child->setTaxPercent($rate);
$child->calcTaxAmount();
if ($discountBefore != $item->getDiscountAmount()) {
$address->setDiscountAmount($address->getDiscountAmount() + ($item->getDiscountAmount() - $discountBefore));
$address->setBaseDiscountAmount($address->getBaseDiscountAmount() + ($item->getBaseDiscountAmount() - $baseDiscountBefore));
$address->setGrandTotal($address->getGrandTotal() - ($item->getDiscountAmount() - $discountBefore));
$address->setBaseGrandTotal($address->getBaseGrandTotal() - ($item->getBaseDiscountAmount() - $baseDiscountBefore));
}
$this->_saveAppliedTaxes($address, $taxCalculationModel->getAppliedRates($request), $child->getTaxAmount(), $child->getBaseTaxAmount(), $rate);
}
$address->setTaxAmount($address->getTaxAmount() + $item->getTaxAmount());
$address->setBaseTaxAmount($address->getBaseTaxAmount() + $item->getBaseTaxAmount());
} else {
$discountBefore = $item->getDiscountAmount();
$baseDiscountBefore = $item->getBaseDiscountAmount();
$product = Mage::getModel('catalog/product')->load($item->getProductId());
$rate = $taxCalculationModel->getRate($request->setProductClassId($product->getTaxClassId()));
if ($item->getTaxExempt()) {
$item->setTaxPercent(0);
} else {
$item->setTaxPercent($rate);
}
$item->calcTaxAmount();
if ($discountBefore != $item->getDiscountAmount()) {
$address->setDiscountAmount($address->getDiscountAmount() + ($item->getDiscountAmount() - $discountBefore));
$address->setBaseDiscountAmount($address->getBaseDiscountAmount() + ($item->getBaseDiscountAmount() - $baseDiscountBefore));
$address->setGrandTotal($address->getGrandTotal() - ($item->getDiscountAmount() - $discountBefore));
$address->setBaseGrandTotal($address->getBaseGrandTotal() - ($item->getBaseDiscountAmount() - $baseDiscountBefore));
}
$address->getOrder()->setTaxAmount($address->getOrder()->getTaxAmount() + $item->getTaxAmount());
$address->getOrder()->setBaseTaxAmount($address->getOrder()->getBaseTaxAmount() + $item->getBaseTaxAmount());
$applied = $taxCalculationModel->getAppliedRates($request);
$this->_saveAppliedTaxes($address, $applied, $item->getTaxAmount(), $item->getBaseTaxAmount(), $rate);
}
}
$shippingTaxClass = Mage::getStoreConfig(Mage_Tax_Model_Config::CONFIG_XML_PATH_SHIPPING_TAX_CLASS, $store);
$shippingTax = 0;
$shippingBaseTax = 0;
if ($shippingTaxClass) {
if ($rate = $taxCalculationModel->getRate($request->setProductClassId($shippingTaxClass))) {
if (!Mage::helper('tax')->shippingPriceIncludesTax()) {
$shippingTax = $address->getShippingAmount() * $rate / 100;
$shippingBaseTax = $address->getBaseShippingAmount() * $rate / 100;
} else {
$shippingTax = $address->getShippingTaxAmount();
$shippingBaseTax = $address->getBaseShippingTaxAmount();
}
$shippingTax = $store->roundPrice($shippingTax);
$shippingBaseTax = $store->roundPrice($shippingBaseTax);
$address->setTaxAmount($address->getOrder()->getTaxAmount() + $shippingTax);
$address->setBaseTaxAmount($address->getBaseTaxAmount() + $shippingBaseTax);
$this->_saveAppliedTaxes($address, $taxCalculationModel->getAppliedRates($request), $shippingTax, $shippingBaseTax, $rate);
}
}
if (!Mage::helper('tax')->shippingPriceIncludesTax()) {
$address->setShippingTaxAmount($shippingTax);
$address->setBaseShippingTaxAmount($shippingBaseTax);
}
$address->setGrandTotal($address->getGrandTotal() + $address->getOrder()->getTaxAmount());
$address->setBaseGrandTotal($address->getBaseGrandTotal() + $address->getOrder()->getBaseTaxAmount());
return $this;
}
/**
* Fetches the discount amount for the customer
* @param TinyBrick_OrderEdit_Model_Order_Address $address
* @return TinyBrick_OrderEdit_Model_Order_Address_Total_Discount
*/
public function fetch(TinyBrick_OrderEdit_Model_Order_Address $address)
{
$amount = $address->getDiscountAmount();
if ($amount != 0) {
$title = Mage::helper('sales')->__('Discount');
if ($code = $address->getCouponCode()) {
$title = Mage::helper('sales')->__('Discount (%s)', $code);
}
$address->addTotal(array('code' => $this->getCode(), 'title' => $title, 'value' => $amount));
}
return $this;
}
