PrestaShop/controllers/admin/AdminAddressesController.php

<?php
/*
* 2007-2011 PrestaShop
*
* NOTICE OF LICENSE
*
* This source file is subject to the Open Software License (OSL 3.0)
* that is bundled with this package in the file LICENSE.txt.
* It is also available through the world-wide-web at this URL:
* http://opensource.org/licenses/osl-3.0.php
* If you did not receive a copy of the license and are unable to
* obtain it through the world-wide-web, please send an email
* to license@prestashop.com so we can send you a copy immediately.
*
* DISCLAIMER
*
* Do not edit or add to this file if you wish to upgrade PrestaShop to newer
* versions in the future. If you wish to customize PrestaShop for your
* needs please refer to http://www.prestashop.com for more information.
*
*  @author PrestaShop SA <contact@prestashop.com>
*  @copyright  2007-2011 PrestaShop SA
*  @version  Release: $Revision$
*  @license    http://opensource.org/licenses/osl-3.0.php  Open Software License (OSL 3.0)
*  International Registered Trademark & Property of PrestaShop SA
*/
//include_once(_PS_ADMIN_DIR_.'/../classes/AdminTab.php');
if (Configuration::get('VATNUMBER_MANAGEMENT') AND file_exists(_PS_MODULE_DIR_.'vatnumber/vatnumber.php'))
	include_once(_PS_MODULE_DIR_.'vatnumber/vatnumber.php');

class AdminAddressesControllerCore extends AdminController
{
	/** @var array countries list */
	private $countriesArray = array();

	public function __construct()
	{
	 	$this->table = 'address';
	 	$this->className = 'Address';
	 	$this->lang = false;
	 	$this->bulk_actions = array('delete' => array('text' => $this->l('Delete selected'), 'confirm' => $this->l('Delete selected items?')));
		$this->requiredDatabase = true;
		$this->addressType = 'customer';
		$this->context = Context::getContext();

		$this->addRowAction('edit');
		$this->addRowAction('delete');

		if (!Tools::getValue('realedit'))
			$this->deleted = true;
		$this->_select = 'cl.`name` as country';
		$this->_join = 'LEFT JOIN `'._DB_PREFIX_.'country_lang` cl ON
			(cl.`id_country` = a.`id_country` AND cl.`id_lang` = '.(int)$this->context->language->id.')';
		$this->_where = 'AND a.id_customer != 0';

		$countries = Country::getCountries($this->context->language->id);
		foreach ($countries AS $country)
			$this->countriesArray[$country['id_country']] = $country['name'];

		$this->fieldsDisplay = array(
		'id_address' => array('title' => $this->l('ID'), 'align' => 'center', 'width' => 25),
		'firstname' => array('title' => $this->l('First name'), 'width' => 80, 'filter_key' => 'a!firstname'),
		'lastname' => array('title' => $this->l('Last name'), 'width' => 100, 'filter_key' => 'a!lastname'),
		'address1' => array('title' => $this->l('Address'), 'width' => 200),
		'postcode' => array('title' => $this->l('Postcode/ Zip Code'), 'align' => 'right', 'width' => 50),
		'city' => array('title' => $this->l('City'), 'width' => 150),
		'country' => array('title' => $this->l('Country'), 'width' => 100, 'type' => 'select', 'select' => $this->countriesArray, 'filter_key' => 'cl!id_country'));

		$this->fields_form = array(
			'legend' => array(
				'title' => $this->l('Addresses'),
				'image' => '../img/admin/contact.gif'
			),
			'input' => array(
				array(
					'type' => 'text',
					'label' => $this->l('Customer'),
					'name' => 'id_customer',
					'size' => 33,
					'required' => true,
				),
				array(
					'type' => 'text',
					'label' => $this->l('Identification Number'),
					'name' => 'dni',
					'size' => 30,
					'required' => false,
					'p' => $this->l('DNI / NIF / NIE')
				),
				array(
					'type' => 'text',
					'label' => $this->l('Alias'),
					'name' => 'alias',
					'size' => 33,
					'required' => true,
					'p' => '<span class="hint" name="help_box">'.$this->l('Invalid characters:').' <>;=#{}<span class="hint-pointer">&nbsp;</span></span>'
				),
				array(
					'type' => 'text',
					'label' => $this->l('Home phone'),
					'name' => 'phone',
					'size' => 33,
					'required' => false,
				),
				array(
					'type' => 'text',
					'label' => $this->l('Mobile phone'),
					'name' => 'phone_mobile',
					'size' => 33,
					'required' => false,
				),
				array(
					'type' => 'textarea',
					'label' => $this->l('Other'),
					'name' => 'other',
					'cols' => 36,
					'rows' => 4,
					'required' => false,
					'p' => '<span class="hint" name="help_box">'.$this->l('Forbidden characters:').' <>;=#{}<span class="hint-pointer">&nbsp;</span></span>'
				),
			),
			'submit' => array(
				'title' => $this->l('   Save   '),
				'class' => 'button'
			)
		);

		parent::__construct();
	}

