Completed
Pull Request — master (#551)
by Maxence
02:47
created

CirclesMemberships::saveMemberships()   B

Complexity

Conditions 8
Paths 18

Size

Total Lines 40

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
dl 0
loc 40
rs 8.0355
c 0
b 0
f 0
cc 8
nc 18
nop 0
1
<?php
2
3
declare(strict_types=1);
4
5
6
/**
7
 * Circles - Bring cloud-users closer together.
8
 *
9
 * This file is licensed under the Affero General Public License version 3 or
10
 * later. See the COPYING file.
11
 *
12
 * @author Maxence Lange <[email protected]>
13
 * @copyright 2021
14
 * @license GNU AGPL version 3 or any later version
15
 *
16
 * This program is free software: you can redistribute it and/or modify
17
 * it under the terms of the GNU Affero General Public License as
18
 * published by the Free Software Foundation, either version 3 of the
19
 * License, or (at your option) any later version.
20
 *
21
 * This program is distributed in the hope that it will be useful,
22
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
23
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
24
 * GNU Affero General Public License for more details.
25
 *
26
 * You should have received a copy of the GNU Affero General Public License
27
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
28
 *
29
 */
30
31
32
namespace OCA\Circles\Command;
33
34
use daita\MySmallPhpTools\Exceptions\InvalidItemException;
35
use daita\MySmallPhpTools\Exceptions\ItemNotFoundException;
36
use daita\MySmallPhpTools\Exceptions\RequestNetworkException;
37
use daita\MySmallPhpTools\Exceptions\SignatoryException;
38
use daita\MySmallPhpTools\Exceptions\UnknownTypeException;
39
use daita\MySmallPhpTools\Model\Nextcloud\nc21\NC21TreeNode;
40
use daita\MySmallPhpTools\Model\SimpleDataStore;
41
use daita\MySmallPhpTools\Traits\Nextcloud\nc21\TNC21ConsoleTree;
42
use daita\MySmallPhpTools\Traits\TArrayTools;
43
use OC\Core\Command\Base;
44
use OCA\Circles\Db\MemberRequest;
45
use OCA\Circles\Db\MembershipRequest;
46
use OCA\Circles\Exceptions\CircleNotFoundException;
47
use OCA\Circles\Exceptions\FederatedUserException;
48
use OCA\Circles\Exceptions\FederatedUserNotFoundException;
49
use OCA\Circles\Exceptions\InvalidIdException;
50
use OCA\Circles\Exceptions\MemberNotFoundException;
51
use OCA\Circles\Exceptions\OwnerNotFoundException;
52
use OCA\Circles\Exceptions\RemoteInstanceException;
53
use OCA\Circles\Exceptions\RemoteNotFoundException;
54
use OCA\Circles\Exceptions\RemoteResourceNotFoundException;
55
use OCA\Circles\Exceptions\UnknownRemoteException;
56
use OCA\Circles\Exceptions\UserTypeNotFoundException;
57
use OCA\Circles\Model\FederatedUser;
58
use OCA\Circles\Model\Member;
59
use OCA\Circles\Model\Membership;
60
use OCA\Circles\Model\ModelManager;
61
use OCA\Circles\Service\ConfigService;
62
use OCA\Circles\Service\FederatedUserService;
63
use OCP\IUserManager;
64
use Symfony\Component\Console\Input\InputArgument;
65
use Symfony\Component\Console\Input\InputInterface;
66
use Symfony\Component\Console\Input\InputOption;
67
use Symfony\Component\Console\Output\OutputInterface;
68
69
70
/**
71
 * Class CirclesMembershipsIndex
72
 *
73
 * @package OCA\Circles\Command
74
 */
75
class CirclesMemberships extends Base {
76
77
78
	use TArrayTools;
79
	use TNC21ConsoleTree;
80
81
82
	/** @var IUserManager */
83
	private $userManager;
84
85
	/** @var MemberRequest */
86
	private $memberRequest;
87
88
	/** @var MembershipRequest */
89
	private $membershipRequest;
90
91
	/** @var ModelManager */
92
	private $modelManager;
93
94
	/** @var FederatedUserService */
95
	private $federatedUserService;
96
97
	/** @var ConfigService */
98
	private $configService;
99
100
101
	/** @var string */
102
	private $singleId;
103
104
	/** @var Membership[] */
105
	private $memberships = [];
106
107
108
	/**
109
	 * CirclesList constructor.
110
	 *
111
	 * @param IUserManager $userManager
112
	 * @param MembershipRequest $membershipRequest
113
	 * @param MemberRequest $memberRequest
114
	 * @param ModelManager $modelManager
115
	 * @param FederatedUserService $federatedUserService
116
	 * @param ConfigService $configService
117
	 */
118
	public function __construct(
119
		IUserManager $userManager,
120
		MembershipRequest $membershipRequest,
121
		MemberRequest $memberRequest,
122
		ModelManager $modelManager,
123
		FederatedUserService $federatedUserService,
124
		ConfigService $configService
125
	) {
126
		parent::__construct();
127
		$this->userManager = $userManager;
128
		$this->memberRequest = $memberRequest;
129
		$this->membershipRequest = $membershipRequest;
130
		$this->modelManager = $modelManager;
131
		$this->federatedUserService = $federatedUserService;
132
		$this->configService = $configService;
133
	}
134
135
136
	protected function configure() {
137
		parent::configure();
138
		$this->setName('circles:memberships')
139
			 ->setDescription('index and display memberships for local and federated users')
140
			 ->addArgument('userId', InputArgument::OPTIONAL, 'userId to generate memberships', '')
141
			 ->addOption('all', '', InputOption::VALUE_NONE, 'index all local users')
142
			 ->addOption(
143
				 'type', '', InputOption::VALUE_REQUIRED, 'type of the user',
144
				 Member::$DEF_TYPE[Member::TYPE_USER]
145
			 );
146
	}
147
148
149
	/**
150
	 * @param InputInterface $input
151
	 * @param OutputInterface $output
152
	 *
153
	 * @return int
154
	 * @throws CircleNotFoundException
155
	 * @throws InvalidItemException
156
	 * @throws UserTypeNotFoundException
157
	 * @throws FederatedUserException
158
	 * @throws FederatedUserNotFoundException
159
	 * @throws InvalidIdException
160
	 * @throws OwnerNotFoundException
161
	 * @throws RemoteInstanceException
162
	 * @throws RemoteNotFoundException
163
	 * @throws RemoteResourceNotFoundException
164
	 * @throws UnknownRemoteException
165
	 * @throws RequestNetworkException
166
	 * @throws SignatoryException
167
	 * @throws MemberNotFoundException
168
	 */
169
	protected function execute(InputInterface $input, OutputInterface $output): int {
170
		$all = $input->getOption('all');
171
		$userId = $input->getArgument('userId');
172
173
		if (!$all && $userId === '') {
174
			$output->writeln('<error>specify a user, or use --all</error>');
175
176
			return 0;
177
		}
178
179
		$type = Member::parseTypeString($input->getOption('type'));
180
		$federatedUser = $this->federatedUserService->getFederatedUser($userId, (int)$type);
181
182
		$output->writeln('Id: <info>' . $federatedUser->getUserId() . '</info>');
183
		$output->writeln('Instance: <info>' . $federatedUser->getInstance() . '</info>');
184
		$output->writeln('Type: <info>' . Member::$DEF_TYPE[$federatedUser->getUserType()] . '</info>');
185
		$output->writeln('SingleId: <info>' . $federatedUser->getSingleId() . '</info>');
186
187
		$this->singleId = $federatedUser->getSingleId();
188
		$tree = new NC21TreeNode(null, new SimpleDataStore(['federatedUser' => $federatedUser]));
189
		$this->generateMemberships($this->singleId, $tree);
190
		$this->saveMemberships();
191
192
		$output->writeln('Memberships:');
193
		foreach ($federatedUser->getMemberships() as $membership) {
194
			$output->writeln(
195
				'- <info>' . $membership->getCircleId() . '</info> ('
196
				. Member::$DEF_LEVEL[$membership->getLevel()] . ')'
197
			);
198
		}
199
		$output->writeln('');
200
201
		$this->drawTree(
202
			$tree, [$this, 'displayLeaf'],
203
			[
204
				'height'       => 3,
205
				'node-spacing' => 0,
206
				'item-spacing' => 1,
207
			]
208
		);
209
210
		return 0;
211
	}
212
213
214
	/**
215
	 * @param string $id
216
	 * @param NC21TreeNode $tree
217
	 * @param array $knownIds
218
	 */
219
	private function generateMemberships(string $id, NC21TreeNode $tree, array $knownIds = []) {
220
		if (in_array($id, $knownIds)) {
221
			return;
222
		}
223
		$knownIds[] = $id;
224
225
		$members = $this->memberRequest->getMembersBySingleId($id);
226
		foreach ($members as $member) {
227
			$membership = new Membership($member);
228
			$membership->setId($this->singleId);
229
			$this->manageMembership($membership);
230
231
			$item = new NC21TreeNode(
232
				$tree, new SimpleDataStore(
233
						 [
234
							 'member'  => $member,
235
							 'cycling' => in_array($member->getCircleId(), $knownIds)
236
						 ]
237
					 )
238
			);
239
			$this->generateMemberships($member->getCircleId(), $item, $knownIds);
240
		}
241
	}
242
243
244
	/**
245
	 * @param SimpleDataStore $data
246
	 * @param int $lineNumber
247
	 *
248
	 * @return string
249
	 * @throws OwnerNotFoundException
250
	 */
251
	public function displayLeaf(SimpleDataStore $data, int $lineNumber): string {
252
		if ($lineNumber === 3) {
253
			return ($data->gBool('cycling')) ? '<comment>(loop detected)</comment>' : '';
254
		}
255
256
		try {
257
			$line = '';
258
			if ($data->hasKey('federatedUser')) {
259
				/** @var FederatedUser $federatedUser */
260
				$federatedUser = $data->gObj('federatedUser', FederatedUser::class);
261
262
				if ($lineNumber === 2) {
263
					return '';
264
				}
265
				$line .= '<info>' . $federatedUser->getSingleId() . '</info>';
266
				if (!$this->configService->isLocalInstance($federatedUser->getInstance())) {
267
					$line .= '@' . $federatedUser->getInstance();
268
				}
269
270
				return $line;
271
			}
272
273
			if ($data->hasKey('member')) {
274
				/** @var Member $member */
275
				$member = $data->gObj('member', Member::class);
276
				$circle = $member->getCircle();
277
278
				if ($lineNumber === 1) {
279
					$line .= '<info>' . $circle->getId() . '</info>';
280
					if (!$this->configService->isLocalInstance($circle->getInstance())) {
281
						$line .= '@' . $circle->getInstance();
282
					}
283
					$line .= ' (' . $circle->getName() . ')';
284
					$line .= ' <info>Level</info>: ' . Member::$DEF_LEVEL[$member->getLevel()];
285
286
					$knownMembership = $this->memberships[$member->getCircleId()];
287
					if ($member->getLevel() !== $knownMembership->getLevel()) {
288
						$line .= ' (' . Member::$DEF_LEVEL[$knownMembership->getLevel()] . ')';
289
					}
290
				}
291
292
				if ($lineNumber === 2) {
293
					$owner = $circle->getOwner();
294
					$line .= '<info>Owner</info>: ' . $owner->getUserId() . '@' . $owner->getInstance() . ' ';
295
					$type =
296
						implode(", ", $this->modelManager->getCircleTypes($circle, ModelManager::TYPES_LONG));
297
					$line .= ($type === '') ? '' : '<info>Config</info>: ' . $type;
298
				}
299
300
				return $line;
301
			}
302
		} catch (InvalidItemException | ItemNotFoundException | UnknownTypeException $e) {
303
		}
304
305
		return '';
306
	}
307
308
309
	/**
310
	 * @param Membership $membership
311
	 */
312
	private function manageMembership(Membership $membership): void {
313
		foreach ($this->memberships as $known) {
314
			if ($known->getCircleId() === $membership->getCircleId()) {
315
				if ($known->getLevel() < $membership->getLevel()) {
316
					$known->setLevel($membership->getLevel());
317
					$known->setMemberId($membership->getMemberId());
318
				}
319
320
				return;
321
			}
322
		}
323
324
		$this->memberships[$membership->getCircleId()] = $membership;
325
	}
326
327
328
	/**
329
	 *
330
	 */
331
	private function saveMemberships() {
332
		$known = $this->membershipRequest->getMemberships($this->singleId);
333
334
		// Check deprecated memberships
335
		$circleIds = array_map(
336
			function(Membership $membership): string {
337
				return $membership->getCircleId();
338
			}, $this->memberships
339
		);
340
341
		foreach ($known as $item) {
342
			if (!in_array($item->getCircleId(), $circleIds)) {
343
				$this->membershipRequest->delete($item);
344
			}
345
		}
346
347
		// Update/Create memberships
348
		$circleIds = array_map(
349
			function(Membership $membership): string {
350
				return $membership->getCircleId();
351
			}, $known
352
		);
353
354
		foreach ($this->memberships as $item) {
355
			if (!in_array($item->getCircleId(), $circleIds)) {
356
				$this->membershipRequest->insert($item);
357
			} else {
358
				foreach ($known as $knownItem) {
359
					if ($knownItem->getCircleId() === $item->getCircleId()) {
360
						if ($knownItem->getLevel() !== $item->getLevel()) {
361
							$this->membershipRequest->update($item);
362
						}
363
364
						break;
365
					}
366
				}
367
			}
368
		}
369
370
	}
371
372
}
373
374