<?php
/*------------------------------------------------------------------------------
  $Id$

  AbanteCart, Ideal OpenSource Ecommerce Solution
  http://www.AbanteCart.com

  Copyright © 2011-2014 Belavier Commerce LLC

  This source file is subject to Open Software License (OSL 3.0)
  License details is bundled with this package in the file LICENSE.txt.
  It is also available at this URL:
  <http://www.opensource.org/licenses/OSL-3.0>

 UPGRADE NOTE:
   Do not edit or add to this file if you wish to upgrade AbanteCart to newer
   versions in the future. If you wish to customize AbanteCart for your
   needs please refer to http://www.AbanteCart.com for more information.
------------------------------------------------------------------------------*/
if (! defined ( 'DIR_CORE' ) || !IS_ADMIN) {
	header ( 'Location: static_pages/' );
}

/**
 * Class ModelSaleCoupon
 */
class ModelSaleCoupon extends Model {
	/**
	 * @param array $data
	 * @return int
	 */
	public function addCoupon($data) {
		if (has_value($data[ 'date_start' ])) {
			$data[ 'date_start' ] = "DATE('" . $data[ 'date_start' ] . "')";

		} else {
			$data[ 'date_start' ] = "NULL";
		}

		if (has_value($data[ 'date_end' ])) {
			$data[ 'date_end' ] = "DATE('" . $data[ 'date_end' ] . "')";
		} else {
			$data[ 'date_end' ] = "NULL";
		}

      	$this->db->query(  "INSERT INTO " . $this->db->table("coupons") . " 
							SET code = '" . $this->db->escape($data['code']) . "',
								discount = '" . (float)$data['discount'] . "',
								type = '" . $this->db->escape($data['type']) . "',
								total = '" . (float)$data['total'] . "',
								logged = '" . (int)$data['logged'] . "',
								shipping = '" . (int)$data['shipping'] . "',
								date_start = " . $data['date_start'] . ",
								date_end = " . $data['date_end'] . ",
								uses_total = '" . (int)$data['uses_total'] . "',
								uses_customer = '" . (int)$data['uses_customer'] . "',
								status = '" . (int)$data['status'] . "',
								date_added = NOW()");

      	$coupon_id = $this->db->getLastId();

      	foreach ($data['coupon_description'] as $language_id => $value) {
			$this->language->replaceDescriptions('coupon_descriptions',
											 array('coupon_id' => (int)$coupon_id),
											 array($language_id => array(
																		'name' => $value['name'],
																		'description' => $value['description']
											 )) );
      	}
		if (isset($data['coupon_product'])) {
      		foreach ($data['coupon_product'] as $product_id) {
        		$this->db->query(  "INSERT INTO " . $this->db->table("coupons_products") . " 
        		                    SET coupon_id = '" . (int)$coupon_id . "', product_id = '" . (int)$product_id . "'");
      		}			
		}
		return $coupon_id;
	}

	/**
	 * @param int $coupon_id
	 * @param array $data
	 */
	public function editCoupon($coupon_id, $data) {
		if (has_value($data[ 'date_start' ])) {
			$data[ 'date_start' ] = "DATE('" . $data[ 'date_start' ] . "')";
		}else{
			if(isset($data[ 'date_start' ])){
				$data[ 'date_start' ]='NULL';
			}

		}

		if (has_value($data[ 'date_end' ])) {
			$data[ 'date_end' ] = "DATE('" . $data[ 'date_end' ] . "')";
		}else{
			if(isset($data[ 'date_end' ])){
				$data[ 'date_end' ] = 'NULL';
			}

		}

		$coupon_table_fields = array('code',
		                             'discount',
		                             'type',
		                             'total',
		                             'logged',
		                             'shipping',
		                             'date_start',
		                             'date_end',
		                             'uses_total',
		                             'uses_customer',
		                             'status');
		$update = array();
		foreach ( $coupon_table_fields as $f ) {
			if ( isset($data[$f]) )
				if(!in_array($f,array('date_start','date_end'))){
					$update[] = $f." = '".$this->db->escape($data[$f])."'";
				}else{
					$update[] = $f." = ".$data[$f]."";
				}
		}
		if ( !empty($update) ) $this->db->query("UPDATE " . $this->db->table("coupons") . " 
												SET ". implode(',', $update) ."
												WHERE coupon_id = '" . (int)$coupon_id . "'");

		if ( !empty($data['coupon_description']) ) {
			foreach ($data['coupon_description'] as $language_id => $value) {
				$update = array();
				if ( isset($value['name']) ) $update["name"] = $value['name'];
				if ( isset($value['description']) ) $update["description"] = $value['description'];
				if ( !empty($update) ){
					$this->language->replaceDescriptions('coupon_descriptions',
														 array('coupon_id' => (int)$coupon_id),
														 array($language_id => array(
																					'name' => $value['name'],
																					'description' => $value['description']
														 )) );
				}
			}
		}

	}

	/**
	 * @param int $coupon_id
	 * @param array $data
	 */
	public function editCouponProducts($coupon_id, $data) {
		$this->db->query("DELETE FROM " . $this->db->table("coupons_products") . " 
						  WHERE coupon_id = '" . (int)$coupon_id . "'");

		if (isset($data['coupon_product'])) {
      		foreach ($data['coupon_product'] as $product_id) {
				$this->db->query(  "INSERT INTO " . $this->db->table("coupons_products") . " 
									SET coupon_id = '" . (int)$coupon_id . "',
										product_id = '" . (int)$product_id . "'");
      		}
		}
	}

	/**
	 * @param int $coupon_id
	 */
	public function deleteCoupon($coupon_id) {
      	$this->db->query("DELETE FROM " . $this->db->table("coupons") . " WHERE coupon_id = '" . (int)$coupon_id . "'");
      	$this->db->query("DELETE FROM " . $this->db->table("coupon_descriptions") . " WHERE coupon_id = '" . (int)$coupon_id . "'");
		$this->db->query("DELETE FROM " . $this->db->table("coupons_products") . " WHERE coupon_id = '" . (int)$coupon_id . "'");
	}

	/**
	 * @param int $coupon_id
	 * @return array
	 */
	public function getCouponByID($coupon_id) {
      	$query = $this->db->query("SELECT DISTINCT * FROM " . $this->db->table("coupons") . " WHERE coupon_id = '" . (int)$coupon_id . "'");
		
		return $query->row;
	}

	/**
	 * @param array $data
	 * @param string $mode
	 * @return array|int
	 */
	public function getCoupons($data = array(), $mode = 'default') {
		if ( !empty($data['content_language_id']) ) {
			$language_id = ( int )$data['content_language_id'];
		} else {
			$language_id = (int)$this->config->get('storefront_language_id');
		}

		//Prepare filter config
		$filter_params = array('status' => 'c.status');
		//Build query string based on GET params first
		$filter_form = new AFilter( array( 'method' => 'get', 'filter_params' => $filter_params ) );
		//Build final filter
		$grid_filter_params = array( 'name' => 'cd.name', 'code' => 'c.code' );
		$filter_grid = new AFilter( array( 'method' => 'post',
										   'grid_filter_params' => $grid_filter_params,
										   'additional_filter_string' => $filter_form->getFilterString()
										  ) );
		$data = array_merge( $filter_grid->getFilterData(), $data);


		if ($mode == 'total_only') {
			$total_sql = 'count(*) as total';
		}
		else {
			$total_sql = "c.coupon_id, cd.name, c.code, c.discount, c.date_start, c.date_end, c.status ";
		}

		$sql = "SELECT ". $total_sql ." 
				FROM " . $this->db->table("coupons") . " c
				JOIN " . $this->db->table("coupon_descriptions") . " cd
					ON (c.coupon_id = cd.coupon_id AND cd.language_id = '" . $language_id . "')";

        if ( !empty($data['search']) ) {
            $sql .= " AND ".$data['search'];
        }
        if ( !empty($data['subsql_filter']) ) {
            $sql .= " AND ".$data['subsql_filter'];
        }

		//If for total, we done bulding the query
		if ($mode == 'total_only') {
		    $query = $this->db->query($sql);
		    return $query->row['total'];
		}

		$sort_data = array(
			'name' => 'cd.name',
			'code' => 'c.code',
			'discount' => 'c.discount',
			'date_start' => 'c.date_start',
			'date_end' => 'c.date_end',
			'status' => 'c.status'
		);	
			
        if (isset($data['sort']) && array_key_exists($data['sort'], $sort_data)) {
            $sql .= " ORDER BY " . $sort_data[$data['sort']];
		} else {
			$sql .= " ORDER BY cd.name";	
		}
			
		if (isset($data['order']) && (strtoupper($data['order']) == 'DESC')) {
			$sql .= " DESC";
		} else {
			$sql .= " ASC";
		}
		
		if (isset($data['start']) || isset($data['limit'])) {
			if ($data['start'] < 0) {
				$data['start'] = 0;
			}			

			if ($data['limit'] < 1) {
				$data['limit'] = 20;
			}	
			
			$sql .= " LIMIT " . (int)$data['start'] . "," . (int)$data['limit'];
		}			
		$query = $this->db->query($sql);
		return $query->rows;
	}

	/**
	 * @param array $data
	 * @return int
	 */
	public function getTotalCoupons( $data ) {
		return $this->getCoupons($data,'total_only');
	}

	/**
	 * @param int $coupon_id
	 * @return array
	 */
	public function getCouponDescriptions($coupon_id) {
		$coupon_description_data = array();
		
		$query = $this->db->query("SELECT *
									FROM " . $this->db->table("coupon_descriptions") . " 
									WHERE coupon_id = '" . (int)$coupon_id . "'");
		
		foreach ($query->rows as $result) {
			$coupon_description_data[$result['language_id']] = array(
				'name'        => $result['name'],
				'description' => $result['description']
			);
		}
		
		return $coupon_description_data;
	}

	/**
	 * @param int $coupon_id
	 * @return array
	 */
	public function getCouponProducts($coupon_id) {
		$coupon_product_data = array();
		
		$query = $this->db->query("SELECT *
									FROM " . $this->db->table("coupons_products") . " 
									WHERE coupon_id = '" . (int)$coupon_id . "'");
		
		foreach ($query->rows as $result) {
			$coupon_product_data[] = $result['product_id'];
		}
		
		return $coupon_product_data;
	}
}