<?php

/**
* ownCloud
*
* @author Michael Gapczynski
* @copyright 2012 Michael Gapczynski mtgap@owncloud.com
*
* 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/>.
*/

require_once 'Dropbox/autoload.php';

class OC_Filestorage_Dropbox extends OC_Filestorage_Common {

	private $dropbox;
	private $metaData = array();

	private static $tempFiles = array();

	public function __construct($params) {
		$oauth = new Dropbox_OAuth_Curl($params['app_key'], $params['app_secret']);
		$oauth->setToken($params['token'], $params['token_secret']);
		$this->dropbox = new Dropbox_API($oauth, 'dropbox');
	}

	private function getMetaData($path, $list = false) {
		if (!$list && isset($this->metaData[$path])) {
			return $this->metaData[$path];
		} else {
			if ($list) {
				try {
					$response = $this->dropbox->getMetaData($path);
				} catch (Exception $exception) {
					return false;
				}
				if ($response && isset($response['contents'])) {
					$contents = $response['contents'];
					// Cache folder's contents
					foreach ($contents as $file) {
						$this->metaData[$path.'/'.basename($file['path'])] = $file;
					}
					unset($response['contents']);
					$this->metaData[$path] = $response;
				}
				$this->metaData[$path] = $response;
				// Return contents of folder only
				return $contents;
			} else {
				try {
					$response = $this->dropbox->getMetaData($path, 'false');
					$this->metaData[$path] = $response;
					return $response;
				} catch (Exception $exception) {
					return false;
				}
			}
		}
	}

	public function mkdir($path) {
		return $this->dropbox->createFolder($path);
	}

	public function rmdir($path) {
		return $this->dropbox->delete($path);
	}

	public function opendir($path) {
		if ($contents = $this->getMetaData($path, true)) {
			$files = array();
			foreach ($contents as $file) {
				$files[] = basename($file['path']);
			}
			OC_FakeDirStream::$dirs['dropbox'.$path] = $files;
			return opendir('fakedir://dropbox'.$path);
		}
		return false;
	}

	public function stat($path) {
		if ($metaData = $this->getMetaData($path)) {
			$stat['size'] = $metaData['bytes'];
			$stat['atime'] = time();
			$stat['mtime'] = (isset($metaData['modified'])) ? strtotime($metaData['modified']) : time();
			$stat['ctime'] = $stat['mtime'];
			return $stat;
		}
		return false;
	}

	public function filetype($path) {
		if ($path == '' || $path == '/') {
			return 'dir';
		} else if ($metaData = $this->getMetaData($path)) {
			if ($metaData['is_dir'] == 'true') {
				return 'dir';
			} else {
				return 'file';
			}
		}
		return false;
	}

	public function is_readable($path) {
		return self::file_exists($path);
	}

	public function is_writable($path) {
		return self::file_exists($path);
	}

	public function file_exists($path) {
		if ($path == '' || $path == '/') {
			return true;
		}
		if ($this->getMetaData($path)) {
			return true;
		}
		return false;
	}

	public function unlink($path) {
		return $this->dropbox->delete($path);
	}

	public function fopen($path, $mode) {
		switch ($mode) {
			case 'r':
			case 'rb':
				$tmpFile = OC_Helper::tmpFile();
				file_put_contents($tmpFile, $this->dropbox->getFile($path));
				return fopen($tmpFile, 'r');
			case 'w':
			case 'wb':
			case 'a':
			case 'ab':
			case 'r+':
			case 'w+':
			case 'wb+':
			case 'a+':
			case 'x':
			case 'x+':
			case 'c':
			case 'c+':
				if (strrpos($path, '.') !== false) {
					$ext = substr($path, strrpos($path, '.'));
				} else {
					$ext = '';
				}
				$tmpFile = OC_Helper::tmpFile($ext);
				OC_CloseStreamWrapper::$callBacks[$tmpFile] = array($this, 'writeBack');
				if ($this->file_exists($path)) {
					$source = $this->fopen($path, 'r');
					file_put_contents($tmpFile, $source);
				}
				self::$tempFiles[$tmpFile] = $path;
				return fopen('close://'.$tmpFile, $mode);
		}
		return false;
	}

	public function writeBack($tmpFile) {
		if (isset(self::$tempFiles[$tmpFile])) {
			$handle = fopen($tmpFile, 'r');
			$response = $this->dropbox->putFile(self::$tempFiles[$tmpFile], $handle);
			if ($response) {
				unlink($tmpFile);
			}
		}
	}

	public function getMimeType($path) {
		if ($this->filetype($path) == 'dir') {
			return 'httpd/unix-directory';
		} else if ($metaData = $this->getMetaData($path)) {
			return $metaData['mime_type'];
		}
		return false;
	}

	public function free_space($path) {
		if ($info = $this->dropbox->getAccountInfo()) {
			return $info['quota_info']['quota'] - $info['quota_info']['normal'];
		}
		return false;
	}

	public function touch($path, $mtime = null) {
		return false;
	}

}

?>