Geolocation class

2014-12-23 18:49:37 +00:00 · 2014-12-23 18:49:37 +00:00 · f561aa4224
parent 6800ee0ec3
commit f561aa4224
1 changed files with 130 additions and 0 deletions
--- a/includes/class-wc-geolocation.php
+++ b/includes/class-wc-geolocation.php
@ -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();