SeaCogs/updatechecker/updatechecker.py

"""
MIT License

Copyright (c) 2018-Present NeuroAssassin

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
"""

# Huge thanks to Sinbad for allowing me to copy parts of his RSS cog
# (https://github.com/mikeshardmind/SinbadCogs/tree/v3/rss), which I
# used to grab the latest commits from repositories.

# Also, the code I use for updating repos I took directly from Red,
# and just took out the message interactions

import asyncio
import traceback
from datetime import datetime

import aiohttp
import discord
from redbot.cogs.downloader.repo_manager import Repo
from redbot.core import Config, commands
from redbot.core.utils.chat_formatting import humanize_list, inline

import feedparser


class UpdateChecker(commands.Cog):
    """Get notices or auto-update cogs when an update is available for it's repo"""

    def __init__(self, bot):
        self.bot = bot
        self.session = aiohttp.ClientSession()
        self.conf = Config.get_conf(self, identifier=473541068378341376)
        default_global = {
            "repos": {},
            "auto": False,
            "gochannel": 0,
            "embed": True,
            "whitelist": [],
            "blacklist": [],
        }
        self.conf.register_global(**default_global)
        self.task = self.bot.loop.create_task(self.bg_task())

    def cog_unload(self):
        self.__unload()

    def __unload(self):
        self.task.cancel()
        self.session.detach()

    async def red_delete_data_for_user(self, **kwargs):
        """This cog does not store user data"""
        return

    async def bg_task(self):
        await self.bot.wait_until_ready()
        # Just in case
        await asyncio.sleep(10)
        while True:
            cog = self.bot.get_cog("Downloader")
            if cog is not None:
                data = await self.conf.all()
                repos = data["repos"]
                auto = data["auto"]
                channel = data["gochannel"]
                use_embed = data["embed"]
                whitelist = data["whitelist"]
                blacklist = data["blacklist"]
                if channel:
                    channel = self.bot.get_channel(channel)
                    if channel is None:
                        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."
                        )
                        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()
                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)
                    if not repo:
                        continue
                    url = repo.url + r"/commits/" + repo.branch + ".atom"
                    response = await self.fetch_feed(url)
                    try:
                        commit = response.entries[0]["id"][33:]
                        hash = "[" + 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
                    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 != commit_saved
                        and cn != commit_saved
                        and commit_saved != "--default--"
                    ):
                        if True:  # KACHOW
                            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=hash)
                                e.set_thumbnail(url=image)
                            else:
                                e = (
                                    "```css\n"
                                    "[Update Checker]"
                                    "``````css\n"
                                    f"    Repo: {repo.name}\n"
                                    f"     URL: {repo.url}\n"
                                    f"  Commit: {cn}\n"
                                    f"    Hash: {commit}\n"
                                    f"    Time: {datetime.utcnow()}"
                                    "```"
                                )
                            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(
                                    "[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."
                                )
                                if use_embed:
                                    await self.bot.send_to_owners(embed=e)
                                else:
                                    await self.bot.send_to_owners(e)
                                await self.conf.gochannel.set(0)
                        else:
                            try:
                                await channel.send(
                                    f"[Update Checker] Update found for repo: {repo.name}.  Updating repos..."
                                )
                            except AttributeError:
                                owner = (await self.bot.application_info()).owner
                                await owner.send(
                                    "[Update Checker] It appears that the channel for this cog has been deleted.  From now on, it will DM you."
                                )
                                channel = owner
                                await self.conf.gochannel.set(0)
                            except discord.errors.Forbidden:
                                owner = (await self.bot.application_info()).owner
                                await owner.send(
                                    "[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)
                            # Just a copy of `[p]cog update`, but without using ctx things
                            try:
                                installed_cogs = set(await cog.installed_cogs())
                                updated = await cog._repo_manager.update_all_repos()
                                updated_cogs = set(
                                    cog for repo in updated for cog in repo.available_cogs
                                )
                                installed_and_updated = updated_cogs & installed_cogs
                                if installed_and_updated:
                                    await cog._reinstall_requirements(installed_and_updated)
                                    await cog._reinstall_cogs(installed_and_updated)
                                    await cog._reinstall_libraries(installed_and_updated)
                                    cognames = {c.name for c in installed_and_updated}
                                    message = humanize_list(tuple(map(inline, cognames)))
                            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)
            await asyncio.sleep(60)

    async def fetch_feed(self, url: str):
        # Thank's to Sinbad's rss cog after which I copied this
        timeout = aiohttp.client.ClientTimeout(total=15)
        try:
            async with self.session.get(url, timeout=timeout) as response:
                data = await response.read()
        except (aiohttp.ClientError, asyncio.TimeoutError):
            return None

        ret = feedparser.parse(data)
        if ret.bozo:
            return None
        return ret

    @commands.is_owner()
    @commands.group(name="cogupdater", aliases=["cu"])
    async def update(self, ctx):
        """Group command for controlling the update checker cog."""
        pass

    @commands.is_owner()
    @update.command()
    async def auto(self, ctx):
        """Changes automatic cog updates to the opposite setting."""
        if False:  # KACHOW
            auto = await self.conf.auto()
            await self.conf.auto.set(not auto)
            status = "disabled" if auto else "enabled"
            await ctx.send(f"Auto cog updates are now {status}")
        else:
            await ctx.send(
                "This command is disabled for the time being.  Cog updates will not run automatically, however notifications will still send."
            )

    @commands.is_owner()
    @update.command()
    async def channel(self, ctx, channel: discord.TextChannel = None):
        """
        Sets a channel for update messages to go to.

        If argument is not supplied, it will be sent to the default notifications channel(s) specified in `[p]set ownernotifications`.
        By default, this goes to owner DMs.
        """
        if channel:
            await self.conf.gochannel.set(channel.id)
            await ctx.send(f"Update messages will now be sent to {channel.mention}")
        else:
            await self.conf.gochannel.set(0)
            await ctx.send("Update messages will now be DMed to you.")

    @commands.is_owner()
    @update.command()
    async def settings(self, ctx):
        """See settings for the Update Checker cog.

        Right now, this shows whether the bot updates cogs automatically and what channel logs are sent to.
        """
        auto = await self.conf.auto()
        channel = await self.conf.gochannel()
        embed = await self.conf.embed()
        if embed:
            e = discord.Embed(title="Update Checker settings", color=0x00FF00)
            e.add_field(name="Automatic Cog Updates", value=str(auto))
            if channel == 0:
                channel = "Direct Messages"
            else:
                try:
                    channel = self.bot.get_channel(channel).name
                except:
                    channel = "Unknown"
            e.add_field(name="Update Channel", value=channel)
            await ctx.send(embed=e)
        else:
            if channel == 0:
                channel = "Direct Messages"
            else:
                try:
                    channel = self.bot.get_channel(channel).name
                except:
                    channel = "Unknown"
            message = (
                "```css\n"
                "[Update Checker settings]"
                "``````css\n"
                f"[Automatic Cog Updates]: {str(auto)}\n"
                f"       [Update Channel]: {channel}"
                "```"
            )
            await ctx.send(message)

    @commands.is_owner()
    @update.command()
    async def embed(self, ctx):
        """Toggles whether to use embeds or colorful codeblock messages when sending an update."""
        c = await self.conf.embed()
        await self.conf.embed.set(not c)
        word = "disabled" if c else "enabled"
        await ctx.send(f"Embeds are now {word}")

    @commands.is_owner()
    @update.group(name="list")
    async def whiteblacklist(self, ctx):
        """Whitelist/blacklist certain repositories from which to receive updates."""
        if ctx.invoked_subcommand is None:
            data = await self.conf.all()
            whitelist = data["whitelist"]
            blacklist = data["blacklist"]
            await ctx.send(
                f"Whitelisted: {humanize_list(tuple(map(inline, whitelist or ['None'])))}\nBlacklisted: {humanize_list(tuple(map(inline, blacklist or ['None'])))}"
            )

    @whiteblacklist.group()
    async def whitelist(self, ctx):
        """Whitelist certain repos from which to receive updates."""
        pass

    @whitelist.command(name="add")
    async def whitelistadd(self, ctx, *repos: Repo):
        """Add repos to the whitelist"""
        data = await self.conf.whitelist()
        ds = set(data)
        ns = set([r.name for r in repos])
        ss = ds | ns
        await self.conf.whitelist.set(list(ss))
        await ctx.send(f"Whitelist update successful: {humanize_list(tuple(map(inline, ss)))}")

    @whitelist.command(name="remove")
    async def whitelistremove(self, ctx, *repos: Repo):
        """Remove repos from the whitelist"""
        data = await self.conf.whitelist()
        ds = set(data)
        ns = set([r.name for r in repos])
        ss = ds - ns
        await self.conf.whitelist.set(list(ss))
        await ctx.send(
            f"Whitelist update successful: {humanize_list(tuple(map(inline, ss or ['None'])))}"
        )

    @whitelist.command(name="clear")
    async def whitelistclear(self, ctx):
        """Removes all repos from the whitelist"""
        await self.conf.whitelist.set([])
        await ctx.send("Whitelist update successful")

    @whiteblacklist.group()
    async def blacklist(self, ctx):
        """Blacklist certain repos from which to receive updates."""
        pass

    @blacklist.command(name="add")
    async def blacklistadd(self, ctx, *repos: Repo):
        """Add repos to the blacklist"""
        data = await self.conf.blacklist()
        ds = set(data)
        ns = set([r.name for r in repos])
        ss = ds | ns
        await self.conf.blacklist.set(list(ss))
        await ctx.send(f"Backlist update successful: {humanize_list(tuple(map(inline, ss)))}")

    @blacklist.command(name="remove")
    async def blacklistremove(self, ctx, *repos: Repo):
        """Remove repos from the blacklist"""
        data = await self.conf.blacklist()
        ds = set(data)
        ns = set([r.name for r in repos])
        ss = ds - ns
        await self.conf.blacklist.set(list(ss))
        await ctx.send(
            f"Blacklist update successful: {humanize_list(tuple(map(inline, ss or ['None'])))}"
        )

    @blacklist.command(name="clear")
    async def blacklistclear(self, ctx):
        """Removes all repos from the blacklist"""
        await self.conf.blacklist.set([])
        await ctx.send("Blacklist update successful")

    @commands.is_owner()
    @update.group(name="task")
    async def _group_update_task(self, ctx):
        """View the status of the task (the one checking for updates)."""
        pass

    @_group_update_task.command()
    async def status(self, ctx):
        """Get the current status of the update task."""
        message = "Task is currently "
        cancelled = self.task.cancelled()
        if cancelled:
            message += "canceled."
        else:
            done = self.task.done()
            if done:
                message += "done."
            else:
                message += "running."
        try:
            self.task.exception()
        except asyncio.exceptions.InvalidStateError:
            message += "  No error has been encountered."
        else:
            message += "  An error has been encountered.  Please run `[p]cogupdater task error` and report it to Neuro Assassin on the help server."
        await ctx.send(message)

    @_group_update_task.command()
    async def error(self, ctx):
        """Gets the latest error of the update task."""
        try:
            e = self.task.exception()
        except asyncio.exceptions.InvalidStateError:
            message = "No error has been encountered."
        else:
            ex = traceback.format_exception(type(e), e, e.__traceback__)
            message = "An error has been encountered: ```py\n" + "".join(ex) + "```"
        await ctx.send(message)