Geolocation class

This commit is contained in:
Mike Jolley 2014-12-23 18:49:37 +00:00
parent 6800ee0ec3
commit f561aa4224
1 changed files with 130 additions and 0 deletions

View File

@ -0,0 +1,130 @@
<?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();