123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253 |
- <?php
- namespace OC\Files\Config;
- use OC\Hooks\Emitter;
- use OC\Hooks\EmitterTrait;
- use OCP\Diagnostics\IEventLogger;
- use OCP\Files\Config\IHomeMountProvider;
- use OCP\Files\Config\IMountProvider;
- use OCP\Files\Config\IMountProviderCollection;
- use OCP\Files\Config\IRootMountProvider;
- use OCP\Files\Config\IUserMountCache;
- use OCP\Files\Mount\IMountManager;
- use OCP\Files\Mount\IMountPoint;
- use OCP\Files\Storage\IStorageFactory;
- use OCP\IUser;
- class MountProviderCollection implements IMountProviderCollection, Emitter {
- use EmitterTrait;
-
- private $homeProviders = [];
-
- private $providers = [];
-
- private $rootProviders = [];
-
- private $loader;
-
- private $mountCache;
-
- private $mountFilters = [];
- private IEventLogger $eventLogger;
-
- public function __construct(
- IStorageFactory $loader,
- IUserMountCache $mountCache,
- IEventLogger $eventLogger
- ) {
- $this->loader = $loader;
- $this->mountCache = $mountCache;
- $this->eventLogger = $eventLogger;
- }
- private function getMountsFromProvider(IMountProvider $provider, IUser $user, IStorageFactory $loader): array {
- $class = str_replace('\\', '_', get_class($provider));
- $uid = $user->getUID();
- $this->eventLogger->start('fs:setup:provider:' . $class, "Getting mounts from $class for $uid");
- $mounts = $provider->getMountsForUser($user, $loader) ?? [];
- $this->eventLogger->end('fs:setup:provider:' . $class);
- return $mounts;
- }
-
- private function getUserMountsForProviders(IUser $user, array $providers): array {
- $loader = $this->loader;
- $mounts = array_map(function (IMountProvider $provider) use ($user, $loader) {
- return $this->getMountsFromProvider($provider, $user, $loader);
- }, $providers);
- $mounts = array_reduce($mounts, function (array $mounts, array $providerMounts) {
- return array_merge($mounts, $providerMounts);
- }, []);
- return $this->filterMounts($user, $mounts);
- }
- public function getMountsForUser(IUser $user): array {
- return $this->getUserMountsForProviders($user, $this->providers);
- }
- public function getUserMountsForProviderClasses(IUser $user, array $mountProviderClasses): array {
- $providers = array_filter(
- $this->providers,
- fn (IMountProvider $mountProvider) => (in_array(get_class($mountProvider), $mountProviderClasses))
- );
- return $this->getUserMountsForProviders($user, $providers);
- }
- public function addMountForUser(IUser $user, IMountManager $mountManager, callable $providerFilter = null) {
-
-
- $firstMounts = [];
- if ($providerFilter) {
- $providers = array_filter($this->providers, $providerFilter);
- } else {
- $providers = $this->providers;
- }
- $firstProviders = array_filter($providers, function (IMountProvider $provider) {
- return (get_class($provider) !== 'OCA\Files_Sharing\MountProvider');
- });
- $lastProviders = array_filter($providers, function (IMountProvider $provider) {
- return (get_class($provider) === 'OCA\Files_Sharing\MountProvider');
- });
- foreach ($firstProviders as $provider) {
- $mounts = $this->getMountsFromProvider($provider, $user, $this->loader);
- $firstMounts = array_merge($firstMounts, $mounts);
- }
- $firstMounts = $this->filterMounts($user, $firstMounts);
- array_walk($firstMounts, [$mountManager, 'addMount']);
- $lateMounts = [];
- foreach ($lastProviders as $provider) {
- $mounts = $this->getMountsFromProvider($provider, $user, $this->loader);
- $lateMounts = array_merge($lateMounts, $mounts);
- }
- $lateMounts = $this->filterMounts($user, $lateMounts);
- $this->eventLogger->start("fs:setup:add-mounts", "Add mounts to the filesystem");
- array_walk($lateMounts, [$mountManager, 'addMount']);
- $this->eventLogger->end("fs:setup:add-mounts");
- return array_merge($lateMounts, $firstMounts);
- }
-
- public function getHomeMountForUser(IUser $user) {
-
- $providers = array_reverse($this->homeProviders);
- foreach ($providers as $homeProvider) {
- if ($mount = $homeProvider->getHomeMountForUser($user, $this->loader)) {
- $mount->setMountPoint('/' . $user->getUID());
- return $mount;
- }
- }
- throw new \Exception('No home storage configured for user ' . $user);
- }
-
- public function registerProvider(IMountProvider $provider) {
- $this->providers[] = $provider;
- $this->emit('\OC\Files\Config', 'registerMountProvider', [$provider]);
- }
- public function registerMountFilter(callable $filter) {
- $this->mountFilters[] = $filter;
- }
- private function filterMounts(IUser $user, array $mountPoints) {
- return array_filter($mountPoints, function (IMountPoint $mountPoint) use ($user) {
- foreach ($this->mountFilters as $filter) {
- if ($filter($mountPoint, $user) === false) {
- return false;
- }
- }
- return true;
- });
- }
-
- public function registerHomeProvider(IHomeMountProvider $provider) {
- $this->homeProviders[] = $provider;
- $this->emit('\OC\Files\Config', 'registerHomeMountProvider', [$provider]);
- }
-
- public function getMountCache() {
- return $this->mountCache;
- }
- public function registerRootProvider(IRootMountProvider $provider) {
- $this->rootProviders[] = $provider;
- }
-
- public function getRootMounts(): array {
- $loader = $this->loader;
- $mounts = array_map(function (IRootMountProvider $provider) use ($loader) {
- return $provider->getRootMounts($loader);
- }, $this->rootProviders);
- $mounts = array_reduce($mounts, function (array $mounts, array $providerMounts) {
- return array_merge($mounts, $providerMounts);
- }, []);
- return $mounts;
- }
- public function clearProviders() {
- $this->providers = [];
- $this->homeProviders = [];
- $this->rootProviders = [];
- }
- public function getProviders(): array {
- return $this->providers;
- }
- }
|