import_export.php

<?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' )) {
	header ( 'Location: static_pages/' );
}
class ControllerPagesToolImportExport extends AController {

	public $tabs = array('export', 'import');
	public $error;
	public $success = '';
	private $data = array();
	private $handler;

	public function main() {
		 //init controller data
        $this->extensions->hk_InitData($this,__FUNCTION__);

		$this->handler = new AData();
		
		$this->loadLanguage('tool/import_export');
        $this->document->setTitle( $this->language->get('import_export_title') );

		$this->data['title'] = $this->language->get('import_export_title');
		$this->data['text_tables'] = $this->language->get('text_tables');
		$this->data['text_options'] = $this->language->get('text_options');

		$this->data['tabs'] = $this->tabs;

		if ( isset($this->request->get['active']) && strpos($this->request->get['active'], '-') !== false ) {
            $this->request->get['active'] = substr($this->request->get['active'], 0, strpos($this->request->get['active'], '-') );
        }
		$this->data['active'] = isset($this->request->get['active']) && in_array($this->request->get['active'], $this->data['tabs']) ?
				$this->request->get['active'] : $this->data['tabs'][0];

		foreach ( $this->data['tabs'] as $tab ) {
			$this->data['tab_'.$tab] = $this->language->get('tab_'.$tab);
			$this->data['link_'.$tab] = $this->html->getSecureURL('p/tool/import_export', '&active='.$tab);
		}

		$this->data['token'] = $this->session->data['token'];

		$this->document->initBreadcrumb( array (
       		'href'      => $this->html->getSecureURL('index/home'),
       		'text'      => $this->language->get('text_home'),
      		'separator' => FALSE
   		 ));
   		$this->document->addBreadcrumb( array (
       		'href'      => $this->html->getSecureURL('tool/import_export'),
       		'text'      => $this->language->get('import_export_title'),
      		'separator' => ' :: '
   		 ));

		$this->getForm();
		
		$this->view->assign('help_url', $this->gen_help_url($this->data['active']) );


		if ( isset($this->session->data['error']) ) {
			$this->data['error'] = $this->session->data['error'];
			unset($this->session->data['error']);
		} else {
			$this->data['error'] = $this->errors;
		}
		$this->data['success'] = $this->success;
		
		$this->view->batchAssign( $this->data );
        $this->processTemplate('pages/tool/import_export.tpl' );

        //update controller data
        $this->extensions->hk_UpdateData($this,__FUNCTION__);
	}

	private function getForm() {
		$form = new AForm('ST');

		$form->setForm(array(
			'form_name' => $this->data['active'].'Frm'
		));

		$this->data['form']['id'] = $this->data['active'].'Frm';

		$this->data['form']['submit'] = $form->getFieldHtml(array(
			'type' => 'button',
			'name' => 'submit',
			'text' => $this->language->get('tab_'.$this->data['active']),
			'style' => 'button1',
		));
		$this->data['form']['cancel'] = $form->getFieldHtml(array(
			'type' => 'button',
			'name' => 'cancel',
			'text' => $this->language->get('button_cancel'),
			'style' => 'button2',
		));

		switch ( $this->data['active'] ) {
			case 'import':
				$this->data['action'] = $this->html->getSecureURL('tool/import_export', '&active='.$this->data['active']);

				$this->data['text_load_file'] = $this->language->get('text_load_file');
				$this->data['text_file_field'] = $this->language->get('text_file_field');

				$fileForm = new AForm('ST');

				$fileForm->setForm(array(
					'form_name' => 'file_import_form'
				));

				$this->data['file_form_open'] = $fileForm->getFieldHtml(array(
					'type' => 'form',
					'name' => 'file_import_form',
					'action' => $this->data['action']
				));

				$this->data['file_field'] = $fileForm->getFieldHtml(array(
					'type' => 'file',
					'name' => 'imported_file'
				));

				$this->data['file_submit'] = $fileForm->getFieldHtml(array(
					'type' => 'button',
					'name' => 'file_submit',
					'text' => $this->language->get('text_load'),
					'style' => 'button1',
				));

				$options['text']['delimiter'] = $this->language->get('text_csv_delimiter') . '<br/>' . $this->language->get('text_export_note_csv_delimiter');
				$options['item']['delimiter'] = $fileForm->getFieldHtml(array(
					'type' => 'selectbox',
					'name' => 'options[delimiter]',
					'value' => 0,
					'options' => array(',', ';', 'TAB')
				));

				$options['text']['test_mode'] = $this->language->get('text_test_mode');
				$options['item']['test_mode'] = $form->getFieldHtml(array(
					'type' => 'checkbox',
					'name' => 'test_mode',
					'value' => 'test',
					'style' => '',
				));

				$this->data['options'] = $options;

				if ( !empty($this->request->files) ) {
					if ( file_exists($this->request->files['imported_file']['tmp_name']) ) {

						$this->data['results'] = $this->import($this->request->files['imported_file']);

						if ( !$this->data['results'] ) {
							$this->success = $this->language->get('text_import_loaded') . '0';
							$this->session->data['error'] = $this->language->get('error_data_corrupted');
						} else {
							$this->data['text_updated'] = $this->language->get('text_updated');
							$this->data['count_updated'] = isset($this->data['results']['update']) ? count($this->data['results']['update']) : 0;
							$this->data['text_created'] = $this->language->get('text_created');
							$this->data['count_created'] = isset($this->data['results']['insert']) ? count($this->data['results']['insert']) : 0;
							$this->data['text_errors'] = $this->language->get('text_errors');
							$this->data['count_errors'] = isset($this->data['results']['error']) ? count($this->data['results']['error']) : 0;
							$this->data['text_some_errors'] = $this->language->get('text_some_errors');
							$this->data['text_loaded'] = $this->language->get('text_import_loaded');
							$this->data['count_loaded'] = $this->data['count_updated'] + $this->data['count_created'] + $this->data['count_errors'];
							$this->data['text_show_details'] = $this->language->get('text_show_details');

							if ( isset($this->data['results']['sql']) ) {
								$this->data['text_test_completed'] = $this->language->get('text_test_completed');
								$this->data['count_test_sqls'] = count($this->data['results']['sql']);
							}
						}
					} elseif ( $this->request->files['imported_file']['error'] != 0 ) {
						$this->success = $this->language->get('text_import_loaded') . '0';
						$this->session->data['error'] = $this->language->get('error_upload_' . $this->request->files['imported_file']['error']);
					}

				}
				break;
				
			case 'export':
				$sections = $this->handler->getSections();
				
				$this->data['action'] = $this->html->getSecureURL('p/tool/export_upload');

				$this->data['form']['fields'] = $this->_build_table_fields($form, (array) $sections);

				$this->data['text_range_from'] = $this->language->get('text_id_range_from');
				$this->data['text_to'] = $this->language->get('text_to');
				break;
		}
		
		$options = array();

		$options['text']['file_format'] = $this->language->get('text_file_format');
		$options['item']['file_format'] = $form->getFieldHtml(array(
			'type' => 'selectbox',
			'name' => 'options[file_format]',
			'value' => 0,
			'options' => array(
				'csv' => '&nbsp;&nbsp;CSV&nbsp;&nbsp;',
				'txt' => 'TXT (delimited)',
				'xml' => '&nbsp;&nbsp;XML&nbsp;&nbsp;'
			)
		));

		$options['text']['file_name'] = $this->language->get('text_file_name');
		$options['item']['file_name'] = $form->getFieldHtml(array(
			'type' => 'input',
			'name' => 'options[file_name]',
			'value' => 'data_export_'.date('dmY_His'),
			'style' => 'large-field',
		));

		$options['text']['delimiter'] = $this->language->get('text_csv_delimiter') . '<br/>' . $this->language->get('text_export_note_csv_delimiter');
		$options['item']['delimiter'] = $form->getFieldHtml(array(
			'type' => 'selectbox',
			'name' => 'options[delimiter]',
			'value' => 0,
			'options' => array(',', ';', 'TAB')
		));

		$this->data['form']['options'] = $options;

		$this->data['form']['form_open'] = $form->getFieldHtml(array(
			'type' => 'form',
			'name' => $this->data['active'].'Frm',
			'action' => $this->data['action'],
		));

	}

