nextcloud/lib/private/archive/zip.php

230 lines
5.7 KiB
PHP
Raw Normal View History

<?php
/**
2015-03-26 13:44:34 +03:00
* @author Bart Visscher <bartv@thisnet.nl>
* @author Christopher Schäpers <kondou@ts.unde.re>
* @author Felix Moeller <mail@felixmoeller.de>
* @author Frank Karlitschek <frank@owncloud.org>
* @author Joas Schilling <nickvergessen@owncloud.com>
* @author Jörn Friedrich Dreyer <jfd@butonic.de>
* @author Morris Jobke <hey@morrisjobke.de>
* @author Robin Appelman <icewind@owncloud.com>
* @author Scrutinizer Auto-Fixer <auto-fixer@scrutinizer-ci.com>
* @author Thomas Müller <thomas.mueller@tmit.eu>
*
* @copyright Copyright (c) 2015, ownCloud, Inc.
* @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/>
*
*/
class OC_Archive_ZIP extends OC_Archive{
/**
* @var ZipArchive zip
*/
private $zip=null;
private $path;
2012-08-29 10:38:33 +04:00
/**
* @param string $source
*/
2012-09-07 17:22:01 +04:00
function __construct($source) {
$this->path=$source;
$this->zip=new ZipArchive();
2012-09-10 13:23:55 +04:00
if($this->zip->open($source, ZipArchive::CREATE)) {
}else{
2012-09-10 13:23:55 +04:00
OCP\Util::writeLog('files_archive', 'Error while opening archive '.$source, OCP\Util::WARN);
}
}
/**
* add an empty folder to the archive
* @param string $path
* @return bool
*/
2012-09-07 17:22:01 +04:00
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
*/
2012-11-02 22:53:02 +04:00
function addFile($path, $source='') {
2012-09-07 17:22:01 +04:00
if($source and $source[0]=='/' and file_exists($source)) {
2012-09-10 13:23:55 +04:00
$result=$this->zip->addFile($source, $path);
}else{
2012-09-10 13:23:55 +04:00
$result=$this->zip->addFromString($path, $source);
}
2012-09-07 17:22:01 +04:00
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
*/
2012-11-02 22:53:02 +04:00
function rename($source, $dest) {
$source=$this->stripPath($source);
$dest=$this->stripPath($dest);
2012-09-10 13:23:55 +04:00
$this->zip->renameName($source, $dest);
}
/**
* get the uncompressed size of a file in the archive
* @param string $path
* @return int
*/
2012-09-07 17:22:01 +04:00
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
*/
2012-09-07 17:22:01 +04:00
function mtime($path) {
return filemtime($this->path);
}
/**
* get the files in a folder
* @param string $path
* @return array
*/
2012-09-07 17:22:01 +04:00
function getFolder($path) {
$files=$this->getFiles();
$folderContent=array();
$pathLength=strlen($path);
2012-09-07 17:22:01 +04:00
foreach($files as $file) {
2012-09-10 13:23:55 +04:00
if(substr($file, 0, $pathLength)==$path and $file!=$path) {
2012-10-05 14:39:43 +04:00
if(strrpos(substr($file, 0, -1), '/')<=$pathLength) {
2012-09-10 13:23:55 +04:00
$folderContent[]=substr($file, $pathLength);
}
}
}
return $folderContent;
}
/**
2013-07-16 07:56:52 +04:00
* get all files in the archive
* @return array
*/
2012-09-07 17:22:01 +04:00
function getFiles() {
$fileCount=$this->zip->numFiles;
$files=array();
2012-09-07 17:22:01 +04:00
for($i=0;$i<$fileCount;$i++) {
$files[]=$this->zip->getNameIndex($i);
}
return $files;
}
/**
* get the content of a file
* @param string $path
* @return string
*/
2012-09-07 17:22:01 +04:00
function getFile($path) {
return $this->zip->getFromName($path);
}
/**
* extract a single file from the archive
* @param string $path
* @param string $dest
* @return boolean|null
*/
2012-11-02 22:53:02 +04:00
function extractFile($path, $dest) {
$fp = $this->zip->getStream($path);
2012-09-10 13:23:55 +04:00
file_put_contents($dest, $fp);
}
2012-03-29 00:31:45 +04:00
/**
* extract the archive
* @param string $dest
2012-03-29 00:31:45 +04:00
* @return bool
*/
2012-09-07 17:22:01 +04:00
function extract($dest) {
2012-03-29 00:31:45 +04:00
return $this->zip->extractTo($dest);
}
/**
* check if a file or folder exists in the archive
* @param string $path
* @return bool
*/
2012-09-07 17:22:01 +04:00
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
*/
2012-09-07 17:22:01 +04:00
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
*/
2012-11-02 22:53:02 +04:00
function getStream($path, $mode) {
2012-09-07 17:22:01 +04:00
if($mode=='r' or $mode=='rb') {
return $this->zip->getStream($path);
2012-10-05 14:39:43 +04:00
} else {
//since we cant directly get a writable stream,
//make a temp copy of the file and put it back
//in the archive when the stream is closed
2012-09-10 13:23:55 +04:00
if(strrpos($path, '.')!==false) {
$ext=substr($path, strrpos($path, '.'));
}else{
$ext='';
}
$tmpFile=OCP\Files::tmpFile($ext);
\OC\Files\Stream\Close::registerCallback($tmpFile, array($this, 'writeBack'));
2012-09-07 17:22:01 +04:00
if($this->fileExists($path)) {
2012-09-10 13:23:55 +04:00
$this->extractFile($path, $tmpFile);
}
self::$tempFiles[$tmpFile]=$path;
2012-09-10 13:23:55 +04:00
return fopen('close://'.$tmpFile, $mode);
}
}
private static $tempFiles=array();
/**
* write back temporary files
*/
2012-09-07 17:22:01 +04:00
function writeBack($tmpFile) {
if(isset(self::$tempFiles[$tmpFile])) {
2012-09-10 13:23:55 +04:00
$this->addFile(self::$tempFiles[$tmpFile], $tmpFile);
unlink($tmpFile);
}
}
/**
* @param string $path
* @return string
*/
2012-09-07 17:22:01 +04:00
private function stripPath($path) {
if(!$path || $path[0]=='/') {
2012-09-10 13:23:55 +04:00
return substr($path, 1);
}else{
return $path;
}
}
}