Location: PHPKode > projects > DotClear > dotclear/plugins/importExport/inc/class.dc.import.flat.php
<?php
# -- BEGIN LICENSE BLOCK ---------------------------------------
#
# This file is part of importExport, a plugin for DotClear2.
#
# Copyright (c) 2003-2012 Olivier Meunier & Association Dotclear
# Licensed under the GPL version 2.0 license.
# See LICENSE file or
# http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
#
# -- END LICENSE BLOCK -----------------------------------------
if (!defined('DC_RC_PATH')) { return; }

class dcImportFlat extends dcIeModule
{
	protected $status = false;
	
	public function setInfo()
	{
		$this->type = 'import';
		$this->name = __('Flat file import');
		$this->description = __('Imports a blog or a full Dotclear installation from flat file.');
	}
	
	public function process($do)
	{
		if ($do == 'single' || $do == 'full') {
			$this->status = $do;
			return;
		}
		
		$to_unlink = false;
		
		# Single blog import
		$files = $this->getPublicFiles();
		$single_upl = null;
		if (!empty($_POST['public_single_file']) && in_array($_POST['public_single_file'],$files)) {
			$single_upl = false;
		} elseif (!empty($_FILES['up_single_file'])) {
			$single_upl = true;
		}
		
		if ($single_upl !== null)
		{
			if ($single_upl) {
				files::uploadStatus($_FILES['up_single_file']);
				$file = DC_TPL_CACHE.'/'.md5(uniqid());
				if (!move_uploaded_file($_FILES['up_single_file']['tmp_name'],$file)) {
					throw new Exception(__('Unable to move uploaded file.'));
				}
				$to_unlink = true;
			} else {
				$file = $_POST['public_single_file'];
			}
			
			try {
				# Try to unzip file
				$unzip_file = $this->unzip($file);
				if (false !== $unzip_file) {
					$bk = new flatImport($this->core,$unzip_file);
				}
				# Else this is a normal file
				else {
					$bk = new flatImport($this->core,$file);
				}
				
				$bk->importSingle();
			} catch (Exception $e) {
				@unlink($unzip_file);
				if ($to_unlink) {
					@unlink($file);
				}
				throw $e;
			}
			@unlink($unzip_file);
			if ($to_unlink) {
				@unlink($file);
			}
			http::redirect($this->getURL().'&do=single');
		}
		
		# Full import
		$full_upl = null;
		if (!empty($_POST['public_full_file']) && in_array($_POST['public_full_file'],$files)) {
			$full_upl = false;
		} elseif (!empty($_FILES['up_full_file'])) {
			$full_upl = true;
		}
		
		if ($full_upl !== null && $this->core->auth->isSuperAdmin())
		{
			if (empty($_POST['your_pwd']) || !$this->core->auth->checkPassword(crypt::hmac(DC_MASTER_KEY,$_POST['your_pwd']))) {
				throw new Exception(__('Password verification failed'));
			}
			
			if ($full_upl) {
				files::uploadStatus($_FILES['up_full_file']);
				$file = DC_TPL_CACHE.'/'.md5(uniqid());
				if (!move_uploaded_file($_FILES['up_full_file']['tmp_name'],$file)) {
					throw new Exception(__('Unable to move uploaded file.'));
				}
				$to_unlink = true;
			} else {
				$file = $_POST['public_full_file'];
			}
			
			try {
				# Try to unzip file
				$unzip_file = $this->unzip($file);
				if (false !== $unzip_file) {
					$bk = new flatImport($this->core,$unzip_file);
				}
				# Else this is a normal file
				else {
					$bk = new flatImport($this->core,$file);
				}
				
				$bk->importFull();
			} catch (Exception $e) {
				@unlink($unzip_file);
				if ($to_unlink) {
					@unlink($file);
				}
				throw $e;
			}
			@unlink($unzip_file);
			if ($to_unlink) {
				@unlink($file);
			}
			http::redirect($this->getURL().'&do=full');
		}
		
		header('content-type:text/plain');
		var_dump($_POST);
		exit;
		
		$this->status = true;
	}
	
