mirror of
https://github.com/miaowware/qrm2.git
synced 2024-11-10 10:23:32 -05:00
234 lines
8.7 KiB
Python
234 lines
8.7 KiB
Python
"""
|
|
Base extension for qrm
|
|
---
|
|
Copyright (C) 2019-2020 Abigail Gold, 0x5c
|
|
|
|
This file is part of discord-qrm2 and is released under the terms of
|
|
the GNU General Public License, version 2.
|
|
"""
|
|
|
|
|
|
import random
|
|
import re
|
|
from collections import OrderedDict
|
|
from typing import Union
|
|
|
|
import discord
|
|
import discord.ext.commands as commands
|
|
|
|
import info
|
|
import common as cmn
|
|
|
|
import data.options as opt
|
|
|
|
|
|
class QrmHelpCommand(commands.HelpCommand):
|
|
def __init__(self):
|
|
super().__init__(command_attrs={"help": "Shows help about qrm or a command", "aliases": ["h"]})
|
|
self.verify_checks = True
|
|
|
|
async def get_bot_mapping(self):
|
|
bot = self.context.bot
|
|
mapping = {}
|
|
|
|
for cmd in await self.filter_commands(bot.commands, sort=True):
|
|
cat = cmd.__original_kwargs__.get("category", None)
|
|
if cat in mapping:
|
|
mapping[cat].append(cmd)
|
|
else:
|
|
mapping[cat] = [cmd]
|
|
return mapping
|
|
|
|
async def get_command_signature(self, command):
|
|
parent = command.full_parent_name
|
|
if command.aliases != []:
|
|
aliases = ", ".join(command.aliases)
|
|
fmt = command.name
|
|
if parent:
|
|
fmt = f"{parent} {fmt}"
|
|
alias = fmt
|
|
return f"{opt.prefix}{alias} {command.signature}\n *Aliases:* {aliases}"
|
|
alias = command.name if not parent else f"{parent} {command.name}"
|
|
return f"{opt.prefix}{alias} {command.signature}"
|
|
|
|
async def send_error_message(self, error):
|
|
embed = cmn.embed_factory(self.context)
|
|
embed.title = "qrm Help Error"
|
|
embed.description = error
|
|
embed.colour = cmn.colours.bad
|
|
await self.context.send(embed=embed)
|
|
|
|
async def send_bot_help(self, mapping):
|
|
embed = cmn.embed_factory(self.context)
|
|
embed.title = "qrm Help"
|
|
embed.description = (f"For command-specific help and usage, use `{opt.prefix}help [command name]`."
|
|
" Many commands have shorter aliases.")
|
|
mapping = await mapping
|
|
|
|
for cat, cmds in mapping.items():
|
|
if cmds == []:
|
|
continue
|
|
names = sorted([cmd.name for cmd in cmds])
|
|
if cat is not None:
|
|
embed.add_field(name=cat.title(), value=", ".join(names), inline=False)
|
|
else:
|
|
embed.add_field(name="Other", value=", ".join(names), inline=False)
|
|
await self.context.send(embed=embed)
|
|
|
|
async def send_command_help(self, command):
|
|
if self.verify_checks:
|
|
if not await command.can_run(self.context):
|
|
raise commands.CheckFailure
|
|
for p in command.parents:
|
|
if not await p.can_run(self.context):
|
|
raise commands.CheckFailure
|
|
embed = cmn.embed_factory(self.context)
|
|
embed.title = await self.get_command_signature(command)
|
|
embed.description = command.help
|
|
await self.context.send(embed=embed)
|
|
|
|
async def send_group_help(self, group):
|
|
if self.verify_checks and not await group.can_run(self.context):
|
|
raise commands.CheckFailure
|
|
embed = cmn.embed_factory(self.context)
|
|
embed.title = await self.get_command_signature(group)
|
|
embed.description = group.help
|
|
for cmd in await self.filter_commands(group.commands, sort=True):
|
|
embed.add_field(name=await self.get_command_signature(cmd), value=cmd.help, inline=False)
|
|
await self.context.send(embed=embed)
|
|
|
|
|
|
class BaseCog(commands.Cog):
|
|
def __init__(self, bot: commands.Bot):
|
|
self.bot = bot
|
|
self.changelog = parse_changelog()
|
|
|
|
@commands.command(name="info", aliases=["about"])
|
|
async def _info(self, ctx: commands.Context):
|
|
"""Shows info about qrm."""
|
|
embed = cmn.embed_factory(ctx)
|
|
embed.title = "About qrm"
|
|
embed.description = info.description
|
|
|
|
embed.add_field(name="Authors", value=", ".join(info.authors))
|
|
embed.add_field(name="License", value=info.license)
|
|
embed.add_field(name="Version", value=f"v{info.release}")
|
|
embed.add_field(name="Contributing", value=info.contributing, inline=False)
|
|
embed.add_field(name="Official Server", value=info.bot_server, inline=False)
|
|
embed.set_thumbnail(url=str(self.bot.user.avatar_url))
|
|
await ctx.send(embed=embed)
|
|
|
|
@commands.command(name="ping", aliases=["beep"])
|
|
async def _ping(self, ctx: commands.Context):
|
|
"""Show the current latency to the discord endpoint."""
|
|
embed = cmn.embed_factory(ctx)
|
|
content = ""
|
|
if ctx.invoked_with == "beep":
|
|
embed.title = "**Boop!**"
|
|
else:
|
|
content = ctx.message.author.mention if random.random() < 0.05 else ""
|
|
embed.title = "🏓 **Pong!**"
|
|
embed.description = f"Current ping is {self.bot.latency*1000:.1f} ms"
|
|
await ctx.send(content, embed=embed)
|
|
|
|
@commands.command(name="changelog", aliases=["clog"])
|
|
async def _changelog(self, ctx: commands.Context, version: str = "latest"):
|
|
"""Show what has changed in a bot version."""
|
|
embed = cmn.embed_factory(ctx)
|
|
embed.title = "qrm Changelog"
|
|
embed.description = ("For a full listing, visit [Github](https://"
|
|
"github.com/classabbyamp/discord-qrm2/blob/master/CHANGELOG.md).")
|
|
changelog = self.changelog
|
|
vers = list(changelog.keys())
|
|
vers.remove("Unreleased")
|
|
|
|
version = version.lower()
|
|
|
|
if version == "latest":
|
|
version = info.release
|
|
if version == "unreleased":
|
|
version = "Unreleased"
|
|
|
|
try:
|
|
log = changelog[version]
|
|
except KeyError:
|
|
embed.title += ": Version Not Found"
|
|
embed.description += "\n\n**Valid versions:** latest, "
|
|
embed.description += ", ".join(vers)
|
|
embed.colour = cmn.colours.bad
|
|
await ctx.send(embed=embed)
|
|
return
|
|
|
|
if "date" in log:
|
|
embed.description += f"\n\n**v{version}** ({log['date']})"
|
|
else:
|
|
embed.description += f"\n\n**v{version}**"
|
|
embed = await format_changelog(log, embed)
|
|
|
|
await ctx.send(embed=embed)
|
|
|
|
@commands.command(name="issue")
|
|
async def _issue(self, ctx: commands.Context):
|
|
"""Shows how to create an issue for the bot."""
|
|
embed = cmn.embed_factory(ctx)
|
|
embed.title = "Found a bug? Have a feature request?"
|
|
embed.description = ("Submit an issue on the [issue tracker]"
|
|
"(https://github.com/classabbyamp/discord-qrm2/issues)!")
|
|
await ctx.send(embed=embed)
|
|
|
|
@commands.command(name="echo", aliases=["e"], category=cmn.cat.admin)
|
|
@commands.check(cmn.check_if_owner)
|
|
async def _echo(self, ctx: commands.Context,
|
|
channel: Union[cmn.GlobalChannelConverter, commands.UserConverter], *, msg: str):
|
|
"""Send a message in a channel as qrm. Accepts channel/user IDs/mentions.
|
|
Channel names are current-guild only.
|
|
Does not work with the ID of the bot user."""
|
|
if isinstance(channel, discord.ClientUser):
|
|
raise commands.BadArgument("Can't send to the bot user!")
|
|
await channel.send(msg)
|
|
|
|
|
|
def parse_changelog():
|
|
changelog = OrderedDict()
|
|
ver = ""
|
|
heading = ""
|
|
|
|
with open("CHANGELOG.md") as changelog_file:
|
|
for line in changelog_file.readlines():
|
|
if line.strip() == "":
|
|
continue
|
|
if re.match(r"##[^#]", line):
|
|
ver_match = re.match(r"\[(.+)\](?: - )?(\d{4}-\d{2}-\d{2})?", line.lstrip("#").strip())
|
|
if ver_match is not None:
|
|
ver = ver_match.group(1)
|
|
changelog[ver] = dict()
|
|
if ver_match.group(2):
|
|
changelog[ver]["date"] = ver_match.group(2)
|
|
elif re.match(r"###[^#]", line):
|
|
heading = line.lstrip("#").strip()
|
|
changelog[ver][heading] = []
|
|
elif ver != "" and heading != "":
|
|
if line.startswith("-"):
|
|
changelog[ver][heading].append(line.lstrip("-").strip())
|
|
return changelog
|
|
|
|
|
|
async def format_changelog(log: dict, embed: discord.Embed):
|
|
for header, lines in log.items():
|
|
formatted = ""
|
|
if header != "date":
|
|
for line in lines:
|
|
formatted += f"- {line}\n"
|
|
embed.add_field(name=f"**{header}**", value=formatted, inline=False)
|
|
return embed
|
|
|
|
|
|
def setup(bot: commands.Bot):
|
|
bot.add_cog(BaseCog(bot))
|
|
bot._original_help_command = bot.help_command
|
|
bot.help_command = QrmHelpCommand()
|
|
|
|
|
|
def teardown(bot: commands.Bot):
|
|
bot.help_command = bot._original_help_command
|