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

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

  Copyright © 2011-2017 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')){
	header('Location: static_pages/');
}

/**
 * Class ModelAccountAddress
 */
class ModelAccountAddress extends Model{
	public $error = array ();

	/**
	 * @param $data
	 * @return int
	 */
	public function addAddress($data = array ()){
		if (!$data){
			return false;
		}
		//encrypt customer data
		$key_sql = '';
		if ($this->dcrypt->active){
			$data = $this->dcrypt->encrypt_data($data, 'addresses');
			$key_sql = ", key_id = '" . (int)$data['key_id'] . "'";
		}

		$this->db->query("INSERT INTO `" . $this->db->table("addresses") . "`
						SET customer_id = '" . (int)$this->customer->getId() . "',
							company = '" . $this->db->escape($data['company']) . "',
							firstname = '" . $this->db->escape($data['firstname']) . "',
							lastname = '" . $this->db->escape($data['lastname']) . "',
							address_1 = '" . $this->db->escape($data['address_1']) . "',
							address_2 = '" . $this->db->escape($data['address_2']) . "',
							postcode = '" . $this->db->escape($data['postcode']) . "',
							city = '" . $this->db->escape($data['city']) . "',
							zone_id = '" . (int)$data['zone_id'] . "',
							country_id = '" . (int)$data['country_id'] . "'" . $key_sql);

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

		if (isset($data['default']) && $data['default'] == '1'){
			$this->db->query("UPDATE " . $this->db->table("customers") . "
							SET address_id = '" . (int)$address_id . "'
							WHERE customer_id = '" . (int)$this->customer->getId() . "'");
		}

		return $address_id;
	}

	/**
	 * @param int $address_id
	 * @param array $data
	 * @return bool
	 */
	public function editAddress($address_id, $data){
		$address_id = (int)$address_id;
		if (!$address_id || !$data){
			return false;
		}

		//encrypt customer data
		$key_sql = '';
		if ($this->dcrypt->active){
			$data = $this->dcrypt->encrypt_data($data, 'addresses');
			$key_sql = ", key_id = '" . (int)$data['key_id'] . "'";
		}

		$this->db->query(
				"UPDATE " . $this->db->table("addresses") . "
				SET company = '" . $this->db->escape($data['company']) . "',
					firstname = '" . $this->db->escape($data['firstname']) . "',
					lastname = '" . $this->db->escape($data['lastname']) . "',
					address_1 = '" . $this->db->escape($data['address_1']) . "',
					address_2 = '" . $this->db->escape($data['address_2']) . "',
					postcode = '" . $this->db->escape($data['postcode']) . "',
					city = '" . $this->db->escape($data['city']) . "',
					zone_id = '" . (int)$data['zone_id'] . "',
					country_id = '" . (int)$data['country_id'] . "'" . $key_sql . "
				WHERE address_id  = '" . (int)$address_id . "' AND customer_id = '" . (int)$this->customer->getId() . "'");

		if (isset($data['default'])){
			$this->db->query(
					"UPDATE " . $this->db->table("customers") . "
					SET address_id = '" . (int)$address_id . "'
					WHERE customer_id = '" . (int)$this->customer->getId() . "'");
		}
		return true;
	}

	/**
	 * @param int $address_id
	 */
	public function deleteAddress($address_id){
		$this->db->query(
				"DELETE FROM " . $this->db->table("addresses") . "
				WHERE address_id = '" . (int)$address_id . "' AND customer_id = '" . (int)$this->customer->getId() . "'");
	}

	/**
	 * @param int $address_id
	 * @return array|bool
	 */
	public function getAddress($address_id){
		$address_query = $this->db->query(
				"SELECT DISTINCT *
				 FROM " . $this->db->table("addresses") . "
				 WHERE address_id = '" . (int)$address_id . "' and customer_id = '" . (int)$this->customer->getId() . "'");

		if ($address_query->num_rows){
			return $this->_build_address_data($address_query->row);
		} else{
			return false;
		}
	}

	/**
	 * @return array
	 */
	public function getAddresses(){
		$address_data = array ();

		$query = $this->db->query(
				"SELECT *
				FROM " . $this->db->table("addresses") . "
				WHERE customer_id = '" . (int)$this->customer->getId() . "'");

		foreach ($query->rows as $result){
			$address_data[] = $this->_build_address_data($result);
		}

		return $address_data;
	}

	/**
	 * @return int
	 */
	public function getTotalAddresses(){
		$query = $this->db->query(
				"SELECT COUNT(*) AS total
				FROM " . $this->db->table("addresses") . "
				WHERE customer_id = '" . (int)$this->customer->getId() . "'");

		return (int)$query->row['total'];
	}

	/**
	 * @param array $data
	 * @return array
	 */
	public function validateAddressData($data){
		$this->error = array ();
		if (mb_strlen($data['firstname']) < 1 || mb_strlen($data['firstname']) > 32){
			$this->error['firstname'] = $this->language->get('error_firstname');
		}

		if (mb_strlen($data['lastname']) < 1 || mb_strlen($data['lastname']) > 32){
			$this->error['lastname'] = $this->language->get('error_lastname');
		}

		if (mb_strlen($data['address_1']) < 3 || mb_strlen($data['address_1']) > 64){
			$this->error['address_1'] = $this->language->get('error_address_1');
		}

		if (mb_strlen($data['city']) < 3 || mb_strlen($data['city']) > 32){
			$this->error['city'] = $this->language->get('error_city');
		}

		if (mb_strlen($data['postcode']) < 3 || mb_strlen($data['postcode']) > 10){
			$this->error['postcode'] = $this->language->get('error_postcode');
		}

		if ($data['country_id'] == 'FALSE'){
			$this->error['country'] = $this->language->get('error_country');
		}

		if ($data['zone_id'] == 'FALSE'){
			$this->error['zone'] = $this->language->get('error_zone');
		}

		if (!$this->error){
			$sql = "SELECT * FROM " . $this->db->table("zones") . "
					WHERE country_id = '" . (int)$data['country_id'] . "'
						AND zone_id = '" . (int)$data['zone_id'] . "';";
			$result = $this->db->query($sql);
			if (!$result->num_rows){
				$this->error['zone'] = $this->language->get('error_zone');
			}
		}

		if (count($this->error)){
			$this->error['warning'] = $this->language->get('gen_data_entry_error');
		}

		$this->extensions->hk_ValidateData($this);

		return $this->error;
	}

	/**
	 * @param array $address_row
	 * @return array
	 * @throws AException
	 */
	private function _build_address_data($address_row){
		$addr_row = $this->dcrypt->decrypt_data($address_row, 'addresses');

		$this->load->model('localisation/country');
		$this->load->model('localisation/zone');
		$country_row = $this->model_localisation_country->getCountry($addr_row['country_id']);

		if ($country_row){
			$country = $country_row['name'];
			$iso_code_2 = $country_row['iso_code_2'];
			$iso_code_3 = $country_row['iso_code_3'];
			$address_format = $country_row['address_format'];
		} else{
			$country = '';
			$iso_code_2 = '';
			$iso_code_3 = '';
			$address_format = '';
		}

		$zone_row = $this->model_localisation_zone->getZone($addr_row['zone_id']);

		if ($zone_row){
			$zone = $zone_row['name'];
			$code = $zone_row['code'];
		} else{
			$zone = '';
			$code = '';
		}

		$address_data = array (
				'address_id'     => $addr_row['address_id'],
				'firstname'      => $addr_row['firstname'],
				'lastname'       => $addr_row['lastname'],
				'company'        => $addr_row['company'],
				'address_1'      => $addr_row['address_1'],
				'address_2'      => $addr_row['address_2'],
				'postcode'       => $addr_row['postcode'],
				'city'           => $addr_row['city'],
				'zone_id'        => $addr_row['zone_id'],
				'zone'           => $zone,
				'zone_code'      => $code,
				'country_id'     => $addr_row['country_id'],
				'country'        => $country,
				'iso_code_2'     => $iso_code_2,
				'iso_code_3'     => $iso_code_3,
				'address_format' => $address_format
		);

		return $address_data;
	}

}