import asyncio import logging import time from typing import Optional import aiohttp import discord from aiohttp import ClientConnectorError, InvalidURL from jishaku.models import copy_context_with from discord.ext import commands from ipinfo.exceptions import RequestQuotaExceededError from structured_config import ConfigFile from tuxbot.cogs.Network.functions.converters import ( IPConverter, IPVersionConverter, DomainConverter, QueryTypeConverter, ASConverter, ) from tuxbot.cogs.Network.functions.exceptions import ( RFC18, InvalidIp, VersionNotFound, InvalidDomain, InvalidQueryType, InvalidAsn, ) from tuxbot.core.bot import Tux from tuxbot.core.i18n import ( Translator, ) from tuxbot.core.utils.data_manager import cogs_data_path from tuxbot.core.utils.functions.extra import ( ContextPlus, command_extra, ) from tuxbot.core.utils.functions.utils import shorten from .config import NetworkConfig from .functions.utils import ( get_ip, get_hostname, get_crimeflare_result, get_ipinfo_result, get_ipwhois_result, get_pydig_result, # get_peeringdb_as_set_result, # get_peeringdb_net_irr_as_set_result, merge_ipinfo_ipwhois, check_query_type_or_raise, check_ip_version_or_raise, check_asn_or_raise, ) log = logging.getLogger("tuxbot.cogs.Network") _ = Translator("Network", __file__) class Network(commands.Cog): def __init__(self, bot: Tux): self.bot = bot self.__config: NetworkConfig = ConfigFile( str(cogs_data_path("Network") / "config.yaml"), NetworkConfig, ).config async def cog_command_error(self, ctx: ContextPlus, error): if isinstance( error, ( RequestQuotaExceededError, RFC18, InvalidIp, InvalidDomain, InvalidQueryType, VersionNotFound, InvalidAsn, ), ): await ctx.send(_(str(error), ctx, self.bot.config)) async def cog_before_invoke(self, ctx: ContextPlus): await ctx.trigger_typing() # ========================================================================= # ========================================================================= @command_extra(name="iplocalise", aliases=["localiseip"], deletable=True) async def _iplocalise( self, ctx: ContextPlus, ip: IPConverter, version: Optional[IPVersionConverter] = None, ): check_ip_version_or_raise(str(version)) ip_address = await get_ip(self.bot.loop, str(ip), str(version)) ip_hostname = await get_hostname(self.bot.loop, str(ip_address)) ipinfo_result = await get_ipinfo_result( self.__config.ipinfoKey, ip_address ) ipwhois_result = await get_ipwhois_result(self.bot.loop, ip_address) merged_results = merge_ipinfo_ipwhois(ipinfo_result, ipwhois_result) e = discord.Embed( title=_( "Information for ``{ip} ({ip_address})``", ctx, self.bot.config ).format(ip=ip, ip_address=ip_address), color=0x5858D7, ) e.add_field( name=_("Belongs to:", ctx, self.bot.config), value=merged_results["belongs"], inline=True, ) e.add_field( name="RIR :", value=merged_results["rir"], inline=True, ) e.add_field( name=_("Region:", ctx, self.bot.config), value=merged_results["region"], inline=False, ) e.set_thumbnail(url=merged_results["flag"]) e.set_footer( text=_("Hostname: {hostname}", ctx, self.bot.config).format( hostname=ip_hostname ), ) await ctx.send(embed=e) @command_extra(name="cloudflare", deletable=True) async def _cloudflare( self, ctx: ContextPlus, ip: DomainConverter, ): crimeflare_result = await get_crimeflare_result( self.bot.session, str(ip) ) if crimeflare_result: alt_ctx = await copy_context_with( ctx, content=f"{ctx.prefix}iplocalise {crimeflare_result}" ) return await alt_ctx.command.reinvoke(alt_ctx) await ctx.send( _( "Unable to collect information through CloudFlare", ctx, self.bot.config, ) ) @command_extra(name="getheaders", aliases=["headers"], deletable=True) async def _getheaders( self, ctx: ContextPlus, ip: DomainConverter, *, user_agent: str = "" ): try: headers = {"User-Agent": user_agent} colors = { "1": 0x17A2B8, "2": 0x28A745, "3": 0xFFC107, "4": 0xDC3545, "5": 0x343A40, } async with self.bot.session.get( str(ip), headers=headers, timeout=aiohttp.ClientTimeout(total=8), ) as s: e = discord.Embed( title=f"Headers : {ip}", color=colors.get(str(s.status)[0], 0x6C757D), ) e.add_field( name="Status", value=f"```{s.status}```", inline=True ) e.set_thumbnail(url=f"https://http.cat/{s.status}") headers = dict(s.headers.items()) headers.pop("Set-Cookie", headers) fail = False for key, value in headers.items(): fail, output = await shorten(ctx.session, value, 50, fail) if output["link"]: value = _( "[show all]({})", ctx, self.bot.config ).format(output["link"]) else: value = f"```\n{output['text']}```" e.add_field(name=key, value=value, inline=True) await ctx.send(embed=e) except ( ClientConnectorError, InvalidURL, asyncio.exceptions.TimeoutError, ): await ctx.send( _("Cannot connect to host {}", ctx, self.bot.config).format(ip) ) @command_extra(name="dig", deletable=True) async def _dig( self, ctx: ContextPlus, domain: IPConverter, query_type: QueryTypeConverter, dnssec: str | bool = False, ): check_query_type_or_raise(str(query_type)) pydig_result = await get_pydig_result( self.bot.loop, str(domain), str(query_type), dnssec ) e = discord.Embed(title=f"DIG {domain} {query_type}", color=0x5858D7) for i, value in enumerate(pydig_result): e.add_field(name=f"#{i}", value=f"```{value}```") if not pydig_result: e.add_field( name=f"DIG {domain} IN {query_type}", value=_("No result...", ctx, self.bot.config), ) await ctx.send(embed=e) @command_extra(name="ping", deletable=True) async def _ping(self, ctx: ContextPlus): start = time.perf_counter() await ctx.trigger_typing() end = time.perf_counter() latency = round(self.bot.latency * 1000, 2) typing = round((end - start) * 1000, 2) e = discord.Embed(title="Ping", color=discord.Color.teal()) e.add_field(name="Websocket", value=f"{latency}ms") e.add_field(name="Typing", value=f"{typing}ms") await ctx.send(embed=e) @command_extra(name="isdown", aliases=["is_down", "down?"], deletable=True) async def _isdown(self, ctx: ContextPlus, domain: IPConverter): try: async with self.bot.session.get( f"https://isitdown.site/api/v3/{domain}", timeout=aiohttp.ClientTimeout(total=8), ) as s: json = await s.json() if json["isitdown"]: title = _("Down...", ctx, self.bot.config) color = 0xDC3545 else: title = _("Up!", ctx, self.bot.config) color = 0x28A745 e = discord.Embed(title=title, color=color) e.set_thumbnail( url=f"https://http.cat/{json['response_code']}" ) await ctx.send(embed=e) except ( ClientConnectorError, InvalidURL, asyncio.exceptions.TimeoutError, ): await ctx.send( _("Cannot connect to host {}", ctx, self.bot.config).format( domain ) ) @command_extra( name="peeringdb", aliases=["peer", "peering"], deletable=True ) async def _peeringdb(self, ctx: ContextPlus, asn: ASConverter): check_asn_or_raise(str(asn)) return await ctx.send("Not implemented yet") # peeringdb_as_set_result = await get_peeringdb_as_set_result( # self.bot.session, str(asn) # ) # peeringdb_net_irr_as_set_result = ( # await get_peeringdb_net_irr_as_set_result( # self.bot.session, peeringdb_as_set_result["data"][0][asn] # ) # )["data"] # # data = peeringdb_net_irr_as_set_result # # self.bot.console.log(data) # # await ctx.send("done")