	private function getImportedData($file) {
		if ( in_array($file['type'], array('text/csv','application/vnd.ms-excel')) ) {
			return $this->handler->csvToArray($file);
		} elseif ( in_array($file['type'], array('text/xml') ) ) {
			if ( $xml = simplexml_load_file($file['tmp_name']) ) {
				return $this->handler->xmlToArray($xml);
			}
		}
		return false;
	}

	private function import($file, $action = 'update_or_insert') {

		$results = array();
		$run_mode = isset($this->request->post['test_mode']) ? $this->request->post['test_mode'] : 'commit';

		if ( in_array($file['type'], array('text/csv','application/vnd.ms-excel','text/plain','application/octet-stream')) ) {
			#NOTE: 'application/octet-stream' is a solution for Windows OS sending unknown file type
			#TODO: Need to add test for the file in case of 'application/octet-stream'
			$csv_array = $this->handler->CSV2ArrayFromFile($file['tmp_name'], $this->request->post['options']['delimiter']);
			$results = $this->handler->importData($csv_array, $run_mode);
			$this->cache->delete('*');

		} elseif ( $file['type'] == 'text/xml' ) {
			$xml_array = $this->handler->XML2ArrayFromFile($file['tmp_name']);
			$results = $this->handler->importData($xml_array, $run_mode);
			$this->cache->delete('*');
		} else {
			$this->session->data['error'] = $this->language->get('error_file_format');
		}
		return $results;
	}

	private function _build_table_fields($form, $data)
	{
		$result = array();

		foreach ( $data as $table_name => $val ) {

			$result[$table_name]['main'] = $form->getFieldHtml(array(
				'type' => 'checkbox',
				'name' => 'section_' . $table_name,
				'value' => $table_name,
				'style' => 'section_input',
			));

			$columns_data = $this->handler->getTableColumns($table_name);
			$columns = array();
			foreach ($columns_data as $column) {
				$columns[$column['Field']] = $column['Field'];
			}

			$result[$table_name]['filter']['columns'] = $form->getFieldHtml(array(
				'type' => 'selectbox',
				'name' => 'data['.$table_name.'][filter][columns]',
				'value' => 0,
				'options' => $columns
			));

			$name = 'data['.$table_name.']';

			if ( isset($val['children']) ) {
				$name .= '[tables]';
				$children = $this->_get_table_children($form, $val['children'], $name);
				$result[$table_name]['children'] = $children;
			}
		}

		$this->data['text_no_children'] = $this->language->get('text_no_children');
		ksort($result);

		return $result;
	}

	private function _get_table_children($form, $data, $name) {

		$children = array();

		foreach ( $data as $key => $val ) {
			$new_name = $name . '['.$key.']';

			$children[$key]['name'] = $new_name;
			$children[$key]['field'] = $form->getFieldHtml(array(
				'type' => 'checkbox',
				'name' => $new_name,
				'value' => $key,
				'style' => '',
			));

			if ( isset($val['children']) ) {
				$new_name .= '[tables]';
				$children = array_merge($children, $this->_get_table_children($form, $val['children'], $new_name));
			}

		}
		return $children;
	}

}