Initial work on Google Docs storage backend

lib/filestorage/google.php

@@ -20,36 +20,252 @@
 * License along with this library.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-class OC_Filestorage_Google extends OC_Filestorage_Common {
+class OC_Filestorage_Google {
 
-	private $auth;
+	private $datadir;
+	private $consumer;
+	private $oauth_token;
+	private $sig_method;
 
-	public function __construct($parameters) {
+	public function __construct($arguments) {
+		$this->datadir = $arguments['datadir'];
+		$consumer_key = isset($arguments['consumer_key']) ? $arguments['consumer_key'] : 'anonymous';
+		$consumer_secret = isset($arguments['consumer_secret']) ? $arguments['consumer_secret'] : 'anonymous';
+		$this->consumer = new OAuthConsumer($consumer_key, $consumer_secret);
+		$this->oauth_token = new OAuthToken($arguments['token'], $arguments['token_secret']);
+		$this->sig_method = new OAuthSignatureMethod_HMAC_SHA1();
+	}
+
+	private function sendRequest($feedUri, $http_method, $postData = null) {
+		$feedUri = trim($feedUri);
+		// create an associative array from each key/value url query param pair.
+		$params = array();
+		$pieces = explode('?', $feedUri);
+		if (isset($pieces[1])) {
+			$params = explode_assoc('=', '&', $pieces[1]);
+		}
+		// urlencode each url parameter key/value pair
+		$tempStr = $pieces[0];
+		foreach ($params as $key => $value) {
+			$tempStr .= '&' . urlencode($key) . '=' . urlencode($value);
+		}
+		$feedUri = preg_replace('/&/', '?', $tempStr, 1);
+		$req = OAuthRequest::from_consumer_and_token($this->consumer, $this->oauth_token, $http_method, $feedUri, $params);
+		$req->sign_request($this->sig_method, $this->consumer, $this->oauth_token);
+		$auth_header = $req->to_header();
+		$result = send_signed_request($http_method, $feedUri, array($auth_header, 'Content-Type: application/atom+xml', 'GData-Version: 3.0'), $postData);
+		// TODO Return false if error is received
+		if (!$result) {
+			return false;
+		}
+		$result = explode('<', $result, 2);
+		$result = isset($result[1]) ? '<'.$result[1] : $result[0];
+		$dom = new DOMDocument();
+		$dom->loadXML($result);
+		return $dom;
+	}
+
+	private function getResource($path) {
+		// TODO Look up google docs query caching/only send back if changes occured
+		// TODO Look inside of collections for specific file
+		// TODO Strip extension
+		$title = basename($path);
+		return $this->sendRequest('https://docs.google.com/feeds/default/private/full?showfolders=true&title='.$title.'&title-exact=true', 'GET');
+	}
+
+	public function mkdir($path) {
+		$dir = dirname($path);
+		// Check if path parent is root directory
+		if ($dir == '/' || $dir == '\.' || $dir == '.') {
+			$feedUri = 'https://docs.google.com/feeds/default/private/full';
+		// Get parent content link
+		} else {
+			$dom = $this->getResource(basename($dir));
+			$feedUri = $dom->getElementsByTagName('content')->item(0)->getAttribute('src');
+		}
+		$title = basename($path);
+		// Construct post data
+		$postData = '<?xml version="1.0" encoding="UTF-8"?>';
+		$postData .= '<entry xmlns="http://www.w3.org/2005/Atom">';
+		$postData .= '<category scheme="http://schemas.google.com/g/2005#kind" term="http://schemas.google.com/docs/2007#folder"/>';
+		$postData .= '<title>'.$title.'</title>';
+		$postData .= '</entry>';
+		if ($dom = $this->sendRequest($feedUri, 'POST', $postData)) {
+			return true;
+		}
+		return false;
+	}
+
+	public function rmdir($path) {
+		return $this->unlink($path);
+	}
+
+	public function opendir($path) {
+		if ($path == '' || $path == '/') {
+			$resource = 'https://docs.google.com/feeds/default/private/full/folder%3Aroot/contents';
+			$dom = $this->sendRequest('https://docs.google.com/feeds/default/private/full/folder%3Aroot/contents', 'GET');
+		} else {
+			$dom = $this->getResource($path); 
+		}
+		global $FAKEDIRS;
+		$files = array();
+// 		while ($next) {
+// 			// send request $next link
+// 			$links = $dom->getElementsByTagName('link');
+// 			foreach ($links as $link) {
+// 				if ($link->getAttribute('rel') == 'next') {
+// 					$next = $link->getAttribute('src');
+// 					break;
+// 				} else {
+// 					$next = false;
+// 				}
+// 			}
+			$entries = $dom->getElementsByTagName('entry');
+			foreach($entries as $entry) {
+				$name = $entry->getElementsByTagName('title')->item(0)->nodeValue;
+				// Native Google resources don't include extensions in title
+				if (!strpos($name, '.')) {
+					if ($ext = $this->filetype('', $entry)) {
+						$name .= '.'.$ext;
+					}
+				}
+				$files[] = $name;
+			}
+// 		}
+		$FAKEDIRS['google'] = $files;
+		return opendir('fakedir://google');
+	}
+
+	public function is_dir($path) {
+		if ($entry = $this->getResource($path)) {
+			$categories = $entry->getElementsByTagName('category');
+			foreach ($categories as $category) {
+				if ($category->getAttribute('scheme') == 'http://schemas.google.com/g/2005#kind') {
+					// Check if label is equal to folder
+					$type = $category->getAttribute('label');
+					if (strlen(strstr($type, 'folder')) > 0) {
+						return true;
+					}
+				}
+			}
+		}
+		return false; 
+	}
+
+	public function is_file($path) {
+		if ($this->getResource($path)) {
+			return true;
+		}
+		return false;
+	}
 
+	public function stat($path) {
+		if ($dom = $this->getResource($path)) {
+			// TODO Native resources don't have a file size
+			$stat['size'] =  $dom->getElementsByTagNameNS('http://schemas.google.com/g/2005', 'quotaBytesUsed')->item(0)->nodeValue;
+			$stat['mtime'] = strtotime($dom->getElementsByTagName('updated')->item(1)->nodeValue);
+			$stat['ctime'] = strtotime($dom->getElementsByTagName('published')->item(0)->nodeValue);
+			return $stat;
 		}
+		return false; 
 		
-	private function connect() {
+	}
 
+	public function filetype($path, $entry = null) {
+		if ($entry == null) {
+			$entry = $this->getResource($path);
+		}
+		$categories = $entry->getElementsByTagName('category');
+		foreach ($categories as $category) {
+			if ($category->getAttribute('scheme') == 'http://schemas.google.com/g/2005#kind') {
+				// Guess extension from label, default to ODF extensions
+				$type = $category->getAttribute('label');
+				if (strlen(strstr($type, 'folder')) > 0) {
+					return '';
+				} else if (strlen(strstr($type, 'document')) > 0) {
+					return 'odt';
+				} else if (strlen(strstr($type, 'presentation')) > 0) {
+					return 'odp';
+				} else if (strlen(strstr($type, 'spreadsheet')) > 0) {
+					return 'ods';
+				} else {
+					return $type;
+				}
 			}
-	public function mkdir($path){}
-	public function rmdir($path){}
-	public function opendir($path){}
-	public function is_dir($path){}
-	public function is_file($path){}
-	public function stat($path){}
-	public function filetype($path){}
-	public function is_readable($path){}
-	public function is_writable($path){}
-	public function file_exists($path){}
-	public function unlink($path){}
-	public function rename($path1,$path2){}
+		}
+	}
+
+	public function is_readable($path) {
+		return true;
+	}
+
+	public function is_writable($path) {
+		// Check if edit or edit-media links exist
+		if ($entry = $this->getResource($path)) {
+			$links = $entry->getElementsByTagName('link');
+			foreach ($links as $link) {
+				if ($link->getAttribute('rel') == 'edit') {
+					return true;
+				} else if ($link->getAttribute('rel') == 'edit-media') {
+					return true;
+				}
+			}
+		}
+		return false;
+	}
+	
+	public function file_exists($path) {
+		if ($this->getResource($path)) {
+			return true;
+		}
+		return false;
+	}
+	
+	public function unlink($path) {
+		// Get resource self link to trash resource
+		if ($entry = $this->getResource($path)) {
+			$links = $entry->getElementsByTagName('link');
+			foreach ($links as $link) {
+				if ($link->getAttribute('rel') == 'self') {
+					$feedUri = $link->getAttribute('href');
+				}
+			}
+		}
+		if (isset($feedUri)) {
+			$this->sendRequest($feedUri, 'DELETE');
+			return true;
+		}
+		return false;
+	}
+
+	public function rename($path1,$path2) {
+
+	}
+
 	public function fopen($path,$mode){}
-	public function toTmpFile($path){}
+
+	public function toTmpFile($path) {
+		$dom = $this->getResource($path);
+		$url = $dom->getElementsByTagName('content')->getAttribute('src');		
+	}
+
 	public function fromTmpFile($tmpPath,$path){}
 	public function fromUploadedFile($tmpPath,$path){}
 	public function getMimeType($path){}
 	public function hash($type,$path,$raw){}
-	public function free_space($path){}
+	
+	public function free_space($path) {
+		if ($dom = $this->sendRequest('https://docs.google.com/feeds/metadata/default', 'GET')) {
+			$total = $dom->getElementsByTagNameNS('http://schemas.google.com/g/2005', 'quotaBytesTotal')->item(0)->nodeValue;
+			$used = $dom->getElementsByTagNameNS('http://schemas.google.com/g/2005', 'quotaBytesUsed')->item(0)->nodeValue;
+			return $total - $used;
+		}
+		return false;
+	}
+  
 	public function search($query){}
-	public function getLocalFile($path){}
+	
+	public function getLocalFile($path) {
+		return false;
+	}
 }
\ No newline at end of file