1 files changed, 154 insertions, 0 deletions
diff --git a/lib/private/Files/AppData/AppData.php b/lib/private/Files/AppData/AppData.php
new file mode 100644
index 00000000000..c13372ae1d9
--- /dev/null
+++ b/lib/private/Files/AppData/AppData.php
@@ -0,0 +1,154 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * SPDX-FileCopyrightText: 2016 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+namespace OC\Files\AppData;
+
+use OC\Files\SimpleFS\SimpleFolder;
+use OC\SystemConfig;
+use OCP\Cache\CappedMemoryCache;
+use OCP\Files\Folder;
+use OCP\Files\IAppData;
+use OCP\Files\IRootFolder;
+use OCP\Files\Node;
+use OCP\Files\NotFoundException;
+use OCP\Files\NotPermittedException;
+use OCP\Files\SimpleFS\ISimpleFolder;
+
+class AppData implements IAppData {
+	private IRootFolder $rootFolder;
+	private SystemConfig $config;
+	private string $appId;
+	private ?Folder $folder = null;
+	/** @var CappedMemoryCache<ISimpleFolder|NotFoundException> */
+	private CappedMemoryCache $folders;
+
+	/**
+	 * AppData constructor.
+	 *
+	 * @param IRootFolder $rootFolder
+	 * @param SystemConfig $systemConfig
+	 * @param string $appId
+	 */
+	public function __construct(IRootFolder $rootFolder,
+		SystemConfig $systemConfig,
+		string $appId) {
+		$this->rootFolder = $rootFolder;
+		$this->config = $systemConfig;
+		$this->appId = $appId;
+		$this->folders = new CappedMemoryCache();
+	}
+
+	private function getAppDataFolderName() {
+		$instanceId = $this->config->getValue('instanceid', null);
+		if ($instanceId === null) {
+			throw new \RuntimeException('no instance id!');
+		}
+
+		return 'appdata_' . $instanceId;
+	}
+
+	protected function getAppDataRootFolder(): Folder {
+		$name = $this->getAppDataFolderName();
+
+		try {
+			/** @var Folder $node */
+			$node = $this->rootFolder->get($name);
+			return $node;
+		} catch (NotFoundException $e) {
+			try {
+				return $this->rootFolder->newFolder($name);
+			} catch (NotPermittedException $e) {
+				throw new \RuntimeException('Could not get appdata folder');
+			}
+		}
+	}
+
+	/**
+	 * @return Folder
+	 * @throws \RuntimeException
+	 */
+	private function getAppDataFolder(): Folder {
+		if ($this->folder === null) {
+			$name = $this->getAppDataFolderName();
+
+			try {
+				$this->folder = $this->rootFolder->get($name . '/' . $this->appId);
+			} catch (NotFoundException $e) {
+				$appDataRootFolder = $this->getAppDataRootFolder();
+
+				try {
+					$this->folder = $appDataRootFolder->get($this->appId);
+				} catch (NotFoundException $e) {
+					try {
+						$this->folder = $appDataRootFolder->newFolder($this->appId);
+					} catch (NotPermittedException $e) {
+						throw new \RuntimeException('Could not get appdata folder for ' . $this->appId);
+					}
+				}
+			}
+		}
+
+		return $this->folder;
+	}
+
+	public function getFolder(string $name): ISimpleFolder {
+		$key = $this->appId . '/' . $name;
+		if ($cachedFolder = $this->folders->get($key)) {
+			if ($cachedFolder instanceof \Exception) {
+				throw $cachedFolder;
+			} else {
+				return $cachedFolder;
+			}
+		}
+		try {
+			// Hardening if somebody wants to retrieve '/'
+			if ($name === '/') {
+				$node = $this->getAppDataFolder();
+			} else {
+				$path = $this->getAppDataFolderName() . '/' . $this->appId . '/' . $name;
+				$node = $this->rootFolder->get($path);
+			}
+		} catch (NotFoundException $e) {
+			$this->folders->set($key, $e);
+			throw $e;
+		}
+
+		/** @var Folder $node */
+		$folder = new SimpleFolder($node);
+		$this->folders->set($key, $folder);
+		return $folder;
+	}
+
+	public function newFolder(string $name): ISimpleFolder {
+		$key = $this->appId . '/' . $name;
+		$folder = $this->getAppDataFolder()->newFolder($name);
+
+		$simpleFolder = new SimpleFolder($folder);
+		$this->folders->set($key, $simpleFolder);
+		return $simpleFolder;
+	}
+
+	public function getDirectoryListing(): array {
+		$listing = $this->getAppDataFolder()->getDirectoryListing();
+
+		$fileListing = array_map(function (Node $folder) {
+			if ($folder instanceof Folder) {
+				return new SimpleFolder($folder);
+			}
+			return null;
+		}, $listing);
+
+		$fileListing = array_filter($fileListing);
+
+		return array_values($fileListing);
+	}
+
+	public function getId(): int {
+		return $this->getAppDataFolder()->getId();
+	}
+}