#include <a_samp>
#include <a_mysql>


#if defined _geoip_included
  #endinput
#endif



#define _geoip_included


AntiDeAMX()
{
    new a[][] =
    {
        "Unarmed (Fist)",
        "Brass K"
    };
    #pragma unused a
}


enum info {
    ISP[170],
    City[170],
    Country[170]
};

new handler;

new CountryThread[MAX_PLAYERS];
new CityThread[MAX_PLAYERS];
new CityThread2[MAX_PLAYERS];
new ISPThread[MAX_PLAYERS];

new GeoIP[MAX_PLAYERS][info];

#define host "localhost"
#define user "root"
#define db "geo_ip"
#define pw ""


public OnFilterScriptInit() {
    AntiDeAMX();
    handler = mysql_connect(host, user, db, pw);
    mysql_debug(1);

    for(new i=0; i < MAX_PLAYERS; ++i) {
        if(IsPlayerConnected(i)) {
            GeoIP_OnPlayerConnect(i);
        }
    }

}
new ticks;
forward GeoIP_OnPlayerConnect(playerid);
public GeoIP_OnPlayerConnect(playerid) {

    strdel(GeoIP[playerid][Country], 0, strlen(GeoIP[playerid][Country]));
    strdel(GeoIP[playerid][City], 0, strlen(GeoIP[playerid][City]));
    strdel(GeoIP[playerid][ISP], 0, strlen(GeoIP[playerid][ISP]));

    ticks = tickcount();
    GetPlayerCountry(playerid);
    GetPlayerISP(playerid);
    GetPlayerCity(playerid);

    return 1;
}

stock GetPlayerCountry(playerid) {
    if(strlen(GeoIP[playerid][Country]) <= 0) {
        new string[170];
        new ip[24];
        GetPlayerIp(playerid, ip, sizeof(ip));
        format(string, sizeof(string), "SELECT cn FROM ip_country WHERE ip_to >= INET_ATON(\"%s\") ORDER BY ip_to ASC LIMIT 1", ip);
        MySQLFix();
        mysql_query(string, playerid+(MAX_PLAYERS * 2), playerid, handler);
        CountryThread[playerid] = playerid+(MAX_PLAYERS * 2);

        return string;
    }
    else
    {
        new str[170];
        strcat(str, GeoIP[playerid][Country]);
        return str;
    }
}

stock GetPlayerISP(playerid) {
    if(strlen(GeoIP[playerid][ISP]) <= 0) {
        new string[170];
        new ip[24];
        GetPlayerIp(playerid, ip, sizeof(ip));

        format(string, sizeof(string), "SELECT internet_service_provider FROM geo_isp WHERE ip_to >= INET_ATON(\"%s\") ORDER BY ip_to ASC LIMIT 1", ip);


        //MySQLFix();
        // for some reason checking the mysql connection in this function crashes it.
        mysql_query(string, playerid+(MAX_PLAYERS * 3), playerid, handler);
        ISPThread[playerid] = playerid+(MAX_PLAYERS * 3);

        return string;
    }
    else
    {
        new str[170], placeholder[1];
        sscanf_(GeoIP[playerid][ISP], "ss", placeholder, str);
        return str;
    }
}

