You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
232 lines
5.7 KiB
232 lines
5.7 KiB
<?php |
|
/** |
|
* @copyright Copyright (c) 2016, ownCloud, Inc. |
|
* |
|
* @author Bart Visscher <bartv@thisnet.nl> |
|
* @author Christopher Schäpers <kondou@ts.unde.re> |
|
* @author Joas Schilling <coding@schilljs.com> |
|
* @author Jörn Friedrich Dreyer <jfd@butonic.de> |
|
* @author Morris Jobke <hey@morrisjobke.de> |
|
* @author Robin Appelman <robin@icewind.nl> |
|
* @author Roeland Jago Douma <roeland@famdouma.nl> |
|
* @author Stefan Weil <sw@weilnetz.de> |
|
* @author Thomas Müller <thomas.mueller@tmit.eu> |
|
* |
|
* @license AGPL-3.0 |
|
* |
|
* This code is free software: you can redistribute it and/or modify |
|
* it under the terms of the GNU Affero General Public License, version 3, |
|
* as published by the Free Software Foundation. |
|
* |
|
* This program is distributed in the hope that it will be useful, |
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
* GNU Affero General Public License for more details. |
|
* |
|
* You should have received a copy of the GNU Affero General Public License, version 3, |
|
* along with this program. If not, see <http://www.gnu.org/licenses/> |
|
* |
|
*/ |
|
|
|
namespace OC\Archive; |
|
|
|
use Icewind\Streams\CallbackWrapper; |
|
use OCP\ILogger; |
|
|
|
class ZIP extends Archive{ |
|
/** |
|
* @var \ZipArchive zip |
|
*/ |
|
private $zip=null; |
|
private $path; |
|
|
|
/** |
|
* @param string $source |
|
*/ |
|
public function __construct($source) { |
|
$this->path=$source; |
|
$this->zip=new \ZipArchive(); |
|
if($this->zip->open($source, \ZipArchive::CREATE)) { |
|
}else{ |
|
\OCP\Util::writeLog('files_archive', 'Error while opening archive '.$source, ILogger::WARN); |
|
} |
|
} |
|
/** |
|
* add an empty folder to the archive |
|
* @param string $path |
|
* @return bool |
|
*/ |
|
public function addFolder($path) { |
|
return $this->zip->addEmptyDir($path); |
|
} |
|
/** |
|
* add a file to the archive |
|
* @param string $path |
|
* @param string $source either a local file or string data |
|
* @return bool |
|
*/ |
|
public function addFile($path, $source='') { |
|
if($source and $source[0]=='/' and file_exists($source)) { |
|
$result=$this->zip->addFile($source, $path); |
|
}else{ |
|
$result=$this->zip->addFromString($path, $source); |
|
} |
|
if($result) { |
|
$this->zip->close();//close and reopen to save the zip |
|
$this->zip->open($this->path); |
|
} |
|
return $result; |
|
} |
|
/** |
|
* rename a file or folder in the archive |
|
* @param string $source |
|
* @param string $dest |
|
* @return boolean|null |
|
*/ |
|
public function rename($source, $dest) { |
|
$source=$this->stripPath($source); |
|
$dest=$this->stripPath($dest); |
|
$this->zip->renameName($source, $dest); |
|
} |
|
/** |
|
* get the uncompressed size of a file in the archive |
|
* @param string $path |
|
* @return int |
|
*/ |
|
public function filesize($path) { |
|
$stat=$this->zip->statName($path); |
|
return $stat['size']; |
|
} |
|
/** |
|
* get the last modified time of a file in the archive |
|
* @param string $path |
|
* @return int |
|
*/ |
|
public function mtime($path) { |
|
return filemtime($this->path); |
|
} |
|
/** |
|
* get the files in a folder |
|
* @param string $path |
|
* @return array |
|
*/ |
|
public function getFolder($path) { |
|
$files=$this->getFiles(); |
|
$folderContent=array(); |
|
$pathLength=strlen($path); |
|
foreach($files as $file) { |
|
if(substr($file, 0, $pathLength)==$path and $file!=$path) { |
|
if(strrpos(substr($file, 0, -1), '/')<=$pathLength) { |
|
$folderContent[]=substr($file, $pathLength); |
|
} |
|
} |
|
} |
|
return $folderContent; |
|
} |
|
/** |
|
* get all files in the archive |
|
* @return array |
|
*/ |
|
public function getFiles() { |
|
$fileCount=$this->zip->numFiles; |
|
$files=array(); |
|
for($i=0;$i<$fileCount;$i++) { |
|
$files[]=$this->zip->getNameIndex($i); |
|
} |
|
return $files; |
|
} |
|
/** |
|
* get the content of a file |
|
* @param string $path |
|
* @return string |
|
*/ |
|
public function getFile($path) { |
|
return $this->zip->getFromName($path); |
|
} |
|
/** |
|
* extract a single file from the archive |
|
* @param string $path |
|
* @param string $dest |
|
* @return boolean|null |
|
*/ |
|
public function extractFile($path, $dest) { |
|
$fp = $this->zip->getStream($path); |
|
file_put_contents($dest, $fp); |
|
} |
|
/** |
|
* extract the archive |
|
* @param string $dest |
|
* @return bool |
|
*/ |
|
public function extract($dest) { |
|
return $this->zip->extractTo($dest); |
|
} |
|
/** |
|
* check if a file or folder exists in the archive |
|
* @param string $path |
|
* @return bool |
|
*/ |
|
public function fileExists($path) { |
|
return ($this->zip->locateName($path)!==false) or ($this->zip->locateName($path.'/')!==false); |
|
} |
|
/** |
|
* remove a file or folder from the archive |
|
* @param string $path |
|
* @return bool |
|
*/ |
|
public function remove($path) { |
|
if($this->fileExists($path.'/')) { |
|
return $this->zip->deleteName($path.'/'); |
|
}else{ |
|
return $this->zip->deleteName($path); |
|
} |
|
} |
|
/** |
|
* get a file handler |
|
* @param string $path |
|
* @param string $mode |
|
* @return resource |
|
*/ |
|
public function getStream($path, $mode) { |
|
if($mode=='r' or $mode=='rb') { |
|
return $this->zip->getStream($path); |
|
} else { |
|
//since we can't directly get a writable stream, |
|
//make a temp copy of the file and put it back |
|
//in the archive when the stream is closed |
|
if(strrpos($path, '.')!==false) { |
|
$ext=substr($path, strrpos($path, '.')); |
|
}else{ |
|
$ext=''; |
|
} |
|
$tmpFile = \OC::$server->getTempManager()->getTemporaryFile($ext); |
|
if($this->fileExists($path)) { |
|
$this->extractFile($path, $tmpFile); |
|
} |
|
$handle = fopen($tmpFile, $mode); |
|
return CallbackWrapper::wrap($handle, null, null, function () use ($path, $tmpFile) { |
|
$this->writeBack($tmpFile, $path); |
|
}); |
|
} |
|
} |
|
|
|
/** |
|
* write back temporary files |
|
*/ |
|
public function writeBack($tmpFile, $path) { |
|
$this->addFile($path, $tmpFile); |
|
unlink($tmpFile); |
|
} |
|
|
|
/** |
|
* @param string $path |
|
* @return string |
|
*/ |
|
private function stripPath($path) { |
|
if(!$path || $path[0]=='/') { |
|
return substr($path, 1); |
|
}else{ |
|
return $path; |
|
} |
|
} |
|
}
|
|
|