You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
320 lines
11 KiB
320 lines
11 KiB
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
The MIT License (MIT)
|
|
|
|
Copyright (c) 2015-2017 Rapptz
|
|
|
|
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.
|
|
"""
|
|
|
|
import discord
|
|
import asyncio
|
|
import re
|
|
import inspect
|
|
|
|
from .errors import BadArgument, NoPrivateMessage
|
|
from .view import StringView
|
|
|
|
__all__ = [ 'Converter', 'MemberConverter', 'UserConverter',
|
|
'TextChannelConverter', 'InviteConverter', 'RoleConverter',
|
|
'GameConverter', 'ColourConverter', 'VoiceChannelConverter',
|
|
'EmojiConverter', 'IDConverter', 'clean_content' ]
|
|
|
|
def _get_from_guilds(bot, getter, argument):
|
|
result = None
|
|
for guild in bot.guilds:
|
|
result = getattr(guild, getter)(argument)
|
|
if result:
|
|
return result
|
|
return result
|
|
|
|
class Converter:
|
|
"""The base class of custom converters that require the :class:`Context`
|
|
to be passed to be useful.
|
|
|
|
This allows you to implement converters that function similar to the
|
|
special cased ``discord`` classes.
|
|
|
|
Classes that derive from this should override the :meth:`convert` method
|
|
to do its conversion logic. This method could be a coroutine or a regular
|
|
function.
|
|
|
|
Before the convert method is called, :meth:`prepare` is called. This
|
|
method must set the attributes below if overwritten.
|
|
|
|
Attributes
|
|
-----------
|
|
ctx: :class:`Context`
|
|
The invocation context that the argument is being used in.
|
|
argument: str
|
|
The argument that is being converted.
|
|
"""
|
|
def prepare(self, ctx, argument):
|
|
self.ctx = ctx
|
|
self.argument = argument
|
|
|
|
def convert(self):
|
|
raise NotImplementedError('Derived classes need to implement this.')
|
|
|
|
class IDConverter(Converter):
|
|
def __init__(self):
|
|
self._id_regex = re.compile(r'([0-9]{15,21})$')
|
|
|
|
def _get_id_match(self):
|
|
return self._id_regex.match(self.argument)
|
|
|
|
class MemberConverter(IDConverter):
|
|
def convert(self):
|
|
message = self.ctx.message
|
|
bot = self.ctx.bot
|
|
match = self._get_id_match() or re.match(r'<@!?([0-9]+)>$', self.argument)
|
|
guild = message.guild
|
|
result = None
|
|
if match is None:
|
|
# not a mention...
|
|
if guild:
|
|
result = guild.get_member_named(self.argument)
|
|
else:
|
|
result = _get_from_guilds(bot, 'get_member_named', self.argument)
|
|
else:
|
|
user_id = int(match.group(1))
|
|
if guild:
|
|
result = guild.get_member(user_id)
|
|
else:
|
|
result = _get_from_guilds(bot, 'get_member', user_id)
|
|
|
|
if result is None:
|
|
raise BadArgument('Member "{}" not found'.format(self.argument))
|
|
|
|
return result
|
|
|
|
class UserConverter(IDConverter):
|
|
def convert(self):
|
|
match = self._get_id_match() or re.match(r'<@!?([0-9]+)>$', self.argument)
|
|
result = None
|
|
state = self.ctx._state
|
|
|
|
if match is not None:
|
|
user_id = int(match.group(1))
|
|
result = self.ctx.bot.get_user(user_id)
|
|
else:
|
|
arg = self.argument
|
|
# check for discriminator if it exists
|
|
if len(arg) > 5 and arg[-5] == '#':
|
|
discrim = arg[-4:]
|
|
name = arg[:-5]
|
|
predicate = lambda u: u.name == name and u.discriminator == discrim
|
|
result = discord.utils.find(predicate, state._users.values())
|
|
if result is not None:
|
|
return result
|
|
|
|
predicate = lambda u: u.name == arg
|
|
result = discord.utils.find(predicate, state._users.values())
|
|
|
|
if result is None:
|
|
raise BadArgument('User "{}" not found'.format(self.argument))
|
|
|
|
return result
|
|
|
|
class TextChannelConverter(IDConverter):
|
|
def convert(self):
|
|
bot = self.ctx.bot
|
|
|
|
match = self._get_id_match() or re.match(r'<#([0-9]+)>$', self.argument)
|
|
result = None
|
|
guild = self.ctx.guild
|
|
|
|
if match is None:
|
|
# not a mention
|
|
if guild:
|
|
result = discord.utils.get(guild.text_channels, name=self.argument)
|
|
else:
|
|
def check(c):
|
|
return isinstance(c, discord.TextChannel) and c.name == self.argument
|
|
result = discord.utils.find(check, bot.get_all_channels())
|
|
else:
|
|
channel_id = int(match.group(1))
|
|
if guild:
|
|
result = guild.get_channel(channel_id)
|
|
else:
|
|
result = _get_from_guilds(bot, 'get_channel', channel_id)
|
|
|
|
if not isinstance(result, discord.TextChannel):
|
|
raise BadArgument('Channel "{}" not found.'.format(self.argument))
|
|
|
|
return result
|
|
|
|
class VoiceChannelConverter(IDConverter):
|
|
def convert(self):
|
|
bot = self.ctx.bot
|
|
|
|
match = self._get_id_match() or re.match(r'<#([0-9]+)>$', self.argument)
|
|
result = None
|
|
guild = self.ctx.guild
|
|
|
|
if match is None:
|
|
# not a mention
|
|
if guild:
|
|
result = discord.utils.get(guild.voice_channels, name=self.argument)
|
|
else:
|
|
def check(c):
|
|
return isinstance(c, discord.VoiceChannel) and c.name == self.argument
|
|
result = discord.utils.find(check, bot.get_all_channels())
|
|
else:
|
|
channel_id = int(match.group(1))
|
|
if guild:
|
|
result = guild.get_channel(channel_id)
|
|
else:
|
|
result = _get_from_guilds(bot, 'get_channel', channel_id)
|
|
|
|
if not isinstance(result, discord.VoiceChannel):
|
|
raise BadArgument('Channel "{}" not found.'.format(self.argument))
|
|
|
|
return result
|
|
|
|
class ColourConverter(Converter):
|
|
def convert(self):
|
|
arg = self.argument.replace('0x', '').lower()
|
|
|
|
if arg[0] == '#':
|
|
arg = arg[1:]
|
|
try:
|
|
value = int(arg, base=16)
|
|
return discord.Colour(value=value)
|
|
except ValueError:
|
|
method = getattr(discord.Colour, arg, None)
|
|
if method is None or not inspect.ismethod(method):
|
|
raise BadArgument('Colour "{}" is invalid.'.format(arg))
|
|
return method()
|
|
|
|
class RoleConverter(IDConverter):
|
|
def convert(self):
|
|
guild = self.ctx.message.guild
|
|
if not guild:
|
|
raise NoPrivateMessage()
|
|
|
|
match = self._get_id_match() or re.match(r'<@&([0-9]+)>$', self.argument)
|
|
params = dict(id=int(match.group(1))) if match else dict(name=self.argument)
|
|
result = discord.utils.get(guild.roles, **params)
|
|
if result is None:
|
|
raise BadArgument('Role "{}" not found.'.format(self.argument))
|
|
return result
|
|
|
|
class GameConverter(Converter):
|
|
def convert(self):
|
|
return discord.Game(name=self.argument)
|
|
|
|
class InviteConverter(Converter):
|
|
@asyncio.coroutine
|
|
def convert(self):
|
|
try:
|
|
invite = yield from self.ctx.bot.get_invite(self.argument)
|
|
return invite
|
|
except Exception as e:
|
|
raise BadArgument('Invite is invalid or expired') from e
|
|
|
|
class EmojiConverter(IDConverter):
|
|
@asyncio.coroutine
|
|
def convert(self):
|
|
message = self.ctx.message
|
|
bot = self.ctx.bot
|
|
|
|
match = self._get_id_match() or re.match(r'<:[a-zA-Z0-9]+:([0-9]+)>$', self.argument)
|
|
result = None
|
|
guild = message.guild
|
|
if match is None:
|
|
# Try to get the emoji by name. Try local guild first.
|
|
if guild:
|
|
result = discord.utils.get(guild.emojis, name=self.argument)
|
|
|
|
if result is None:
|
|
result = discord.utils.get(bot.emojis, name=self.argument)
|
|
else:
|
|
emoji_id = int(match.group(1))
|
|
|
|
# Try to look up emoji by id.
|
|
if guild:
|
|
result = discord.utils.get(guild.emojis, id=emoji_id)
|
|
|
|
if result is None:
|
|
result = discord.utils.get(bot.emojis, id=emoji_id)
|
|
|
|
if result is None:
|
|
raise BadArgument('Emoji "{}" not found.'.format(self.argument))
|
|
|
|
return result
|
|
|
|
class clean_content(Converter):
|
|
def __init__(self, *, fix_channel_mentions=False, use_nicknames=True):
|
|
self.fix_channel_mentions = fix_channel_mentions
|
|
self.use_nicknames = use_nicknames
|
|
|
|
def convert(self):
|
|
message = self.ctx.message
|
|
transformations = {}
|
|
|
|
if self.fix_channel_mentions:
|
|
transformations.update(
|
|
('<#%s>' % channel.id, '#' + channel.name)
|
|
for channel in message.channel_mentions
|
|
)
|
|
|
|
if self.use_nicknames:
|
|
transformations.update(
|
|
('<@%s>' % member.id, '@' + member.display_name)
|
|
for member in message.mentions
|
|
)
|
|
|
|
transformations.update(
|
|
('<@!%s>' % member.id, '@' + member.display_name)
|
|
for member in message.mentions
|
|
)
|
|
else:
|
|
transformations.update(
|
|
('<@%s>' % member.id, '@' + member.name)
|
|
for member in message.mentions
|
|
)
|
|
|
|
transformations.update(
|
|
('<@!%s>' % member.id, '@' + member.name)
|
|
for member in message.mentions
|
|
)
|
|
|
|
transformations.update(
|
|
('<@&%s>' % role.id, '@' + role.name)
|
|
for role in message.role_mentions
|
|
)
|
|
|
|
def repl(obj):
|
|
return transformations.get(obj.group(0), '')
|
|
|
|
pattern = re.compile('|'.join(transformations.keys()))
|
|
result = pattern.sub(repl, self.argument)
|
|
|
|
transformations = {
|
|
'@everyone': '@\u200beveryone',
|
|
'@here': '@\u200bhere'
|
|
}
|
|
|
|
def repl2(obj):
|
|
return transformations.get(obj.group(0), '')
|
|
|
|
pattern = re.compile('|'.join(transformations.keys()))
|
|
return pattern.sub(repl2, result)
|
|
|