stock GetPlayerCity(playerid) {
    if(strlen(GeoIP[playerid][City]) <= 0) {
        new string[170];
        new ip[24];
        GetPlayerIp(playerid, ip, sizeof(ip));
        format(string, sizeof(string), "SELECT locid FROM ip_city WHERE ip_to >= INET_ATON(\"%s\") ORDER BY ip_to ASC LIMIT 1", ip);
        MySQLFix();
        mysql_query(string, playerid+(MAX_PLAYERS * 4), playerid, handler);
        CityThread[playerid] = playerid+(MAX_PLAYERS * 4);

        return string;
    }
    else
    {
        new str[170];
        strcat(str, GeoIP[playerid][City]);
        return str;
    }
}
/*
stock GetPlayerCountry2(ip[]) {
    new playerid = 0;
    if(strlen(GeoIP[playerid][Country]) <= 0) {
        new string[170];
        format(string, sizeof(string), "SELECT cn FROM ip_country WHERE ip_to >= INET_ATON(\"%s\") ORDER BY ip_to ASC LIMIT 1", ip);
        MySQLFix();
        mysql_query(string, playerid+(MAX_PLAYERS * 2), playerid, handler);
        CountryThread[playerid] = playerid+(MAX_PLAYERS * 2);

        return string;
    }
    else
    {
        new str[170];
        strcat(str, GeoIP[playerid][Country]);
        return str;
    }
}

stock GetPlayerISP2(ip[]) {
    new playerid = 0;
    if(strlen(GeoIP[playerid][ISP]) <= 0) {
        new string[170];

        format(string, sizeof(string), "SELECT internet_service_provider FROM geo_isp WHERE ip_to >= INET_ATON(\"%s\") ORDER BY ip_to ASC LIMIT 1", ip);

        //MySQLFix();
        // for some reason checking the mysql connection in this function crashes it.
        mysql_query(string, playerid+(MAX_PLAYERS * 3), playerid, handler);
        ISPThread[playerid] = playerid+(MAX_PLAYERS * 3);

        return string;
    }
    else
    {
        new str[170], placeholder[1];
        sscanf_(GeoIP[playerid][ISP], "ss", placeholder, str);
        return str;
    }
}

stock GetPlayerCity2(ip[]) {
    new playerid = 0;
    if(strlen(GeoIP[playerid][City]) <= 0) {
        new string[170];
        format(string, sizeof(string), "SELECT locid FROM ip_city WHERE ip_to >= INET_ATON(\"%s\") ORDER BY ip_to ASC LIMIT 1", ip);
        MySQLFix();
        mysql_query(string, playerid+(MAX_PLAYERS * 4), playerid, handler);
        CityThread[playerid] = playerid+(MAX_PLAYERS * 4);

        return string;
    }
    else
    {
        new str[170];
        strcat(str, GeoIP[playerid][City]);
        return str;
    }
}*/

public OnQueryFinish(query[], resultid, extraid, connectionHandle) {
    if(connectionHandle != handler) return 1;
    for(new i=0; i < MAX_PLAYERS; ++i) /*if(IsPlayerConnected(i))*/ {
        if(resultid == CountryThread[i]) {
            mysql_store_result(handler);
            new templine[170];
            mysql_fetch_row_format(templine, "|", handler);
            mysql_free_result(handler);
            GeoIP[i][Country] = templine;
            break;
        }
        else if(resultid == CityThread[i]) {

            new templine[170];
            mysql_store_result(handler);
            mysql_fetch_row_format(templine, "|", handler);
            mysql_free_result(handler);

            GeoIP[i][City] = templine;
            new string[150];

            format(string, sizeof(string), "SELECT city FROM ip_city_locations WHERE locid = %d", strval(templine));
            MySQLFix();
            mysql_query(string, CityThread[i] * 2, extraid, handler);
            CityThread2[i] = CityThread[i] * 2;
            break;

        }
        else if(resultid == ISPThread[i]) {
            mysql_store_result(handler);
            new templine[170];
            mysql_fetch_row_format(templine, "|", handler);
            mysql_free_result(handler);
            GeoIP[i][ISP] = templine;
            break;
        }
        else if(resultid == CityThread2[i]) {
            new templine[170];
            mysql_store_result(handler);
            mysql_fetch_row_format(templine, "|", handler);
            mysql_free_result(handler);

            GeoIP[i][City] = templine;
            CallLocalFunction("OnGeoIPCityFinish", "d", extraid);

            printf("MySQL took %d ms", tickcount()-ticks);
            break;
        }
    }
    return 1;
}


