Merge pull request #19782 from nextcloud/quota-wrapper-include-external

Use global used space in quota wrappen when external storage is included

1 files changed, 18 insertions, 9 deletions

diff --git a/lib/private/Files/Storage/Wrapper/Quota.php b/lib/private/Files/Storage/Wrapper/Quota.php
index b5a1be95aaf..d4e4be41f71 100644
--- a/lib/private/Files/Storage/Wrapper/Quota.php
+++ b/lib/private/Files/Storage/Wrapper/Quota.php
@@ -29,6 +29,7 @@
 
 namespace OC\Files\Storage\Wrapper;
 
+use OC\Files\Filesystem;
 use OCP\Files\Cache\ICacheEntry;
 use OCP\Files\Storage\IStorage;
 
@@ -44,6 +45,8 @@ class Quota extends Wrapper {
 	 */
 	protected $sizeRoot;
 
+	private $config;
+
 	/**
 	 * @param array $parameters
 	 */
@@ -51,6 +54,7 @@ class Quota extends Wrapper {
 		parent::__construct($parameters);
 		$this->quota = $parameters['quota'];
 		$this->sizeRoot = isset($parameters['root']) ? $parameters['root'] : '';
+		$this->config = \OC::$server->getSystemConfig();
 	}
 
 	/**
@@ -65,16 +69,21 @@ class Quota extends Wrapper {
 	 * @param \OC\Files\Storage\Storage $storage
 	 */
 	protected function getSize($path, $storage = null) {
-		if (is_null($storage)) {
-			$cache = $this->getCache();
-		} else {
-			$cache = $storage->getCache();
-		}
-		$data = $cache->get($path);
-		if ($data instanceof ICacheEntry and isset($data['size'])) {
-			return $data['size'];
+		if ($this->config->getValue('quota_include_external_storage', false)) {
+			$rootInfo = Filesystem::getFileInfo('', 'ext');
+			return $rootInfo->getSize(true);
 		} else {
-			return \OCP\Files\FileInfo::SPACE_NOT_COMPUTED;
+			if (is_null($storage)) {
+				$cache = $this->getCache();
+			} else {
+				$cache = $storage->getCache();
+			}
+			$data = $cache->get($path);
+			if ($data instanceof ICacheEntry and isset($data['size'])) {
+				return $data['size'];
+			} else {
+				return \OCP\Files\FileInfo::SPACE_NOT_COMPUTED;
+			}
 		}
 	}