import contextlib
from urllib.parse import urlparse

from pagermaid.listener import listener
from pagermaid.enums import Message
from pagermaid.services import client as requests


async def get_ip_info(url: str) -> str:
    data = await requests.get(
        f"http://ip-api.com/json/{url}?fields=status,message,country,regionName,"
        f"city,lat,lon,isp,org,as,mobile,proxy,hosting,query")
    ipinfo_json = data.json()
    if ipinfo_json['status'] == 'fail':
        return ""
    elif ipinfo_json['status'] == 'success':
        ipinfo_list = [f"查询目标： `{url}`"]
        if ipinfo_json['query'] != url:
            ipinfo_list.extend(["解析地址： `" + ipinfo_json['query'] + "`"])
        ipinfo_list.extend(
            [(("地区： `" + ipinfo_json['country'] + ' - ' + ipinfo_json['regionName'] + ' - ' + ipinfo_json['city']) + "`"),
             "经纬度： `" + str(ipinfo_json['lat']) + ',' + str(ipinfo_json['lon']) + "`",
             "ISP： `" + ipinfo_json['isp'] + "`"])
        if ipinfo_json['org'] != '':
            ipinfo_list.extend(["组织： `" + ipinfo_json['org'] + "`"])
        with contextlib.suppress(Exception):
            ipinfo_list.extend(['[' + ipinfo_json['as'] + '](https://bgp.he.net/' + ipinfo_json['as'].split()[0] + ')'])
        if ipinfo_json['mobile']:
            ipinfo_list.extend(['此 IP 可能为**蜂窝移动数据 IP**'])
        if ipinfo_json['proxy']:
            ipinfo_list.extend(['此 IP 可能为**代理 IP**'])
        if ipinfo_json['hosting']:
            ipinfo_list.extend(['此 IP 可能为**数据中心 IP**'])
        return "\n".join(ipinfo_list)


@listener(command="ip",
          description="IPINFO （或者回复一句话）",
          parameters="<ip/域名>")
async def ipinfo(message: Message):
    reply = message.reply_to_message
    message: Message = await message.edit('正在查询中...')
    try:
        if reply:
            for num in range(len(reply.entities)):
                url = reply.text[reply.entities[num].offset:reply.entities[num].offset + reply.entities[num].length]
                url = urlparse(url)
                url = url.hostname or url.path
                await message.edit(await get_ip_info(url))
                return
        elif message.arguments:
            url_ = urlparse(message.arguments)
            url = url_.hostname or url_.path
            if ":" in url and "." not in url and not url_.hostname:
                url = message.arguments
            await message.edit(await get_ip_info(url))
            return
        await message.edit('没有找到要查询的 ip/域名 ...')
    except Exception:
        await message.edit('没有找到要查询的 ip/域名 ...')