stock sscanf_(string[], format[], {Float,_}:...)
{
    #if defined isnull
        if (isnull(string))
    #else
        if (string[0] == 0 || (string[0] == 1 && string[1] == 0))
    #endif
        {
            return format[0];
        }
    #pragma tabsize 4
    new
        formatPos = 0,
        stringPos = 0,
        paramPos = 2,
        paramCount = numargs(),
        delim = ' ';
    while (string[stringPos] && string[stringPos] <= ' ')
    {
        stringPos++;
    }
    while (paramPos < paramCount && string[stringPos])
    {
        switch (format[formatPos++])
        {
            case '\0':
            {
                return 0;
            }
            case 'i', 'd':
            {
                new
                    neg = 1,
                    num = 0,
                    ch = string[stringPos];
                if (ch == '-')
                {
                    neg = -1;
                    ch = string[++stringPos];
                }
                do
                {
                    stringPos++;
                    if ('0' <= ch <= '9')
                    {
                        num = (num * 10) + (ch - '0');
                    }
                    else
                    {
                        return -1;
                    }
                }
                while ((ch = string[stringPos]) > ' ' && ch != delim);
                setarg(paramPos, 0, num * neg);
            }
            case 'h', 'x':
            {
                new
                    num = 0,
                    ch = string[stringPos];
                do
                {
                    stringPos++;
                    switch (ch)
                    {
                        case 'x', 'X':
                        {
                            num = 0;
                            continue;
                        }
                        case '0' .. '9':
                        {
                            num = (num << 4) | (ch - '0');
                        }
                        case 'a' .. 'f':
                        {
                            num = (num << 4) | (ch - ('a' - 10));
                        }
                        case 'A' .. 'F':
                        {
                            num = (num << 4) | (ch - ('A' - 10));
                        }
                        default:
                        {
                            return -1;
                        }
                    }
                }
                while ((ch = string[stringPos]) > ' ' && ch != delim);
                setarg(paramPos, 0, num);
            }
            case 'c':
            {
                setarg(paramPos, 0, string[stringPos++]);
            }
            case 'f':
            {

                new changestr[16], changepos = 0, strpos = stringPos;
                while(changepos < 16 && string[strpos] && string[strpos] != delim)
                {
                    changestr[changepos++] = string[strpos++];
                    }
                changestr[changepos] = '\0';
                setarg(paramPos,0,_:floatstr(changestr));
            }
            case 'p':
            {
                delim = format[formatPos++];
                continue;
            }
            case '\'':
            {
                new
                    end = formatPos - 1,
                    ch;
                while ((ch = format[++end]) && ch != '\'') {}
                if (!ch)
                {
                    return -1;
                }
                format[end] = '\0';
                if ((ch = strfind(string, format[formatPos], false, stringPos)) == -1)
                {
                    if (format[end + 1])
                    {
                        return -1;
                    }
                    return 0;
                }
                format[end] = '\'';
                stringPos = ch + (end - formatPos);
                formatPos = end + 1;
            }
            case 'u':
            {
                new
                    end = stringPos - 1,
                    id = 0,
                    bool:num = true,
                    ch;
                while ((ch = string[++end]) && ch != delim)
                {
                    if (num)
                    {
                        if ('0' <= ch <= '9')
                        {
                            id = (id * 10) + (ch - '0');
                        }
                        else
                        {
                            num = false;
                        }
                    }
                }
                if (num && IsPlayerConnected(id))
                {
                    setarg(paramPos, 0, id);
                }
                else
                {
                    #if !defined foreach
                        #define foreach(%1,%2) for (new %2 = 0; %2 < MAX_PLAYERS; %2++) if (IsPlayerConnected(%2))
                        #define __SSCANF_FOREACH__
                    #endif
                    string[end] = '\0';
                    num = false;
                    new
                        name[MAX_PLAYER_NAME];
                    id = end - stringPos;
                    foreach (Player, playerid)
                    {
                        GetPlayerName(playerid, name, sizeof (name));
                        if (!strcmp(name, string[stringPos], true, id))
                        {
                            setarg(paramPos, 0, playerid);
                            num = true;
                            break;
                        }
                    }
                    if (!num)
                    {
                        setarg(paramPos, 0, INVALID_PLAYER_ID);
                    }
                    string[end] = ch;
                    #if defined __SSCANF_FOREACH__
                        #undef foreach
                        #undef __SSCANF_FOREACH__
                    #endif
                }
                stringPos = end;
            }
            case 's', 'z':
            {
                new
                    i = 0,
                    ch;
                if (format[formatPos])
                {
                    while ((ch = string[stringPos++]) && ch != delim)
                    {
                        setarg(paramPos, i++, ch);
                    }
                    if (!i)
                    {
                        return -1;
                    }
                }
                else
                {
                    while ((ch = string[stringPos++]))
                    {
                        setarg(paramPos, i++, ch);
                    }
                }
                stringPos--;
                setarg(paramPos, i, '\0');
            }
            default:
            {
                continue;
            }
        }
        while (string[stringPos] && string[stringPos] != delim && string[stringPos] > ' ')
        {
            stringPos++;
        }
        while (string[stringPos] && (string[stringPos] == delim || string[stringPos] <= ' '))
        {
            stringPos++;
        }
        paramPos++;
    }
    do
    {
        if ((delim = format[formatPos++]) > ' ')
        {
            if (delim == '\'')
            {
                while ((delim = format[formatPos++]) && delim != '\'') {}
            }
            else if (delim != 'z')
            {
                return delim;
            }
        }
    }
    while (delim > ' ');
    return 0;
}

MySQLFix()
{
    if(mysql_ping(handler) == -1)
    {
        handler = mysql_connect(host, user, db, pw);
    }
    return 1;
}