	public function gui()
	{
		if ($this->status == 'single')
		{
			dcPage::message(__('Single blog successfully imported.'));
			return;
		}
		if ($this->status == 'full')
		{
			dcPage::message(__('Content successfully imported.'));
			return;
		}
		
		$public_files = array_merge(array('-' => ''),$this->getPublicFiles());
		$has_files = (boolean) (count($public_files) - 1);
		
		echo
		'<script type="text/javascript">'."\n".
		"//<![CDATA[\n".
		dcPage::jsVar('dotclear.msg.confirm_full_import',
			__('Are you sure you want to import a full backup file?')).
		"$(function() {".
			"$('#up_single_file').change(function() { ".
				"if (this.value != '') { $('#public_single_file').val(''); } ".
			"}); ".
			"$('#public_single_file').change(function() { ".
				"if (this.value != '') { $('#up_single_file').val(''); } ".
			"}); ".
			"$('#up_full_file').change(function() { ".
				"if (this.value != '') { $('#public_full_file').val(''); } ".
			"}); ".
			"$('#public_full_file').change(function() { ".
				"if (this.value != '') { $('#up_full_file').val(''); } ".
			"}); ".
			"$('#formfull').submit(function() { ".
				"return window.confirm(dotclear.msg.confirm_full_import); ".
			"}); ".
		"});\n".
		"//]]>\n".
		"</script>\n";
		
		echo
		'<form action="'.$this->getURL(true).'" method="post" enctype="multipart/form-data">'.
		'<fieldset><legend>'.__('Single blog').'</legend>'.
		'<p>'.sprintf(__('This will import a single blog backup as new content in the current blog: %s.'),html::escapeHTML($this->core->blog->name)).'</p>'.
		
		'<p><label for="up_single_file">'.__('Upload a backup file').
		' ('.sprintf(__('Maximum size %s'),files::size(DC_MAX_UPLOAD_SIZE)).')'.'</label>'.
		'<input type="file" id="up_single_file" name="up_single_file" size="20" />'.
		'</p>';
		
		if ($has_files) {
			echo 
			'<p><label for="public_single_file">'.__('or pick up a local file in your public directory').' '.
			form::combo('public_single_file',$public_files).
			'</label></p>';
		}
		
		echo 
		'<p>'.
		$this->core->formNonce().
		form::hidden(array('do'),1).
		form::hidden(array('MAX_FILE_SIZE'),DC_MAX_UPLOAD_SIZE).
		'<input type="submit" value="'.__('Import').'" /></p>'.
		
		'</fieldset>'.
		'</form>';
		
		if ($this->core->auth->isSuperAdmin())
		{
			echo
			'<form action="'.$this->getURL(true).'" method="post" enctype="multipart/form-data" id="formfull">'.
			'<fieldset><legend>'.__('Multiple blogs').'</legend>'.
			'<p>'.__('This will reset all the content of your database, except users.').'</p>'.
			
			'<p><label for="up_full_file">'.__('Upload a backup file').
			' ('.sprintf(__('Maximum size %s'),files::size(DC_MAX_UPLOAD_SIZE)).')'.'</label>'.
			'<input type="file" id="up_full_file" name="up_full_file" size="20" />'.
			'</p>';
			
			if ($has_files) {
				echo 
				'<p><label for="public_full_file">'.__('or pick up a local file in your public directory').'</label>'.
				form::combo('public_full_file',$public_files).
				'</p>';
			}
			
			echo
			'<p><label for="your_pwd" class="required"><abbr title="'.__('Required field').'">*</abbr> '.__('Your password:').'</label>'.
			form::password('your_pwd',20,255).'</p>'.
			
			'<p>'.
			$this->core->formNonce().
			form::hidden(array('do'),1).
			form::hidden(array('MAX_FILE_SIZE'),DC_MAX_UPLOAD_SIZE).
			'<input type="submit" value="'.__('Import').'" /></p>'.
			
			'</fieldset>'.
			'</form>';
		}
	}
	
	protected function getPublicFiles()
	{
		$public_files = array();
		$dir = @dir($this->core->blog->public_path);
		if ($dir)
		{
			while (($entry = $dir->read()) !== false) {
				$entry_path = $dir->path.'/'.$entry;
				
				if (is_file($entry_path) && is_readable($entry_path))
				{
					# Do not test each zip file content here, its too long
					if (substr($entry_path,-4) == '.zip') {
							$public_files[$entry] = $entry_path;
					}
					elseif (self::checkFileContent($entry_path)) {
						$public_files[$entry] = $entry_path;
					}
				}
			}
		}
		return $public_files;
	}
	
	protected static function checkFileContent($entry_path)
	{
		$ret = false;
		
		$fp = fopen($entry_path,'rb');
		$ret = strpos(fgets($fp),'///DOTCLEAR|') === 0;
		fclose($fp);
		
		return $ret;
	}
	
	private function unzip($file)
	{
		$zip = new fileUnzip($file);
		
		if ($zip->isEmpty()) {
			$zip->close();
			return false;//throw new Exception(__('File is empty or not a compressed file.'));
		}
		
		foreach($zip->getFilesList() as $zip_file)
		{
			# Check zipped file name
			if (substr($zip_file,-4) != '.txt') {
				continue;
			}
			
			# Check zipped file contents
			$content = $zip->unzip($zip_file);
			if (strpos($content,'///DOTCLEAR|') !== 0) {
				unset($content);
				continue;
			}
			
			$target = path::fullFromRoot($zip_file,dirname($file));
			
			# Check existing files with same name
			if (file_exists($target)) {
				$zip->close();
				unset($content);
				throw new Exception(__('Another file with same name exists.'));
			}
			
			# Extract backup content
			if (file_put_contents($target,$content) === false) {
				$zip->close();
				unset($content);
				throw new Exception(__('Failed to extract backup file.'));
			}
			
			$zip->close();
			unset($content);
			
			# Return extracted file name
			return $target;
		}
		
		$zip->close();
		throw new Exception(__('No backup in compressed file.'));
	}
}
?>
Return current item: DotClear