102 lines
2.4 KiB
PHP
102 lines
2.4 KiB
PHP
<?php
|
|
/**
|
|
* Wrapper for MaxMind GeoLite2 Reader
|
|
*
|
|
* This class provide an interface to handle geolocation and error handling.
|
|
*
|
|
* Requires PHP 5.4+.
|
|
*
|
|
* @package WooCommerce\Classes
|
|
* @since 3.4.0
|
|
*/
|
|
|
|
defined( 'ABSPATH' ) || exit;
|
|
|
|
/**
|
|
* Geolite integration class.
|
|
*/
|
|
class WC_Geolite_Integration {
|
|
|
|
/**
|
|
* MaxMind GeoLite2 database path.
|
|
*
|
|
* @var string
|
|
*/
|
|
private $database = '';
|
|
|
|
/**
|
|
* Logger instance.
|
|
*
|
|
* @var WC_Logger
|
|
*/
|
|
private $log = null;
|
|
|
|
/**
|
|
* Constructor.
|
|
*
|
|
* @param string $database MaxMind GeoLite2 database path.
|
|
*/
|
|
public function __construct( $database ) {
|
|
$this->database = $database;
|
|
|
|
if ( ! class_exists( 'MaxMind\\Db\\Reader', false ) ) {
|
|
$this->require_geolite_library();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get country 2-letters ISO by IP address.
|
|
* Returns empty string when not able to find any ISO code.
|
|
*
|
|
* @param string $ip_address User IP address.
|
|
* @return string
|
|
*/
|
|
public function get_country_iso( $ip_address ) {
|
|
$iso_code = '';
|
|
|
|
try {
|
|
$reader = new MaxMind\Db\Reader( $this->database ); // phpcs:ignore PHPCompatibility.PHP.NewLanguageConstructs.t_ns_separatorFound
|
|
$data = $reader->get( $ip_address );
|
|
|
|
if ( isset( $data['country']['iso_code'] ) ) {
|
|
$iso_code = $data['country']['iso_code'];
|
|
}
|
|
|
|
$reader->close();
|
|
} catch ( Exception $e ) {
|
|
$this->log( $e->getMessage(), 'warning' );
|
|
}
|
|
|
|
return sanitize_text_field( strtoupper( $iso_code ) );
|
|
}
|
|
|
|
/**
|
|
* Logging method.
|
|
*
|
|
* @param string $message Log message.
|
|
* @param string $level Log level.
|
|
* Available options: 'emergency', 'alert',
|
|
* 'critical', 'error', 'warning', 'notice',
|
|
* 'info' and 'debug'.
|
|
* Defaults to 'info'.
|
|
*/
|
|
private function log( $message, $level = 'info' ) {
|
|
if ( is_null( $this->log ) ) {
|
|
$this->log = wc_get_logger();
|
|
}
|
|
|
|
$this->log->log( $level, $message, array( 'source' => 'geoip' ) );
|
|
}
|
|
|
|
/**
|
|
* Require geolite library.
|
|
*/
|
|
private function require_geolite_library() {
|
|
require_once WC_ABSPATH . 'includes/libraries/geolite2/Reader/Decoder.php';
|
|
require_once WC_ABSPATH . 'includes/libraries/geolite2/Reader/InvalidDatabaseException.php';
|
|
require_once WC_ABSPATH . 'includes/libraries/geolite2/Reader/Metadata.php';
|
|
require_once WC_ABSPATH . 'includes/libraries/geolite2/Reader/Util.php';
|
|
require_once WC_ABSPATH . 'includes/libraries/geolite2/Reader.php';
|
|
}
|
|
}
|