1 files changed, 35 insertions, 0 deletions
diff --git a/public/catalog/model/extension/total/credit.php b/public/catalog/model/extension/total/credit.php
new file mode 100644
index 0000000..6fe51ee
--- /dev/null
+++ b/public/catalog/model/extension/total/credit.php
@@ -0,0 +1,35 @@
+<?php
+class ModelExtensionTotalCredit extends Model {
+	public function getTotal($total) {
+		$this->load->language('extension/total/credit');
+
+		$balance = $this->customer->getBalance();
+
+		if ((float)$balance) {
+			$credit = min($balance, $total['total']);
+
+			if ((float)$credit > 0) {
+				$total['totals'][] = array(
+					'code'       => 'credit',
+					'title'      => $this->language->get('text_credit'),
+					'value'      => -$credit,
+					'sort_order' => $this->config->get('total_credit_sort_order')
+				);
+
+				$total['total'] -= $credit;
+			}
+		}
+	}
+
+	public function confirm($order_info, $order_total) {
+		$this->load->language('extension/total/credit');
+
+		if ($order_info['customer_id']) {
+			$this->db->query("INSERT INTO " . DB_PREFIX . "customer_transaction SET customer_id = '" . (int)$order_info['customer_id'] . "', order_id = '" . (int)$order_info['order_id'] . "', description = '" . $this->db->escape(sprintf($this->language->get('text_order_id'), (int)$order_info['order_id'])) . "', amount = '" . (float)$order_total['value'] . "', date_added = NOW()");
+		}
+	}
+
+	public function unconfirm($order_id) {
+		$this->db->query("DELETE FROM " . DB_PREFIX . "customer_transaction WHERE order_id = '" . (int)$order_id . "'");
+	}
+}