	public function postProcess()
	{
		if ($this->action == 'save')
		{
			// Transform e-mail in id_customer for parent processing
			if (Validate::isEmail(Tools::getValue('email')))
			{
				$customer = new Customer();
				$customer->getByEmail(Tools::getValue('email'));
				if (Validate::isLoadedObject($customer))
					$_POST['id_customer'] = $customer->id;
				else
					$this->_errors[] = Tools::displayError('This e-mail address is not registered.');
			}
			elseif ($id_customer = Tools::getValue('id_customer'))
			{
				$customer = new Customer((int)($id_customer));
				if (Validate::isLoadedObject($customer))
					$_POST['id_customer'] = $customer->id;
				else
					$this->_errors[] = Tools::displayError('Unknown customer');
			}
			else
				$this->_errors[] = Tools::displayError('Unknown customer');
			if (Country::isNeedDniByCountryId(Tools::getValue('id_country')) AND !Tools::getValue('dni'))
				$this->_errors[] = Tools::displayError('Identification number is incorrect or has already been used.');

			/* If the selected country does not contain states */
			$id_state = (int)(Tools::getValue('id_state'));
			if ($id_country = Tools::getValue('id_country') AND $country = new Country((int)($id_country)) AND !(int)($country->contains_states) AND $id_state)
				$this->_errors[] = Tools::displayError('You have selected a state for a country that does not contain states.');

			/* If the selected country contains states, then a state have to be selected */
			if ((int)($country->contains_states) AND !$id_state)
				$this->_errors[] = Tools::displayError('An address located in a country containing states must have a state selected.');

			/* Check zip code */
			if ($country->need_zip_code)
			{
				$zip_code_format = $country->zip_code_format;
				if (($postcode = Tools::getValue('postcode')) AND $zip_code_format)
				{
					$zip_regexp = '/^'.$zip_code_format.'$/ui';
					$zip_regexp = str_replace(' ', '( |)', $zip_regexp);
					$zip_regexp = str_replace('-', '(-|)', $zip_regexp);
					$zip_regexp = str_replace('N', '[0-9]', $zip_regexp);
					$zip_regexp = str_replace('L', '[a-zA-Z]', $zip_regexp);
					$zip_regexp = str_replace('C', $country->iso_code, $zip_regexp);
					if (!preg_match($zip_regexp, $postcode))
						$this->_errors[] = Tools::displayError('Your zip/postal code is incorrect.').'<br />'.Tools::displayError('Must be typed as follows:').' '.str_replace('C', $country->iso_code, str_replace('N', '0', str_replace('L', 'A', $zip_code_format)));
				}
				elseif ($zip_code_format)
					$this->_errors[] = Tools::displayError('Postcode required.');
				elseif ($postcode AND !preg_match('/^[0-9a-zA-Z -]{4,9}$/ui', $postcode))
					$this->_errors[] = Tools::displayError('Your zip/postal code is incorrect.');
			}

			/* If this address come from order's edition and is the same as the other one (invoice or delivery one)
			** we delete its id_address to force the creation of a new one */
			if ((int)(Tools::getValue('id_order')))
			{
				$this->_redirect = false;
				if (isset($_POST['address_type']))
					$_POST['id_address'] = '';
			}
		}
		if (!sizeof($this->_errors))
			parent::postProcess();

		/* Reassignation of the order's new (invoice or delivery) address */
		$address_type = ((int)(Tools::getValue('address_type')) == 2 ? 'invoice' : ((int)(Tools::getValue('address_type')) == 1 ? 'delivery' : ''));
		if ($this->action == 'save' AND ($id_order = (int)(Tools::getValue('id_order'))) AND !sizeof($this->_errors) AND !empty($address_type))
		{
			if(!Db::getInstance()->execute('UPDATE '._DB_PREFIX_.'orders SET `id_address_'.$address_type.'` = '.Db::getInstance()->Insert_ID().' WHERE `id_order` = '.$id_order))
				$this->_errors[] = Tools::displayError('An error occurred while linking this address to its order.');
			else
				Tools::redirectAdmin(Tools::getValue('back').'&conf=4');
		}
	}

	/**
	 * Get Address formats used by the country where the address id retrieved from POST/GET is.
	 *
	 * @return array address formats
	 */
	protected function processAddressFormat()
	{
		$tmp_addr = new Address((int)Tools::getValue("id_address"));

		$selectedCountry = ($tmp_addr && $tmp_addr->id_country) ? $tmp_addr->id_country :
				(int)(Configuration::get('PS_COUNTRY_DEFAULT'));

		$inv_adr_fields = AddressFormat::getOrderedAddressFields($selectedCountry, false, true);
		$dlv_adr_fields = AddressFormat::getOrderedAddressFields($selectedCountry, false, true);

		$inv_all_fields = array();
		$dlv_all_fields = array();

		$out = array();

		foreach (array('inv','dlv') as $adr_type)
		{
			foreach (${$adr_type.'_adr_fields'} as $fields_line)
				foreach(explode(' ',$fields_line) as $field_item)
					${$adr_type.'_all_fields'}[] = trim($field_item);


			$out[$adr_type.'_adr_fields'] =  ${$adr_type.'_adr_fields'};
			$out[$adr_type.'_all_fields'] =  ${$adr_type.'_all_fields'};
		}

		return $out;
	}

