@@ -17,209 +17,209 @@ |
||
17 | 17 | * @psalm-type ObjectStoreConfig array{class: class-string<IObjectStore>, arguments: array{multibucket: bool, ...}} |
18 | 18 | */ |
19 | 19 | class PrimaryObjectStoreConfig { |
20 | - public function __construct( |
|
21 | - private readonly IConfig $config, |
|
22 | - private readonly IAppManager $appManager, |
|
23 | - ) { |
|
24 | - } |
|
25 | - |
|
26 | - /** |
|
27 | - * @param ObjectStoreConfig $config |
|
28 | - */ |
|
29 | - public function buildObjectStore(array $config): IObjectStore { |
|
30 | - return new $config['class']($config['arguments']); |
|
31 | - } |
|
32 | - |
|
33 | - /** |
|
34 | - * @return ?ObjectStoreConfig |
|
35 | - */ |
|
36 | - public function getObjectStoreConfigForRoot(): ?array { |
|
37 | - if (!$this->hasObjectStore()) { |
|
38 | - return null; |
|
39 | - } |
|
40 | - |
|
41 | - $config = $this->getObjectStoreConfiguration('root'); |
|
42 | - |
|
43 | - if ($config['arguments']['multibucket']) { |
|
44 | - if (!isset($config['arguments']['bucket'])) { |
|
45 | - $config['arguments']['bucket'] = ''; |
|
46 | - } |
|
47 | - |
|
48 | - // put the root FS always in first bucket for multibucket configuration |
|
49 | - $config['arguments']['bucket'] .= '0'; |
|
50 | - } |
|
51 | - return $config; |
|
52 | - } |
|
53 | - |
|
54 | - /** |
|
55 | - * @return ?ObjectStoreConfig |
|
56 | - */ |
|
57 | - public function getObjectStoreConfigForUser(IUser $user): ?array { |
|
58 | - if (!$this->hasObjectStore()) { |
|
59 | - return null; |
|
60 | - } |
|
61 | - |
|
62 | - $store = $this->getObjectStoreForUser($user); |
|
63 | - $config = $this->getObjectStoreConfiguration($store); |
|
64 | - |
|
65 | - if ($config['arguments']['multibucket']) { |
|
66 | - $config['arguments']['bucket'] = $this->getBucketForUser($user, $config); |
|
67 | - } |
|
68 | - return $config; |
|
69 | - } |
|
70 | - |
|
71 | - /** |
|
72 | - * @param string $name |
|
73 | - * @return ObjectStoreConfig |
|
74 | - */ |
|
75 | - public function getObjectStoreConfiguration(string $name): array { |
|
76 | - $configs = $this->getObjectStoreConfigs(); |
|
77 | - $name = $this->resolveAlias($name); |
|
78 | - if (!isset($configs[$name])) { |
|
79 | - throw new \Exception("Object store configuration for '$name' not found"); |
|
80 | - } |
|
81 | - if (is_string($configs[$name])) { |
|
82 | - throw new \Exception("Object store configuration for '{$configs[$name]}' not found"); |
|
83 | - } |
|
84 | - return $configs[$name]; |
|
85 | - } |
|
86 | - |
|
87 | - public function resolveAlias(string $name): string { |
|
88 | - $configs = $this->getObjectStoreConfigs(); |
|
89 | - |
|
90 | - while (isset($configs[$name]) && is_string($configs[$name])) { |
|
91 | - $name = $configs[$name]; |
|
92 | - } |
|
93 | - return $name; |
|
94 | - } |
|
95 | - |
|
96 | - public function hasObjectStore(): bool { |
|
97 | - $objectStore = $this->config->getSystemValue('objectstore', null); |
|
98 | - $objectStoreMultiBucket = $this->config->getSystemValue('objectstore_multibucket', null); |
|
99 | - return $objectStore || $objectStoreMultiBucket; |
|
100 | - } |
|
101 | - |
|
102 | - public function hasMultipleObjectStorages(): bool { |
|
103 | - $objectStore = $this->config->getSystemValue('objectstore', []); |
|
104 | - return isset($objectStore['default']); |
|
105 | - } |
|
106 | - |
|
107 | - /** |
|
108 | - * @return ?array<string, ObjectStoreConfig|string> |
|
109 | - * @throws InvalidObjectStoreConfigurationException |
|
110 | - */ |
|
111 | - public function getObjectStoreConfigs(): ?array { |
|
112 | - $objectStore = $this->config->getSystemValue('objectstore', null); |
|
113 | - $objectStoreMultiBucket = $this->config->getSystemValue('objectstore_multibucket', null); |
|
114 | - |
|
115 | - // new-style multibucket config uses the same 'objectstore' key but sets `'multibucket' => true`, transparently upgrade older style config |
|
116 | - if ($objectStoreMultiBucket) { |
|
117 | - $objectStoreMultiBucket['arguments']['multibucket'] = true; |
|
118 | - return [ |
|
119 | - 'default' => 'server1', |
|
120 | - 'server1' => $this->validateObjectStoreConfig($objectStoreMultiBucket), |
|
121 | - 'root' => 'server1', |
|
122 | - ]; |
|
123 | - } elseif ($objectStore) { |
|
124 | - if (!isset($objectStore['default'])) { |
|
125 | - $objectStore = [ |
|
126 | - 'default' => 'server1', |
|
127 | - 'root' => 'server1', |
|
128 | - 'server1' => $objectStore, |
|
129 | - ]; |
|
130 | - } |
|
131 | - if (!isset($objectStore['root'])) { |
|
132 | - $objectStore['root'] = 'default'; |
|
133 | - } |
|
134 | - |
|
135 | - if (!is_string($objectStore['default'])) { |
|
136 | - throw new InvalidObjectStoreConfigurationException('The \'default\' object storage configuration is required to be a reference to another configuration.'); |
|
137 | - } |
|
138 | - return array_map($this->validateObjectStoreConfig(...), $objectStore); |
|
139 | - } else { |
|
140 | - return null; |
|
141 | - } |
|
142 | - } |
|
143 | - |
|
144 | - /** |
|
145 | - * @param array|string $config |
|
146 | - * @return string|ObjectStoreConfig |
|
147 | - */ |
|
148 | - private function validateObjectStoreConfig(array|string $config): array|string { |
|
149 | - if (is_string($config)) { |
|
150 | - return $config; |
|
151 | - } |
|
152 | - if (!isset($config['class'])) { |
|
153 | - throw new InvalidObjectStoreConfigurationException('No class configured for object store'); |
|
154 | - } |
|
155 | - if (!isset($config['arguments'])) { |
|
156 | - $config['arguments'] = []; |
|
157 | - } |
|
158 | - $class = $config['class']; |
|
159 | - $arguments = $config['arguments']; |
|
160 | - if (!is_array($arguments)) { |
|
161 | - throw new InvalidObjectStoreConfigurationException('Configured object store arguments are not an array'); |
|
162 | - } |
|
163 | - if (!isset($arguments['multibucket'])) { |
|
164 | - $arguments['multibucket'] = false; |
|
165 | - } |
|
166 | - if (!is_bool($arguments['multibucket'])) { |
|
167 | - throw new InvalidObjectStoreConfigurationException('arguments.multibucket must be a boolean in object store configuration'); |
|
168 | - } |
|
169 | - |
|
170 | - if (!is_string($class)) { |
|
171 | - throw new InvalidObjectStoreConfigurationException('Configured class for object store is not a string'); |
|
172 | - } |
|
173 | - |
|
174 | - if (str_starts_with($class, 'OCA\\') && substr_count($class, '\\') >= 2) { |
|
175 | - [$appId] = explode('\\', $class); |
|
176 | - $this->appManager->loadApp(strtolower($appId)); |
|
177 | - } |
|
178 | - |
|
179 | - if (!is_a($class, IObjectStore::class, true)) { |
|
180 | - throw new InvalidObjectStoreConfigurationException('Configured class for object store is not an object store'); |
|
181 | - } |
|
182 | - return [ |
|
183 | - 'class' => $class, |
|
184 | - 'arguments' => $arguments, |
|
185 | - ]; |
|
186 | - } |
|
187 | - |
|
188 | - public function getBucketForUser(IUser $user, array $config): string { |
|
189 | - $bucket = $this->getSetBucketForUser($user); |
|
190 | - |
|
191 | - if ($bucket === null) { |
|
192 | - /* |
|
20 | + public function __construct( |
|
21 | + private readonly IConfig $config, |
|
22 | + private readonly IAppManager $appManager, |
|
23 | + ) { |
|
24 | + } |
|
25 | + |
|
26 | + /** |
|
27 | + * @param ObjectStoreConfig $config |
|
28 | + */ |
|
29 | + public function buildObjectStore(array $config): IObjectStore { |
|
30 | + return new $config['class']($config['arguments']); |
|
31 | + } |
|
32 | + |
|
33 | + /** |
|
34 | + * @return ?ObjectStoreConfig |
|
35 | + */ |
|
36 | + public function getObjectStoreConfigForRoot(): ?array { |
|
37 | + if (!$this->hasObjectStore()) { |
|
38 | + return null; |
|
39 | + } |
|
40 | + |
|
41 | + $config = $this->getObjectStoreConfiguration('root'); |
|
42 | + |
|
43 | + if ($config['arguments']['multibucket']) { |
|
44 | + if (!isset($config['arguments']['bucket'])) { |
|
45 | + $config['arguments']['bucket'] = ''; |
|
46 | + } |
|
47 | + |
|
48 | + // put the root FS always in first bucket for multibucket configuration |
|
49 | + $config['arguments']['bucket'] .= '0'; |
|
50 | + } |
|
51 | + return $config; |
|
52 | + } |
|
53 | + |
|
54 | + /** |
|
55 | + * @return ?ObjectStoreConfig |
|
56 | + */ |
|
57 | + public function getObjectStoreConfigForUser(IUser $user): ?array { |
|
58 | + if (!$this->hasObjectStore()) { |
|
59 | + return null; |
|
60 | + } |
|
61 | + |
|
62 | + $store = $this->getObjectStoreForUser($user); |
|
63 | + $config = $this->getObjectStoreConfiguration($store); |
|
64 | + |
|
65 | + if ($config['arguments']['multibucket']) { |
|
66 | + $config['arguments']['bucket'] = $this->getBucketForUser($user, $config); |
|
67 | + } |
|
68 | + return $config; |
|
69 | + } |
|
70 | + |
|
71 | + /** |
|
72 | + * @param string $name |
|
73 | + * @return ObjectStoreConfig |
|
74 | + */ |
|
75 | + public function getObjectStoreConfiguration(string $name): array { |
|
76 | + $configs = $this->getObjectStoreConfigs(); |
|
77 | + $name = $this->resolveAlias($name); |
|
78 | + if (!isset($configs[$name])) { |
|
79 | + throw new \Exception("Object store configuration for '$name' not found"); |
|
80 | + } |
|
81 | + if (is_string($configs[$name])) { |
|
82 | + throw new \Exception("Object store configuration for '{$configs[$name]}' not found"); |
|
83 | + } |
|
84 | + return $configs[$name]; |
|
85 | + } |
|
86 | + |
|
87 | + public function resolveAlias(string $name): string { |
|
88 | + $configs = $this->getObjectStoreConfigs(); |
|
89 | + |
|
90 | + while (isset($configs[$name]) && is_string($configs[$name])) { |
|
91 | + $name = $configs[$name]; |
|
92 | + } |
|
93 | + return $name; |
|
94 | + } |
|
95 | + |
|
96 | + public function hasObjectStore(): bool { |
|
97 | + $objectStore = $this->config->getSystemValue('objectstore', null); |
|
98 | + $objectStoreMultiBucket = $this->config->getSystemValue('objectstore_multibucket', null); |
|
99 | + return $objectStore || $objectStoreMultiBucket; |
|
100 | + } |
|
101 | + |
|
102 | + public function hasMultipleObjectStorages(): bool { |
|
103 | + $objectStore = $this->config->getSystemValue('objectstore', []); |
|
104 | + return isset($objectStore['default']); |
|
105 | + } |
|
106 | + |
|
107 | + /** |
|
108 | + * @return ?array<string, ObjectStoreConfig|string> |
|
109 | + * @throws InvalidObjectStoreConfigurationException |
|
110 | + */ |
|
111 | + public function getObjectStoreConfigs(): ?array { |
|
112 | + $objectStore = $this->config->getSystemValue('objectstore', null); |
|
113 | + $objectStoreMultiBucket = $this->config->getSystemValue('objectstore_multibucket', null); |
|
114 | + |
|
115 | + // new-style multibucket config uses the same 'objectstore' key but sets `'multibucket' => true`, transparently upgrade older style config |
|
116 | + if ($objectStoreMultiBucket) { |
|
117 | + $objectStoreMultiBucket['arguments']['multibucket'] = true; |
|
118 | + return [ |
|
119 | + 'default' => 'server1', |
|
120 | + 'server1' => $this->validateObjectStoreConfig($objectStoreMultiBucket), |
|
121 | + 'root' => 'server1', |
|
122 | + ]; |
|
123 | + } elseif ($objectStore) { |
|
124 | + if (!isset($objectStore['default'])) { |
|
125 | + $objectStore = [ |
|
126 | + 'default' => 'server1', |
|
127 | + 'root' => 'server1', |
|
128 | + 'server1' => $objectStore, |
|
129 | + ]; |
|
130 | + } |
|
131 | + if (!isset($objectStore['root'])) { |
|
132 | + $objectStore['root'] = 'default'; |
|
133 | + } |
|
134 | + |
|
135 | + if (!is_string($objectStore['default'])) { |
|
136 | + throw new InvalidObjectStoreConfigurationException('The \'default\' object storage configuration is required to be a reference to another configuration.'); |
|
137 | + } |
|
138 | + return array_map($this->validateObjectStoreConfig(...), $objectStore); |
|
139 | + } else { |
|
140 | + return null; |
|
141 | + } |
|
142 | + } |
|
143 | + |
|
144 | + /** |
|
145 | + * @param array|string $config |
|
146 | + * @return string|ObjectStoreConfig |
|
147 | + */ |
|
148 | + private function validateObjectStoreConfig(array|string $config): array|string { |
|
149 | + if (is_string($config)) { |
|
150 | + return $config; |
|
151 | + } |
|
152 | + if (!isset($config['class'])) { |
|
153 | + throw new InvalidObjectStoreConfigurationException('No class configured for object store'); |
|
154 | + } |
|
155 | + if (!isset($config['arguments'])) { |
|
156 | + $config['arguments'] = []; |
|
157 | + } |
|
158 | + $class = $config['class']; |
|
159 | + $arguments = $config['arguments']; |
|
160 | + if (!is_array($arguments)) { |
|
161 | + throw new InvalidObjectStoreConfigurationException('Configured object store arguments are not an array'); |
|
162 | + } |
|
163 | + if (!isset($arguments['multibucket'])) { |
|
164 | + $arguments['multibucket'] = false; |
|
165 | + } |
|
166 | + if (!is_bool($arguments['multibucket'])) { |
|
167 | + throw new InvalidObjectStoreConfigurationException('arguments.multibucket must be a boolean in object store configuration'); |
|
168 | + } |
|
169 | + |
|
170 | + if (!is_string($class)) { |
|
171 | + throw new InvalidObjectStoreConfigurationException('Configured class for object store is not a string'); |
|
172 | + } |
|
173 | + |
|
174 | + if (str_starts_with($class, 'OCA\\') && substr_count($class, '\\') >= 2) { |
|
175 | + [$appId] = explode('\\', $class); |
|
176 | + $this->appManager->loadApp(strtolower($appId)); |
|
177 | + } |
|
178 | + |
|
179 | + if (!is_a($class, IObjectStore::class, true)) { |
|
180 | + throw new InvalidObjectStoreConfigurationException('Configured class for object store is not an object store'); |
|
181 | + } |
|
182 | + return [ |
|
183 | + 'class' => $class, |
|
184 | + 'arguments' => $arguments, |
|
185 | + ]; |
|
186 | + } |
|
187 | + |
|
188 | + public function getBucketForUser(IUser $user, array $config): string { |
|
189 | + $bucket = $this->getSetBucketForUser($user); |
|
190 | + |
|
191 | + if ($bucket === null) { |
|
192 | + /* |
|
193 | 193 | * Use any provided bucket argument as prefix |
194 | 194 | * and add the mapping from username => bucket |
195 | 195 | */ |
196 | - if (!isset($config['arguments']['bucket'])) { |
|
197 | - $config['arguments']['bucket'] = ''; |
|
198 | - } |
|
199 | - $mapper = new Mapper($user, $this->config); |
|
200 | - $numBuckets = $config['arguments']['num_buckets'] ?? 64; |
|
201 | - $bucket = $config['arguments']['bucket'] . $mapper->getBucket($numBuckets); |
|
202 | - |
|
203 | - $this->config->setUserValue($user->getUID(), 'homeobjectstore', 'bucket', $bucket); |
|
204 | - } |
|
205 | - |
|
206 | - return $bucket; |
|
207 | - } |
|
208 | - |
|
209 | - public function getSetBucketForUser(IUser $user): ?string { |
|
210 | - return $this->config->getUserValue($user->getUID(), 'homeobjectstore', 'bucket', null); |
|
211 | - } |
|
212 | - |
|
213 | - public function getObjectStoreForUser(IUser $user): string { |
|
214 | - if ($this->hasMultipleObjectStorages()) { |
|
215 | - $value = $this->config->getUserValue($user->getUID(), 'homeobjectstore', 'objectstore', null); |
|
216 | - if ($value === null) { |
|
217 | - $value = $this->resolveAlias('default'); |
|
218 | - $this->config->setUserValue($user->getUID(), 'homeobjectstore', 'objectstore', $value); |
|
219 | - } |
|
220 | - return $value; |
|
221 | - } else { |
|
222 | - return 'default'; |
|
223 | - } |
|
224 | - } |
|
196 | + if (!isset($config['arguments']['bucket'])) { |
|
197 | + $config['arguments']['bucket'] = ''; |
|
198 | + } |
|
199 | + $mapper = new Mapper($user, $this->config); |
|
200 | + $numBuckets = $config['arguments']['num_buckets'] ?? 64; |
|
201 | + $bucket = $config['arguments']['bucket'] . $mapper->getBucket($numBuckets); |
|
202 | + |
|
203 | + $this->config->setUserValue($user->getUID(), 'homeobjectstore', 'bucket', $bucket); |
|
204 | + } |
|
205 | + |
|
206 | + return $bucket; |
|
207 | + } |
|
208 | + |
|
209 | + public function getSetBucketForUser(IUser $user): ?string { |
|
210 | + return $this->config->getUserValue($user->getUID(), 'homeobjectstore', 'bucket', null); |
|
211 | + } |
|
212 | + |
|
213 | + public function getObjectStoreForUser(IUser $user): string { |
|
214 | + if ($this->hasMultipleObjectStorages()) { |
|
215 | + $value = $this->config->getUserValue($user->getUID(), 'homeobjectstore', 'objectstore', null); |
|
216 | + if ($value === null) { |
|
217 | + $value = $this->resolveAlias('default'); |
|
218 | + $this->config->setUserValue($user->getUID(), 'homeobjectstore', 'objectstore', $value); |
|
219 | + } |
|
220 | + return $value; |
|
221 | + } else { |
|
222 | + return 'default'; |
|
223 | + } |
|
224 | + } |
|
225 | 225 | } |
@@ -18,81 +18,81 @@ |
||
18 | 18 | use Symfony\Component\Console\Output\OutputInterface; |
19 | 19 | |
20 | 20 | class Users extends Base { |
21 | - public function __construct( |
|
22 | - private readonly IUserManager $userManager, |
|
23 | - private readonly PrimaryObjectStoreConfig $objectStoreConfig, |
|
24 | - private readonly IConfig $config, |
|
25 | - ) { |
|
26 | - parent::__construct(); |
|
27 | - } |
|
21 | + public function __construct( |
|
22 | + private readonly IUserManager $userManager, |
|
23 | + private readonly PrimaryObjectStoreConfig $objectStoreConfig, |
|
24 | + private readonly IConfig $config, |
|
25 | + ) { |
|
26 | + parent::__construct(); |
|
27 | + } |
|
28 | 28 | |
29 | - protected function configure(): void { |
|
30 | - parent::configure(); |
|
31 | - $this |
|
32 | - ->setName('files:object:multi:users') |
|
33 | - ->setDescription('Get the mapping between users and object store buckets') |
|
34 | - ->addOption('bucket', 'b', InputOption::VALUE_REQUIRED, 'Only list users using the specified bucket') |
|
35 | - ->addOption('object-store', 'o', InputOption::VALUE_REQUIRED, 'Only list users using the specified object store configuration') |
|
36 | - ->addOption('user', 'u', InputOption::VALUE_REQUIRED, 'Only show the mapping for the specified user, ignores all other options'); |
|
37 | - } |
|
29 | + protected function configure(): void { |
|
30 | + parent::configure(); |
|
31 | + $this |
|
32 | + ->setName('files:object:multi:users') |
|
33 | + ->setDescription('Get the mapping between users and object store buckets') |
|
34 | + ->addOption('bucket', 'b', InputOption::VALUE_REQUIRED, 'Only list users using the specified bucket') |
|
35 | + ->addOption('object-store', 'o', InputOption::VALUE_REQUIRED, 'Only list users using the specified object store configuration') |
|
36 | + ->addOption('user', 'u', InputOption::VALUE_REQUIRED, 'Only show the mapping for the specified user, ignores all other options'); |
|
37 | + } |
|
38 | 38 | |
39 | - public function execute(InputInterface $input, OutputInterface $output): int { |
|
40 | - if ($userId = $input->getOption('user')) { |
|
41 | - $user = $this->userManager->get($userId); |
|
42 | - if (!$user) { |
|
43 | - $output->writeln("<error>User $userId not found</error>"); |
|
44 | - return 1; |
|
45 | - } |
|
46 | - $users = new \ArrayIterator([$user]); |
|
47 | - } else { |
|
48 | - $bucket = (string)$input->getOption('bucket'); |
|
49 | - $objectStore = (string)$input->getOption('object-store'); |
|
50 | - if ($bucket !== '' && $objectStore === '') { |
|
51 | - $users = $this->getUsers($this->config->getUsersForUserValue('homeobjectstore', 'bucket', $bucket)); |
|
52 | - } elseif ($bucket === '' && $objectStore !== '') { |
|
53 | - $users = $this->getUsers($this->config->getUsersForUserValue('homeobjectstore', 'objectstore', $objectStore)); |
|
54 | - } elseif ($bucket) { |
|
55 | - $users = $this->getUsers(array_intersect( |
|
56 | - $this->config->getUsersForUserValue('homeobjectstore', 'bucket', $bucket), |
|
57 | - $this->config->getUsersForUserValue('homeobjectstore', 'objectstore', $objectStore) |
|
58 | - )); |
|
59 | - } else { |
|
60 | - $users = $this->userManager->getSeenUsers(); |
|
61 | - } |
|
62 | - } |
|
39 | + public function execute(InputInterface $input, OutputInterface $output): int { |
|
40 | + if ($userId = $input->getOption('user')) { |
|
41 | + $user = $this->userManager->get($userId); |
|
42 | + if (!$user) { |
|
43 | + $output->writeln("<error>User $userId not found</error>"); |
|
44 | + return 1; |
|
45 | + } |
|
46 | + $users = new \ArrayIterator([$user]); |
|
47 | + } else { |
|
48 | + $bucket = (string)$input->getOption('bucket'); |
|
49 | + $objectStore = (string)$input->getOption('object-store'); |
|
50 | + if ($bucket !== '' && $objectStore === '') { |
|
51 | + $users = $this->getUsers($this->config->getUsersForUserValue('homeobjectstore', 'bucket', $bucket)); |
|
52 | + } elseif ($bucket === '' && $objectStore !== '') { |
|
53 | + $users = $this->getUsers($this->config->getUsersForUserValue('homeobjectstore', 'objectstore', $objectStore)); |
|
54 | + } elseif ($bucket) { |
|
55 | + $users = $this->getUsers(array_intersect( |
|
56 | + $this->config->getUsersForUserValue('homeobjectstore', 'bucket', $bucket), |
|
57 | + $this->config->getUsersForUserValue('homeobjectstore', 'objectstore', $objectStore) |
|
58 | + )); |
|
59 | + } else { |
|
60 | + $users = $this->userManager->getSeenUsers(); |
|
61 | + } |
|
62 | + } |
|
63 | 63 | |
64 | - $this->writeStreamingTableInOutputFormat($input, $output, $this->infoForUsers($users), 100); |
|
65 | - return 0; |
|
66 | - } |
|
64 | + $this->writeStreamingTableInOutputFormat($input, $output, $this->infoForUsers($users), 100); |
|
65 | + return 0; |
|
66 | + } |
|
67 | 67 | |
68 | - /** |
|
69 | - * @param string[] $userIds |
|
70 | - * @return \Iterator<IUser> |
|
71 | - */ |
|
72 | - private function getUsers(array $userIds): \Iterator { |
|
73 | - foreach ($userIds as $userId) { |
|
74 | - $user = $this->userManager->get($userId); |
|
75 | - if ($user) { |
|
76 | - yield $user; |
|
77 | - } |
|
78 | - } |
|
79 | - } |
|
68 | + /** |
|
69 | + * @param string[] $userIds |
|
70 | + * @return \Iterator<IUser> |
|
71 | + */ |
|
72 | + private function getUsers(array $userIds): \Iterator { |
|
73 | + foreach ($userIds as $userId) { |
|
74 | + $user = $this->userManager->get($userId); |
|
75 | + if ($user) { |
|
76 | + yield $user; |
|
77 | + } |
|
78 | + } |
|
79 | + } |
|
80 | 80 | |
81 | - /** |
|
82 | - * @param \Iterator<IUser> $users |
|
83 | - * @return \Iterator<array> |
|
84 | - */ |
|
85 | - private function infoForUsers(\Iterator $users): \Iterator { |
|
86 | - foreach ($users as $user) { |
|
87 | - yield $this->infoForUser($user); |
|
88 | - } |
|
89 | - } |
|
81 | + /** |
|
82 | + * @param \Iterator<IUser> $users |
|
83 | + * @return \Iterator<array> |
|
84 | + */ |
|
85 | + private function infoForUsers(\Iterator $users): \Iterator { |
|
86 | + foreach ($users as $user) { |
|
87 | + yield $this->infoForUser($user); |
|
88 | + } |
|
89 | + } |
|
90 | 90 | |
91 | - private function infoForUser(IUser $user): array { |
|
92 | - return [ |
|
93 | - 'user' => $user->getUID(), |
|
94 | - 'object-store' => $this->objectStoreConfig->getObjectStoreForUser($user), |
|
95 | - 'bucket' => $this->objectStoreConfig->getSetBucketForUser($user) ?? 'unset', |
|
96 | - ]; |
|
97 | - } |
|
91 | + private function infoForUser(IUser $user): array { |
|
92 | + return [ |
|
93 | + 'user' => $user->getUID(), |
|
94 | + 'object-store' => $this->objectStoreConfig->getObjectStoreForUser($user), |
|
95 | + 'bucket' => $this->objectStoreConfig->getSetBucketForUser($user) ?? 'unset', |
|
96 | + ]; |
|
97 | + } |
|
98 | 98 | } |
@@ -19,90 +19,90 @@ |
||
19 | 19 | use Symfony\Component\Console\Question\ConfirmationQuestion; |
20 | 20 | |
21 | 21 | class Rename extends Base { |
22 | - public function __construct( |
|
23 | - private readonly IDBConnection $connection, |
|
24 | - private readonly PrimaryObjectStoreConfig $objectStoreConfig, |
|
25 | - private readonly IConfig $config, |
|
26 | - ) { |
|
27 | - parent::__construct(); |
|
28 | - } |
|
22 | + public function __construct( |
|
23 | + private readonly IDBConnection $connection, |
|
24 | + private readonly PrimaryObjectStoreConfig $objectStoreConfig, |
|
25 | + private readonly IConfig $config, |
|
26 | + ) { |
|
27 | + parent::__construct(); |
|
28 | + } |
|
29 | 29 | |
30 | - protected function configure(): void { |
|
31 | - parent::configure(); |
|
32 | - $this |
|
33 | - ->setName('files:object:multi:rename-config') |
|
34 | - ->setDescription('Rename an object store configuration and move all users over to the new configuration,') |
|
35 | - ->addArgument('source', InputArgument::REQUIRED, 'Object store configuration to rename') |
|
36 | - ->addArgument('target', InputArgument::REQUIRED, 'New name for the object store configuration'); |
|
37 | - } |
|
30 | + protected function configure(): void { |
|
31 | + parent::configure(); |
|
32 | + $this |
|
33 | + ->setName('files:object:multi:rename-config') |
|
34 | + ->setDescription('Rename an object store configuration and move all users over to the new configuration,') |
|
35 | + ->addArgument('source', InputArgument::REQUIRED, 'Object store configuration to rename') |
|
36 | + ->addArgument('target', InputArgument::REQUIRED, 'New name for the object store configuration'); |
|
37 | + } |
|
38 | 38 | |
39 | - public function execute(InputInterface $input, OutputInterface $output): int { |
|
40 | - $source = $input->getArgument('source'); |
|
41 | - $target = $input->getArgument('target'); |
|
39 | + public function execute(InputInterface $input, OutputInterface $output): int { |
|
40 | + $source = $input->getArgument('source'); |
|
41 | + $target = $input->getArgument('target'); |
|
42 | 42 | |
43 | - $configs = $this->objectStoreConfig->getObjectStoreConfigs(); |
|
44 | - if (!isset($configs[$source])) { |
|
45 | - $output->writeln('<error>Unknown object store configuration: ' . $source . '</error>'); |
|
46 | - return 1; |
|
47 | - } |
|
43 | + $configs = $this->objectStoreConfig->getObjectStoreConfigs(); |
|
44 | + if (!isset($configs[$source])) { |
|
45 | + $output->writeln('<error>Unknown object store configuration: ' . $source . '</error>'); |
|
46 | + return 1; |
|
47 | + } |
|
48 | 48 | |
49 | - if ($source === 'root') { |
|
50 | - $output->writeln('<error>Renaming the root configuration is not supported.</error>'); |
|
51 | - return 1; |
|
52 | - } |
|
49 | + if ($source === 'root') { |
|
50 | + $output->writeln('<error>Renaming the root configuration is not supported.</error>'); |
|
51 | + return 1; |
|
52 | + } |
|
53 | 53 | |
54 | - if ($source === 'default') { |
|
55 | - $output->writeln('<error>Renaming the default configuration is not supported.</error>'); |
|
56 | - return 1; |
|
57 | - } |
|
54 | + if ($source === 'default') { |
|
55 | + $output->writeln('<error>Renaming the default configuration is not supported.</error>'); |
|
56 | + return 1; |
|
57 | + } |
|
58 | 58 | |
59 | - if (!isset($configs[$target])) { |
|
60 | - $output->writeln('<comment>Target object store configuration ' . $target . ' doesn\'t exist yet.</comment>'); |
|
61 | - $output->writeln('The target configuration can be created automatically.'); |
|
62 | - $output->writeln('However, as this depends on modifying the config.php, this only works as long as the instance runs on a single node or all nodes in a clustered setup have a shared config file (such as from a shared network mount).'); |
|
63 | - $output->writeln('If the different nodes have a separate copy of the config.php file, the automatic object store configuration creation will lead to the configuration going out of sync.'); |
|
64 | - $output->writeln('If these requirements are not met, you can manually create the target object store configuration in each node\'s configuration before running the command.'); |
|
65 | - $output->writeln(''); |
|
66 | - $output->writeln('<error>Failure to check these requirements will lead to data loss for users.</error>'); |
|
59 | + if (!isset($configs[$target])) { |
|
60 | + $output->writeln('<comment>Target object store configuration ' . $target . ' doesn\'t exist yet.</comment>'); |
|
61 | + $output->writeln('The target configuration can be created automatically.'); |
|
62 | + $output->writeln('However, as this depends on modifying the config.php, this only works as long as the instance runs on a single node or all nodes in a clustered setup have a shared config file (such as from a shared network mount).'); |
|
63 | + $output->writeln('If the different nodes have a separate copy of the config.php file, the automatic object store configuration creation will lead to the configuration going out of sync.'); |
|
64 | + $output->writeln('If these requirements are not met, you can manually create the target object store configuration in each node\'s configuration before running the command.'); |
|
65 | + $output->writeln(''); |
|
66 | + $output->writeln('<error>Failure to check these requirements will lead to data loss for users.</error>'); |
|
67 | 67 | |
68 | - /** @var QuestionHelper $helper */ |
|
69 | - $helper = $this->getHelper('question'); |
|
70 | - $question = new ConfirmationQuestion('Automatically create target object store configuration? [y/N] ', false); |
|
71 | - if ($helper->ask($input, $output, $question)) { |
|
72 | - $configs[$target] = $configs[$source]; |
|
68 | + /** @var QuestionHelper $helper */ |
|
69 | + $helper = $this->getHelper('question'); |
|
70 | + $question = new ConfirmationQuestion('Automatically create target object store configuration? [y/N] ', false); |
|
71 | + if ($helper->ask($input, $output, $question)) { |
|
72 | + $configs[$target] = $configs[$source]; |
|
73 | 73 | |
74 | - // update all aliases |
|
75 | - foreach ($configs as &$config) { |
|
76 | - if ($config === $source) { |
|
77 | - $config = $target; |
|
78 | - } |
|
79 | - } |
|
80 | - $this->config->setSystemValue('objectstore', $configs); |
|
81 | - } else { |
|
82 | - return 0; |
|
83 | - } |
|
84 | - } elseif (($configs[$source] !== $configs[$target]) || $configs[$source] !== $target) { |
|
85 | - $output->writeln('<error>Source and target configuration differ.</error>'); |
|
86 | - $output->writeln(''); |
|
87 | - $output->writeln('To ensure proper migration of users, the source and target configuration must be the same to ensure that the objects for the moved users exist on the target configuration.'); |
|
88 | - $output->writeln('The usual migration process consists of creating a clone of the old configuration, moving the users from the old configuration to the new one, and then adjust the old configuration that is longer used.'); |
|
89 | - return 1; |
|
90 | - } |
|
74 | + // update all aliases |
|
75 | + foreach ($configs as &$config) { |
|
76 | + if ($config === $source) { |
|
77 | + $config = $target; |
|
78 | + } |
|
79 | + } |
|
80 | + $this->config->setSystemValue('objectstore', $configs); |
|
81 | + } else { |
|
82 | + return 0; |
|
83 | + } |
|
84 | + } elseif (($configs[$source] !== $configs[$target]) || $configs[$source] !== $target) { |
|
85 | + $output->writeln('<error>Source and target configuration differ.</error>'); |
|
86 | + $output->writeln(''); |
|
87 | + $output->writeln('To ensure proper migration of users, the source and target configuration must be the same to ensure that the objects for the moved users exist on the target configuration.'); |
|
88 | + $output->writeln('The usual migration process consists of creating a clone of the old configuration, moving the users from the old configuration to the new one, and then adjust the old configuration that is longer used.'); |
|
89 | + return 1; |
|
90 | + } |
|
91 | 91 | |
92 | - $query = $this->connection->getQueryBuilder(); |
|
93 | - $query->update('preferences') |
|
94 | - ->set('configvalue', $query->createNamedParameter($target)) |
|
95 | - ->where($query->expr()->eq('appid', $query->createNamedParameter('homeobjectstore'))) |
|
96 | - ->andWhere($query->expr()->eq('configkey', $query->createNamedParameter('objectstore'))) |
|
97 | - ->andWhere($query->expr()->eq('configvalue', $query->createNamedParameter($source))); |
|
98 | - $count = $query->executeStatement(); |
|
92 | + $query = $this->connection->getQueryBuilder(); |
|
93 | + $query->update('preferences') |
|
94 | + ->set('configvalue', $query->createNamedParameter($target)) |
|
95 | + ->where($query->expr()->eq('appid', $query->createNamedParameter('homeobjectstore'))) |
|
96 | + ->andWhere($query->expr()->eq('configkey', $query->createNamedParameter('objectstore'))) |
|
97 | + ->andWhere($query->expr()->eq('configvalue', $query->createNamedParameter($source))); |
|
98 | + $count = $query->executeStatement(); |
|
99 | 99 | |
100 | - if ($count > 0) { |
|
101 | - $output->writeln('Moved <info>' . $count . '</info> users'); |
|
102 | - } else { |
|
103 | - $output->writeln('No users moved'); |
|
104 | - } |
|
100 | + if ($count > 0) { |
|
101 | + $output->writeln('Moved <info>' . $count . '</info> users'); |
|
102 | + } else { |
|
103 | + $output->writeln('No users moved'); |
|
104 | + } |
|
105 | 105 | |
106 | - return 0; |
|
107 | - } |
|
106 | + return 0; |
|
107 | + } |
|
108 | 108 | } |
@@ -17,269 +17,269 @@ |
||
17 | 17 | use Test\TestCase; |
18 | 18 | |
19 | 19 | class PrimaryObjectStoreConfigTest extends TestCase { |
20 | - private array $systemConfig = []; |
|
21 | - private array $userConfig = []; |
|
22 | - private IConfig&MockObject $config; |
|
23 | - private IAppManager&MockObject $appManager; |
|
24 | - private PrimaryObjectStoreConfig $objectStoreConfig; |
|
25 | - |
|
26 | - protected function setUp(): void { |
|
27 | - parent::setUp(); |
|
28 | - |
|
29 | - $this->systemConfig = []; |
|
30 | - $this->config = $this->createMock(IConfig::class); |
|
31 | - $this->appManager = $this->createMock(IAppManager::class); |
|
32 | - $this->config->method('getSystemValue') |
|
33 | - ->willReturnCallback(function ($key, $default = '') { |
|
34 | - if (isset($this->systemConfig[$key])) { |
|
35 | - return $this->systemConfig[$key]; |
|
36 | - } else { |
|
37 | - return $default; |
|
38 | - } |
|
39 | - }); |
|
40 | - $this->config->method('getUserValue') |
|
41 | - ->willReturnCallback(function ($userId, $appName, $key, $default = '') { |
|
42 | - if (isset($this->userConfig[$userId][$appName][$key])) { |
|
43 | - return $this->userConfig[$userId][$appName][$key]; |
|
44 | - } else { |
|
45 | - return $default; |
|
46 | - } |
|
47 | - }); |
|
48 | - $this->config->method('setUserValue') |
|
49 | - ->willReturnCallback(function ($userId, $appName, $key, $value) { |
|
50 | - $this->userConfig[$userId][$appName][$key] = $value; |
|
51 | - }); |
|
52 | - |
|
53 | - $this->objectStoreConfig = new PrimaryObjectStoreConfig($this->config, $this->appManager); |
|
54 | - } |
|
55 | - |
|
56 | - private function getUser(string $uid): IUser { |
|
57 | - $user = $this->createMock(IUser::class); |
|
58 | - $user->method('getUID') |
|
59 | - ->willReturn($uid); |
|
60 | - return $user; |
|
61 | - } |
|
62 | - |
|
63 | - private function setConfig(string $key, $value) { |
|
64 | - $this->systemConfig[$key] = $value; |
|
65 | - } |
|
66 | - |
|
67 | - public function testNewUserGetsDefault() { |
|
68 | - $this->setConfig('objectstore', [ |
|
69 | - 'default' => 'server1', |
|
70 | - 'server1' => [ |
|
71 | - 'class' => StorageObjectStore::class, |
|
72 | - 'arguments' => [ |
|
73 | - 'host' => 'server1', |
|
74 | - ], |
|
75 | - ], |
|
76 | - ]); |
|
77 | - |
|
78 | - $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test')); |
|
79 | - $this->assertEquals('server1', $result['arguments']['host']); |
|
80 | - |
|
81 | - $this->assertEquals('server1', $this->config->getUserValue('test', 'homeobjectstore', 'objectstore', null)); |
|
82 | - } |
|
83 | - |
|
84 | - public function testExistingUserKeepsStorage() { |
|
85 | - // setup user with `server1` as storage |
|
86 | - $this->testNewUserGetsDefault(); |
|
87 | - |
|
88 | - $this->setConfig('objectstore', [ |
|
89 | - 'default' => 'server2', |
|
90 | - 'server1' => [ |
|
91 | - 'class' => StorageObjectStore::class, |
|
92 | - 'arguments' => [ |
|
93 | - 'host' => 'server1', |
|
94 | - ], |
|
95 | - ], |
|
96 | - 'server2' => [ |
|
97 | - 'class' => StorageObjectStore::class, |
|
98 | - 'arguments' => [ |
|
99 | - 'host' => 'server2', |
|
100 | - ], |
|
101 | - ], |
|
102 | - ]); |
|
103 | - |
|
104 | - $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test')); |
|
105 | - $this->assertEquals('server1', $result['arguments']['host']); |
|
106 | - |
|
107 | - $this->assertEquals('server1', $this->config->getUserValue('test', 'homeobjectstore', 'objectstore', null)); |
|
108 | - |
|
109 | - $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('other-user')); |
|
110 | - $this->assertEquals('server2', $result['arguments']['host']); |
|
111 | - } |
|
112 | - |
|
113 | - public function testNestedAliases() { |
|
114 | - $this->setConfig('objectstore', [ |
|
115 | - 'default' => 'a1', |
|
116 | - 'a1' => 'a2', |
|
117 | - 'a2' => 'server1', |
|
118 | - 'server1' => [ |
|
119 | - 'class' => StorageObjectStore::class, |
|
120 | - 'arguments' => [ |
|
121 | - 'host' => 'server1', |
|
122 | - ], |
|
123 | - ], |
|
124 | - ]); |
|
125 | - $this->assertEquals('server1', $this->objectStoreConfig->resolveAlias('default')); |
|
126 | - } |
|
127 | - |
|
128 | - public function testMultibucketChangedConfig() { |
|
129 | - $this->setConfig('objectstore', [ |
|
130 | - 'default' => 'server1', |
|
131 | - 'server1' => [ |
|
132 | - 'class' => StorageObjectStore::class, |
|
133 | - 'arguments' => [ |
|
134 | - 'host' => 'server1', |
|
135 | - 'multibucket' => true, |
|
136 | - 'num_buckets' => 8, |
|
137 | - 'bucket' => 'bucket-' |
|
138 | - ], |
|
139 | - ], |
|
140 | - ]); |
|
141 | - |
|
142 | - $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test')); |
|
143 | - $this->assertEquals('server1', $result['arguments']['host']); |
|
144 | - $this->assertEquals('bucket-7', $result['arguments']['bucket']); |
|
145 | - |
|
146 | - $this->setConfig('objectstore', [ |
|
147 | - 'default' => 'server1', |
|
148 | - 'server1' => [ |
|
149 | - 'class' => StorageObjectStore::class, |
|
150 | - 'arguments' => [ |
|
151 | - 'host' => 'server1', |
|
152 | - 'multibucket' => true, |
|
153 | - 'num_buckets' => 64, |
|
154 | - 'bucket' => 'bucket-' |
|
155 | - ], |
|
156 | - ], |
|
157 | - ]); |
|
158 | - |
|
159 | - $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test')); |
|
160 | - $this->assertEquals('server1', $result['arguments']['host']); |
|
161 | - $this->assertEquals('bucket-7', $result['arguments']['bucket']); |
|
162 | - |
|
163 | - $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test-foo')); |
|
164 | - $this->assertEquals('server1', $result['arguments']['host']); |
|
165 | - $this->assertEquals('bucket-40', $result['arguments']['bucket']); |
|
166 | - |
|
167 | - $this->setConfig('objectstore', [ |
|
168 | - 'default' => 'server2', |
|
169 | - 'server1' => [ |
|
170 | - 'class' => StorageObjectStore::class, |
|
171 | - 'arguments' => [ |
|
172 | - 'host' => 'server1', |
|
173 | - 'multibucket' => true, |
|
174 | - 'num_buckets' => 64, |
|
175 | - 'bucket' => 'bucket-' |
|
176 | - ], |
|
177 | - ], |
|
178 | - 'server2' => [ |
|
179 | - 'class' => StorageObjectStore::class, |
|
180 | - 'arguments' => [ |
|
181 | - 'host' => 'server2', |
|
182 | - 'multibucket' => true, |
|
183 | - 'num_buckets' => 16, |
|
184 | - 'bucket' => 'bucket-' |
|
185 | - ], |
|
186 | - ], |
|
187 | - ]); |
|
188 | - |
|
189 | - $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test')); |
|
190 | - $this->assertEquals('server1', $result['arguments']['host']); |
|
191 | - $this->assertEquals('bucket-7', $result['arguments']['bucket']); |
|
192 | - |
|
193 | - $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test-bar')); |
|
194 | - $this->assertEquals('server2', $result['arguments']['host']); |
|
195 | - $this->assertEquals('bucket-4', $result['arguments']['bucket']); |
|
196 | - } |
|
197 | - |
|
198 | - public function testMultibucketOldConfig() { |
|
199 | - $this->setConfig('objectstore_multibucket', [ |
|
200 | - 'class' => StorageObjectStore::class, |
|
201 | - 'arguments' => [ |
|
202 | - 'host' => 'server1', |
|
203 | - 'multibucket' => true, |
|
204 | - 'num_buckets' => 8, |
|
205 | - 'bucket' => 'bucket-' |
|
206 | - ], |
|
207 | - ]); |
|
208 | - $configs = $this->objectStoreConfig->getObjectStoreConfigs(); |
|
209 | - $this->assertEquals([ |
|
210 | - 'default' => 'server1', |
|
211 | - 'root' => 'server1', |
|
212 | - 'server1' => [ |
|
213 | - 'class' => StorageObjectStore::class, |
|
214 | - 'arguments' => [ |
|
215 | - 'host' => 'server1', |
|
216 | - 'multibucket' => true, |
|
217 | - 'num_buckets' => 8, |
|
218 | - 'bucket' => 'bucket-' |
|
219 | - ], |
|
220 | - ], |
|
221 | - ], $configs); |
|
222 | - } |
|
223 | - |
|
224 | - public function testSingleObjectStore() { |
|
225 | - $this->setConfig('objectstore', [ |
|
226 | - 'class' => StorageObjectStore::class, |
|
227 | - 'arguments' => [ |
|
228 | - 'host' => 'server1', |
|
229 | - ], |
|
230 | - ]); |
|
231 | - $configs = $this->objectStoreConfig->getObjectStoreConfigs(); |
|
232 | - $this->assertEquals([ |
|
233 | - 'default' => 'server1', |
|
234 | - 'root' => 'server1', |
|
235 | - 'server1' => [ |
|
236 | - 'class' => StorageObjectStore::class, |
|
237 | - 'arguments' => [ |
|
238 | - 'host' => 'server1', |
|
239 | - 'multibucket' => false, |
|
240 | - ], |
|
241 | - ], |
|
242 | - ], $configs); |
|
243 | - } |
|
244 | - |
|
245 | - public function testRoot() { |
|
246 | - $this->setConfig('objectstore', [ |
|
247 | - 'default' => 'server1', |
|
248 | - 'server1' => [ |
|
249 | - 'class' => StorageObjectStore::class, |
|
250 | - 'arguments' => [ |
|
251 | - 'host' => 'server1', |
|
252 | - ], |
|
253 | - ], |
|
254 | - 'server2' => [ |
|
255 | - 'class' => StorageObjectStore::class, |
|
256 | - 'arguments' => [ |
|
257 | - 'host' => 'server2', |
|
258 | - ], |
|
259 | - ], |
|
260 | - ]); |
|
261 | - |
|
262 | - $result = $this->objectStoreConfig->getObjectStoreConfigForRoot(); |
|
263 | - $this->assertEquals('server1', $result['arguments']['host']); |
|
264 | - |
|
265 | - $this->setConfig('objectstore', [ |
|
266 | - 'default' => 'server1', |
|
267 | - 'root' => 'server2', |
|
268 | - 'server1' => [ |
|
269 | - 'class' => StorageObjectStore::class, |
|
270 | - 'arguments' => [ |
|
271 | - 'host' => 'server1', |
|
272 | - ], |
|
273 | - ], |
|
274 | - 'server2' => [ |
|
275 | - 'class' => StorageObjectStore::class, |
|
276 | - 'arguments' => [ |
|
277 | - 'host' => 'server2', |
|
278 | - ], |
|
279 | - ], |
|
280 | - ]); |
|
281 | - |
|
282 | - $result = $this->objectStoreConfig->getObjectStoreConfigForRoot(); |
|
283 | - $this->assertEquals('server2', $result['arguments']['host']); |
|
284 | - } |
|
20 | + private array $systemConfig = []; |
|
21 | + private array $userConfig = []; |
|
22 | + private IConfig&MockObject $config; |
|
23 | + private IAppManager&MockObject $appManager; |
|
24 | + private PrimaryObjectStoreConfig $objectStoreConfig; |
|
25 | + |
|
26 | + protected function setUp(): void { |
|
27 | + parent::setUp(); |
|
28 | + |
|
29 | + $this->systemConfig = []; |
|
30 | + $this->config = $this->createMock(IConfig::class); |
|
31 | + $this->appManager = $this->createMock(IAppManager::class); |
|
32 | + $this->config->method('getSystemValue') |
|
33 | + ->willReturnCallback(function ($key, $default = '') { |
|
34 | + if (isset($this->systemConfig[$key])) { |
|
35 | + return $this->systemConfig[$key]; |
|
36 | + } else { |
|
37 | + return $default; |
|
38 | + } |
|
39 | + }); |
|
40 | + $this->config->method('getUserValue') |
|
41 | + ->willReturnCallback(function ($userId, $appName, $key, $default = '') { |
|
42 | + if (isset($this->userConfig[$userId][$appName][$key])) { |
|
43 | + return $this->userConfig[$userId][$appName][$key]; |
|
44 | + } else { |
|
45 | + return $default; |
|
46 | + } |
|
47 | + }); |
|
48 | + $this->config->method('setUserValue') |
|
49 | + ->willReturnCallback(function ($userId, $appName, $key, $value) { |
|
50 | + $this->userConfig[$userId][$appName][$key] = $value; |
|
51 | + }); |
|
52 | + |
|
53 | + $this->objectStoreConfig = new PrimaryObjectStoreConfig($this->config, $this->appManager); |
|
54 | + } |
|
55 | + |
|
56 | + private function getUser(string $uid): IUser { |
|
57 | + $user = $this->createMock(IUser::class); |
|
58 | + $user->method('getUID') |
|
59 | + ->willReturn($uid); |
|
60 | + return $user; |
|
61 | + } |
|
62 | + |
|
63 | + private function setConfig(string $key, $value) { |
|
64 | + $this->systemConfig[$key] = $value; |
|
65 | + } |
|
66 | + |
|
67 | + public function testNewUserGetsDefault() { |
|
68 | + $this->setConfig('objectstore', [ |
|
69 | + 'default' => 'server1', |
|
70 | + 'server1' => [ |
|
71 | + 'class' => StorageObjectStore::class, |
|
72 | + 'arguments' => [ |
|
73 | + 'host' => 'server1', |
|
74 | + ], |
|
75 | + ], |
|
76 | + ]); |
|
77 | + |
|
78 | + $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test')); |
|
79 | + $this->assertEquals('server1', $result['arguments']['host']); |
|
80 | + |
|
81 | + $this->assertEquals('server1', $this->config->getUserValue('test', 'homeobjectstore', 'objectstore', null)); |
|
82 | + } |
|
83 | + |
|
84 | + public function testExistingUserKeepsStorage() { |
|
85 | + // setup user with `server1` as storage |
|
86 | + $this->testNewUserGetsDefault(); |
|
87 | + |
|
88 | + $this->setConfig('objectstore', [ |
|
89 | + 'default' => 'server2', |
|
90 | + 'server1' => [ |
|
91 | + 'class' => StorageObjectStore::class, |
|
92 | + 'arguments' => [ |
|
93 | + 'host' => 'server1', |
|
94 | + ], |
|
95 | + ], |
|
96 | + 'server2' => [ |
|
97 | + 'class' => StorageObjectStore::class, |
|
98 | + 'arguments' => [ |
|
99 | + 'host' => 'server2', |
|
100 | + ], |
|
101 | + ], |
|
102 | + ]); |
|
103 | + |
|
104 | + $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test')); |
|
105 | + $this->assertEquals('server1', $result['arguments']['host']); |
|
106 | + |
|
107 | + $this->assertEquals('server1', $this->config->getUserValue('test', 'homeobjectstore', 'objectstore', null)); |
|
108 | + |
|
109 | + $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('other-user')); |
|
110 | + $this->assertEquals('server2', $result['arguments']['host']); |
|
111 | + } |
|
112 | + |
|
113 | + public function testNestedAliases() { |
|
114 | + $this->setConfig('objectstore', [ |
|
115 | + 'default' => 'a1', |
|
116 | + 'a1' => 'a2', |
|
117 | + 'a2' => 'server1', |
|
118 | + 'server1' => [ |
|
119 | + 'class' => StorageObjectStore::class, |
|
120 | + 'arguments' => [ |
|
121 | + 'host' => 'server1', |
|
122 | + ], |
|
123 | + ], |
|
124 | + ]); |
|
125 | + $this->assertEquals('server1', $this->objectStoreConfig->resolveAlias('default')); |
|
126 | + } |
|
127 | + |
|
128 | + public function testMultibucketChangedConfig() { |
|
129 | + $this->setConfig('objectstore', [ |
|
130 | + 'default' => 'server1', |
|
131 | + 'server1' => [ |
|
132 | + 'class' => StorageObjectStore::class, |
|
133 | + 'arguments' => [ |
|
134 | + 'host' => 'server1', |
|
135 | + 'multibucket' => true, |
|
136 | + 'num_buckets' => 8, |
|
137 | + 'bucket' => 'bucket-' |
|
138 | + ], |
|
139 | + ], |
|
140 | + ]); |
|
141 | + |
|
142 | + $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test')); |
|
143 | + $this->assertEquals('server1', $result['arguments']['host']); |
|
144 | + $this->assertEquals('bucket-7', $result['arguments']['bucket']); |
|
145 | + |
|
146 | + $this->setConfig('objectstore', [ |
|
147 | + 'default' => 'server1', |
|
148 | + 'server1' => [ |
|
149 | + 'class' => StorageObjectStore::class, |
|
150 | + 'arguments' => [ |
|
151 | + 'host' => 'server1', |
|
152 | + 'multibucket' => true, |
|
153 | + 'num_buckets' => 64, |
|
154 | + 'bucket' => 'bucket-' |
|
155 | + ], |
|
156 | + ], |
|
157 | + ]); |
|
158 | + |
|
159 | + $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test')); |
|
160 | + $this->assertEquals('server1', $result['arguments']['host']); |
|
161 | + $this->assertEquals('bucket-7', $result['arguments']['bucket']); |
|
162 | + |
|
163 | + $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test-foo')); |
|
164 | + $this->assertEquals('server1', $result['arguments']['host']); |
|
165 | + $this->assertEquals('bucket-40', $result['arguments']['bucket']); |
|
166 | + |
|
167 | + $this->setConfig('objectstore', [ |
|
168 | + 'default' => 'server2', |
|
169 | + 'server1' => [ |
|
170 | + 'class' => StorageObjectStore::class, |
|
171 | + 'arguments' => [ |
|
172 | + 'host' => 'server1', |
|
173 | + 'multibucket' => true, |
|
174 | + 'num_buckets' => 64, |
|
175 | + 'bucket' => 'bucket-' |
|
176 | + ], |
|
177 | + ], |
|
178 | + 'server2' => [ |
|
179 | + 'class' => StorageObjectStore::class, |
|
180 | + 'arguments' => [ |
|
181 | + 'host' => 'server2', |
|
182 | + 'multibucket' => true, |
|
183 | + 'num_buckets' => 16, |
|
184 | + 'bucket' => 'bucket-' |
|
185 | + ], |
|
186 | + ], |
|
187 | + ]); |
|
188 | + |
|
189 | + $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test')); |
|
190 | + $this->assertEquals('server1', $result['arguments']['host']); |
|
191 | + $this->assertEquals('bucket-7', $result['arguments']['bucket']); |
|
192 | + |
|
193 | + $result = $this->objectStoreConfig->getObjectStoreConfigForUser($this->getUser('test-bar')); |
|
194 | + $this->assertEquals('server2', $result['arguments']['host']); |
|
195 | + $this->assertEquals('bucket-4', $result['arguments']['bucket']); |
|
196 | + } |
|
197 | + |
|
198 | + public function testMultibucketOldConfig() { |
|
199 | + $this->setConfig('objectstore_multibucket', [ |
|
200 | + 'class' => StorageObjectStore::class, |
|
201 | + 'arguments' => [ |
|
202 | + 'host' => 'server1', |
|
203 | + 'multibucket' => true, |
|
204 | + 'num_buckets' => 8, |
|
205 | + 'bucket' => 'bucket-' |
|
206 | + ], |
|
207 | + ]); |
|
208 | + $configs = $this->objectStoreConfig->getObjectStoreConfigs(); |
|
209 | + $this->assertEquals([ |
|
210 | + 'default' => 'server1', |
|
211 | + 'root' => 'server1', |
|
212 | + 'server1' => [ |
|
213 | + 'class' => StorageObjectStore::class, |
|
214 | + 'arguments' => [ |
|
215 | + 'host' => 'server1', |
|
216 | + 'multibucket' => true, |
|
217 | + 'num_buckets' => 8, |
|
218 | + 'bucket' => 'bucket-' |
|
219 | + ], |
|
220 | + ], |
|
221 | + ], $configs); |
|
222 | + } |
|
223 | + |
|
224 | + public function testSingleObjectStore() { |
|
225 | + $this->setConfig('objectstore', [ |
|
226 | + 'class' => StorageObjectStore::class, |
|
227 | + 'arguments' => [ |
|
228 | + 'host' => 'server1', |
|
229 | + ], |
|
230 | + ]); |
|
231 | + $configs = $this->objectStoreConfig->getObjectStoreConfigs(); |
|
232 | + $this->assertEquals([ |
|
233 | + 'default' => 'server1', |
|
234 | + 'root' => 'server1', |
|
235 | + 'server1' => [ |
|
236 | + 'class' => StorageObjectStore::class, |
|
237 | + 'arguments' => [ |
|
238 | + 'host' => 'server1', |
|
239 | + 'multibucket' => false, |
|
240 | + ], |
|
241 | + ], |
|
242 | + ], $configs); |
|
243 | + } |
|
244 | + |
|
245 | + public function testRoot() { |
|
246 | + $this->setConfig('objectstore', [ |
|
247 | + 'default' => 'server1', |
|
248 | + 'server1' => [ |
|
249 | + 'class' => StorageObjectStore::class, |
|
250 | + 'arguments' => [ |
|
251 | + 'host' => 'server1', |
|
252 | + ], |
|
253 | + ], |
|
254 | + 'server2' => [ |
|
255 | + 'class' => StorageObjectStore::class, |
|
256 | + 'arguments' => [ |
|
257 | + 'host' => 'server2', |
|
258 | + ], |
|
259 | + ], |
|
260 | + ]); |
|
261 | + |
|
262 | + $result = $this->objectStoreConfig->getObjectStoreConfigForRoot(); |
|
263 | + $this->assertEquals('server1', $result['arguments']['host']); |
|
264 | + |
|
265 | + $this->setConfig('objectstore', [ |
|
266 | + 'default' => 'server1', |
|
267 | + 'root' => 'server2', |
|
268 | + 'server1' => [ |
|
269 | + 'class' => StorageObjectStore::class, |
|
270 | + 'arguments' => [ |
|
271 | + 'host' => 'server1', |
|
272 | + ], |
|
273 | + ], |
|
274 | + 'server2' => [ |
|
275 | + 'class' => StorageObjectStore::class, |
|
276 | + 'arguments' => [ |
|
277 | + 'host' => 'server2', |
|
278 | + ], |
|
279 | + ], |
|
280 | + ]); |
|
281 | + |
|
282 | + $result = $this->objectStoreConfig->getObjectStoreConfigForRoot(); |
|
283 | + $this->assertEquals('server2', $result['arguments']['host']); |
|
284 | + } |
|
285 | 285 | } |
@@ -16,256 +16,256 @@ |
||
16 | 16 | use OCP\IUser; |
17 | 17 | |
18 | 18 | class ObjectHomeMountProviderTest extends \Test\TestCase { |
19 | - /** @var ObjectHomeMountProvider */ |
|
20 | - protected $provider; |
|
21 | - |
|
22 | - /** @var IConfig|\PHPUnit\Framework\MockObject\MockObject */ |
|
23 | - protected $config; |
|
24 | - |
|
25 | - /** @var IUser|\PHPUnit\Framework\MockObject\MockObject */ |
|
26 | - protected $user; |
|
27 | - |
|
28 | - /** @var IStorageFactory|\PHPUnit\Framework\MockObject\MockObject */ |
|
29 | - protected $loader; |
|
30 | - |
|
31 | - protected function setUp(): void { |
|
32 | - parent::setUp(); |
|
33 | - |
|
34 | - $this->config = $this->createMock(IConfig::class); |
|
35 | - $this->user = $this->createMock(IUser::class); |
|
36 | - $this->loader = $this->createMock(IStorageFactory::class); |
|
37 | - |
|
38 | - $objectStoreConfig = new PrimaryObjectStoreConfig($this->config, $this->createMock(IAppManager::class)); |
|
39 | - $this->provider = new ObjectHomeMountProvider($objectStoreConfig); |
|
40 | - } |
|
41 | - |
|
42 | - public function testSingleBucket(): void { |
|
43 | - $this->config->method('getSystemValue') |
|
44 | - ->willReturnCallback(function ($key, $default) { |
|
45 | - if ($key === 'objectstore') { |
|
46 | - return [ |
|
47 | - 'class' => 'Test\Files\Mount\FakeObjectStore', |
|
48 | - 'arguments' => [ |
|
49 | - 'foo' => 'bar' |
|
50 | - ], |
|
51 | - ]; |
|
52 | - } else { |
|
53 | - return $default; |
|
54 | - } |
|
55 | - }); |
|
56 | - |
|
57 | - $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
58 | - $arguments = $this->invokePrivate($mount, 'arguments'); |
|
59 | - |
|
60 | - $objectStore = $arguments['objectstore']; |
|
61 | - $this->assertInstanceOf(FakeObjectStore::class, $objectStore); |
|
62 | - $this->assertEquals(['foo' => 'bar', 'multibucket' => false], $objectStore->getArguments()); |
|
63 | - } |
|
64 | - |
|
65 | - public function testMultiBucket(): void { |
|
66 | - $this->config->method('getSystemValue') |
|
67 | - ->willReturnCallback(function ($key, $default) { |
|
68 | - if ($key === 'objectstore_multibucket') { |
|
69 | - return [ |
|
70 | - 'class' => 'Test\Files\Mount\FakeObjectStore', |
|
71 | - 'arguments' => [ |
|
72 | - 'foo' => 'bar' |
|
73 | - ], |
|
74 | - ]; |
|
75 | - } else { |
|
76 | - return $default; |
|
77 | - } |
|
78 | - }); |
|
79 | - |
|
80 | - $this->user->method('getUID') |
|
81 | - ->willReturn('uid'); |
|
82 | - $this->loader->expects($this->never())->method($this->anything()); |
|
83 | - |
|
84 | - $this->config->method('getUserValue') |
|
85 | - ->willReturn(null); |
|
86 | - |
|
87 | - $this->config |
|
88 | - ->method('setUserValue') |
|
89 | - ->with( |
|
90 | - $this->equalTo('uid'), |
|
91 | - $this->equalTo('homeobjectstore'), |
|
92 | - $this->equalTo('bucket'), |
|
93 | - $this->equalTo('49'), |
|
94 | - $this->equalTo(null) |
|
95 | - ); |
|
96 | - |
|
97 | - $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
98 | - $arguments = $this->invokePrivate($mount, 'arguments'); |
|
99 | - |
|
100 | - $objectStore = $arguments['objectstore']; |
|
101 | - $this->assertInstanceOf(FakeObjectStore::class, $objectStore); |
|
102 | - $this->assertEquals(['foo' => 'bar', 'bucket' => 49, 'multibucket' => true], $objectStore->getArguments()); |
|
103 | - } |
|
104 | - |
|
105 | - public function testMultiBucketWithPrefix(): void { |
|
106 | - $this->config->method('getSystemValue') |
|
107 | - ->willReturnCallback(function ($key, $default) { |
|
108 | - if ($key === 'objectstore_multibucket') { |
|
109 | - return [ |
|
110 | - 'class' => 'Test\Files\Mount\FakeObjectStore', |
|
111 | - 'arguments' => [ |
|
112 | - 'foo' => 'bar', |
|
113 | - 'bucket' => 'myBucketPrefix', |
|
114 | - ], |
|
115 | - ]; |
|
116 | - } else { |
|
117 | - return $default; |
|
118 | - } |
|
119 | - }); |
|
120 | - |
|
121 | - $this->user->method('getUID') |
|
122 | - ->willReturn('uid'); |
|
123 | - $this->loader->expects($this->never())->method($this->anything()); |
|
124 | - |
|
125 | - $this->config |
|
126 | - ->method('getUserValue') |
|
127 | - ->willReturn(null); |
|
128 | - |
|
129 | - $this->config->expects($this->once()) |
|
130 | - ->method('setUserValue') |
|
131 | - ->with( |
|
132 | - $this->equalTo('uid'), |
|
133 | - $this->equalTo('homeobjectstore'), |
|
134 | - $this->equalTo('bucket'), |
|
135 | - $this->equalTo('myBucketPrefix49'), |
|
136 | - $this->equalTo(null) |
|
137 | - ); |
|
138 | - |
|
139 | - $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
140 | - $arguments = $this->invokePrivate($mount, 'arguments'); |
|
141 | - |
|
142 | - $objectStore = $arguments['objectstore']; |
|
143 | - $this->assertInstanceOf(FakeObjectStore::class, $objectStore); |
|
144 | - $this->assertEquals(['foo' => 'bar', 'bucket' => 'myBucketPrefix49', 'multibucket' => true], $objectStore->getArguments()); |
|
145 | - } |
|
146 | - |
|
147 | - public function testMultiBucketBucketAlreadySet(): void { |
|
148 | - $this->config->method('getSystemValue') |
|
149 | - ->willReturnCallback(function ($key, $default) { |
|
150 | - if ($key === 'objectstore_multibucket') { |
|
151 | - return [ |
|
152 | - 'class' => 'Test\Files\Mount\FakeObjectStore', |
|
153 | - 'arguments' => [ |
|
154 | - 'foo' => 'bar', |
|
155 | - 'bucket' => 'myBucketPrefix', |
|
156 | - ], |
|
157 | - ]; |
|
158 | - } else { |
|
159 | - return $default; |
|
160 | - } |
|
161 | - }); |
|
162 | - |
|
163 | - $this->user->method('getUID') |
|
164 | - ->willReturn('uid'); |
|
165 | - $this->loader->expects($this->never())->method($this->anything()); |
|
166 | - |
|
167 | - $this->config |
|
168 | - ->method('getUserValue') |
|
169 | - ->willReturnCallback(function ($uid, $app, $key, $default) { |
|
170 | - if ($uid === 'uid' && $app === 'homeobjectstore' && $key === 'bucket') { |
|
171 | - return 'awesomeBucket1'; |
|
172 | - } else { |
|
173 | - return $default; |
|
174 | - } |
|
175 | - }); |
|
176 | - |
|
177 | - $this->config->expects($this->never()) |
|
178 | - ->method('setUserValue'); |
|
179 | - |
|
180 | - $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
181 | - $arguments = $this->invokePrivate($mount, 'arguments'); |
|
182 | - |
|
183 | - $objectStore = $arguments['objectstore']; |
|
184 | - $this->assertInstanceOf(FakeObjectStore::class, $objectStore); |
|
185 | - $this->assertEquals(['foo' => 'bar', 'bucket' => 'awesomeBucket1', 'multibucket' => true], $objectStore->getArguments()); |
|
186 | - } |
|
187 | - |
|
188 | - public function testMultiBucketConfigFirst(): void { |
|
189 | - $this->config->method('getSystemValue') |
|
190 | - ->willReturnCallback(function ($key, $default) { |
|
191 | - if ($key === 'objectstore_multibucket') { |
|
192 | - return [ |
|
193 | - 'class' => 'Test\Files\Mount\FakeObjectStore', |
|
194 | - 'arguments' => [ |
|
195 | - 'foo' => 'bar', |
|
196 | - 'bucket' => 'myBucketPrefix', |
|
197 | - ], |
|
198 | - ]; |
|
199 | - } else { |
|
200 | - return $default; |
|
201 | - } |
|
202 | - }); |
|
203 | - |
|
204 | - $this->user->method('getUID') |
|
205 | - ->willReturn('uid'); |
|
206 | - $this->loader->expects($this->never())->method($this->anything()); |
|
207 | - |
|
208 | - $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
209 | - $this->assertInstanceOf('OC\Files\Mount\MountPoint', $mount); |
|
210 | - } |
|
211 | - |
|
212 | - public function testMultiBucketConfigFirstFallBackSingle(): void { |
|
213 | - $this->config |
|
214 | - ->method('getSystemValue')->willReturnMap([ |
|
215 | - ['objectstore_multibucket', null, null], |
|
216 | - ['objectstore', null, [ |
|
217 | - 'class' => 'Test\Files\Mount\FakeObjectStore', |
|
218 | - 'arguments' => [ |
|
219 | - 'foo' => 'bar', |
|
220 | - 'bucket' => 'myBucketPrefix', |
|
221 | - ], |
|
222 | - ]], |
|
223 | - ]); |
|
224 | - |
|
225 | - $this->user->method('getUID') |
|
226 | - ->willReturn('uid'); |
|
227 | - $this->loader->expects($this->never())->method($this->anything()); |
|
228 | - |
|
229 | - $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
230 | - $this->assertInstanceOf('OC\Files\Mount\MountPoint', $mount); |
|
231 | - } |
|
232 | - |
|
233 | - public function testNoObjectStore(): void { |
|
234 | - $this->config->method('getSystemValue') |
|
235 | - ->willReturnCallback(function ($key, $default) { |
|
236 | - return $default; |
|
237 | - }); |
|
238 | - |
|
239 | - $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
240 | - $this->assertNull($mount); |
|
241 | - } |
|
19 | + /** @var ObjectHomeMountProvider */ |
|
20 | + protected $provider; |
|
21 | + |
|
22 | + /** @var IConfig|\PHPUnit\Framework\MockObject\MockObject */ |
|
23 | + protected $config; |
|
24 | + |
|
25 | + /** @var IUser|\PHPUnit\Framework\MockObject\MockObject */ |
|
26 | + protected $user; |
|
27 | + |
|
28 | + /** @var IStorageFactory|\PHPUnit\Framework\MockObject\MockObject */ |
|
29 | + protected $loader; |
|
30 | + |
|
31 | + protected function setUp(): void { |
|
32 | + parent::setUp(); |
|
33 | + |
|
34 | + $this->config = $this->createMock(IConfig::class); |
|
35 | + $this->user = $this->createMock(IUser::class); |
|
36 | + $this->loader = $this->createMock(IStorageFactory::class); |
|
37 | + |
|
38 | + $objectStoreConfig = new PrimaryObjectStoreConfig($this->config, $this->createMock(IAppManager::class)); |
|
39 | + $this->provider = new ObjectHomeMountProvider($objectStoreConfig); |
|
40 | + } |
|
41 | + |
|
42 | + public function testSingleBucket(): void { |
|
43 | + $this->config->method('getSystemValue') |
|
44 | + ->willReturnCallback(function ($key, $default) { |
|
45 | + if ($key === 'objectstore') { |
|
46 | + return [ |
|
47 | + 'class' => 'Test\Files\Mount\FakeObjectStore', |
|
48 | + 'arguments' => [ |
|
49 | + 'foo' => 'bar' |
|
50 | + ], |
|
51 | + ]; |
|
52 | + } else { |
|
53 | + return $default; |
|
54 | + } |
|
55 | + }); |
|
56 | + |
|
57 | + $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
58 | + $arguments = $this->invokePrivate($mount, 'arguments'); |
|
59 | + |
|
60 | + $objectStore = $arguments['objectstore']; |
|
61 | + $this->assertInstanceOf(FakeObjectStore::class, $objectStore); |
|
62 | + $this->assertEquals(['foo' => 'bar', 'multibucket' => false], $objectStore->getArguments()); |
|
63 | + } |
|
64 | + |
|
65 | + public function testMultiBucket(): void { |
|
66 | + $this->config->method('getSystemValue') |
|
67 | + ->willReturnCallback(function ($key, $default) { |
|
68 | + if ($key === 'objectstore_multibucket') { |
|
69 | + return [ |
|
70 | + 'class' => 'Test\Files\Mount\FakeObjectStore', |
|
71 | + 'arguments' => [ |
|
72 | + 'foo' => 'bar' |
|
73 | + ], |
|
74 | + ]; |
|
75 | + } else { |
|
76 | + return $default; |
|
77 | + } |
|
78 | + }); |
|
79 | + |
|
80 | + $this->user->method('getUID') |
|
81 | + ->willReturn('uid'); |
|
82 | + $this->loader->expects($this->never())->method($this->anything()); |
|
83 | + |
|
84 | + $this->config->method('getUserValue') |
|
85 | + ->willReturn(null); |
|
86 | + |
|
87 | + $this->config |
|
88 | + ->method('setUserValue') |
|
89 | + ->with( |
|
90 | + $this->equalTo('uid'), |
|
91 | + $this->equalTo('homeobjectstore'), |
|
92 | + $this->equalTo('bucket'), |
|
93 | + $this->equalTo('49'), |
|
94 | + $this->equalTo(null) |
|
95 | + ); |
|
96 | + |
|
97 | + $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
98 | + $arguments = $this->invokePrivate($mount, 'arguments'); |
|
99 | + |
|
100 | + $objectStore = $arguments['objectstore']; |
|
101 | + $this->assertInstanceOf(FakeObjectStore::class, $objectStore); |
|
102 | + $this->assertEquals(['foo' => 'bar', 'bucket' => 49, 'multibucket' => true], $objectStore->getArguments()); |
|
103 | + } |
|
104 | + |
|
105 | + public function testMultiBucketWithPrefix(): void { |
|
106 | + $this->config->method('getSystemValue') |
|
107 | + ->willReturnCallback(function ($key, $default) { |
|
108 | + if ($key === 'objectstore_multibucket') { |
|
109 | + return [ |
|
110 | + 'class' => 'Test\Files\Mount\FakeObjectStore', |
|
111 | + 'arguments' => [ |
|
112 | + 'foo' => 'bar', |
|
113 | + 'bucket' => 'myBucketPrefix', |
|
114 | + ], |
|
115 | + ]; |
|
116 | + } else { |
|
117 | + return $default; |
|
118 | + } |
|
119 | + }); |
|
120 | + |
|
121 | + $this->user->method('getUID') |
|
122 | + ->willReturn('uid'); |
|
123 | + $this->loader->expects($this->never())->method($this->anything()); |
|
124 | + |
|
125 | + $this->config |
|
126 | + ->method('getUserValue') |
|
127 | + ->willReturn(null); |
|
128 | + |
|
129 | + $this->config->expects($this->once()) |
|
130 | + ->method('setUserValue') |
|
131 | + ->with( |
|
132 | + $this->equalTo('uid'), |
|
133 | + $this->equalTo('homeobjectstore'), |
|
134 | + $this->equalTo('bucket'), |
|
135 | + $this->equalTo('myBucketPrefix49'), |
|
136 | + $this->equalTo(null) |
|
137 | + ); |
|
138 | + |
|
139 | + $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
140 | + $arguments = $this->invokePrivate($mount, 'arguments'); |
|
141 | + |
|
142 | + $objectStore = $arguments['objectstore']; |
|
143 | + $this->assertInstanceOf(FakeObjectStore::class, $objectStore); |
|
144 | + $this->assertEquals(['foo' => 'bar', 'bucket' => 'myBucketPrefix49', 'multibucket' => true], $objectStore->getArguments()); |
|
145 | + } |
|
146 | + |
|
147 | + public function testMultiBucketBucketAlreadySet(): void { |
|
148 | + $this->config->method('getSystemValue') |
|
149 | + ->willReturnCallback(function ($key, $default) { |
|
150 | + if ($key === 'objectstore_multibucket') { |
|
151 | + return [ |
|
152 | + 'class' => 'Test\Files\Mount\FakeObjectStore', |
|
153 | + 'arguments' => [ |
|
154 | + 'foo' => 'bar', |
|
155 | + 'bucket' => 'myBucketPrefix', |
|
156 | + ], |
|
157 | + ]; |
|
158 | + } else { |
|
159 | + return $default; |
|
160 | + } |
|
161 | + }); |
|
162 | + |
|
163 | + $this->user->method('getUID') |
|
164 | + ->willReturn('uid'); |
|
165 | + $this->loader->expects($this->never())->method($this->anything()); |
|
166 | + |
|
167 | + $this->config |
|
168 | + ->method('getUserValue') |
|
169 | + ->willReturnCallback(function ($uid, $app, $key, $default) { |
|
170 | + if ($uid === 'uid' && $app === 'homeobjectstore' && $key === 'bucket') { |
|
171 | + return 'awesomeBucket1'; |
|
172 | + } else { |
|
173 | + return $default; |
|
174 | + } |
|
175 | + }); |
|
176 | + |
|
177 | + $this->config->expects($this->never()) |
|
178 | + ->method('setUserValue'); |
|
179 | + |
|
180 | + $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
181 | + $arguments = $this->invokePrivate($mount, 'arguments'); |
|
182 | + |
|
183 | + $objectStore = $arguments['objectstore']; |
|
184 | + $this->assertInstanceOf(FakeObjectStore::class, $objectStore); |
|
185 | + $this->assertEquals(['foo' => 'bar', 'bucket' => 'awesomeBucket1', 'multibucket' => true], $objectStore->getArguments()); |
|
186 | + } |
|
187 | + |
|
188 | + public function testMultiBucketConfigFirst(): void { |
|
189 | + $this->config->method('getSystemValue') |
|
190 | + ->willReturnCallback(function ($key, $default) { |
|
191 | + if ($key === 'objectstore_multibucket') { |
|
192 | + return [ |
|
193 | + 'class' => 'Test\Files\Mount\FakeObjectStore', |
|
194 | + 'arguments' => [ |
|
195 | + 'foo' => 'bar', |
|
196 | + 'bucket' => 'myBucketPrefix', |
|
197 | + ], |
|
198 | + ]; |
|
199 | + } else { |
|
200 | + return $default; |
|
201 | + } |
|
202 | + }); |
|
203 | + |
|
204 | + $this->user->method('getUID') |
|
205 | + ->willReturn('uid'); |
|
206 | + $this->loader->expects($this->never())->method($this->anything()); |
|
207 | + |
|
208 | + $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
209 | + $this->assertInstanceOf('OC\Files\Mount\MountPoint', $mount); |
|
210 | + } |
|
211 | + |
|
212 | + public function testMultiBucketConfigFirstFallBackSingle(): void { |
|
213 | + $this->config |
|
214 | + ->method('getSystemValue')->willReturnMap([ |
|
215 | + ['objectstore_multibucket', null, null], |
|
216 | + ['objectstore', null, [ |
|
217 | + 'class' => 'Test\Files\Mount\FakeObjectStore', |
|
218 | + 'arguments' => [ |
|
219 | + 'foo' => 'bar', |
|
220 | + 'bucket' => 'myBucketPrefix', |
|
221 | + ], |
|
222 | + ]], |
|
223 | + ]); |
|
224 | + |
|
225 | + $this->user->method('getUID') |
|
226 | + ->willReturn('uid'); |
|
227 | + $this->loader->expects($this->never())->method($this->anything()); |
|
228 | + |
|
229 | + $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
230 | + $this->assertInstanceOf('OC\Files\Mount\MountPoint', $mount); |
|
231 | + } |
|
232 | + |
|
233 | + public function testNoObjectStore(): void { |
|
234 | + $this->config->method('getSystemValue') |
|
235 | + ->willReturnCallback(function ($key, $default) { |
|
236 | + return $default; |
|
237 | + }); |
|
238 | + |
|
239 | + $mount = $this->provider->getHomeMountForUser($this->user, $this->loader); |
|
240 | + $this->assertNull($mount); |
|
241 | + } |
|
242 | 242 | } |
243 | 243 | |
244 | 244 | class FakeObjectStore implements IObjectStore { |
245 | - public function __construct( |
|
246 | - private array $arguments, |
|
247 | - ) { |
|
248 | - } |
|
245 | + public function __construct( |
|
246 | + private array $arguments, |
|
247 | + ) { |
|
248 | + } |
|
249 | 249 | |
250 | - public function getArguments() { |
|
251 | - return $this->arguments; |
|
252 | - } |
|
250 | + public function getArguments() { |
|
251 | + return $this->arguments; |
|
252 | + } |
|
253 | 253 | |
254 | - public function getStorageId() { |
|
255 | - } |
|
254 | + public function getStorageId() { |
|
255 | + } |
|
256 | 256 | |
257 | - public function readObject($urn) { |
|
258 | - } |
|
257 | + public function readObject($urn) { |
|
258 | + } |
|
259 | 259 | |
260 | - public function writeObject($urn, $stream, ?string $mimetype = null) { |
|
261 | - } |
|
260 | + public function writeObject($urn, $stream, ?string $mimetype = null) { |
|
261 | + } |
|
262 | 262 | |
263 | - public function deleteObject($urn) { |
|
264 | - } |
|
263 | + public function deleteObject($urn) { |
|
264 | + } |
|
265 | 265 | |
266 | - public function objectExists($urn) { |
|
267 | - } |
|
266 | + public function objectExists($urn) { |
|
267 | + } |
|
268 | 268 | |
269 | - public function copyObject($from, $to) { |
|
270 | - } |
|
269 | + public function copyObject($from, $to) { |
|
270 | + } |
|
271 | 271 | } |