ryuko-ng/robocop_ng/cogs/remind.py

import asyncio
import time
from datetime import datetime

import discord
from discord.ext import commands
from discord.ext.commands import Cog

from robocop_ng.helpers.robocronp import add_job, get_crontab


class Remind(Cog):
    def __init__(self, bot):
        self.bot = bot

    @commands.cooldown(1, 60, type=commands.BucketType.user)
    @commands.command()
    async def remindlist(self, ctx):
        """Lists your reminders."""
        ctab = get_crontab(self.bot)
        uid = str(ctx.author.id)
        embed = discord.Embed(title=f"Active robocronp jobs")
        for jobtimestamp in ctab["remind"]:
            if uid not in ctab["remind"][jobtimestamp]:
                continue
            job_details = ctab["remind"][jobtimestamp][uid]
            expiry_timestr = datetime.utcfromtimestamp(int(jobtimestamp)).strftime(
                "%Y-%m-%d %H:%M:%S (UTC)"
            )
            embed.add_field(
                name=f"Reminder for {expiry_timestr}",
                value=f"Added on: {job_details['added']}, "
                f"Text: {job_details['text']}",
                inline=False,
            )
        await ctx.send(embed=embed)

    @commands.cooldown(1, 60, type=commands.BucketType.user)
    @commands.command(aliases=["remindme"])
    async def remind(self, ctx, when: str, *, text: str = "something"):
        """Reminds you about something."""
        ref_message = None
        if ctx.message.reference is not None:
            ref_message = await ctx.channel.fetch_message(
                ctx.message.reference.message_id
            )

        if ctx.guild:
            await ctx.message.delete()

        current_timestamp = time.time()
        expiry_timestamp = self.bot.parse_time(when)

        if current_timestamp + 5 > expiry_timestamp:
            msg = await ctx.send(
                f"{ctx.author.mention}: Minimum remind interval is 5 seconds."
            )
            await asyncio.sleep(5)
            await msg.delete()
            return

        expiry_datetime = datetime.utcfromtimestamp(expiry_timestamp)
        duration_text = self.bot.get_relative_timestamp(
            time_to=expiry_datetime, include_to=True, humanized=True
        )

        safe_text = await commands.clean_content().convert(ctx, str(text))
        if ref_message is not None:
            safe_text += f"\nMessage reference: {ref_message.jump_url}"

        added_on = datetime.utcnow().strftime("%Y-%m-%d %H:%M:%S (UTC)")

        add_job(
            self.bot,
            "remind",
            ctx.author.id,
            {"text": safe_text, "added": added_on},
            expiry_timestamp,
        )

        await ctx.send(
            f"{ctx.author.mention}: I'll remind you in "
            f"DMs about `{safe_text}` in {duration_text}."
        )


async def setup(bot):
    await bot.add_cog(Remind(bot))