import os.path
import json
import gzip
from io import BytesIO

import pygeoip

from cloudbot import hook, http


@hook.onload()
def load_regions(bot):
    global regions, geo
    # load region database
    with open(os.path.join(bot.data_dir, "geoip_regions.json"), "rb") as f:
        regions = json.loads(f.read().decode())

    if os.path.isfile(os.path.join(bot.data_dir, "GeoLiteCity.dat")):
        # initialise geolocation database
        geo = pygeoip.GeoIP(os.path.join(bot.data_dir, "GeoLiteCity.dat"))
    else:
        print("Downloading GeoIP database")
        download = http.get("http://geolite.maxmind.com/download/geoip/database/GeoLiteCity.dat.gz", decode=False)
        print("Download complete")
        bytes_io = BytesIO(download)
        geoip_file = gzip.GzipFile(fileobj=bytes_io, mode='rb')

        output = open(os.path.join(bot.data_dir, "GeoLiteCity.dat"), 'wb')
        output.write(geoip_file.read())
        output.close()

        geo = pygeoip.GeoIP(os.path.join(bot.data_dir, "GeoLiteCity.dat"))


@hook.command()
def geoip(text):
    """<host/ip> - gets the location of <host/ip>"""

    try:
        record = geo.record_by_name(text)
    except Exception:
        return "Sorry, I can't locate that in my database."

    data = {}

    if "region_name" in record:
        # we try catching an exception here because the region DB is missing a few areas
        # it's a lazy patch, but it should do the job
        try:
            data["region"] = ", " + regions[record["country_code"]][record["region_name"]]
        except Exception:
            data["region"] = ""
    else:
        data["region"] = ""

    data["cc"] = record["country_code"] or "N/A"
    data["country"] = record["country_name"] or "Unknown"
    data["city"] = record["city"] or "Unknown"

    return "\x02Country:\x02 {country} ({cc}), \x02City:\x02 {city}{region}".format(**data)