Code Coverage for /workspace/src/extensions/IPInfo/src/InfoRetriever/IPCountInfoRetriever.php

1	<?php
2	namespace MediaWiki\IPInfo\InfoRetriever;
3
4	use MediaWiki\IPInfo\Info\IPCountInfo;
5	use MediaWiki\IPInfo\TempUserIPLookup;
6	use MediaWiki\User\UserIdentity;
7
8	/**
9	* Retrieves the number of unique IP addresses used by a temporary user account.
10	*/
11	class IPCountInfoRetriever extends BaseInfoRetriever {
12
13	public const NAME = 'ipinfo-source-ip-count';
14
15	private TempUserIPLookup $tempUserIPLookup;
16
17	public function __construct( TempUserIPLookup $tempUserIPLookup ) {
18	$this->tempUserIPLookup = $tempUserIPLookup;
19	}
20
21	public function getName(): string {
22	return self::NAME;
23	}
24
25	public function retrieveFor( UserIdentity $user, ?string $ip ): IPCountInfo {
26	// Showing the count of unique IP addresses only makes sense for temporary users,
27	// since anonymous users are identified by their IP address and therefore by definition
28	// have a unique IP address count of 1.
29	// So, don't attempt to fetch or show this data for them.
30	if ( !$user->isRegistered() ) {
31	return new IPCountInfo( null );
32	}
33
34	$count = $this->tempUserIPLookup->getDistinctAddressCount( $user );
35	return new IPCountInfo( $count );
36	}
37	}