* * @author Robin Appelman * @author Roeland Jago Douma * * @license GNU AGPL version 3 or any later version * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as * published by the Free Software Foundation, either version 3 of the * License, or (at your option) any later version. * * This program 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 program. If not, see . * */ namespace OC\Files\AppData; use OCP\Cache\CappedMemoryCache; use OC\Files\SimpleFS\SimpleFolder; use OC\SystemConfig; 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 */ 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(); } }