	public function initContent()
	{
		if ($this->display == 'edit' || $this->display == 'add')
		{
			if (!($address = $this->loadObject(true)))
				return false;

			if (Validate::isLoadedObject($address))
			{
				$customer = new Customer($address->id_customer);
				$tokenCustomer = Tools::getAdminToken('AdminCustomers'.(int)(Tab::getIdFromClassName('AdminCustomers')).(int)$this->context->employee->id);
			}

			if (Configuration::get('VATNUMBER_MANAGEMENT'))
				if (file_exists(_PS_MODULE_DIR_.'vatnumber/vatnumber.php') && VatNumber::isApplicable(Configuration::get('PS_COUNTRY_DEFAULT')))
					$vat = 'is_applicable';
				else
					$vat = 'management';

			$this->context->smarty->assign(array(
				'vat' => isset($vat) ? $vat : null,
				'customer' => isset($customer) ? $customer : null,
				'tokenCustomer' => isset ($tokenCustomer) ? $tokenCustomer : null
			));

			// Order address fields depending on country format
			$addresses_fields = $this->processAddressFormat();
			$addresses_fields = $addresses_fields["dlv_all_fields"];	// we use  delivery address

			$temp_fields = array();

			foreach($addresses_fields as $addr_field_item)
			{
				if ($addr_field_item == 'company')
				{
					$temp_fields[] = array(
						'type' => 'text',
						'label' => $this->l('Company'),
						'name' => 'company',
						'size' => 33,
						'required' => false,
						'p' => '<span class="hint" name="help_box">'.$this->l('Invalid characters:').' <>;=#{}<span class="hint-pointer">&nbsp;</span></span>'
					);
					$temp_fields[] = array(
						'type' => 'text',
						'label' => $this->l('VAT number'),
						'name' => 'vat_number',
						'size' => 33,
						''
					);
				}
				elseif ($addr_field_item == 'lastname')
				{
					$temp_fields[] = array(
						'type' => 'text',
						'label' => $this->l('Last name'),
						'name' => 'lastname',
						'size' => 33,
						'required' => true,
						'p' => '<span class="hint" name="help_box">'.$this->l('Invalid characters:').' 0-9!<>,;?=+()@#"<22>{}_$%:<span class="hint-pointer">&nbsp;</span></span>'
					);
				}
				elseif ($addr_field_item == 'firstname')
				{
					$temp_fields[] = array(
						'type' => 'text',
						'label' => $this->l('First name'),
						'name' => 'firstname',
						'size' => 33,
						'required' => true,
						'p' => '<span class="hint" name="help_box">'.$this->l('Invalid characters:').' 0-9!<>,;?=+()@#"<22>{}_$%:<span class="hint-pointer">&nbsp;</span></span>'
					);
				}
				elseif ($addr_field_item == 'address1')
				{
					$temp_fields[] = array(
						'type' => 'text',
						'label' => $this->l('Address'),
						'name' => 'address1',
						'size' => 33,
						'required' => true,
					);
				}
				elseif ($addr_field_item == 'address2')
				{
					$temp_fields[] = array(
						'type' => 'text',
						'label' => $this->l('Address').' (2)',
						'name' => 'address2',
						'size' => 33,
						'required' => false,
					);
				}
				elseif ($addr_field_item == 'postcode')
				{
					$temp_fields[] = array(
						'type' => 'text',
						'label' => $this->l('Postcode/ Zip Code'),
						'name' => 'postcode',
						'size' => 33,
						'required' => false,
					);
				}
				elseif ($addr_field_item == 'city')
				{
					$temp_fields[] = array(
						'type' => 'text',
						'label' => $this->l('City'),
						'name' => 'city',
						'size' => 33,
						'required' => true,
					);
				}
				elseif ($addr_field_item == 'country' || $addr_field_item == 'Country:name')
				{
					$temp_fields[] = array(
						'type' => 'select',
						'label' => $this->l('Country:'),
						'name' => 'id_country',
						'required' => false,
						'options' => array(
							'query' => Country::getCountries($this->context->language->id),
							'id' => 'id_country',
							'name' => 'name'
						)
					);
					$temp_fields[] = array(
						'type' => 'select',
						'label' => $this->l('State'),
						'name' => 'id_state',
						'required' => false,
						'options' => array(
							'id' => 'id_state',
							'name' => 'name'
						)
					);

				}
			}
			// merge address format with the rest of the form
			array_splice($this->fields_form['input'], 3, 0, $temp_fields);
		}
		else
			$this->display = 'list';

		parent::initContent();
	}
}