<?php
  /**
   * Copyright (c) 2012 Robin Appelman <icewind@owncloud.com>
   * This file is licensed under the Affero General Public License version 3 or
   * later.
   * See the COPYING-README file.
   */
  
  class OC_Archive_ZIP extends OC_Archive{
  	/**
  	 * @var ZipArchive zip
  	 */
  	private $zip=null;
  	private $path;
  
  	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, OCP\Util::WARN);
  		}
  	}
  	/**
  	 * add an empty folder to the archive
  	 * @param string path
  	 * @return bool
  	 */
  	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
  	 */
  	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 bool
  	 */
  	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
  	 */
  	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
  	 */
  	function mtime($path) {
  		return filemtime($this->path);
  	}
  	/**
  	 * get the files in a folder
  	 * @param path
  	 * @return array
  	 */
  	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
  	 */
  	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
  	 */
  	function getFile($path) {
  		return $this->zip->getFromName($path);
  	}
  	/**
  	 * extract a single file from the archive
  	 * @param string path
  	 * @param string dest
  	 * @return bool
  	 */
  	function extractFile($path, $dest) {
  		$fp = $this->zip->getStream($path);
  		file_put_contents($dest, $fp);
  	}
  	/**
  	 * extract the archive
  	 * @param string path
  	 * @param string dest
  	 * @return bool
  	 */
  	function extract($dest) {
  		return $this->zip->extractTo($dest);
  	}
  	/**
  	 * check if a file or folder exists in the archive
  	 * @param string path
  	 * @return bool
  	 */
  	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
  	 */
  	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
  	 */
  	function getStream($path, $mode) {
  		if($mode=='r' or $mode=='rb') {
  			return $this->zip->getStream($path);
  		} 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
  			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'));
  			if($this->fileExists($path)) {
  				$this->extractFile($path, $tmpFile);
  			}
  			self::$tempFiles[$tmpFile]=$path;
  			return fopen('close://'.$tmpFile, $mode);
  		}
  	}
  
  	private static $tempFiles=array();
  	/**
  	 * write back temporary files
  	 */
  	function writeBack($tmpFile) {
  		if(isset(self::$tempFiles[$tmpFile])) {
  			$this->addFile(self::$tempFiles[$tmpFile], $tmpFile);
  			unlink($tmpFile);
  		}
  	}
  
  	private function stripPath($path) {
  		if(!$path || $path[0]=='/') {
  			return substr($path, 1);
  		}else{
  			return $path;
  		}
  	}
  }