diff --git a/extensions/__init__.py b/extensions/__init__.py index 50583da..f26c4db 100644 --- a/extensions/__init__.py +++ b/extensions/__init__.py @@ -10,3 +10,4 @@ bot.load_extension("extensions.presentation") bot.load_extension("extensions.rorec") bot.load_extension("extensions.warn") bot.load_extension("extensions.pcp") +bot.load_extension("extensions.tex") diff --git a/extensions/tex.py b/extensions/tex.py new file mode 100644 index 0000000..f94620d --- /dev/null +++ b/extensions/tex.py @@ -0,0 +1,54 @@ +from urllib.parse import urlencode + +from discord import Embed +from discord.ext import commands + +from administrator.logger import logger + + +extension_name = "TeX" + + +class TeX(commands.Cog): + def __init__(self, bot: commands.Bot): + self.bot = bot + self.polls = {} + + def description(self): + return "Render TeX formula" + + @commands.group("tex", pass_context=True) + async def tex(self, ctx: commands.Context): + if ctx.message.content.count("`") == 2: + start = ctx.message.content.find("`") + end = ctx.message.content.find("`", start+1) + command = ctx.message.content[start+1:end] + await ctx.send(f"https://chart.apis.google.com/chart?cht=tx&chs=40&{urlencode({'chl': command})}") + elif ctx.invoked_subcommand is None: + await ctx.invoke(self.tex_help) + + @tex.group("help", pass_context=True) + async def tex_help(self, ctx: commands.Context): + embed = Embed(title="TeX help") + embed.add_field(name="tex \`formula\`", value="Render a TeX formula", inline=False) + await ctx.send(embed=embed) + + +def setup(bot): + logger.info(f"Loading...") + try: + bot.add_cog(TeX(bot)) + except Exception as e: + logger.error(f"Error loading: {e}") + else: + logger.info(f"Load successful") + + +def teardown(bot): + logger.info(f"Unloading...") + try: + bot.remove_cog("TeX") + except Exception as e: + logger.error(f"Error unloading: {e}") + else: + logger.info(f"Unload successful")