nextcloud / server

apps/dav/lib/Connector/Sabre/QuotaPlugin.php

Indentation +168 added lines, -168 removed lines

@@ -42,172 +42,172 @@
  */
 class QuotaPlugin extends \Sabre\DAV\ServerPlugin {
 
-	/** @var \OC\Files\View */
-	private $view;
-
-	/**
-	 * Reference to main server object
-	 *
-	 * @var \Sabre\DAV\Server
-	 */
-	private $server;
-
-	/**
-	 * @param \OC\Files\View $view
-	 */
-	public function __construct($view) {
-		$this->view = $view;
-	}
-
-	/**
-	 * This initializes the plugin.
-	 *
-	 * This function is called by \Sabre\DAV\Server, after
-	 * addPlugin is called.
-	 *
-	 * This method should set up the requires event subscriptions.
-	 *
-	 * @param \Sabre\DAV\Server $server
-	 * @return void
-	 */
-	public function initialize(\Sabre\DAV\Server $server) {
-
-		$this->server = $server;
-
-		$server->on('beforeWriteContent', [$this, 'beforeWriteContent'], 10);
-		$server->on('beforeCreateFile', [$this, 'beforeCreateFile'], 10);
-		$server->on('beforeMove', [$this, 'beforeMove'], 10);
-	}
-
-	/**
-	 * Check quota before creating file
-	 *
-	 * @param string $uri target file URI
-	 * @param resource $data data
-	 * @param INode $parent Sabre Node
-	 * @param bool $modified modified
-	 */
-	public function beforeCreateFile($uri, $data, INode $parent, $modified) {
-		if (!$parent instanceof Node) {
-			return;
-		}
-
-		return $this->checkQuota($parent->getPath() . '/' . basename($uri));
-	}
-
-	/**
-	 * Check quota before writing content
-	 *
-	 * @param string $uri target file URI
-	 * @param INode $node Sabre Node
-	 * @param resource $data data
-	 * @param bool $modified modified
-	 */
-	public function beforeWriteContent($uri, INode $node, $data, $modified) {
-		if (!$node instanceof Node) {
-			return;
-		}
-
-		return $this->checkQuota($node->getPath());
-	}
-
-	/**
-	 * Check if we're moving a Futurefile in which case we need to check
-	 * the quota on the target destination.
-	 *
-	 * @param string $source source path
-	 * @param string $destination destination path
-	 */
-	public function beforeMove($source, $destination) {
-		$sourceNode = $this->server->tree->getNodeForPath($source);
-		if (!$sourceNode instanceof FutureFile) {
-			return;
-		}
-
-		// get target node for proper path conversion
-		if ($this->server->tree->nodeExists($destination)) {
-			$destinationNode = $this->server->tree->getNodeForPath($destination);
-			$path = $destinationNode->getPath();
-		} else {
-			$parentNode = $this->server->tree->getNodeForPath(dirname($destination));
-			$path = $parentNode->getPath();
-		}
-
-		return $this->checkQuota($path, $sourceNode->getSize());
-	}
-
-
-	/**
-	 * This method is called before any HTTP method and validates there is enough free space to store the file
-	 *
-	 * @param string $path relative to the users home
-	 * @param int $length
-	 * @throws InsufficientStorage
-	 * @return bool
-	 */
-	public function checkQuota($path, $length = null) {
-		if ($length === null) {
-			$length = $this->getLength();
-		}
-
-		if ($length) {
-			list($parentPath, $newName) = \Sabre\Uri\split($path);
-			if(is_null($parentPath)) {
-				$parentPath = '';
-			}
-			$req = $this->server->httpRequest;
-			if ($req->getHeader('OC-Chunked')) {
-				$info = \OC_FileChunking::decodeName($newName);
-				$chunkHandler = $this->getFileChunking($info);
-				// subtract the already uploaded size to see whether
-				// there is still enough space for the remaining chunks
-				$length -= $chunkHandler->getCurrentSize();
-				// use target file name for free space check in case of shared files
-				$path = rtrim($parentPath, '/') . '/' . $info['name'];
-			}
-			$freeSpace = $this->getFreeSpace($path);
-			if ($freeSpace >= 0 && $length > $freeSpace) {
-				if (isset($chunkHandler)) {
-					$chunkHandler->cleanup();
-				}
-				throw new InsufficientStorage("Insufficient space in $path, $length required, $freeSpace available");
-			}
-		}
-		return true;
-	}
-
-	public function getFileChunking($info) {
-		// FIXME: need a factory for better mocking support
-		return new \OC_FileChunking($info);
-	}
-
-	public function getLength() {
-		$req = $this->server->httpRequest;
-		$length = $req->getHeader('X-Expected-Entity-Length');
-		if (!is_numeric($length)) {
-			$length = $req->getHeader('Content-Length');
-			$length = is_numeric($length) ? $length : null;
-		}
-
-		$ocLength = $req->getHeader('OC-Total-Length');
-		if (is_numeric($length) && is_numeric($ocLength)) {
-			return max($length, $ocLength);
-		}
-
-		return $length;
-	}
-
-	/**
-	 * @param string $uri
-	 * @return mixed
-	 * @throws ServiceUnavailable
-	 */
-	public function getFreeSpace($uri) {
-		try {
-			$freeSpace = $this->view->free_space(ltrim($uri, '/'));
-			return $freeSpace;
-		} catch (StorageNotAvailableException $e) {
-			throw new ServiceUnavailable($e->getMessage());
-		}
-	}
+    /** @var \OC\Files\View */
+    private $view;
+
+    /**
+     * Reference to main server object
+     *
+     * @var \Sabre\DAV\Server
+     */
+    private $server;
+
+    /**
+     * @param \OC\Files\View $view
+     */
+    public function __construct($view) {
+        $this->view = $view;
+    }
+
+    /**
+     * This initializes the plugin.
+     *
+     * This function is called by \Sabre\DAV\Server, after
+     * addPlugin is called.
+     *
+     * This method should set up the requires event subscriptions.
+     *
+     * @param \Sabre\DAV\Server $server
+     * @return void
+     */
+    public function initialize(\Sabre\DAV\Server $server) {
+
+        $this->server = $server;
+
+        $server->on('beforeWriteContent', [$this, 'beforeWriteContent'], 10);
+        $server->on('beforeCreateFile', [$this, 'beforeCreateFile'], 10);
+        $server->on('beforeMove', [$this, 'beforeMove'], 10);
+    }
+
+    /**
+     * Check quota before creating file
+     *
+     * @param string $uri target file URI
+     * @param resource $data data
+     * @param INode $parent Sabre Node
+     * @param bool $modified modified
+     */
+    public function beforeCreateFile($uri, $data, INode $parent, $modified) {
+        if (!$parent instanceof Node) {
+            return;
+        }
+
+        return $this->checkQuota($parent->getPath() . '/' . basename($uri));
+    }
+
+    /**
+     * Check quota before writing content
+     *
+     * @param string $uri target file URI
+     * @param INode $node Sabre Node
+     * @param resource $data data
+     * @param bool $modified modified
+     */
+    public function beforeWriteContent($uri, INode $node, $data, $modified) {
+        if (!$node instanceof Node) {
+            return;
+        }
+
+        return $this->checkQuota($node->getPath());
+    }
+
+    /**
+     * Check if we're moving a Futurefile in which case we need to check
+     * the quota on the target destination.
+     *
+     * @param string $source source path
+     * @param string $destination destination path
+     */
+    public function beforeMove($source, $destination) {
+        $sourceNode = $this->server->tree->getNodeForPath($source);
+        if (!$sourceNode instanceof FutureFile) {
+            return;
+        }
+
+        // get target node for proper path conversion
+        if ($this->server->tree->nodeExists($destination)) {
+            $destinationNode = $this->server->tree->getNodeForPath($destination);
+            $path = $destinationNode->getPath();
+        } else {
+            $parentNode = $this->server->tree->getNodeForPath(dirname($destination));
+            $path = $parentNode->getPath();
+        }
+
+        return $this->checkQuota($path, $sourceNode->getSize());
+    }
+
+
+    /**
+     * This method is called before any HTTP method and validates there is enough free space to store the file
+     *
+     * @param string $path relative to the users home
+     * @param int $length
+     * @throws InsufficientStorage
+     * @return bool
+     */
+    public function checkQuota($path, $length = null) {
+        if ($length === null) {
+            $length = $this->getLength();
+        }
+
+        if ($length) {
+            list($parentPath, $newName) = \Sabre\Uri\split($path);
+            if(is_null($parentPath)) {
+                $parentPath = '';
+            }
+            $req = $this->server->httpRequest;
+            if ($req->getHeader('OC-Chunked')) {
+                $info = \OC_FileChunking::decodeName($newName);
+                $chunkHandler = $this->getFileChunking($info);
+                // subtract the already uploaded size to see whether
+                // there is still enough space for the remaining chunks
+                $length -= $chunkHandler->getCurrentSize();
+                // use target file name for free space check in case of shared files
+                $path = rtrim($parentPath, '/') . '/' . $info['name'];
+            }
+            $freeSpace = $this->getFreeSpace($path);
+            if ($freeSpace >= 0 && $length > $freeSpace) {
+                if (isset($chunkHandler)) {
+                    $chunkHandler->cleanup();
+                }
+                throw new InsufficientStorage("Insufficient space in $path, $length required, $freeSpace available");
+            }
+        }
+        return true;
+    }
+
+    public function getFileChunking($info) {
+        // FIXME: need a factory for better mocking support
+        return new \OC_FileChunking($info);
+    }
+
+    public function getLength() {
+        $req = $this->server->httpRequest;
+        $length = $req->getHeader('X-Expected-Entity-Length');
+        if (!is_numeric($length)) {
+            $length = $req->getHeader('Content-Length');
+            $length = is_numeric($length) ? $length : null;
+        }
+
+        $ocLength = $req->getHeader('OC-Total-Length');
+        if (is_numeric($length) && is_numeric($ocLength)) {
+            return max($length, $ocLength);
+        }
+
+        return $length;
+    }
+
+    /**
+     * @param string $uri
+     * @return mixed
+     * @throws ServiceUnavailable
+     */
+    public function getFreeSpace($uri) {
+        try {
+            $freeSpace = $this->view->free_space(ltrim($uri, '/'));
+            return $freeSpace;
+        } catch (StorageNotAvailableException $e) {
+            throw new ServiceUnavailable($e->getMessage());
+        }
+    }
 }

Spacing +3 added lines, -3 removed lines

@@ -92,7 +92,7 @@ 
 			return;
 		}
 
-		return $this->checkQuota($parent->getPath() . '/' . basename($uri));
+		return $this->checkQuota($parent->getPath().'/'.basename($uri));
 	}
 
 	/**
@@ -152,7 +152,7 @@ 
 
 		if ($length) {
 			list($parentPath, $newName) = \Sabre\Uri\split($path);
-			if(is_null($parentPath)) {
+			if (is_null($parentPath)) {
 				$parentPath = '';
 			}
 			$req = $this->server->httpRequest;
@@ -163,7 +163,7 @@ 
 				// there is still enough space for the remaining chunks
 				$length -= $chunkHandler->getCurrentSize();
 				// use target file name for free space check in case of shared files
-				$path = rtrim($parentPath, '/') . '/' . $info['name'];
+				$path = rtrim($parentPath, '/').'/'.$info['name'];
 			}
 			$freeSpace = $this->getFreeSpace($path);
 			if ($freeSpace >= 0 && $length > $freeSpace) {