<?php

/**
 * Class for get whois data for IP address.
 */
class Qs_GeoIpCountryWhois
{
    /**
     * Get country code for IP address
     *
     * @param string $ip IP address
     * @param string $default Default data if ip not found
     * @return string
     */
    public static function getCountry($ip, $default = null)
    {
        $country = null;
        $db = Qs_Db::getInstance();
        $long = sprintf('%u', ip2long($ip));

        if ($long != -1 && $long !== false) {
            $select = $db->select()
                ->from(Qs_Db::getTableName('Ip2Country'), ['code'])
                ->where('`longFrom` <= ?', $long, Zend_Db::BIGINT_TYPE)
                ->where('`longTo` >= ?', $long, Zend_Db::BIGINT_TYPE);
            $country = $db->fetchOne($select);
        }
        if (!$country) {
            if (false === ($country = Qs_GeoIpCountryWhois::_getWhoisCountryCodeFromCache($ip))) {
                $country = Qs_GeoIpCountryWhois::_getWhoisCountry($ip);
            }
        }
        return ($country) ? $country : $default;
    }

    /**
     * Get country code from cached data
     *
     * @param string $ip IP address
     * @return mixed  If country found then return country code, else false
     */
    protected static function _getWhoisCountryCodeFromCache($ip)
    {
        $db = Qs_Db::getInstance();
        $select = $db->select()
            ->from(Qs_Db::getTableName('WhoisCache'), ['countryCode'])
            ->where('`ip` = ?', $ip);
        $country = $db->fetchOne($select);
        if ($country) {
            return $country;
        }
        return false;
    }

    /**
     * Save contry code to cache
     *
     * @param string $ip IP address
     * @param string $code Country code
     * @return Qs_GeoIpCountryWhois
     */
    protected static function _saveWhoisCountryCode2Cache($ip, $code)
    {
        $db = Qs_Db::getInstance();
        $data = [
            'ip'          => $ip,
            'countryCode' => $code,
        ];
        $tableWhoisCache = new Qs_Db_Table(['name' => Qs_Db::getTableName('WhoisCache'), 'db' => $db]);
        $tableWhoisCache->insert($data);
        return true;
    }

    /**
     * Get country from whois (use whois linux command)
     *
     * @param string $ip IP address
     * @return string Country code
     */
    protected static function _getWhoisCountry($ip)
    {
        $country = null;
        $output = [];
        exec('whois ' . escapeshellarg($ip) . ' | grep -i \'Country:\'', $output);
        if (!empty($output)) {
            list(, $country) = explode(':', $output[0]);
            $country = trim($country);
        }
        if ($country) {
            Qs_GeoIpCountryWhois::_saveWhoisCountryCode2Cache($ip, $country);
        }
        return $country;
    }
}