You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
3.0 KiB
97 lines
3.0 KiB
import logging
|
|
import re
|
|
|
|
import aiohttp
|
|
import discord
|
|
from redbot.core import Config, commands
|
|
from redbot.core.bot import Red
|
|
|
|
log = logging.getLogger("red.fox_v3.isitdown")
|
|
|
|
|
|
class IsItDown(commands.Cog):
|
|
"""
|
|
Cog Description
|
|
|
|
Less important information about the cog
|
|
"""
|
|
|
|
def __init__(self, bot: Red):
|
|
super().__init__()
|
|
self.bot = bot
|
|
self.config = Config.get_conf(self, identifier=0, force_registration=True)
|
|
|
|
default_guild = {"iids": []} # List of tuple pairs (channel_id, website)
|
|
|
|
self.config.register_guild(**default_guild)
|
|
|
|
async def red_delete_data_for_user(self, **kwargs):
|
|
"""Nothing to delete"""
|
|
return
|
|
|
|
@commands.command(alias=["iid"])
|
|
async def isitdown(self, ctx: commands.Context, url_to_check):
|
|
"""
|
|
Check if the provided url is down
|
|
|
|
Alias: iid
|
|
"""
|
|
try:
|
|
resp = await self._check_if_down(url_to_check)
|
|
except AssertionError:
|
|
await ctx.maybe_send_embed("Invalid URL provided. Make sure not to include `http://`")
|
|
return
|
|
|
|
if resp["isitdown"]:
|
|
await ctx.maybe_send_embed(f"{url_to_check} is DOWN!")
|
|
else:
|
|
await ctx.maybe_send_embed(f"{url_to_check} is UP!")
|
|
|
|
@commands.admin()
|
|
@commands.command(alias=["iidmonitor"])
|
|
async def isitdownmonitor(self, ctx: commands.Context, announce_channel: discord.TextChannel,
|
|
url_to_check):
|
|
"""
|
|
Add a continuous monitoring for a url and a channel to post updates in.
|
|
"""
|
|
try:
|
|
resp = await self._check_if_down(url_to_check)
|
|
except AssertionError:
|
|
await ctx.maybe_send_embed("Invalid URL provided. Make sure not to include `http://`")
|
|
return
|
|
|
|
async with self.config.guild(ctx.guild).iids() as iids:
|
|
iids.append((announce_channel.id, url_to_check))
|
|
|
|
await ctx.maybe_send_embed(
|
|
f"{announce_channel.mention} will now receive alerts when the status of {url_to_check} changes")
|
|
|
|
@commands.admin()
|
|
@commands.command(alias=["iidlist"])
|
|
async def isitdownlist(self, ctx: commands.Context):
|
|
"""
|
|
List all checks that have been setup across all guilds
|
|
"""
|
|
if not await self.config.guild(ctx.guild).iids():
|
|
await ctx.maybe_send_embed("No urls are configured to be checked")
|
|
|
|
em = discord.Embed("")
|
|
|
|
@commands.admin()
|
|
@commands.command(alias=["iidclear"])
|
|
async def isitdownclear(self, ctx: commands.Context):
|
|
"""
|
|
Clear all checks that have been setup across all guilds
|
|
"""
|
|
|
|
async def _check_if_down(self, url_to_check):
|
|
url = re.compile(r"https?://(www\.)?")
|
|
url.sub("", url_to_check).strip().strip("/")
|
|
|
|
url = f"https://isitdown.site/api/v3/{url}"
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(url) as response:
|
|
assert response.status == 200
|
|
resp = await response.json()
|
|
return resp
|