<?php

require_once(dirname(__FILE__) . '/../vendor/autoload.php');

use GeoIp2\Database\Reader;

if (!defined('WFWAF_RUN_COMPLETE')) {
class wfWAFGeoIP2 {
	private $_reader;
	
	/**
	 * Returns the singleton wfWAFGeoIP2.
	 *
	 * @return wfWAFGeoIP2
	 */
	public static function shared() {
		static $_geoip = null;
		if ($_geoip === null) {
			$_geoip = new wfWAFGeoIP2();
		}
		return $_geoip;
	}
	
	public function __construct() {
		try {
			if (file_exists(WFWAF_LOG_PATH . '/GeoLite2-Country.mmdb')) {
				$this->_reader = new Reader(WFWAF_LOG_PATH . '/GeoLite2-Country.mmdb');
				return;
			}
		}
		catch (Exception $e) {
			//Fall through to bundled copy
		}
		
		$this->_reader = new Reader(__DIR__ . '/../lib/GeoLite2-Country.mmdb'); //Can throw, but we don't catch it because it means the installation is likely corrupt and needs fixed anyway
	}
	
	/**
	 * Returns the database version in use. This is the timestamp of when it was packaged.
	 *
	 * @return null|int
	 */
	public function version() {
		try {
			return $this->_reader->metadata()->buildEpoch;
		}
		catch (Exception $e) {
			//Fall through
		}
		return null;
	}
	
	/**
	 * Returns the country code for the IP if known.
	 *
	 * @param string $ip
	 * @return null|string
	 */
	public function countryCode($ip) {
		try {
			$record = $this->_reader->country($ip);
			return $record->country->isoCode;
		}
		catch (Exception $e) {
			//Fall through
		}
		return null;
	}
}
}