feat(external-storage): Implement IPartialMountPoint

Signed-off-by: Carl Schwan <[email protected]>

Author: CarlSchwan

apps/files_external/lib/Config/ConfigAdapter.php

@@ -18,6 +18,7 @@
 use OCA\Files_External\Service\UserStoragesService;
 use OCP\AppFramework\QueryException;
 use OCP\Files\Config\IMountProvider;
+use OCP\Files\Config\IPartialMountProvider;
 use OCP\Files\Mount\IMountPoint;
 use OCP\Files\ObjectStore\IObjectStore;
 use OCP\Files\Storage\IConstructableStorage;
@@ -26,13 +27,14 @@
 use OCP\Files\StorageNotAvailableException;
 use OCP\IUser;
 use OCP\Server;
+use Override;
 use Psr\Clock\ClockInterface;
 use Psr\Log\LoggerInterface;
 
 /**
  * Make the old files_external config work with the new public mount config api
  */
-class ConfigAdapter implements IMountProvider {
+class ConfigAdapter implements IMountProvider, IPartialMountProvider {
 	public function __construct(
 		private UserStoragesService $userStoragesService,
 		private UserGlobalStoragesService $userGlobalStoragesService,
@@ -168,4 +170,98 @@ public function getMountsForUser(IUser $user, IStorageFactory $loader) {
 
 		return $mounts;
 	}
+
+	#[Override]
+	public function getMountsForPath(string $path, bool $forChildren, array $mountProviderArgs, IStorageFactory $loader): array {
+		if (empty($mountProviderArgs)) {
+			return [];
+		}
+
+		$userId = null;
+		$user = null;
+		foreach ($mountProviderArgs as $mountProviderArg) {
+			if ($userId === null) {
+				$user = $mountProviderArg->mountInfo->getUser();
+				$userId = $user->getUID();
+			} elseif ($userId !== $mountProviderArg->mountInfo->getUser()->getUID()) {
+				throw new \LogicException('Mounts must belong to the same user!');
+			}
+		}
+
+		$this->userStoragesService->setUser($user);
+		$this->userGlobalStoragesService->setUser($user);
+
+		$storageConfigs = $this->userGlobalStoragesService->getAllStoragesForUserWithPath($path, $forChildren);
+
+		$storages = array_map(function (StorageConfig $storageConfig) use ($user) {
+			try {
+				$this->prepareStorageConfig($storageConfig, $user);
+				return $this->constructStorage($storageConfig);
+			} catch (\Exception $e) {
+				// propagate exception into filesystem
+				return new FailedStorage(['exception' => $e]);
+			}
+		}, $storageConfigs);
+
+
+		Storage::getGlobalCache()->loadForStorageIds(array_map(function (IStorage $storage) {
+			return $storage->getId();
+		}, $storages));
+
+		$availableStorages = array_map(function (IStorage $storage, StorageConfig $storageConfig): IStorage {
+			try {
+				$availability = $storage->getAvailability();
+				if (!$availability['available'] && !Availability::shouldRecheck($availability)) {
+					$storage = new FailedStorage([
+						'exception' => new StorageNotAvailableException('Storage with mount id ' . $storageConfig->getId() . ' is not available')
+					]);
+				}
+			} catch (\Exception $e) {
+				// propagate exception into filesystem
+				$storage = new FailedStorage(['exception' => $e]);
+			}
+			return $storage;
+		}, $storages, $storageConfigs);
+
+		$mounts = [];
+
+		$i = 0;
+		foreach ($storageConfigs as $storageConfig) {
+			$storage = $availableStorages[$i];
+			$i++;
+			$storage->setOwner($user->getUID());
+			$mountPoint = '/' . $user->getUID() . '/files' . $storageConfig->getMountPoint();
+			if ($storageConfig->getType() === StorageConfig::MOUNT_TYPE_PERSONAL) {
+				$mounts[$mountPoint] = new PersonalMount(
+					$this->userStoragesService,
+					$storageConfig,
+					$storageConfig->getId(),
+					new KnownMtime([
+						'storage' => $storage,
+						'clock' => $this->clock,
+					]),
+					$mountPoint,
+					null,
+					$loader,
+					$storageConfig->getMountOptions(),
+					$storageConfig->getId()
+				);
+			} else {
+				$mounts[$mountPoint] = new SystemMountPoint(
+					$storageConfig,
+					$storage,
+					$mountPoint,
+					null,
+					$loader,
+					$storageConfig->getMountOptions(),
+					$storageConfig->getId()
+				);
+			}
+		}
+
+		$this->userStoragesService->resetUser();
+		$this->userGlobalStoragesService->resetUser();
+
+		return $mounts;
+	}
 }