Code Coverage for /workspace/src/extensions/CookieWarning/includes/HttpGeoLocation.php

1	<?php
2
3	namespace CookieWarning;
4
5	use Http;
6	use InvalidArgumentException;
7	use Wikimedia\IPUtils;
8
9	/**
10	* Implements the GeoLocation class, which allows to locate the user based on the IP address.
11	*/
12	class HttpGeoLocation implements GeoLocation {
13	private $geoIPServiceURL;
14	private $locatedIPs = [];
15
16	/**
17	* @param string $geoIPServiceURL
18	*/
19	public function __construct( $geoIPServiceURL ) {
20	if ( !is_string( $geoIPServiceURL ) \|\| !$geoIPServiceURL ) {
21	throw new InvalidArgumentException( 'The geoIPServiceUL is invalid' );
22	}
23	$this->geoIPServiceURL = $geoIPServiceURL;
24	}
25
26	/**
27	* {@inheritdoc}
28	* @param string $ip The IP address to lookup
29	* @return string\|null
30	*/
31	public function locate( $ip ) {
32	if ( isset( $this->locatedIPs[$ip] ) ) {
33	return $this->locatedIPs[$ip];
34	}
35	if ( !IPUtils::isValid( $ip ) ) {
36	throw new InvalidArgumentException( "$ip is not a valid IP address." );
37	}
38	if ( substr( $this->geoIPServiceURL, -1 ) !== '/' ) {
39	$this->geoIPServiceURL .= '/';
40	}
41	$json = Http::get( $this->geoIPServiceURL . $ip, [
42	'timeout' => '2',
43	] );
44	if ( !$json ) {
45	return null;
46	}
47	$returnObject = json_decode( $json );
48	if ( $returnObject === null \|\| !property_exists( $returnObject, 'country_code' ) ) {
49	return null;
50	}
51	$this->locatedIPs[$ip] = $returnObject->country_code;
52
53	return $this->locatedIPs[$ip];
54	}
55	}