63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
#!/usr/bin/env python3
|
|
# encoding: utf-8
|
|
|
|
import contextlib
|
|
|
|
import discord
|
|
from discord.ext import commands
|
|
|
|
class Meta(commands.Cog):
|
|
# TODO does this need to be configurable?
|
|
INVITE_DURATION_SECONDS = 60 * 60 * 3
|
|
MAX_INVITE_USES = 5
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
@commands.command(aliases=['inv'])
|
|
async def invite(self, context):
|
|
"""Gives you a link to add me to your server."""
|
|
permissions = discord.Permissions()
|
|
permissions.update(**dict.fromkeys((
|
|
'read_messages',
|
|
'send_messages',
|
|
'add_reactions',
|
|
'external_emojis',
|
|
'manage_emojis',
|
|
'embed_links',
|
|
'attach_files',
|
|
), True))
|
|
|
|
await context.send('<%s>' % discord.utils.oauth_url(self.bot.user.id, permissions))
|
|
|
|
@commands.command()
|
|
async def support(self, context):
|
|
"""Directs you to the support server."""
|
|
ch = self.bot.get_channel(self.bot.config['support_server_invite_channel'])
|
|
if ch is None:
|
|
await context.send('This command is temporarily unavailable. Try again later?')
|
|
return
|
|
|
|
reason = f'Created for {context.author} (ID: {context.author.id})'
|
|
invite = await ch.create_invite(max_age=self.INVITE_DURATION_SECONDS, max_uses=self.MAX_INVITE_USES, reason=reason)
|
|
|
|
try:
|
|
await context.author.send(f'Official support server invite: {invite}')
|
|
except discord.Forbidden:
|
|
with contextlib.suppress(discord.HTTPException):
|
|
await context.message.add_reaction(utils.SUCCESS_EMOJIS[True])
|
|
with contextlib.suppress(discord.HTTPException):
|
|
await context.send('Unable to send invite in DMs. Please allow DMs from server members.')
|
|
else:
|
|
try:
|
|
await context.message.add_reaction('📬')
|
|
except discord.HTTPException:
|
|
with contextlib.suppress(discord.HTTPException):
|
|
await context.send('📬')
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Meta(bot))
|
|
|
|
if not bot.config.get('support_server_invite_channel'):
|
|
bot.remove_command('support')
|