WIP: Add UpdateChecker cog #14
1 changed files with 162 additions and 169 deletions
|
|
@ -75,171 +75,169 @@ class UpdateChecker(commands.Cog):
|
||||||
|
|
||||||
@tasks.loop(minutes=1)
|
@tasks.loop(minutes=1)
|
||||||
async def bg_task(self):
|
async def bg_task(self):
|
||||||
while True:
|
cog = self.bot.get_cog("Downloader")
|
||||||
cog = self.bot.get_cog("Downloader")
|
if cog is not None:
|
||||||
if cog is not None:
|
data = await self.conf.all()
|
||||||
data = await self.conf.all()
|
repos = data["repos"]
|
||||||
repos = data["repos"]
|
channel = data["gochannel"]
|
||||||
auto = data["auto"]
|
use_embed = data["embed"]
|
||||||
channel = data["gochannel"]
|
whitelist = data["whitelist"]
|
||||||
use_embed = data["embed"]
|
blacklist = data["blacklist"]
|
||||||
whitelist = data["whitelist"]
|
if channel:
|
||||||
blacklist = data["blacklist"]
|
channel = self.bot.get_channel(channel)
|
||||||
if channel:
|
if channel is None:
|
||||||
channel = self.bot.get_channel(channel)
|
await self.bot.send_to_owners(
|
||||||
if channel is None:
|
"[Update Checker] It appears that I am no longer allowed to send messages to the designated update channel. "
|
||||||
|
"From now on, it will DM you."
|
||||||
|
)
|
||||||
|
await self.conf.gochannel.set(0)
|
||||||
|
send = self.bot.send_to_owners
|
||||||
|
else:
|
||||||
|
use_embed = (
|
||||||
|
use_embed and channel.permissions_for(channel.guild.me).embed_links
|
||||||
|
)
|
||||||
|
send = channel.send
|
||||||
|
else:
|
||||||
|
send = self.bot.send_to_owners
|
||||||
|
|
||||||
|
all_repos = cog._repo_manager.get_all_repo_names() # pylint: disable=protected-access
|
||||||
|
for repo in all_repos:
|
||||||
|
if not (repo in list(repos.keys())):
|
||||||
|
repos[repo] = "--default--"
|
||||||
|
await self.conf.repos.set(repos)
|
||||||
|
|
||||||
|
saving_dict = {k: v for k, v in repos.items() if k in all_repos}
|
||||||
|
for repo_name, commit_saved in saving_dict.items():
|
||||||
|
repo = cog._repo_manager.get_repo(repo_name) # pylint: disable=protected-access
|
||||||
|
if not repo:
|
||||||
|
continue
|
||||||
|
parsed_url = urlparse(repo.url)
|
||||||
|
if parsed_url.netloc == "github.com":
|
||||||
|
url = repo.url + r"/commits/" + repo.branch + ".atom"
|
||||||
|
response = await self.fetch_feed(url)
|
||||||
|
try:
|
||||||
|
commit = response.entries[0]["id"][33:]
|
||||||
|
chash = "[" + commit + "](" + response.entries[0]["link"] + ")"
|
||||||
|
cn = response.entries[0]["title"] + " - " + response.entries[0]["author"]
|
||||||
|
image = response.entries[0]["media_thumbnail"][0]["url"].split("?")[0]
|
||||||
|
except AttributeError:
|
||||||
|
continue
|
||||||
|
else:
|
||||||
|
url = repo.url + r"/rss/branch/" + repo.branch
|
||||||
|
response = await self.fetch_feed(url)
|
||||||
|
try:
|
||||||
|
commit = response.entries[0]["id"][33:]
|
||||||
|
chash = "[" + commit + "](" + response.entries[0]["link"] + ")"
|
||||||
|
cn = response.entries[0]["title"] + " - " + response.entries[0]["author"]
|
||||||
|
image = await self.fetch_gitea_thumbnail(parsed_url.scheme + "://" + parsed_url.netloc + "/api/v1/repos" + parsed_url.path)
|
||||||
|
except AttributeError:
|
||||||
|
continue
|
||||||
|
saving_dict[repo_name] = commit
|
||||||
|
if whitelist:
|
||||||
|
if repo_name not in whitelist:
|
||||||
|
continue
|
||||||
|
if repo_name in blacklist:
|
||||||
|
continue
|
||||||
|
# CN is used here for backwards compatability, don't want people to get an
|
||||||
|
# update for each and every one of their cogs when updating this cog
|
||||||
|
if commit_saved not in (commit, cn, '--default--'):
|
||||||
|
if use_embed:
|
||||||
|
e = discord.Embed(
|
||||||
|
title="Update Checker",
|
||||||
|
description=f"Update available for repo: {repo.name}",
|
||||||
|
timestamp=datetime.utcnow(),
|
||||||
|
color=0x00FF00,
|
||||||
|
)
|
||||||
|
e.add_field(name="URL", value=repo.url)
|
||||||
|
e.add_field(name="Branch", value=repo.branch)
|
||||||
|
e.add_field(name="Commit", value=cn)
|
||||||
|
e.add_field(name="Hash", value=chash)
|
||||||
|
if image is not None:
|
||||||
|
e.set_thumbnail(url=image)
|
||||||
|
else:
|
||||||
|
e = box(
|
||||||
|
"[Update Checker]"
|
||||||
|
f" Repo: {repo.name}\n"
|
||||||
|
f" URL: {repo.url}\n"
|
||||||
|
f" Commit: {cn}\n"
|
||||||
|
f" Hash: {commit}\n"
|
||||||
|
f" Time: {datetime.utcnow()}",
|
||||||
|
'css'
|
||||||
|
)
|
||||||
|
try:
|
||||||
|
if use_embed:
|
||||||
|
await send(embed=e)
|
||||||
|
else:
|
||||||
|
await send(e)
|
||||||
|
except discord.Forbidden:
|
||||||
|
# send_to_owners suppresses Forbidden, logging it to console.
|
||||||
|
# As a result, this will only happen if a channel was set.
|
||||||
await self.bot.send_to_owners(
|
await self.bot.send_to_owners(
|
||||||
"[Update Checker] It appears that I am no longer allowed to send messages to the designated update channel. "
|
"[Update Checker] It appears that I am no longer allowed to send messages to the designated update channel. "
|
||||||
"From now on, it will DM you."
|
"From now on, it will DM you."
|
||||||
)
|
)
|
||||||
await self.conf.gochannel.set(0)
|
|
||||||
send = self.bot.send_to_owners
|
|
||||||
else:
|
|
||||||
use_embed = (
|
|
||||||
use_embed and channel.permissions_for(channel.guild.me).embed_links
|
|
||||||
)
|
|
||||||
send = channel.send
|
|
||||||
else:
|
|
||||||
send = self.bot.send_to_owners
|
|
||||||
|
|
||||||
all_repos = cog._repo_manager.get_all_repo_names() # pylint: disable=protected-access
|
|
||||||
for repo in all_repos:
|
|
||||||
if not (repo in list(repos.keys())):
|
|
||||||
repos[repo] = "--default--"
|
|
||||||
await self.conf.repos.set(repos)
|
|
||||||
|
|
||||||
saving_dict = {k: v for k, v in repos.items() if k in all_repos}
|
|
||||||
for repo_name, commit_saved in saving_dict.items():
|
|
||||||
repo = cog._repo_manager.get_repo(repo_name) # pylint: disable=protected-access
|
|
||||||
if not repo:
|
|
||||||
continue
|
|
||||||
parsed_url = urlparse(repo.url)
|
|
||||||
if parsed_url.netloc == "github.com":
|
|
||||||
url = repo.url + r"/commits/" + repo.branch + ".atom"
|
|
||||||
response = await self.fetch_feed(url)
|
|
||||||
try:
|
|
||||||
commit = response.entries[0]["id"][33:]
|
|
||||||
chash = "[" + commit + "](" + response.entries[0]["link"] + ")"
|
|
||||||
cn = response.entries[0]["title"] + " - " + response.entries[0]["author"]
|
|
||||||
image = response.entries[0]["media_thumbnail"][0]["url"].split("?")[0]
|
|
||||||
except AttributeError:
|
|
||||||
continue
|
|
||||||
else:
|
|
||||||
url = repo.url + r"/rss/branch/" + repo.branch
|
|
||||||
response = await self.fetch_feed(url)
|
|
||||||
try:
|
|
||||||
commit = response.entries[0]["id"][33:]
|
|
||||||
chash = "[" + commit + "](" + response.entries[0]["link"] + ")"
|
|
||||||
cn = response.entries[0]["title"] + " - " + response.entries[0]["author"]
|
|
||||||
image = await self.fetch_gitea_thumbnail(parsed_url.scheme + "://" + parsed_url.netloc + "/api/v1/repos" + parsed_url.path)
|
|
||||||
except AttributeError:
|
|
||||||
continue
|
|
||||||
saving_dict[repo_name] = commit
|
|
||||||
if whitelist:
|
|
||||||
if repo_name not in whitelist:
|
|
||||||
continue
|
|
||||||
if repo_name in blacklist:
|
|
||||||
continue
|
|
||||||
# CN is used here for backwards compatability, don't want people to get an
|
|
||||||
# update for each and every one of their cogs when updating this cog
|
|
||||||
if commit_saved not in (commit, cn, '--default--'):
|
|
||||||
if use_embed:
|
if use_embed:
|
||||||
e = discord.Embed(
|
await self.bot.send_to_owners(embed=e)
|
||||||
title="Update Checker",
|
|
||||||
description=f"Update available for repo: {repo.name}",
|
|
||||||
timestamp=datetime.utcnow(),
|
|
||||||
color=0x00FF00,
|
|
||||||
)
|
|
||||||
e.add_field(name="URL", value=repo.url)
|
|
||||||
e.add_field(name="Branch", value=repo.branch)
|
|
||||||
e.add_field(name="Commit", value=cn)
|
|
||||||
e.add_field(name="Hash", value=chash)
|
|
||||||
if image is not None:
|
|
||||||
e.set_thumbnail(url=image)
|
|
||||||
else:
|
else:
|
||||||
e = box(
|
await self.bot.send_to_owners(e)
|
||||||
"[Update Checker]"
|
await self.conf.gochannel.set(0)
|
||||||
f" Repo: {repo.name}\n"
|
# Was already inaccessible before I got here, so I'm just gonna leave it and look at it later -- Sea
|
||||||
f" URL: {repo.url}\n"
|
# try:
|
||||||
f" Commit: {cn}\n"
|
# await channel.send(
|
||||||
f" Hash: {commit}\n"
|
# f"[Update Checker] Update found for repo: {repo.name}. Updating repos..."
|
||||||
f" Time: {datetime.utcnow()}",
|
# )
|
||||||
'css'
|
# except AttributeError:
|
||||||
)
|
# owner = (await self.bot.application_info()).owner
|
||||||
try:
|
# await owner.send(
|
||||||
if use_embed:
|
# "[Update Checker] It appears that the channel for this cog has been deleted. From now on, it will DM you."
|
||||||
await send(embed=e)
|
# )
|
||||||
else:
|
# channel = owner
|
||||||
await send(e)
|
# await self.conf.gochannel.set(0)
|
||||||
except discord.Forbidden:
|
# except discord.errors.Forbidden:
|
||||||
# send_to_owners suppresses Forbidden, logging it to console.
|
# owner = (await self.bot.application_info()).owner
|
||||||
# As a result, this will only happen if a channel was set.
|
# await owner.send(
|
||||||
await self.bot.send_to_owners(
|
# "[Update Checker] It appears that I am no longer allowed to send messages to the designated update channel. From now on, it will DM you."
|
||||||
"[Update Checker] It appears that I am no longer allowed to send messages to the designated update channel. "
|
# )
|
||||||
"From now on, it will DM you."
|
# channel = owner
|
||||||
)
|
# await self.conf.gochannel.set(0)
|
||||||
if use_embed:
|
# # Just a copy of `[p]cog update`, but without using ctx things
|
||||||
await self.bot.send_to_owners(embed=e)
|
# try:
|
||||||
else:
|
# installed_cogs = set(await cog.installed_cogs())
|
||||||
await self.bot.send_to_owners(e)
|
# updated = await cog._repo_manager.update_all_repos()
|
||||||
await self.conf.gochannel.set(0)
|
# updated_cogs = set(
|
||||||
# Was already inaccessible before I got here, so I'm just gonna leave it and look at it later -- Sea
|
# cog for repo in updated for cog in repo.available_cogs
|
||||||
# try:
|
# )
|
||||||
# await channel.send(
|
# installed_and_updated = updated_cogs & installed_cogs
|
||||||
# f"[Update Checker] Update found for repo: {repo.name}. Updating repos..."
|
# if installed_and_updated:
|
||||||
# )
|
# await cog._reinstall_requirements(installed_and_updated)
|
||||||
# except AttributeError:
|
# await cog._reinstall_cogs(installed_and_updated)
|
||||||
# owner = (await self.bot.application_info()).owner
|
# await cog._reinstall_libraries(installed_and_updated)
|
||||||
# await owner.send(
|
# cognames = {c.name for c in installed_and_updated}
|
||||||
# "[Update Checker] It appears that the channel for this cog has been deleted. From now on, it will DM you."
|
# message = humanize_list(tuple(map(inline, cognames)))
|
||||||
# )
|
# except Exception as error:
|
||||||
# channel = owner
|
# exception_log = (
|
||||||
# await self.conf.gochannel.set(0)
|
# "Exception while updating repos in Update Checker \n"
|
||||||
# except discord.errors.Forbidden:
|
# )
|
||||||
# owner = (await self.bot.application_info()).owner
|
# exception_log += "".join(
|
||||||
# await owner.send(
|
# traceback.format_exception(
|
||||||
# "[Update Checker] It appears that I am no longer allowed to send messages to the designated update channel. From now on, it will DM you."
|
# type(error), error, error.__traceback__
|
||||||
# )
|
# )
|
||||||
# channel = owner
|
# )
|
||||||
# await self.conf.gochannel.set(0)
|
# try:
|
||||||
# # Just a copy of `[p]cog update`, but without using ctx things
|
# await channel.send(
|
||||||
# try:
|
# f"[Update Checker]: Error while updating repos.\n\n{exception_log}"
|
||||||
# installed_cogs = set(await cog.installed_cogs())
|
# )
|
||||||
# updated = await cog._repo_manager.update_all_repos()
|
# except discord.errors.Forbidden:
|
||||||
# updated_cogs = set(
|
# pass
|
||||||
# cog for repo in updated for cog in repo.available_cogs
|
# else:
|
||||||
# )
|
# try:
|
||||||
# installed_and_updated = updated_cogs & installed_cogs
|
# await channel.send(
|
||||||
# if installed_and_updated:
|
# f"[Update Checker]: Ran cog update. Updated cogs: {message}"
|
||||||
# await cog._reinstall_requirements(installed_and_updated)
|
# )
|
||||||
# await cog._reinstall_cogs(installed_and_updated)
|
# except discord.errors.Forbidden:
|
||||||
# await cog._reinstall_libraries(installed_and_updated)
|
# pass
|
||||||
# cognames = {c.name for c in installed_and_updated}
|
await asyncio.sleep(1)
|
||||||
# message = humanize_list(tuple(map(inline, cognames)))
|
await self.conf.repos.set(saving_dict)
|
||||||
# except Exception as error:
|
|
||||||
# exception_log = (
|
|
||||||
# "Exception while updating repos in Update Checker \n"
|
|
||||||
# )
|
|
||||||
# exception_log += "".join(
|
|
||||||
# traceback.format_exception(
|
|
||||||
# type(error), error, error.__traceback__
|
|
||||||
# )
|
|
||||||
# )
|
|
||||||
# try:
|
|
||||||
# await channel.send(
|
|
||||||
# f"[Update Checker]: Error while updating repos.\n\n{exception_log}"
|
|
||||||
# )
|
|
||||||
# except discord.errors.Forbidden:
|
|
||||||
# pass
|
|
||||||
# else:
|
|
||||||
# try:
|
|
||||||
# await channel.send(
|
|
||||||
# f"[Update Checker]: Ran cog update. Updated cogs: {message}"
|
|
||||||
# )
|
|
||||||
# except discord.errors.Forbidden:
|
|
||||||
# pass
|
|
||||||
await asyncio.sleep(1)
|
|
||||||
await self.conf.repos.set(saving_dict)
|
|
||||||
|
|
||||||
async def fetch_feed(self, url: str):
|
async def fetch_feed(self, url: str):
|
||||||
# Thank's to Sinbad's rss cog after which I copied this
|
# Thank's to Sinbad's rss cog after which I copied this
|
||||||
|
|
@ -315,9 +313,8 @@ class UpdateChecker(commands.Cog):
|
||||||
if channel == 0:
|
if channel == 0:
|
||||||
channel = "Direct Messages"
|
channel = "Direct Messages"
|
||||||
else:
|
else:
|
||||||
try:
|
channel = self.bot.get_channel(channel).name
|
||||||
channel = self.bot.get_channel(channel).name
|
if channel is None:
|
||||||
except:
|
|
||||||
channel = "Unknown"
|
channel = "Unknown"
|
||||||
e.add_field(name="Update Channel", value=channel)
|
e.add_field(name="Update Channel", value=channel)
|
||||||
await ctx.send(embed=e)
|
await ctx.send(embed=e)
|
||||||
|
|
@ -325,9 +322,8 @@ class UpdateChecker(commands.Cog):
|
||||||
if channel == 0:
|
if channel == 0:
|
||||||
channel = "Direct Messages"
|
channel = "Direct Messages"
|
||||||
else:
|
else:
|
||||||
try:
|
channel = self.bot.get_channel(channel).name
|
||||||
channel = self.bot.get_channel(channel).name
|
if channel is None:
|
||||||
except:
|
|
||||||
channel = "Unknown"
|
channel = "Unknown"
|
||||||
message = (
|
message = (
|
||||||
"```css\n"
|
"```css\n"
|
||||||
|
|
@ -363,7 +359,6 @@ class UpdateChecker(commands.Cog):
|
||||||
@whiteblacklist.group()
|
@whiteblacklist.group()
|
||||||
async def whitelist(self, ctx):
|
async def whitelist(self, ctx):
|
||||||
"""Whitelist certain repos from which to receive updates."""
|
"""Whitelist certain repos from which to receive updates."""
|
||||||
pass
|
|
||||||
|
|
||||||
@whitelist.command(name="add")
|
@whitelist.command(name="add")
|
||||||
async def whitelistadd(self, ctx, *repos: Repo):
|
async def whitelistadd(self, ctx, *repos: Repo):
|
||||||
|
|
@ -396,7 +391,6 @@ class UpdateChecker(commands.Cog):
|
||||||
@whiteblacklist.group()
|
@whiteblacklist.group()
|
||||||
async def blacklist(self, ctx):
|
async def blacklist(self, ctx):
|
||||||
"""Blacklist certain repos from which to receive updates."""
|
"""Blacklist certain repos from which to receive updates."""
|
||||||
pass
|
|
||||||
|
|
||||||
@blacklist.command(name="add")
|
@blacklist.command(name="add")
|
||||||
async def blacklistadd(self, ctx, *repos: Repo):
|
async def blacklistadd(self, ctx, *repos: Repo):
|
||||||
|
|
@ -430,7 +424,6 @@ class UpdateChecker(commands.Cog):
|
||||||
@update.group(name="task")
|
@update.group(name="task")
|
||||||
async def _group_update_task(self, ctx):
|
async def _group_update_task(self, ctx):
|
||||||
"""View the status of the task (the one checking for updates)."""
|
"""View the status of the task (the one checking for updates)."""
|
||||||
pass
|
|
||||||
|
|
||||||
@_group_update_task.command()
|
@_group_update_task.command()
|
||||||
async def status(self, ctx):
|
async def status(self, ctx):
|
||||||
|
|
@ -462,5 +455,5 @@ class UpdateChecker(commands.Cog):
|
||||||
message = "No error has been encountered."
|
message = "No error has been encountered."
|
||||||
else:
|
else:
|
||||||
ex = traceback.format_exception(type(e), e, e.__traceback__)
|
ex = traceback.format_exception(type(e), e, e.__traceback__)
|
||||||
message = "An error has been encountered: ```py\n" + "".join(ex) + "```"
|
message = "An error has been encountered:" + box("".join(ex), "py")
|
||||||
await ctx.send(message)
|
await ctx.send(message)
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue