*/ class Mage_Sales_Model_Order_Invoice_Comment extends Mage_Sales_Model_Abstract { /** * Invoice instance * * @var Mage_Sales_Model_Order_Invoice */ protected $_invoice; /** * Initialize resource model */ protected function _construct() { $this->_init('sales/order_invoice_comment'); } /** * Declare invoice instance * * @param Mage_Sales_Model_Order_Invoice $invoice * @return Mage_Sales_Model_Order_Invoice_Comment */ public function setInvoice(Mage_Sales_Model_Order_Invoice $invoice) { $this->_invoice = $invoice; return $this; } /** * Retrieve invoice instance * * @return Mage_Sales_Model_Order_Invoice */ public function getInvoice() { return $this->_invoice; } /** * Get store object * * @return Mage_Core_Model_Store */ public function getStore() { if ($this->getInvoice()) { return $this->getInvoice()->getStore(); } return Mage::app()->getStore(); } /** * Before object save * * @return Mage_Sales_Model_Order_Invoice_Comment */ protected function _beforeSave() { parent::_beforeSave(); if (!$this->getParentId() && $this->getInvoice()) { $this->setParentId($this->getInvoice()->getId()); } return $this; } }