Signed-off-by: Robin Appelman <robin@icewind.nl>pull/52786/head
parent
385dd36ff8
commit
2d4bba7b0c
@ -0,0 +1,98 @@ |
||||
<?php |
||||
|
||||
declare(strict_types=1); |
||||
/** |
||||
* SPDX-FileCopyrightText: 2025 Robin Appelman <robin@icewind.nl> |
||||
* SPDX-License-Identifier: AGPL-3.0-or-later |
||||
*/ |
||||
|
||||
namespace OCA\Files\Command\Object\Multi; |
||||
|
||||
use OC\Core\Command\Base; |
||||
use OC\Files\ObjectStore\PrimaryObjectStoreConfig; |
||||
use OCP\IConfig; |
||||
use OCP\IUser; |
||||
use OCP\IUserManager; |
||||
use Symfony\Component\Console\Input\InputInterface; |
||||
use Symfony\Component\Console\Input\InputOption; |
||||
use Symfony\Component\Console\Output\OutputInterface; |
||||
|
||||
class Users extends Base { |
||||
public function __construct( |
||||
private readonly IUserManager $userManager, |
||||
private readonly PrimaryObjectStoreConfig $objectStoreConfig, |
||||
private readonly IConfig $config, |
||||
) { |
||||
parent::__construct(); |
||||
} |
||||
|
||||
protected function configure(): void { |
||||
parent::configure(); |
||||
$this |
||||
->setName('files:object:multi:users') |
||||
->setDescription('Get the mapping between users and object store buckets') |
||||
->addOption('bucket', 'b', InputOption::VALUE_REQUIRED, 'Only list users using the specified bucket') |
||||
->addOption('object-store', 'o', InputOption::VALUE_REQUIRED, 'Only list users using the specified object store configuration') |
||||
->addOption('user', 'u', InputOption::VALUE_REQUIRED, 'Only show the mapping for the specified user, ignores all other options'); |
||||
} |
||||
|
||||
public function execute(InputInterface $input, OutputInterface $output): int { |
||||
if ($userId = $input->getOption('user')) { |
||||
$user = $this->userManager->get($userId); |
||||
if (!$user) { |
||||
$output->writeln("<error>User $userId not found</error>"); |
||||
return 1; |
||||
} |
||||
$users = new \ArrayIterator([$user]); |
||||
} else { |
||||
$bucket = (string)$input->getOption('bucket'); |
||||
$objectStore = (string)$input->getOption('object-store'); |
||||
if ($bucket !== '' && $objectStore === '') { |
||||
$users = $this->getUsers($this->config->getUsersForUserValue('homeobjectstore', 'bucket', $bucket)); |
||||
} elseif ($bucket === '' && $objectStore !== '') { |
||||
$users = $this->getUsers($this->config->getUsersForUserValue('homeobjectstore', 'objectstore', $objectStore)); |
||||
} elseif ($bucket) { |
||||
$users = $this->getUsers(array_intersect( |
||||
$this->config->getUsersForUserValue('homeobjectstore', 'bucket', $bucket), |
||||
$this->config->getUsersForUserValue('homeobjectstore', 'objectstore', $objectStore) |
||||
)); |
||||
} else { |
||||
$users = $this->userManager->getSeenUsers(); |
||||
} |
||||
} |
||||
|
||||
$this->writeStreamingTableInOutputFormat($input, $output, $this->infoForUsers($users), 100); |
||||
return 0; |
||||
} |
||||
|
||||
/** |
||||
* @param string[] $userIds |
||||
* @return \Iterator<IUser> |
||||
*/ |
||||
private function getUsers(array $userIds): \Iterator { |
||||
foreach ($userIds as $userId) { |
||||
$user = $this->userManager->get($userId); |
||||
if ($user) { |
||||
yield $user; |
||||
} |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* @param \Iterator<IUser> $users |
||||
* @return \Iterator<array> |
||||
*/ |
||||
private function infoForUsers(\Iterator $users): \Iterator { |
||||
foreach ($users as $user) { |
||||
yield $this->infoForUser($user); |
||||
} |
||||
} |
||||
|
||||
private function infoForUser(IUser $user): array { |
||||
return [ |
||||
'user' => $user->getUID(), |
||||
'object-store' => $this->objectStoreConfig->getObjectStoreForUser($user), |
||||
'bucket' => $this->objectStoreConfig->getSetBucketForUser($user) ?? 'unset', |
||||
]; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue