#!/usr/bin/php
<?php

require_once 'geoip2.phar';

try {
	$dir = realpath(dirname(__FILE__)) . '/';
	$city_db = new GeoIp2\Database\Reader($dir . 'GeoIP2-City.mmdb');
	$isp_db = new GeoIp2\Database\Reader($dir . 'GeoIP2-ISP.mmdb');
} catch (Exception $e) {
	die($e);
}

function ipinfo($ip, $isatty)
{
	global $city_db, $isp_db, $domain_db;
	$data = filter_var($ip, FILTER_VALIDATE_IP, FILTER_FLAG_NO_PRIV_RANGE | FILTER_FLAG_NO_RES_RANGE);
	if (!$data) {
		return $ip;
	}
	$info = '';
	try {
		$city = $city_db->city($ip);
		if ($s = $city->country->names)
			$info .= $s['en'];
		else
			$info .= 'Unknown';
		if ($s = $city->city->names)
			$info .= ',' . $s['en'];
		if ($s = $isp_db->isp($ip)->isp)
			$info .= ' - ' . preg_replace('/\s*[\-\(].+$/', '', $s);
	} catch (GeoIp2\Exception\AddressNotFoundException $e) {
	}
	if (!$info)
		return $ip;
	if (!$isatty)
		return "$ip($info)";
	return "$ip\033[1;34m($info)\033[0m";
}

$isatty = posix_isatty(STDOUT);
while($line = fgets(STDIN)){
	$pattern = '/(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}|\d{4}:[0-9a-f:]+)/';
	echo preg_replace_callback($pattern, function($matches) { return ipinfo($matches[0], $GLOBALS['isatty']); }, $line);
}