diff --git a/lib/private/Files/Mount/Manager.php b/lib/private/Files/Mount/Manager.php index b763697387dd7..72b38f38575b7 100644 --- a/lib/private/Files/Mount/Manager.php +++ b/lib/private/Files/Mount/Manager.php @@ -230,11 +230,11 @@ public function getSetupManager(): SetupManager { } /** - * Return all mounts in a path from a specific mount provider + * Return all mounts in a path from a specific mount provider, indexed by mount point * * @param string $path * @param string[] $mountProviders - * @return IMountPoint[] + * @return array */ public function getMountsByMountProvider(string $path, array $mountProviders) { $this->getSetupManager()->setupForProvider($path, $mountProviders); diff --git a/lib/private/Files/Node/Root.php b/lib/private/Files/Node/Root.php index 9b0a48fa296fd..81587fe99c1b8 100644 --- a/lib/private/Files/Node/Root.php +++ b/lib/private/Files/Node/Root.php @@ -19,6 +19,7 @@ use OCP\Cache\CappedMemoryCache; use OCP\EventDispatcher\IEventDispatcher; use OCP\Files\Cache\ICacheEntry; +use OCP\Files\Config\ICachedMountInfo; use OCP\Files\Config\IUserMountCache; use OCP\Files\Events\Node\FilesystemTornDownEvent; use OCP\Files\IRootFolder; @@ -405,17 +406,18 @@ public function getFirstNodeByIdInPath(int $id, string $path): ?INode { */ public function getByIdInPath(int $id, string $path): array { $mountCache = $this->getUserMountCache(); + $setupManager = $this->mountManager->getSetupManager(); if ($path !== '' && strpos($path, '/', 1) > 0) { [, $user] = explode('/', $path); } else { $user = null; } - $mountsContainingFile = $mountCache->getMountsForFileId($id, $user); + $mountsContainingFileInfos = $mountCache->getMountsForFileId($id, $user); // if the mount isn't in the cache yet, perform a setup first, then try again - if (count($mountsContainingFile) === 0) { - $this->mountManager->getSetupManager()->setupForPath($path, true); - $mountsContainingFile = $mountCache->getMountsForFileId($id, $user); + if (count($mountsContainingFileInfos) === 0) { + $setupManager->setupForPath($path, true); + $mountsContainingFileInfos = $mountCache->getMountsForFileId($id, $user); } // when a user has access through the same storage through multiple paths @@ -427,20 +429,21 @@ public function getByIdInPath(int $id, string $path): array { $mountRootIds = array_map(function ($mount) { return $mount->getRootId(); - }, $mountsContainingFile); + }, $mountsContainingFileInfos); $mountRootPaths = array_map(function ($mount) { return $mount->getRootInternalPath(); - }, $mountsContainingFile); + }, $mountsContainingFileInfos); $mountProviders = array_unique(array_map(function ($mount) { return $mount->getMountProvider(); - }, $mountsContainingFile)); + }, $mountsContainingFileInfos)); + $mountPoints = array_map(fn(ICachedMountInfo $mountInfo) => $mountInfo->getMountPoint(), $mountsContainingFileInfos); $mountRoots = array_combine($mountRootIds, $mountRootPaths); $mounts = $this->mountManager->getMountsByMountProvider($path, $mountProviders); - - $mountsContainingFile = array_filter($mounts, function ($mount) use ($mountRoots) { - return isset($mountRoots[$mount->getStorageRootId()]); - }); + $mountsContainingFile = array_filter($mounts, fn (IMountPoint $mount) => in_array($mount->getMountPoint(), $mountPoints)); + if (count($mountsContainingFile) == 0) { + $mountsContainingFile = array_filter(array_map($this->mountManager->getMountFromMountInfo(...), $mountsContainingFileInfos)); + } if (count($mountsContainingFile) === 0) { if ($user === $this->getAppDataDirectoryName()) { diff --git a/lib/private/Files/SetupManager.php b/lib/private/Files/SetupManager.php index df0fd5a2d52d9..d23314ace2199 100644 --- a/lib/private/Files/SetupManager.php +++ b/lib/private/Files/SetupManager.php @@ -434,8 +434,8 @@ public function setupRoot(): void { * @param string $path * @return IUser|null */ - private function getUserForPath(string $path): ?IUser { - if ($path === '' || $path === '/') { + private function getUserForPath(string $path, bool $includeChildren = false): ?IUser { + if (($path === '' || $path === '/') && !$includeChildren) { return null; } elseif (str_starts_with($path, '/__groupfolders')) { return null; @@ -456,7 +456,7 @@ private function getUserForPath(string $path): ?IUser { #[Override] public function setupForPath(string $path, bool $includeChildren = false): void { - $user = $this->getUserForPath($path); + $user = $this->getUserForPath($path, $includeChildren); if (!$user) { $this->setupRoot(); return; diff --git a/tests/lib/Files/Node/FolderTest.php b/tests/lib/Files/Node/FolderTest.php index ad60cb535558c..19374fee6c128 100644 --- a/tests/lib/Files/Node/FolderTest.php +++ b/tests/lib/Files/Node/FolderTest.php @@ -29,6 +29,7 @@ use OC\Files\View; use OCP\Constants; use OCP\Files\Cache\ICacheEntry; +use OCP\Files\Config\ICachedMountInfo; use OCP\Files\InvalidPathException; use OCP\Files\IRootFolder; use OCP\Files\Mount\IMountPoint; @@ -538,8 +539,8 @@ public function testGetById(): void { ->with('/bar/foo') ->willReturn([]); - $manager->method('getMountsByMountProvider') - ->willReturn([$mount]); + $manager->method('getMountFromMountInfo') + ->willReturn($mount); $node = new Folder($root, $view, '/bar/foo'); $result = $node->getById(1); @@ -583,8 +584,8 @@ public function testGetByIdMountRoot(): void { ->with(1) ->willReturn($fileInfo); - $manager->method('getMountsByMountProvider') - ->willReturn([$mount]); + $manager->method('getMountFromMountInfo') + ->willReturn($mount); $node = new Folder($root, $view, '/bar'); $result = $node->getById(1); @@ -628,8 +629,8 @@ public function testGetByIdOutsideFolder(): void { ->with(1) ->willReturn($fileInfo); - $manager->method('getMountsByMountProvider') - ->willReturn([$mount]); + $manager->method('getMountFromMountInfo') + ->willReturn($mount); $node = new Folder($root, $view, '/bar/foo'); $result = $node->getById(1); @@ -668,14 +669,31 @@ public function testGetByIdMultipleStorages(): void { 1, '' ), + new CachedMountInfo( + $this->user, + 1, + 0, + '/bar/foo/asd/', + 'test', + 1, + '' + ), ]); $cache->method('get') ->with(1) ->willReturn($fileInfo); - $manager->method('getMountsByMountProvider') - ->willReturn([$mount1, $mount2]); + $manager->method('getMountFromMountInfo') + ->willReturnCallback(function (ICachedMountInfo $mountInfo) use ($mount1, $mount2) { + if ($mountInfo->getMountPoint() === $mount1->getMountPoint()) { + return $mount1; + } + if ($mountInfo->getMountPoint() === $mount2->getMountPoint()) { + return $mount2; + } + return null; + }); $node = new Folder($root, $view, '/bar/foo'); $result = $node->getById(1);