131 lines
3.2 KiB
PHP
131 lines
3.2 KiB
PHP
|
<?php
|
||
|
/**
|
||
|
* Geolocation class.
|
||
|
*
|
||
|
* Handles geolocation and updating the geolocation database.
|
||
|
*
|
||
|
* This product uses GeoLite2 data created by MaxMind, available from http://www.maxmind.com
|
||
|
*
|
||
|
* @author WooThemes
|
||
|
* @category Admin
|
||
|
* @package WooCommerce/Classes
|
||
|
* @version 2.3.0
|
||
|
*/
|
||
|
|
||
|
if ( ! defined( 'ABSPATH' ) ) {
|
||
|
exit;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* WC_Geolocation Class
|
||
|
*/
|
||
|
class WC_Geolocation {
|
||
|
|
||
|
/** URL to the geolocation database we're using */
|
||
|
const GEOLITE_DB = 'http://geolite.maxmind.com/download/geoip/database/GeoLiteCountry/GeoIP.dat.gz';
|
||
|
|
||
|
/**
|
||
|
* Hook in tabs.
|
||
|
*/
|
||
|
public static function init() {
|
||
|
add_action( 'woocommerce_geoip_updater', array( __CLASS__, 'update_database' ) );
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Update geoip database. Adapted from https://wordpress.org/plugins/geoip-detect/.
|
||
|
*/
|
||
|
public static function update_database() {
|
||
|
require_once( ABSPATH . 'wp-admin/includes/file.php' );
|
||
|
|
||
|
$upload_dir = wp_upload_dir();
|
||
|
$database = $upload_dir['basedir'] . '/GeoIP.dat';
|
||
|
$tmp_database = download_url( self::GEOLITE_DB );
|
||
|
$logger = new WC_Logger();
|
||
|
|
||
|
if ( ! is_wp_error( $tmp_database ) ) {
|
||
|
$gzhandle = @gzopen( $tmp_database, 'r' );
|
||
|
$handle = @fopen( $database, 'w' );
|
||
|
|
||
|
if ( $gzhandle && $handle ) {
|
||
|
while ( ( $string = gzread( $gzhandle, 4096 ) ) != false ) {
|
||
|
fwrite( $handle, $string, strlen( $string ) );
|
||
|
}
|
||
|
gzclose( $gzhandle );
|
||
|
fclose( $handle );
|
||
|
} else {
|
||
|
$logger->add( 'geolocation', 'Unable to open database file' );
|
||
|
}
|
||
|
|
||
|
@unlink( $tmp_database );
|
||
|
} else {
|
||
|
$logger->add( 'geolocation', 'Unable to download GeoIP Database: ' . $tmp_database->get_message() );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Geolocation an IP address
|
||
|
* @param string $ip_address
|
||
|
* @return array
|
||
|
*/
|
||
|
public static function geolocate_ip( $ip_address ) {
|
||
|
$country_code = self::geolite_geolocation( $ip_address );
|
||
|
|
||
|
// Fallback to API
|
||
|
if ( empty( $country_code ) ) {
|
||
|
$country_code = self::freegeoip_geolocation( $ip_address );
|
||
|
}
|
||
|
|
||
|
return array(
|
||
|
'country' => $country_code,
|
||
|
'state' => ''
|
||
|
);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Use MAXMIND GeoLite database to geolocation the user.
|
||
|
* @param string $ip_address
|
||
|
* @return string
|
||
|
*/
|
||
|
private static function geolite_geolocation( $ip_address ) {
|
||
|
$country_code = '';
|
||
|
$upload_dir = wp_upload_dir();
|
||
|
$database = $upload_dir['basedir'] . '/GeoIP.dat';
|
||
|
|
||
|
if ( file_exists( $database ) ) {
|
||
|
if ( ! class_exists( 'GeoIP' ) ) {
|
||
|
include_once( 'libraries/geoip.php' );
|
||
|
}
|
||
|
$gi = geoip_open( $database, GEOIP_STANDARD );
|
||
|
$country_code = geoip_country_code_by_addr( $gi, $ip_address );
|
||
|
geoip_close( $gi );
|
||
|
}
|
||
|
|
||
|
return $country_code;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Use freegeoip.net
|
||
|
* @param string $ip_address
|
||
|
* @return string
|
||
|
*/
|
||
|
private static function freegeoip_geolocation( $ip_address ) {
|
||
|
$country_code = get_transient( 'geoip_' . $ip_address );
|
||
|
|
||
|
if ( false === $country_code ) {
|
||
|
$json_geo_ip = wp_remote_get( 'https://freegeoip.net/json/' . $ip_address );
|
||
|
|
||
|
if ( ! is_wp_error( $json_geo_ip ) ) {
|
||
|
$data = json_decode( $json_geo_ip['body'] );
|
||
|
$country_code = sanitize_text_field( $data->country_code );
|
||
|
set_transient( 'geoip_' . $ip_address, $country_code, WEEK_IN_SECONDS );
|
||
|
} else {
|
||
|
$country_code = '';
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return $country_code;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
WC_Geolocation::init();
|