<?php
  
  /**
   * ownCloud
   *
   * @author Frank Karlitschek
   * @copyright 2012 Frank Karlitschek frank@owncloud.org
   *
   * This library is free software; you can redistribute it and/or
   * modify it under the terms of the GNU AFFERO GENERAL PUBLIC LICENSE
   * License as published by the Free Software Foundation; either
   * version 3 of the License, or any later version.
   *
   * This library 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 along with this library.  If not, see <http://www.gnu.org/licenses/>.
   *
   */

  // TODO: get rid of this using proper composer packages
  require_once 'mcnetic/phpzipstreamer/ZipStreamer.php';
  
  class GET_TYPE {
  	const FILE = 1;
  	const ZIP_FILES = 2;
  	const ZIP_DIR = 3;
  }

  /**

   * Class for file server access

   *
   */
  class OC_Files {

  	/**
  	 * @param string $filename
  	 * @param string $name
  	 * @param bool $zip
  	 */
  	private static function sendHeaders($filename, $name, $zip = false) {
  		OC_Response::setContentDispositionHeader($name, 'attachment');
  		header('Content-Transfer-Encoding: binary');
  		OC_Response::disableCaching();
  		if ($zip) {
  			header('Content-Type: application/zip');
  		} else {
  			$filesize = \OC\Files\Filesystem::filesize($filename);
  			header('Content-Type: '.\OC\Files\Filesystem::getMimeType($filename));
  			if ($filesize > -1) {
  				header("Content-Length: ".$filesize);
  			}
  		}

  	}
  
  	/**
  	 * return the content of a file or return a zip file containing multiple files
  	 *
  	 * @param string $dir

  	 * @param string $files ; separated list of files to download

  	 * @param boolean $only_header ; boolean to only send header of the request
  	 */
  	public static function get($dir, $files, $only_header = false) {
  		$xsendfile = false;
  		if (isset($_SERVER['MOD_X_SENDFILE_ENABLED']) ||

  			isset($_SERVER['MOD_X_SENDFILE2_ENABLED']) ||

  			isset($_SERVER['MOD_X_ACCEL_REDIRECT_ENABLED'])) {
  			$xsendfile = true;
  		}

  		if (is_array($files) && count($files) === 1) {

  			$files = $files[0];
  		}
  
  		if (is_array($files)) {

  			$get_type = GET_TYPE::ZIP_FILES;

  			$basename = basename($dir);
  			if ($basename) {
  				$name = $basename . '.zip';
  			} else {

  				$name = 'download.zip';

  			}

  
  			$filename = $dir . '/' . $name;

  		} else {

  			$filename = $dir . '/' . $files;

  			if (\OC\Files\Filesystem::is_dir($dir . '/' . $files)) {
  				$get_type = GET_TYPE::ZIP_DIR;
  				// downloading root ?
  				if ($files === '') {
  					$name = 'download.zip';
  				} else {
  					$name = $files . '.zip';
  				}
  
  			} else {
  				$get_type = GET_TYPE::FILE;
  				$name = $files;
  			}
  		}
  
  		if ($get_type === GET_TYPE::FILE) {
  			$zip = false;

  			if ($xsendfile && OC_App::isEnabled('files_encryption')) {
  				$xsendfile = false;
  			}

  		} else {
  			$zip = new ZipStreamer(false);

  		}
  		OC_Util::obEnd();
  		if ($zip or \OC\Files\Filesystem::isReadable($filename)) {

  			self::sendHeaders($filename, $name, $zip);
  		} elseif (!\OC\Files\Filesystem::file_exists($filename)) {

  			header("HTTP/1.0 404 Not Found");
  			$tmpl = new OC_Template('', '404', 'guest');
  			$tmpl->assign('file', $name);
  			$tmpl->printPage();
  		} else {
  			header("HTTP/1.0 403 Forbidden");
  			die('403 Forbidden');
  		}
  		if($only_header) {
  			return ;
  		}
  		if ($zip) {

  			$executionTime = intval(ini_get('max_execution_time'));
  			set_time_limit(0);
  			if ($get_type === GET_TYPE::ZIP_FILES) {
  				foreach ($files as $file) {
  					$file = $dir . '/' . $file;
  					if (\OC\Files\Filesystem::is_file($file)) {
  						$fh = \OC\Files\Filesystem::fopen($file, 'r');
  						$zip->addFileFromStream($fh, basename($file));
  						fclose($fh);
  					} elseif (\OC\Files\Filesystem::is_dir($file)) {
  						self::zipAddDir($file, $zip);
  					}

  				}

  			} elseif ($get_type === GET_TYPE::ZIP_DIR) {
  				$file = $dir . '/' . $files;
  				self::zipAddDir($file, $zip);

  			}

  			$zip->finalize();
  			set_time_limit($executionTime);
  		} else {
  			if ($xsendfile) {
  				$view = \OC\Files\Filesystem::getView();
  				/** @var $storage \OC\Files\Storage\Storage */
  				list($storage) = $view->resolvePath($filename);
  				if ($storage->isLocal()) {
  					self::addSendfileHeader($filename);
  				} else {
  					\OC\Files\Filesystem::readfile($filename);
  				}
  			} else {
  				\OC\Files\Filesystem::readfile($filename);

  			}
  		}
  	}

  	/**
  	 * @param false|string $filename
  	 */

  	private static function addSendfileHeader($filename) {
  		if (isset($_SERVER['MOD_X_SENDFILE_ENABLED'])) {

  			$filename = \OC\Files\Filesystem::getLocalFile($filename);

  			header("X-Sendfile: " . $filename);

   		}
   		if (isset($_SERVER['MOD_X_SENDFILE2_ENABLED'])) {

  			$filename = \OC\Files\Filesystem::getLocalFile($filename);
  			if (isset($_SERVER['HTTP_RANGE']) &&

  				preg_match("/^bytes=([0-9]+)-([0-9]*)$/", $_SERVER['HTTP_RANGE'], $range)) {
  				$filelength = filesize($filename);

   				if ($range[2] === "") {

   					$range[2] = $filelength - 1;
   				}
   				header("Content-Range: bytes $range[1]-$range[2]/" . $filelength);
   				header("HTTP/1.1 206 Partial content");
   				header("X-Sendfile2: " . str_replace(",", "%2c", rawurlencode($filename)) . " $range[1]-$range[2]");
   			} else {
   				header("X-Sendfile: " . $filename);
   			}

  		}

  		if (isset($_SERVER['MOD_X_ACCEL_REDIRECT_ENABLED'])) {

  			$filename = \OC::$WEBROOT . '/data' . \OC\Files\Filesystem::getRoot() . $filename;

  			header("X-Accel-Redirect: " . $filename);
  		}
  	}

  	/**
  	 * @param string $dir
  	 * @param ZipStreamer $zip
  	 * @param string $internalDir
  	 */

  	public static function zipAddDir($dir, $zip, $internalDir='') {
  		$dirname=basename($dir);

  		$rootDir = $internalDir.$dirname;
  		if (!empty($rootDir)) {
  			$zip->addEmptyDir($rootDir);
  		}

  		$internalDir.=$dirname.='/';

  		// prevent absolute dirs
  		$internalDir = ltrim($internalDir, '/');
  
  		$files=\OC\Files\Filesystem::getDirectoryContent($dir);

  		foreach($files as $file) {
  			$filename=$file['name'];
  			$file=$dir.'/'.$filename;
  			if(\OC\Files\Filesystem::is_file($file)) {

  				$fh = \OC\Files\Filesystem::fopen($file, 'r');
  				$zip->addFileFromStream($fh, $internalDir.$filename);
  				fclose($fh);

  			}elseif(\OC\Files\Filesystem::is_dir($file)) {
  				self::zipAddDir($file, $zip, $internalDir);
  			}
  		}
  	}
  
  	/**

  	 * set the maximum upload size limit for apache hosts using .htaccess
  	 *

  	 * @param int $size file size in bytes
  	 * @return bool false on failure, size on success

  	 */
  	static function setUploadLimit($size) {
  		//don't allow user to break his config -- upper boundary
  		if ($size > PHP_INT_MAX) {
  			//max size is always 1 byte lower than computerFileSize returns
  			if ($size > PHP_INT_MAX + 1)
  				return false;
  			$size -= 1;
  		} else {

  			$size = OC_Helper::phpFileSize($size);

  		}
  
  		//don't allow user to break his config -- broken or malicious size input

  		if (intval($size) === 0) {

  			return false;
  		}

  		//suppress errors in case we don't have permissions for
  		$htaccess = @file_get_contents(OC::$SERVERROOT . '/.htaccess');

  		if (!$htaccess) {
  			return false;
  		}
  
  		$phpValueKeys = array(
  			'upload_max_filesize',
  			'post_max_size'
  		);
  
  		foreach ($phpValueKeys as $key) {
  			$pattern = '/php_value ' . $key . ' (\S)*/';
  			$setting = 'php_value ' . $key . ' ' . $size;
  			$hasReplaced = 0;
  			$content = preg_replace($pattern, $setting, $htaccess, 1, $hasReplaced);
  			if ($content !== null) {
  				$htaccess = $content;
  			}

  			if ($hasReplaced === 0) {

  				$htaccess .= "
  " . $setting;
  			}
  		}
  
  		//check for write permissions
  		if (is_writable(OC::$SERVERROOT . '/.htaccess')) {
  			file_put_contents(OC::$SERVERROOT . '/.htaccess', $htaccess);
  			return OC_Helper::computerFileSize($size);
  		} else {
  			OC_Log::write('files',
  				'Can\'t write upload limit to ' . OC::$SERVERROOT . '/.htaccess. Please check the file permissions',
  				OC_Log::WARN);
  		}
  		return false;
  	}
  }