@@ -57,304 +57,304 @@ |
||
| 57 | 57 | |
| 58 | 58 | class OwnershipTransferService { |
| 59 | 59 | |
| 60 | - /** @var IEncryptionManager */ |
|
| 61 | - private $encryptionManager; |
|
| 62 | - |
|
| 63 | - /** @var IShareManager */ |
|
| 64 | - private $shareManager; |
|
| 65 | - |
|
| 66 | - /** @var IMountManager */ |
|
| 67 | - private $mountManager; |
|
| 68 | - |
|
| 69 | - /** @var IUserMountCache */ |
|
| 70 | - private $userMountCache; |
|
| 71 | - |
|
| 72 | - public function __construct(IEncryptionManager $manager, |
|
| 73 | - IShareManager $shareManager, |
|
| 74 | - IMountManager $mountManager, |
|
| 75 | - IUserMountCache $userMountCache) { |
|
| 76 | - $this->encryptionManager = $manager; |
|
| 77 | - $this->shareManager = $shareManager; |
|
| 78 | - $this->mountManager = $mountManager; |
|
| 79 | - $this->userMountCache = $userMountCache; |
|
| 80 | - } |
|
| 81 | - |
|
| 82 | - /** |
|
| 83 | - * @param IUser $sourceUser |
|
| 84 | - * @param IUser $destinationUser |
|
| 85 | - * @param string $path |
|
| 86 | - * |
|
| 87 | - * @param OutputInterface|null $output |
|
| 88 | - * @param bool $move |
|
| 89 | - * @throws TransferOwnershipException |
|
| 90 | - * @throws \OC\User\NoUserException |
|
| 91 | - */ |
|
| 92 | - public function transfer(IUser $sourceUser, |
|
| 93 | - IUser $destinationUser, |
|
| 94 | - string $path, |
|
| 95 | - ?OutputInterface $output = null, |
|
| 96 | - bool $move = false, |
|
| 97 | - bool $firstLogin = false): void { |
|
| 98 | - $output = $output ?? new NullOutput(); |
|
| 99 | - $sourceUid = $sourceUser->getUID(); |
|
| 100 | - $destinationUid = $destinationUser->getUID(); |
|
| 101 | - $sourcePath = rtrim($sourceUid . '/files/' . $path, '/'); |
|
| 102 | - |
|
| 103 | - // If encryption is on we have to ensure the user has logged in before and that all encryption modules are ready |
|
| 104 | - if (($this->encryptionManager->isEnabled() && $destinationUser->getLastLogin() === 0) |
|
| 105 | - || !$this->encryptionManager->isReadyForUser($destinationUid)) { |
|
| 106 | - throw new TransferOwnershipException("The target user is not ready to accept files. The user has at least to have logged in once.", 2); |
|
| 107 | - } |
|
| 108 | - |
|
| 109 | - // setup filesystem |
|
| 110 | - // Requesting the user folder will set it up if the user hasn't logged in before |
|
| 111 | - \OC::$server->getUserFolder($destinationUser->getUID()); |
|
| 112 | - Filesystem::initMountPoints($sourceUid); |
|
| 113 | - Filesystem::initMountPoints($destinationUid); |
|
| 114 | - |
|
| 115 | - $view = new View(); |
|
| 116 | - |
|
| 117 | - if ($move) { |
|
| 118 | - $finalTarget = "$destinationUid/files/"; |
|
| 119 | - } else { |
|
| 120 | - $date = date('Y-m-d H-i-s'); |
|
| 121 | - |
|
| 122 | - // Remove some characters which are prone to cause errors |
|
| 123 | - $cleanUserName = str_replace(['\\', '/', ':', '.', '?', '#', '\'', '"'], '-', $sourceUser->getDisplayName()); |
|
| 124 | - // Replace multiple dashes with one dash |
|
| 125 | - $cleanUserName = preg_replace('/-{2,}/s', '-', $cleanUserName); |
|
| 126 | - $cleanUserName = $cleanUserName ?: $sourceUid; |
|
| 127 | - |
|
| 128 | - $finalTarget = "$destinationUid/files/transferred from $cleanUserName on $date"; |
|
| 129 | - try { |
|
| 130 | - $view->verifyPath(dirname($finalTarget), basename($finalTarget)); |
|
| 131 | - } catch (InvalidPathException $e) { |
|
| 132 | - $finalTarget = "$destinationUid/files/transferred from $sourceUid on $date"; |
|
| 133 | - } |
|
| 134 | - } |
|
| 135 | - |
|
| 136 | - if (!($view->is_dir($sourcePath) || $view->is_file($sourcePath))) { |
|
| 137 | - throw new TransferOwnershipException("Unknown path provided: $path", 1); |
|
| 138 | - } |
|
| 139 | - |
|
| 140 | - if ($move && ( |
|
| 141 | - !$view->is_dir($finalTarget) || ( |
|
| 142 | - !$firstLogin && |
|
| 143 | - count($view->getDirectoryContent($finalTarget)) > 0 |
|
| 144 | - ) |
|
| 145 | - ) |
|
| 146 | - ) { |
|
| 147 | - throw new TransferOwnershipException("Destination path does not exists or is not empty", 1); |
|
| 148 | - } |
|
| 149 | - |
|
| 150 | - |
|
| 151 | - // analyse source folder |
|
| 152 | - $this->analyse( |
|
| 153 | - $sourceUid, |
|
| 154 | - $destinationUid, |
|
| 155 | - $sourcePath, |
|
| 156 | - $view, |
|
| 157 | - $output |
|
| 158 | - ); |
|
| 159 | - |
|
| 160 | - // collect all the shares |
|
| 161 | - $shares = $this->collectUsersShares( |
|
| 162 | - $sourceUid, |
|
| 163 | - $output, |
|
| 164 | - $view, |
|
| 165 | - $sourcePath |
|
| 166 | - ); |
|
| 167 | - |
|
| 168 | - // transfer the files |
|
| 169 | - $this->transferFiles( |
|
| 170 | - $sourceUid, |
|
| 171 | - $sourcePath, |
|
| 172 | - $finalTarget, |
|
| 173 | - $view, |
|
| 174 | - $output |
|
| 175 | - ); |
|
| 176 | - |
|
| 177 | - // restore the shares |
|
| 178 | - $this->restoreShares( |
|
| 179 | - $sourceUid, |
|
| 180 | - $destinationUid, |
|
| 181 | - $shares, |
|
| 182 | - $output |
|
| 183 | - ); |
|
| 184 | - } |
|
| 185 | - |
|
| 186 | - private function walkFiles(View $view, $path, Closure $callBack) { |
|
| 187 | - foreach ($view->getDirectoryContent($path) as $fileInfo) { |
|
| 188 | - if (!$callBack($fileInfo)) { |
|
| 189 | - return; |
|
| 190 | - } |
|
| 191 | - if ($fileInfo->getType() === FileInfo::TYPE_FOLDER) { |
|
| 192 | - $this->walkFiles($view, $fileInfo->getPath(), $callBack); |
|
| 193 | - } |
|
| 194 | - } |
|
| 195 | - } |
|
| 196 | - |
|
| 197 | - /** |
|
| 198 | - * @param OutputInterface $output |
|
| 199 | - * |
|
| 200 | - * @throws \Exception |
|
| 201 | - */ |
|
| 202 | - protected function analyse(string $sourceUid, |
|
| 203 | - string $destinationUid, |
|
| 204 | - string $sourcePath, |
|
| 205 | - View $view, |
|
| 206 | - OutputInterface $output): void { |
|
| 207 | - $output->writeln('Validating quota'); |
|
| 208 | - $size = $view->getFileInfo($sourcePath, false)->getSize(false); |
|
| 209 | - $freeSpace = $view->free_space($destinationUid . '/files/'); |
|
| 210 | - if ($size > $freeSpace && $freeSpace !== FileInfo::SPACE_UNKNOWN) { |
|
| 211 | - $output->writeln('<error>Target user does not have enough free space available.</error>'); |
|
| 212 | - throw new \Exception('Execution terminated.'); |
|
| 213 | - } |
|
| 214 | - |
|
| 215 | - $output->writeln("Analysing files of $sourceUid ..."); |
|
| 216 | - $progress = new ProgressBar($output); |
|
| 217 | - $progress->start(); |
|
| 218 | - |
|
| 219 | - $encryptedFiles = []; |
|
| 220 | - $this->walkFiles($view, $sourcePath, |
|
| 221 | - function (FileInfo $fileInfo) use ($progress) { |
|
| 222 | - if ($fileInfo->getType() === FileInfo::TYPE_FOLDER) { |
|
| 223 | - // only analyze into folders from main storage, |
|
| 224 | - if (!$fileInfo->getStorage()->instanceOfStorage(IHomeStorage::class)) { |
|
| 225 | - return false; |
|
| 226 | - } |
|
| 227 | - return true; |
|
| 228 | - } |
|
| 229 | - $progress->advance(); |
|
| 230 | - if ($fileInfo->isEncrypted()) { |
|
| 231 | - $encryptedFiles[] = $fileInfo; |
|
| 232 | - } |
|
| 233 | - return true; |
|
| 234 | - }); |
|
| 235 | - $progress->finish(); |
|
| 236 | - $output->writeln(''); |
|
| 237 | - |
|
| 238 | - // no file is allowed to be encrypted |
|
| 239 | - if (!empty($encryptedFiles)) { |
|
| 240 | - $output->writeln("<error>Some files are encrypted - please decrypt them first.</error>"); |
|
| 241 | - foreach ($encryptedFiles as $encryptedFile) { |
|
| 242 | - /** @var FileInfo $encryptedFile */ |
|
| 243 | - $output->writeln(" " . $encryptedFile->getPath()); |
|
| 244 | - } |
|
| 245 | - throw new \Exception('Execution terminated.'); |
|
| 246 | - } |
|
| 247 | - } |
|
| 248 | - |
|
| 249 | - private function collectUsersShares(string $sourceUid, |
|
| 250 | - OutputInterface $output, |
|
| 251 | - View $view, |
|
| 252 | - string $path): array { |
|
| 253 | - $output->writeln("Collecting all share information for files and folders of $sourceUid ..."); |
|
| 254 | - |
|
| 255 | - $shares = []; |
|
| 256 | - $progress = new ProgressBar($output); |
|
| 257 | - foreach ([IShare::TYPE_GROUP, IShare::TYPE_USER, IShare::TYPE_LINK, IShare::TYPE_REMOTE, IShare::TYPE_ROOM, IShare::TYPE_EMAIL, IShare::TYPE_CIRCLE, IShare::TYPE_DECK] as $shareType) { |
|
| 258 | - $offset = 0; |
|
| 259 | - while (true) { |
|
| 260 | - $sharePage = $this->shareManager->getSharesBy($sourceUid, $shareType, null, true, 50, $offset); |
|
| 261 | - $progress->advance(count($sharePage)); |
|
| 262 | - if (empty($sharePage)) { |
|
| 263 | - break; |
|
| 264 | - } |
|
| 265 | - if ($path !== "$sourceUid/files") { |
|
| 266 | - $sharePage = array_filter($sharePage, function (IShare $share) use ($view, $path) { |
|
| 267 | - try { |
|
| 268 | - $relativePath = $view->getPath($share->getNodeId()); |
|
| 269 | - $singleFileTranfer = $view->is_file($path); |
|
| 270 | - if ($singleFileTranfer) { |
|
| 271 | - return Filesystem::normalizePath($relativePath) === Filesystem::normalizePath($path); |
|
| 272 | - } |
|
| 273 | - |
|
| 274 | - return mb_strpos( |
|
| 275 | - Filesystem::normalizePath($relativePath . '/', false), |
|
| 276 | - Filesystem::normalizePath($path . '/', false)) === 0; |
|
| 277 | - } catch (\Exception $e) { |
|
| 278 | - return false; |
|
| 279 | - } |
|
| 280 | - }); |
|
| 281 | - } |
|
| 282 | - $shares = array_merge($shares, $sharePage); |
|
| 283 | - $offset += 50; |
|
| 284 | - } |
|
| 285 | - } |
|
| 286 | - |
|
| 287 | - $progress->finish(); |
|
| 288 | - $output->writeln(''); |
|
| 289 | - return $shares; |
|
| 290 | - } |
|
| 291 | - |
|
| 292 | - /** |
|
| 293 | - * @throws TransferOwnershipException |
|
| 294 | - */ |
|
| 295 | - protected function transferFiles(string $sourceUid, |
|
| 296 | - string $sourcePath, |
|
| 297 | - string $finalTarget, |
|
| 298 | - View $view, |
|
| 299 | - OutputInterface $output): void { |
|
| 300 | - $output->writeln("Transferring files to $finalTarget ..."); |
|
| 301 | - |
|
| 302 | - // This change will help user to transfer the folder specified using --path option. |
|
| 303 | - // Else only the content inside folder is transferred which is not correct. |
|
| 304 | - if ($sourcePath !== "$sourceUid/files") { |
|
| 305 | - $view->mkdir($finalTarget); |
|
| 306 | - $finalTarget = $finalTarget . '/' . basename($sourcePath); |
|
| 307 | - } |
|
| 308 | - if ($view->rename($sourcePath, $finalTarget) === false) { |
|
| 309 | - throw new TransferOwnershipException("Could not transfer files.", 1); |
|
| 310 | - } |
|
| 311 | - if (!is_dir("$sourceUid/files")) { |
|
| 312 | - // because the files folder is moved away we need to recreate it |
|
| 313 | - $view->mkdir("$sourceUid/files"); |
|
| 314 | - } |
|
| 315 | - } |
|
| 316 | - |
|
| 317 | - private function restoreShares(string $sourceUid, |
|
| 318 | - string $destinationUid, |
|
| 319 | - array $shares, |
|
| 320 | - OutputInterface $output) { |
|
| 321 | - $output->writeln("Restoring shares ..."); |
|
| 322 | - $progress = new ProgressBar($output, count($shares)); |
|
| 323 | - |
|
| 324 | - foreach ($shares as $share) { |
|
| 325 | - try { |
|
| 326 | - if ($share->getShareType() === IShare::TYPE_USER && |
|
| 327 | - $share->getSharedWith() === $destinationUid) { |
|
| 328 | - // Unmount the shares before deleting, so we don't try to get the storage later on. |
|
| 329 | - $shareMountPoint = $this->mountManager->find('/' . $destinationUid . '/files' . $share->getTarget()); |
|
| 330 | - if ($shareMountPoint) { |
|
| 331 | - $this->mountManager->removeMount($shareMountPoint->getMountPoint()); |
|
| 332 | - } |
|
| 333 | - $this->shareManager->deleteShare($share); |
|
| 334 | - } else { |
|
| 335 | - if ($share->getShareOwner() === $sourceUid) { |
|
| 336 | - $share->setShareOwner($destinationUid); |
|
| 337 | - } |
|
| 338 | - if ($share->getSharedBy() === $sourceUid) { |
|
| 339 | - $share->setSharedBy($destinationUid); |
|
| 340 | - } |
|
| 341 | - |
|
| 342 | - |
|
| 343 | - // trigger refetching of the node so that the new owner and mountpoint are taken into account |
|
| 344 | - // otherwise the checks on the share update will fail due to the original node not being available in the new user scope |
|
| 345 | - $this->userMountCache->clear(); |
|
| 346 | - $share->setNodeId($share->getNode()->getId()); |
|
| 347 | - |
|
| 348 | - $this->shareManager->updateShare($share); |
|
| 349 | - } |
|
| 350 | - } catch (\OCP\Files\NotFoundException $e) { |
|
| 351 | - $output->writeln('<error>Share with id ' . $share->getId() . ' points at deleted file, skipping</error>'); |
|
| 352 | - } catch (\Throwable $e) { |
|
| 353 | - $output->writeln('<error>Could not restore share with id ' . $share->getId() . ':' . $e->getTraceAsString() . '</error>'); |
|
| 354 | - } |
|
| 355 | - $progress->advance(); |
|
| 356 | - } |
|
| 357 | - $progress->finish(); |
|
| 358 | - $output->writeln(''); |
|
| 359 | - } |
|
| 60 | + /** @var IEncryptionManager */ |
|
| 61 | + private $encryptionManager; |
|
| 62 | + |
|
| 63 | + /** @var IShareManager */ |
|
| 64 | + private $shareManager; |
|
| 65 | + |
|
| 66 | + /** @var IMountManager */ |
|
| 67 | + private $mountManager; |
|
| 68 | + |
|
| 69 | + /** @var IUserMountCache */ |
|
| 70 | + private $userMountCache; |
|
| 71 | + |
|
| 72 | + public function __construct(IEncryptionManager $manager, |
|
| 73 | + IShareManager $shareManager, |
|
| 74 | + IMountManager $mountManager, |
|
| 75 | + IUserMountCache $userMountCache) { |
|
| 76 | + $this->encryptionManager = $manager; |
|
| 77 | + $this->shareManager = $shareManager; |
|
| 78 | + $this->mountManager = $mountManager; |
|
| 79 | + $this->userMountCache = $userMountCache; |
|
| 80 | + } |
|
| 81 | + |
|
| 82 | + /** |
|
| 83 | + * @param IUser $sourceUser |
|
| 84 | + * @param IUser $destinationUser |
|
| 85 | + * @param string $path |
|
| 86 | + * |
|
| 87 | + * @param OutputInterface|null $output |
|
| 88 | + * @param bool $move |
|
| 89 | + * @throws TransferOwnershipException |
|
| 90 | + * @throws \OC\User\NoUserException |
|
| 91 | + */ |
|
| 92 | + public function transfer(IUser $sourceUser, |
|
| 93 | + IUser $destinationUser, |
|
| 94 | + string $path, |
|
| 95 | + ?OutputInterface $output = null, |
|
| 96 | + bool $move = false, |
|
| 97 | + bool $firstLogin = false): void { |
|
| 98 | + $output = $output ?? new NullOutput(); |
|
| 99 | + $sourceUid = $sourceUser->getUID(); |
|
| 100 | + $destinationUid = $destinationUser->getUID(); |
|
| 101 | + $sourcePath = rtrim($sourceUid . '/files/' . $path, '/'); |
|
| 102 | + |
|
| 103 | + // If encryption is on we have to ensure the user has logged in before and that all encryption modules are ready |
|
| 104 | + if (($this->encryptionManager->isEnabled() && $destinationUser->getLastLogin() === 0) |
|
| 105 | + || !$this->encryptionManager->isReadyForUser($destinationUid)) { |
|
| 106 | + throw new TransferOwnershipException("The target user is not ready to accept files. The user has at least to have logged in once.", 2); |
|
| 107 | + } |
|
| 108 | + |
|
| 109 | + // setup filesystem |
|
| 110 | + // Requesting the user folder will set it up if the user hasn't logged in before |
|
| 111 | + \OC::$server->getUserFolder($destinationUser->getUID()); |
|
| 112 | + Filesystem::initMountPoints($sourceUid); |
|
| 113 | + Filesystem::initMountPoints($destinationUid); |
|
| 114 | + |
|
| 115 | + $view = new View(); |
|
| 116 | + |
|
| 117 | + if ($move) { |
|
| 118 | + $finalTarget = "$destinationUid/files/"; |
|
| 119 | + } else { |
|
| 120 | + $date = date('Y-m-d H-i-s'); |
|
| 121 | + |
|
| 122 | + // Remove some characters which are prone to cause errors |
|
| 123 | + $cleanUserName = str_replace(['\\', '/', ':', '.', '?', '#', '\'', '"'], '-', $sourceUser->getDisplayName()); |
|
| 124 | + // Replace multiple dashes with one dash |
|
| 125 | + $cleanUserName = preg_replace('/-{2,}/s', '-', $cleanUserName); |
|
| 126 | + $cleanUserName = $cleanUserName ?: $sourceUid; |
|
| 127 | + |
|
| 128 | + $finalTarget = "$destinationUid/files/transferred from $cleanUserName on $date"; |
|
| 129 | + try { |
|
| 130 | + $view->verifyPath(dirname($finalTarget), basename($finalTarget)); |
|
| 131 | + } catch (InvalidPathException $e) { |
|
| 132 | + $finalTarget = "$destinationUid/files/transferred from $sourceUid on $date"; |
|
| 133 | + } |
|
| 134 | + } |
|
| 135 | + |
|
| 136 | + if (!($view->is_dir($sourcePath) || $view->is_file($sourcePath))) { |
|
| 137 | + throw new TransferOwnershipException("Unknown path provided: $path", 1); |
|
| 138 | + } |
|
| 139 | + |
|
| 140 | + if ($move && ( |
|
| 141 | + !$view->is_dir($finalTarget) || ( |
|
| 142 | + !$firstLogin && |
|
| 143 | + count($view->getDirectoryContent($finalTarget)) > 0 |
|
| 144 | + ) |
|
| 145 | + ) |
|
| 146 | + ) { |
|
| 147 | + throw new TransferOwnershipException("Destination path does not exists or is not empty", 1); |
|
| 148 | + } |
|
| 149 | + |
|
| 150 | + |
|
| 151 | + // analyse source folder |
|
| 152 | + $this->analyse( |
|
| 153 | + $sourceUid, |
|
| 154 | + $destinationUid, |
|
| 155 | + $sourcePath, |
|
| 156 | + $view, |
|
| 157 | + $output |
|
| 158 | + ); |
|
| 159 | + |
|
| 160 | + // collect all the shares |
|
| 161 | + $shares = $this->collectUsersShares( |
|
| 162 | + $sourceUid, |
|
| 163 | + $output, |
|
| 164 | + $view, |
|
| 165 | + $sourcePath |
|
| 166 | + ); |
|
| 167 | + |
|
| 168 | + // transfer the files |
|
| 169 | + $this->transferFiles( |
|
| 170 | + $sourceUid, |
|
| 171 | + $sourcePath, |
|
| 172 | + $finalTarget, |
|
| 173 | + $view, |
|
| 174 | + $output |
|
| 175 | + ); |
|
| 176 | + |
|
| 177 | + // restore the shares |
|
| 178 | + $this->restoreShares( |
|
| 179 | + $sourceUid, |
|
| 180 | + $destinationUid, |
|
| 181 | + $shares, |
|
| 182 | + $output |
|
| 183 | + ); |
|
| 184 | + } |
|
| 185 | + |
|
| 186 | + private function walkFiles(View $view, $path, Closure $callBack) { |
|
| 187 | + foreach ($view->getDirectoryContent($path) as $fileInfo) { |
|
| 188 | + if (!$callBack($fileInfo)) { |
|
| 189 | + return; |
|
| 190 | + } |
|
| 191 | + if ($fileInfo->getType() === FileInfo::TYPE_FOLDER) { |
|
| 192 | + $this->walkFiles($view, $fileInfo->getPath(), $callBack); |
|
| 193 | + } |
|
| 194 | + } |
|
| 195 | + } |
|
| 196 | + |
|
| 197 | + /** |
|
| 198 | + * @param OutputInterface $output |
|
| 199 | + * |
|
| 200 | + * @throws \Exception |
|
| 201 | + */ |
|
| 202 | + protected function analyse(string $sourceUid, |
|
| 203 | + string $destinationUid, |
|
| 204 | + string $sourcePath, |
|
| 205 | + View $view, |
|
| 206 | + OutputInterface $output): void { |
|
| 207 | + $output->writeln('Validating quota'); |
|
| 208 | + $size = $view->getFileInfo($sourcePath, false)->getSize(false); |
|
| 209 | + $freeSpace = $view->free_space($destinationUid . '/files/'); |
|
| 210 | + if ($size > $freeSpace && $freeSpace !== FileInfo::SPACE_UNKNOWN) { |
|
| 211 | + $output->writeln('<error>Target user does not have enough free space available.</error>'); |
|
| 212 | + throw new \Exception('Execution terminated.'); |
|
| 213 | + } |
|
| 214 | + |
|
| 215 | + $output->writeln("Analysing files of $sourceUid ..."); |
|
| 216 | + $progress = new ProgressBar($output); |
|
| 217 | + $progress->start(); |
|
| 218 | + |
|
| 219 | + $encryptedFiles = []; |
|
| 220 | + $this->walkFiles($view, $sourcePath, |
|
| 221 | + function (FileInfo $fileInfo) use ($progress) { |
|
| 222 | + if ($fileInfo->getType() === FileInfo::TYPE_FOLDER) { |
|
| 223 | + // only analyze into folders from main storage, |
|
| 224 | + if (!$fileInfo->getStorage()->instanceOfStorage(IHomeStorage::class)) { |
|
| 225 | + return false; |
|
| 226 | + } |
|
| 227 | + return true; |
|
| 228 | + } |
|
| 229 | + $progress->advance(); |
|
| 230 | + if ($fileInfo->isEncrypted()) { |
|
| 231 | + $encryptedFiles[] = $fileInfo; |
|
| 232 | + } |
|
| 233 | + return true; |
|
| 234 | + }); |
|
| 235 | + $progress->finish(); |
|
| 236 | + $output->writeln(''); |
|
| 237 | + |
|
| 238 | + // no file is allowed to be encrypted |
|
| 239 | + if (!empty($encryptedFiles)) { |
|
| 240 | + $output->writeln("<error>Some files are encrypted - please decrypt them first.</error>"); |
|
| 241 | + foreach ($encryptedFiles as $encryptedFile) { |
|
| 242 | + /** @var FileInfo $encryptedFile */ |
|
| 243 | + $output->writeln(" " . $encryptedFile->getPath()); |
|
| 244 | + } |
|
| 245 | + throw new \Exception('Execution terminated.'); |
|
| 246 | + } |
|
| 247 | + } |
|
| 248 | + |
|
| 249 | + private function collectUsersShares(string $sourceUid, |
|
| 250 | + OutputInterface $output, |
|
| 251 | + View $view, |
|
| 252 | + string $path): array { |
|
| 253 | + $output->writeln("Collecting all share information for files and folders of $sourceUid ..."); |
|
| 254 | + |
|
| 255 | + $shares = []; |
|
| 256 | + $progress = new ProgressBar($output); |
|
| 257 | + foreach ([IShare::TYPE_GROUP, IShare::TYPE_USER, IShare::TYPE_LINK, IShare::TYPE_REMOTE, IShare::TYPE_ROOM, IShare::TYPE_EMAIL, IShare::TYPE_CIRCLE, IShare::TYPE_DECK] as $shareType) { |
|
| 258 | + $offset = 0; |
|
| 259 | + while (true) { |
|
| 260 | + $sharePage = $this->shareManager->getSharesBy($sourceUid, $shareType, null, true, 50, $offset); |
|
| 261 | + $progress->advance(count($sharePage)); |
|
| 262 | + if (empty($sharePage)) { |
|
| 263 | + break; |
|
| 264 | + } |
|
| 265 | + if ($path !== "$sourceUid/files") { |
|
| 266 | + $sharePage = array_filter($sharePage, function (IShare $share) use ($view, $path) { |
|
| 267 | + try { |
|
| 268 | + $relativePath = $view->getPath($share->getNodeId()); |
|
| 269 | + $singleFileTranfer = $view->is_file($path); |
|
| 270 | + if ($singleFileTranfer) { |
|
| 271 | + return Filesystem::normalizePath($relativePath) === Filesystem::normalizePath($path); |
|
| 272 | + } |
|
| 273 | + |
|
| 274 | + return mb_strpos( |
|
| 275 | + Filesystem::normalizePath($relativePath . '/', false), |
|
| 276 | + Filesystem::normalizePath($path . '/', false)) === 0; |
|
| 277 | + } catch (\Exception $e) { |
|
| 278 | + return false; |
|
| 279 | + } |
|
| 280 | + }); |
|
| 281 | + } |
|
| 282 | + $shares = array_merge($shares, $sharePage); |
|
| 283 | + $offset += 50; |
|
| 284 | + } |
|
| 285 | + } |
|
| 286 | + |
|
| 287 | + $progress->finish(); |
|
| 288 | + $output->writeln(''); |
|
| 289 | + return $shares; |
|
| 290 | + } |
|
| 291 | + |
|
| 292 | + /** |
|
| 293 | + * @throws TransferOwnershipException |
|
| 294 | + */ |
|
| 295 | + protected function transferFiles(string $sourceUid, |
|
| 296 | + string $sourcePath, |
|
| 297 | + string $finalTarget, |
|
| 298 | + View $view, |
|
| 299 | + OutputInterface $output): void { |
|
| 300 | + $output->writeln("Transferring files to $finalTarget ..."); |
|
| 301 | + |
|
| 302 | + // This change will help user to transfer the folder specified using --path option. |
|
| 303 | + // Else only the content inside folder is transferred which is not correct. |
|
| 304 | + if ($sourcePath !== "$sourceUid/files") { |
|
| 305 | + $view->mkdir($finalTarget); |
|
| 306 | + $finalTarget = $finalTarget . '/' . basename($sourcePath); |
|
| 307 | + } |
|
| 308 | + if ($view->rename($sourcePath, $finalTarget) === false) { |
|
| 309 | + throw new TransferOwnershipException("Could not transfer files.", 1); |
|
| 310 | + } |
|
| 311 | + if (!is_dir("$sourceUid/files")) { |
|
| 312 | + // because the files folder is moved away we need to recreate it |
|
| 313 | + $view->mkdir("$sourceUid/files"); |
|
| 314 | + } |
|
| 315 | + } |
|
| 316 | + |
|
| 317 | + private function restoreShares(string $sourceUid, |
|
| 318 | + string $destinationUid, |
|
| 319 | + array $shares, |
|
| 320 | + OutputInterface $output) { |
|
| 321 | + $output->writeln("Restoring shares ..."); |
|
| 322 | + $progress = new ProgressBar($output, count($shares)); |
|
| 323 | + |
|
| 324 | + foreach ($shares as $share) { |
|
| 325 | + try { |
|
| 326 | + if ($share->getShareType() === IShare::TYPE_USER && |
|
| 327 | + $share->getSharedWith() === $destinationUid) { |
|
| 328 | + // Unmount the shares before deleting, so we don't try to get the storage later on. |
|
| 329 | + $shareMountPoint = $this->mountManager->find('/' . $destinationUid . '/files' . $share->getTarget()); |
|
| 330 | + if ($shareMountPoint) { |
|
| 331 | + $this->mountManager->removeMount($shareMountPoint->getMountPoint()); |
|
| 332 | + } |
|
| 333 | + $this->shareManager->deleteShare($share); |
|
| 334 | + } else { |
|
| 335 | + if ($share->getShareOwner() === $sourceUid) { |
|
| 336 | + $share->setShareOwner($destinationUid); |
|
| 337 | + } |
|
| 338 | + if ($share->getSharedBy() === $sourceUid) { |
|
| 339 | + $share->setSharedBy($destinationUid); |
|
| 340 | + } |
|
| 341 | + |
|
| 342 | + |
|
| 343 | + // trigger refetching of the node so that the new owner and mountpoint are taken into account |
|
| 344 | + // otherwise the checks on the share update will fail due to the original node not being available in the new user scope |
|
| 345 | + $this->userMountCache->clear(); |
|
| 346 | + $share->setNodeId($share->getNode()->getId()); |
|
| 347 | + |
|
| 348 | + $this->shareManager->updateShare($share); |
|
| 349 | + } |
|
| 350 | + } catch (\OCP\Files\NotFoundException $e) { |
|
| 351 | + $output->writeln('<error>Share with id ' . $share->getId() . ' points at deleted file, skipping</error>'); |
|
| 352 | + } catch (\Throwable $e) { |
|
| 353 | + $output->writeln('<error>Could not restore share with id ' . $share->getId() . ':' . $e->getTraceAsString() . '</error>'); |
|
| 354 | + } |
|
| 355 | + $progress->advance(); |
|
| 356 | + } |
|
| 357 | + $progress->finish(); |
|
| 358 | + $output->writeln(''); |
|
| 359 | + } |
|
| 360 | 360 | } |
@@ -98,7 +98,7 @@ discard block |
||
| 98 | 98 | $output = $output ?? new NullOutput(); |
| 99 | 99 | $sourceUid = $sourceUser->getUID(); |
| 100 | 100 | $destinationUid = $destinationUser->getUID(); |
| 101 | - $sourcePath = rtrim($sourceUid . '/files/' . $path, '/'); |
|
| 101 | + $sourcePath = rtrim($sourceUid.'/files/'.$path, '/'); |
|
| 102 | 102 | |
| 103 | 103 | // If encryption is on we have to ensure the user has logged in before and that all encryption modules are ready |
| 104 | 104 | if (($this->encryptionManager->isEnabled() && $destinationUser->getLastLogin() === 0) |
@@ -206,7 +206,7 @@ discard block |
||
| 206 | 206 | OutputInterface $output): void { |
| 207 | 207 | $output->writeln('Validating quota'); |
| 208 | 208 | $size = $view->getFileInfo($sourcePath, false)->getSize(false); |
| 209 | - $freeSpace = $view->free_space($destinationUid . '/files/'); |
|
| 209 | + $freeSpace = $view->free_space($destinationUid.'/files/'); |
|
| 210 | 210 | if ($size > $freeSpace && $freeSpace !== FileInfo::SPACE_UNKNOWN) { |
| 211 | 211 | $output->writeln('<error>Target user does not have enough free space available.</error>'); |
| 212 | 212 | throw new \Exception('Execution terminated.'); |
@@ -218,7 +218,7 @@ discard block |
||
| 218 | 218 | |
| 219 | 219 | $encryptedFiles = []; |
| 220 | 220 | $this->walkFiles($view, $sourcePath, |
| 221 | - function (FileInfo $fileInfo) use ($progress) { |
|
| 221 | + function(FileInfo $fileInfo) use ($progress) { |
|
| 222 | 222 | if ($fileInfo->getType() === FileInfo::TYPE_FOLDER) { |
| 223 | 223 | // only analyze into folders from main storage, |
| 224 | 224 | if (!$fileInfo->getStorage()->instanceOfStorage(IHomeStorage::class)) { |
@@ -240,7 +240,7 @@ discard block |
||
| 240 | 240 | $output->writeln("<error>Some files are encrypted - please decrypt them first.</error>"); |
| 241 | 241 | foreach ($encryptedFiles as $encryptedFile) { |
| 242 | 242 | /** @var FileInfo $encryptedFile */ |
| 243 | - $output->writeln(" " . $encryptedFile->getPath()); |
|
| 243 | + $output->writeln(" ".$encryptedFile->getPath()); |
|
| 244 | 244 | } |
| 245 | 245 | throw new \Exception('Execution terminated.'); |
| 246 | 246 | } |
@@ -263,7 +263,7 @@ discard block |
||
| 263 | 263 | break; |
| 264 | 264 | } |
| 265 | 265 | if ($path !== "$sourceUid/files") { |
| 266 | - $sharePage = array_filter($sharePage, function (IShare $share) use ($view, $path) { |
|
| 266 | + $sharePage = array_filter($sharePage, function(IShare $share) use ($view, $path) { |
|
| 267 | 267 | try { |
| 268 | 268 | $relativePath = $view->getPath($share->getNodeId()); |
| 269 | 269 | $singleFileTranfer = $view->is_file($path); |
@@ -272,8 +272,8 @@ discard block |
||
| 272 | 272 | } |
| 273 | 273 | |
| 274 | 274 | return mb_strpos( |
| 275 | - Filesystem::normalizePath($relativePath . '/', false), |
|
| 276 | - Filesystem::normalizePath($path . '/', false)) === 0; |
|
| 275 | + Filesystem::normalizePath($relativePath.'/', false), |
|
| 276 | + Filesystem::normalizePath($path.'/', false)) === 0; |
|
| 277 | 277 | } catch (\Exception $e) { |
| 278 | 278 | return false; |
| 279 | 279 | } |
@@ -303,7 +303,7 @@ discard block |
||
| 303 | 303 | // Else only the content inside folder is transferred which is not correct. |
| 304 | 304 | if ($sourcePath !== "$sourceUid/files") { |
| 305 | 305 | $view->mkdir($finalTarget); |
| 306 | - $finalTarget = $finalTarget . '/' . basename($sourcePath); |
|
| 306 | + $finalTarget = $finalTarget.'/'.basename($sourcePath); |
|
| 307 | 307 | } |
| 308 | 308 | if ($view->rename($sourcePath, $finalTarget) === false) { |
| 309 | 309 | throw new TransferOwnershipException("Could not transfer files.", 1); |
@@ -326,7 +326,7 @@ discard block |
||
| 326 | 326 | if ($share->getShareType() === IShare::TYPE_USER && |
| 327 | 327 | $share->getSharedWith() === $destinationUid) { |
| 328 | 328 | // Unmount the shares before deleting, so we don't try to get the storage later on. |
| 329 | - $shareMountPoint = $this->mountManager->find('/' . $destinationUid . '/files' . $share->getTarget()); |
|
| 329 | + $shareMountPoint = $this->mountManager->find('/'.$destinationUid.'/files'.$share->getTarget()); |
|
| 330 | 330 | if ($shareMountPoint) { |
| 331 | 331 | $this->mountManager->removeMount($shareMountPoint->getMountPoint()); |
| 332 | 332 | } |
@@ -348,9 +348,9 @@ discard block |
||
| 348 | 348 | $this->shareManager->updateShare($share); |
| 349 | 349 | } |
| 350 | 350 | } catch (\OCP\Files\NotFoundException $e) { |
| 351 | - $output->writeln('<error>Share with id ' . $share->getId() . ' points at deleted file, skipping</error>'); |
|
| 351 | + $output->writeln('<error>Share with id '.$share->getId().' points at deleted file, skipping</error>'); |
|
| 352 | 352 | } catch (\Throwable $e) { |
| 353 | - $output->writeln('<error>Could not restore share with id ' . $share->getId() . ':' . $e->getTraceAsString() . '</error>'); |
|
| 353 | + $output->writeln('<error>Could not restore share with id '.$share->getId().':'.$e->getTraceAsString().'</error>'); |
|
| 354 | 354 | } |
| 355 | 355 | $progress->advance(); |
| 356 | 356 | } |