mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-06 11:18:54 -05:00
d13bf37845
This also wraps some fields of the modlog with the same sanitization, as well as the `[p]names` command.
1839 lines
68 KiB
Python
1839 lines
68 KiB
Python
import asyncio
|
|
import contextlib
|
|
from datetime import datetime, timedelta
|
|
from collections import deque, defaultdict, namedtuple
|
|
from typing import Optional, Union, cast
|
|
|
|
import discord
|
|
|
|
from discord.ext.commands.converter import Converter, Greedy
|
|
from discord.ext.commands.errors import BadArgument
|
|
from redbot.core import checks, Config, modlog, commands
|
|
from redbot.core.bot import Red
|
|
from redbot.core.i18n import Translator, cog_i18n
|
|
from redbot.core.utils.chat_formatting import box, escape, pagify, format_perms_list
|
|
from redbot.core.utils.common_filters import (
|
|
filter_invites,
|
|
filter_various_mentions,
|
|
escape_spoilers,
|
|
)
|
|
from redbot.core.utils.mod import is_mod_or_superior, is_allowed_by_hierarchy, get_audit_reason
|
|
from .log import log
|
|
|
|
_ = T_ = Translator("Mod", __file__)
|
|
|
|
|
|
class RawUserIds(Converter):
|
|
async def convert(self, ctx, argument):
|
|
# This is for the hackban command, where we receive IDs that
|
|
# are most likely not in the guild.
|
|
# As long as it's numeric and long enough, it makes a good candidate
|
|
# to attempt a ban on
|
|
if argument.isnumeric() and len(argument) >= 17:
|
|
return int(argument)
|
|
|
|
raise BadArgument("{} doesn't look like a valid user ID.".format(argument))
|
|
|
|
|
|
@cog_i18n(_)
|
|
class Mod(commands.Cog):
|
|
"""Moderation tools."""
|
|
|
|
default_guild_settings = {
|
|
"ban_mention_spam": False,
|
|
"delete_repeats": False,
|
|
"ignored": False,
|
|
"respect_hierarchy": True,
|
|
"delete_delay": -1,
|
|
"reinvite_on_unban": False,
|
|
"current_tempbans": [],
|
|
}
|
|
|
|
default_channel_settings = {"ignored": False}
|
|
|
|
default_member_settings = {"past_nicks": [], "perms_cache": {}, "banned_until": False}
|
|
|
|
default_user_settings = {"past_names": []}
|
|
|
|
def __init__(self, bot: Red):
|
|
super().__init__()
|
|
self.bot = bot
|
|
self.settings = Config.get_conf(self, 4961522000, force_registration=True)
|
|
|
|
self.settings.register_guild(**self.default_guild_settings)
|
|
self.settings.register_channel(**self.default_channel_settings)
|
|
self.settings.register_member(**self.default_member_settings)
|
|
self.settings.register_user(**self.default_user_settings)
|
|
self.ban_queue = []
|
|
self.unban_queue = []
|
|
self.cache = defaultdict(lambda: deque(maxlen=3))
|
|
|
|
self.registration_task = self.bot.loop.create_task(self._casetype_registration())
|
|
self.tban_expiry_task = self.bot.loop.create_task(self.check_tempban_expirations())
|
|
self.last_case = defaultdict(dict)
|
|
|
|
def __unload(self):
|
|
self.registration_task.cancel()
|
|
self.tban_expiry_task.cancel()
|
|
|
|
@staticmethod
|
|
async def _casetype_registration():
|
|
casetypes_to_register = [
|
|
{
|
|
"name": "ban",
|
|
"default_setting": True,
|
|
"image": "\N{HAMMER}",
|
|
"case_str": "Ban",
|
|
"audit_type": "ban",
|
|
},
|
|
{
|
|
"name": "kick",
|
|
"default_setting": True,
|
|
"image": "\N{WOMANS BOOTS}",
|
|
"case_str": "Kick",
|
|
"audit_type": "kick",
|
|
},
|
|
{
|
|
"name": "hackban",
|
|
"default_setting": True,
|
|
"image": "\N{BUST IN SILHOUETTE}\N{HAMMER}",
|
|
"case_str": "Hackban",
|
|
"audit_type": "ban",
|
|
},
|
|
{
|
|
"name": "tempban",
|
|
"default_setting": True,
|
|
"image": "\N{ALARM CLOCK}\N{HAMMER}",
|
|
"case_str": "Tempban",
|
|
"audit_type": "ban",
|
|
},
|
|
{
|
|
"name": "softban",
|
|
"default_setting": True,
|
|
"image": "\N{DASH SYMBOL}\N{HAMMER}",
|
|
"case_str": "Softban",
|
|
"audit_type": "ban",
|
|
},
|
|
{
|
|
"name": "unban",
|
|
"default_setting": True,
|
|
"image": "\N{DOVE OF PEACE}",
|
|
"case_str": "Unban",
|
|
"audit_type": "unban",
|
|
},
|
|
{
|
|
"name": "voiceban",
|
|
"default_setting": True,
|
|
"image": "\N{SPEAKER WITH CANCELLATION STROKE}",
|
|
"case_str": "Voice Ban",
|
|
"audit_type": "member_update",
|
|
},
|
|
{
|
|
"name": "voiceunban",
|
|
"default_setting": True,
|
|
"image": "\N{SPEAKER}",
|
|
"case_str": "Voice Unban",
|
|
"audit_type": "member_update",
|
|
},
|
|
{
|
|
"name": "vmute",
|
|
"default_setting": False,
|
|
"image": "\N{SPEAKER WITH CANCELLATION STROKE}",
|
|
"case_str": "Voice Mute",
|
|
"audit_type": "overwrite_update",
|
|
},
|
|
{
|
|
"name": "cmute",
|
|
"default_setting": False,
|
|
"image": "\N{SPEAKER WITH CANCELLATION STROKE}",
|
|
"case_str": "Channel Mute",
|
|
"audit_type": "overwrite_update",
|
|
},
|
|
{
|
|
"name": "smute",
|
|
"default_setting": True,
|
|
"image": "\N{SPEAKER WITH CANCELLATION STROKE}",
|
|
"case_str": "Server Mute",
|
|
"audit_type": "overwrite_update",
|
|
},
|
|
{
|
|
"name": "vunmute",
|
|
"default_setting": False,
|
|
"image": "\N{SPEAKER}",
|
|
"case_str": "Voice Unmute",
|
|
"audit_type": "overwrite_update",
|
|
},
|
|
{
|
|
"name": "cunmute",
|
|
"default_setting": False,
|
|
"image": "\N{SPEAKER}",
|
|
"case_str": "Channel Unmute",
|
|
"audit_type": "overwrite_update",
|
|
},
|
|
{
|
|
"name": "sunmute",
|
|
"default_setting": True,
|
|
"image": "\N{SPEAKER}",
|
|
"case_str": "Server Unmute",
|
|
"audit_type": "overwrite_update",
|
|
},
|
|
]
|
|
try:
|
|
await modlog.register_casetypes(casetypes_to_register)
|
|
except RuntimeError:
|
|
pass
|
|
|
|
@commands.group()
|
|
@commands.guild_only()
|
|
@checks.guildowner_or_permissions(administrator=True)
|
|
async def modset(self, ctx: commands.Context):
|
|
"""Manage server administration settings."""
|
|
if ctx.invoked_subcommand is None:
|
|
guild = ctx.guild
|
|
# Display current settings
|
|
delete_repeats = await self.settings.guild(guild).delete_repeats()
|
|
ban_mention_spam = await self.settings.guild(guild).ban_mention_spam()
|
|
respect_hierarchy = await self.settings.guild(guild).respect_hierarchy()
|
|
delete_delay = await self.settings.guild(guild).delete_delay()
|
|
reinvite_on_unban = await self.settings.guild(guild).reinvite_on_unban()
|
|
msg = ""
|
|
msg += _("Delete repeats: {yes_or_no}\n").format(
|
|
yes_or_no=_("Yes") if delete_repeats else _("No")
|
|
)
|
|
msg += _("Ban mention spam: {num_mentions}\n").format(
|
|
num_mentions=_("{num} mentions").format(num=ban_mention_spam)
|
|
if ban_mention_spam
|
|
else _("No")
|
|
)
|
|
msg += _("Respects hierarchy: {yes_or_no}\n").format(
|
|
yes_or_no=_("Yes") if respect_hierarchy else _("No")
|
|
)
|
|
msg += _("Delete delay: {num_seconds}\n").format(
|
|
num_seconds=_("{num} seconds").format(num=delete_delay)
|
|
if delete_delay != -1
|
|
else _("None")
|
|
)
|
|
msg += _("Reinvite on unban: {yes_or_no}\n").format(
|
|
yes_or_no=_("Yes") if reinvite_on_unban else _("No")
|
|
)
|
|
await ctx.send(box(msg))
|
|
|
|
@modset.command()
|
|
@commands.guild_only()
|
|
async def hierarchy(self, ctx: commands.Context):
|
|
"""Toggle role hierarchy check for mods and admins.
|
|
|
|
**WARNING**: Disabling this setting will allow mods to take
|
|
actions on users above them in the role hierarchy!
|
|
|
|
This is enabled by default.
|
|
"""
|
|
guild = ctx.guild
|
|
toggled = await self.settings.guild(guild).respect_hierarchy()
|
|
if not toggled:
|
|
await self.settings.guild(guild).respect_hierarchy.set(True)
|
|
await ctx.send(
|
|
_("Role hierarchy will be checked when moderation commands are issued.")
|
|
)
|
|
else:
|
|
await self.settings.guild(guild).respect_hierarchy.set(False)
|
|
await ctx.send(
|
|
_("Role hierarchy will be ignored when moderation commands are issued.")
|
|
)
|
|
|
|
@modset.command()
|
|
@commands.guild_only()
|
|
async def banmentionspam(self, ctx: commands.Context, max_mentions: int = 0):
|
|
"""Set the autoban conditions for mention spam.
|
|
|
|
Users will be banned if they send any message which contains more than
|
|
`<max_mentions>` mentions.
|
|
|
|
`<max_mentions>` must be at least 5. Set to 0 to disable.
|
|
"""
|
|
guild = ctx.guild
|
|
if max_mentions:
|
|
if max_mentions < 5:
|
|
max_mentions = 5
|
|
await self.settings.guild(guild).ban_mention_spam.set(max_mentions)
|
|
await ctx.send(
|
|
_(
|
|
"Autoban for mention spam enabled. "
|
|
"Anyone mentioning {max_mentions} or more different people "
|
|
"in a single message will be autobanned."
|
|
).format(max_mentions=max_mentions)
|
|
)
|
|
else:
|
|
cur_setting = await self.settings.guild(guild).ban_mention_spam()
|
|
if not cur_setting:
|
|
await ctx.send_help()
|
|
return
|
|
await self.settings.guild(guild).ban_mention_spam.set(False)
|
|
await ctx.send(_("Autoban for mention spam disabled."))
|
|
|
|
@modset.command()
|
|
@commands.guild_only()
|
|
async def deleterepeats(self, ctx: commands.Context):
|
|
"""Enable auto-deletion of repeated messages."""
|
|
guild = ctx.guild
|
|
cur_setting = await self.settings.guild(guild).delete_repeats()
|
|
if not cur_setting:
|
|
await self.settings.guild(guild).delete_repeats.set(True)
|
|
await ctx.send(_("Messages repeated up to 3 times will be deleted."))
|
|
else:
|
|
await self.settings.guild(guild).delete_repeats.set(False)
|
|
await ctx.send(_("Repeated messages will be ignored."))
|
|
|
|
@modset.command()
|
|
@commands.guild_only()
|
|
async def deletedelay(self, ctx: commands.Context, time: int = None):
|
|
"""Set the delay until the bot removes the command message.
|
|
|
|
Must be between -1 and 60.
|
|
|
|
Set to -1 to disable this feature.
|
|
"""
|
|
guild = ctx.guild
|
|
if time is not None:
|
|
time = min(max(time, -1), 60) # Enforces the time limits
|
|
await self.settings.guild(guild).delete_delay.set(time)
|
|
if time == -1:
|
|
await ctx.send(_("Command deleting disabled."))
|
|
else:
|
|
await ctx.send(_("Delete delay set to {num} seconds.").format(num=time))
|
|
else:
|
|
delay = await self.settings.guild(guild).delete_delay()
|
|
if delay != -1:
|
|
await ctx.send(
|
|
_(
|
|
"Bot will delete command messages after"
|
|
" {num} seconds. Set this value to -1 to"
|
|
" stop deleting messages"
|
|
).format(num=delay)
|
|
)
|
|
else:
|
|
await ctx.send(_("I will not delete command messages."))
|
|
|
|
@modset.command()
|
|
@commands.guild_only()
|
|
async def reinvite(self, ctx: commands.Context):
|
|
"""Toggle whether an invite will be sent to a user when unbanned.
|
|
|
|
If this is True, the bot will attempt to create and send a single-use invite
|
|
to the newly-unbanned user.
|
|
"""
|
|
guild = ctx.guild
|
|
cur_setting = await self.settings.guild(guild).reinvite_on_unban()
|
|
if not cur_setting:
|
|
await self.settings.guild(guild).reinvite_on_unban.set(True)
|
|
await ctx.send(
|
|
_("Users unbanned with {command} will be reinvited.").format(
|
|
command=f"{ctx.prefix}unban"
|
|
)
|
|
)
|
|
else:
|
|
await self.settings.guild(guild).reinvite_on_unban.set(False)
|
|
await ctx.send(
|
|
_("Users unbanned with {command} will not be reinvited.").format(
|
|
command=f"{ctx.prefix}unban"
|
|
)
|
|
)
|
|
|
|
@commands.command()
|
|
@commands.guild_only()
|
|
@commands.bot_has_permissions(kick_members=True)
|
|
@checks.admin_or_permissions(kick_members=True)
|
|
async def kick(self, ctx: commands.Context, user: discord.Member, *, reason: str = None):
|
|
"""Kick a user.
|
|
|
|
If a reason is specified, it will be the reason that shows up
|
|
in the audit log.
|
|
"""
|
|
author = ctx.author
|
|
guild = ctx.guild
|
|
|
|
if author == user:
|
|
await ctx.send(
|
|
_("I cannot let you do that. Self-harm is bad {emoji}").format(
|
|
emoji="\N{PENSIVE FACE}"
|
|
)
|
|
)
|
|
return
|
|
elif not await is_allowed_by_hierarchy(self.bot, self.settings, guild, author, user):
|
|
await ctx.send(
|
|
_(
|
|
"I cannot let you do that. You are "
|
|
"not higher than the user in the role "
|
|
"hierarchy."
|
|
)
|
|
)
|
|
return
|
|
elif ctx.guild.me.top_role <= user.top_role or user == ctx.guild.owner:
|
|
await ctx.send(_("I cannot do that due to discord hierarchy rules"))
|
|
return
|
|
audit_reason = get_audit_reason(author, reason)
|
|
try:
|
|
await guild.kick(user, reason=audit_reason)
|
|
log.info("{}({}) kicked {}({})".format(author.name, author.id, user.name, user.id))
|
|
except discord.errors.Forbidden:
|
|
await ctx.send(_("I'm not allowed to do that."))
|
|
except Exception as e:
|
|
print(e)
|
|
else:
|
|
await ctx.send(_("Done. That felt good."))
|
|
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"kick",
|
|
user,
|
|
author,
|
|
reason,
|
|
until=None,
|
|
channel=None,
|
|
)
|
|
except RuntimeError as e:
|
|
await ctx.send(e)
|
|
|
|
@commands.command()
|
|
@commands.guild_only()
|
|
@commands.bot_has_permissions(ban_members=True)
|
|
@checks.admin_or_permissions(ban_members=True)
|
|
async def ban(
|
|
self,
|
|
ctx: commands.Context,
|
|
user: discord.Member,
|
|
days: Optional[int] = 0,
|
|
*,
|
|
reason: str = None,
|
|
):
|
|
"""Ban a user from this server.
|
|
|
|
If days is not a number, it's treated as the first word of the reason.
|
|
Minimum 0 days, maximum 7. Defaults to 0."""
|
|
result = await self.ban_user(
|
|
user=user, ctx=ctx, days=days, reason=reason, create_modlog_case=True
|
|
)
|
|
|
|
if result is True:
|
|
await ctx.send(_("Done. It was about time."))
|
|
elif isinstance(result, str):
|
|
await ctx.send(result)
|
|
|
|
@commands.command()
|
|
@commands.guild_only()
|
|
@commands.bot_has_permissions(ban_members=True)
|
|
@checks.admin_or_permissions(ban_members=True)
|
|
async def hackban(
|
|
self,
|
|
ctx: commands.Context,
|
|
user_ids: Greedy[RawUserIds],
|
|
days: Optional[int] = 0,
|
|
*,
|
|
reason: str = None,
|
|
):
|
|
"""Preemptively bans user(s) from the server
|
|
|
|
User IDs need to be provided in order to ban
|
|
using this command"""
|
|
|
|
banned = []
|
|
errors = {}
|
|
|
|
async def show_results():
|
|
text = _("Banned {} users from the server.".format(len(banned)))
|
|
if errors:
|
|
text += _("\nErrors:\n")
|
|
text += "\n".join(errors.values())
|
|
|
|
for p in pagify(text):
|
|
await ctx.send(p)
|
|
|
|
def remove_processed(ids):
|
|
return [_id for _id in ids if _id not in banned and _id not in errors]
|
|
|
|
user_ids = list(set(user_ids)) # No dupes
|
|
|
|
author = ctx.author
|
|
guild = ctx.guild
|
|
|
|
if not user_ids:
|
|
await ctx.send_help()
|
|
return
|
|
|
|
if not (days >= 0 and days <= 7):
|
|
await ctx.send(_("Invalid days. Must be between 0 and 7."))
|
|
return
|
|
|
|
if not guild.me.guild_permissions.ban_members:
|
|
return await ctx.send(_("I lack the permissions to do this."))
|
|
|
|
ban_list = await guild.bans()
|
|
for entry in ban_list:
|
|
for user_id in user_ids:
|
|
if entry.user.id == user_id:
|
|
errors[user_id] = _("User {} is already banned.".format(user_id))
|
|
|
|
user_ids = remove_processed(user_ids)
|
|
|
|
if not user_ids:
|
|
await show_results()
|
|
return
|
|
|
|
for user_id in user_ids:
|
|
user = guild.get_member(user_id)
|
|
if user is not None:
|
|
# Instead of replicating all that handling... gets attr from decorator
|
|
try:
|
|
result = await self.ban_user(
|
|
user=user, ctx=ctx, days=days, reason=reason, create_modlog_case=True
|
|
)
|
|
if result is True:
|
|
banned.append(user_id)
|
|
else:
|
|
errors[user_id] = _("Failed to ban user {}: {}".format(user_id, result))
|
|
except Exception as e:
|
|
errors[user_id] = _("Failed to ban user {}: {}".format(user_id, e))
|
|
|
|
user_ids = remove_processed(user_ids)
|
|
|
|
if not user_ids:
|
|
await show_results()
|
|
return
|
|
|
|
for user_id in user_ids:
|
|
user = discord.Object(id=user_id)
|
|
audit_reason = get_audit_reason(author, reason)
|
|
queue_entry = (guild.id, user_id)
|
|
self.ban_queue.append(queue_entry)
|
|
try:
|
|
await guild.ban(user, reason=audit_reason, delete_message_days=days)
|
|
log.info("{}({}) hackbanned {}".format(author.name, author.id, user_id))
|
|
except discord.NotFound:
|
|
self.ban_queue.remove(queue_entry)
|
|
errors[user_id] = _("User {} does not exist.".format(user_id))
|
|
continue
|
|
except discord.Forbidden:
|
|
self.ban_queue.remove(queue_entry)
|
|
errors[user_id] = _("Could not ban {}: missing permissions.".format(user_id))
|
|
continue
|
|
else:
|
|
banned.append(user_id)
|
|
|
|
user_info = await self.bot.get_user_info(user_id)
|
|
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"hackban",
|
|
user_info,
|
|
author,
|
|
reason,
|
|
until=None,
|
|
channel=None,
|
|
)
|
|
except RuntimeError as e:
|
|
errors["0"] = _("Failed to create modlog case: {}".format(e))
|
|
|
|
await show_results()
|
|
|
|
@commands.command()
|
|
@commands.guild_only()
|
|
@commands.bot_has_permissions(ban_members=True)
|
|
@checks.admin_or_permissions(ban_members=True)
|
|
async def tempban(
|
|
self, ctx: commands.Context, user: discord.Member, days: int = 1, *, reason: str = None
|
|
):
|
|
"""Temporarily ban a user from this server."""
|
|
guild = ctx.guild
|
|
author = ctx.author
|
|
days_delta = timedelta(days=int(days))
|
|
unban_time = datetime.utcnow() + days_delta
|
|
|
|
invite = await self.get_invite_for_reinvite(ctx, int(days_delta.total_seconds() + 86400))
|
|
if invite is None:
|
|
invite = ""
|
|
|
|
queue_entry = (guild.id, user.id)
|
|
await self.settings.member(user).banned_until.set(unban_time.timestamp())
|
|
cur_tbans = await self.settings.guild(guild).current_tempbans()
|
|
cur_tbans.append(user.id)
|
|
await self.settings.guild(guild).current_tempbans.set(cur_tbans)
|
|
|
|
with contextlib.suppress(discord.HTTPException):
|
|
# We don't want blocked DMs preventing us from banning
|
|
await user.send(
|
|
_(
|
|
"You have been temporarily banned from {server_name} until {date}. "
|
|
"Here is an invite for when your ban expires: {invite_link}"
|
|
).format(
|
|
server_name=guild.name,
|
|
date=unban_time.strftime("%m-%d-%Y %H:%M:%S"),
|
|
invite_link=invite,
|
|
)
|
|
)
|
|
self.ban_queue.append(queue_entry)
|
|
try:
|
|
await guild.ban(user)
|
|
except discord.Forbidden:
|
|
await ctx.send(_("I can't do that for some reason."))
|
|
except discord.HTTPException:
|
|
await ctx.send(_("Something went wrong while banning"))
|
|
else:
|
|
await ctx.send(_("Done. Enough chaos for now"))
|
|
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"tempban",
|
|
user,
|
|
author,
|
|
reason,
|
|
unban_time,
|
|
)
|
|
except RuntimeError as e:
|
|
await ctx.send(e)
|
|
|
|
@commands.command()
|
|
@commands.guild_only()
|
|
@commands.bot_has_permissions(ban_members=True)
|
|
@checks.admin_or_permissions(ban_members=True)
|
|
async def softban(self, ctx: commands.Context, user: discord.Member, *, reason: str = None):
|
|
"""Kick a user and delete 1 day's worth of their messages."""
|
|
guild = ctx.guild
|
|
author = ctx.author
|
|
|
|
if author == user:
|
|
await ctx.send(
|
|
_("I cannot let you do that. Self-harm is bad {emoji}").format(
|
|
emoji="\N{PENSIVE FACE}"
|
|
)
|
|
)
|
|
return
|
|
elif not await is_allowed_by_hierarchy(self.bot, self.settings, guild, author, user):
|
|
await ctx.send(
|
|
_(
|
|
"I cannot let you do that. You are "
|
|
"not higher than the user in the role "
|
|
"hierarchy."
|
|
)
|
|
)
|
|
return
|
|
|
|
audit_reason = get_audit_reason(author, reason)
|
|
|
|
invite = await self.get_invite_for_reinvite(ctx)
|
|
if invite is None:
|
|
invite = ""
|
|
|
|
queue_entry = (guild.id, user.id)
|
|
try: # We don't want blocked DMs preventing us from banning
|
|
msg = await user.send(
|
|
_(
|
|
"You have been banned and "
|
|
"then unbanned as a quick way to delete your messages.\n"
|
|
"You can now join the server again. {invite_link}"
|
|
).format(invite_link=invite)
|
|
)
|
|
except discord.HTTPException:
|
|
msg = None
|
|
self.ban_queue.append(queue_entry)
|
|
try:
|
|
await guild.ban(user, reason=audit_reason, delete_message_days=1)
|
|
except discord.errors.Forbidden:
|
|
self.ban_queue.remove(queue_entry)
|
|
await ctx.send(_("My role is not high enough to softban that user."))
|
|
if msg is not None:
|
|
await msg.delete()
|
|
return
|
|
except discord.HTTPException as e:
|
|
self.ban_queue.remove(queue_entry)
|
|
print(e)
|
|
return
|
|
self.unban_queue.append(queue_entry)
|
|
try:
|
|
await guild.unban(user)
|
|
except discord.HTTPException as e:
|
|
self.unban_queue.remove(queue_entry)
|
|
print(e)
|
|
return
|
|
else:
|
|
await ctx.send(_("Done. Enough chaos."))
|
|
log.info(
|
|
"{}({}) softbanned {}({}), deleting 1 day worth "
|
|
"of messages".format(author.name, author.id, user.name, user.id)
|
|
)
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"softban",
|
|
user,
|
|
author,
|
|
reason,
|
|
until=None,
|
|
channel=None,
|
|
)
|
|
except RuntimeError as e:
|
|
await ctx.send(e)
|
|
|
|
@commands.command()
|
|
@commands.guild_only()
|
|
@commands.bot_has_permissions(ban_members=True)
|
|
@checks.admin_or_permissions(ban_members=True)
|
|
async def unban(self, ctx: commands.Context, user_id: int, *, reason: str = None):
|
|
"""Unban a user from this server.
|
|
|
|
Requires specifying the target user's ID. To find this, you may either:
|
|
1. Copy it from the mod log case (if one was created), or
|
|
2. enable developer mode, go to Bans in this server's settings, right-
|
|
click the user and select 'Copy ID'."""
|
|
guild = ctx.guild
|
|
author = ctx.author
|
|
user = await self.bot.get_user_info(user_id)
|
|
if not user:
|
|
await ctx.send(_("Couldn't find a user with that ID!"))
|
|
return
|
|
audit_reason = get_audit_reason(ctx.author, reason)
|
|
bans = await guild.bans()
|
|
bans = [be.user for be in bans]
|
|
if user not in bans:
|
|
await ctx.send(_("It seems that user isn't banned!"))
|
|
return
|
|
queue_entry = (guild.id, user.id)
|
|
self.unban_queue.append(queue_entry)
|
|
try:
|
|
await guild.unban(user, reason=audit_reason)
|
|
except discord.HTTPException:
|
|
self.unban_queue.remove(queue_entry)
|
|
await ctx.send(_("Something went wrong while attempting to unban that user"))
|
|
return
|
|
else:
|
|
await ctx.send(_("Unbanned that user from this server"))
|
|
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"unban",
|
|
user,
|
|
author,
|
|
reason,
|
|
until=None,
|
|
channel=None,
|
|
)
|
|
except RuntimeError as e:
|
|
await ctx.send(e)
|
|
|
|
if await self.settings.guild(guild).reinvite_on_unban():
|
|
invite = await self.get_invite_for_reinvite(ctx)
|
|
if invite:
|
|
try:
|
|
await user.send(
|
|
_(
|
|
"You've been unbanned from {server}.\n"
|
|
"Here is an invite for that server: {invite_link}"
|
|
).format(server=guild.name, invite_link=invite.url)
|
|
)
|
|
except discord.Forbidden:
|
|
await ctx.send(
|
|
_(
|
|
"I failed to send an invite to that user. "
|
|
"Perhaps you may be able to send it for me?\n"
|
|
"Here's the invite link: {invite_link}"
|
|
).format(invite_link=invite.url)
|
|
)
|
|
except discord.HTTPException:
|
|
await ctx.send(
|
|
_(
|
|
"Something went wrong when attempting to send that user"
|
|
"an invite. Here's the link so you can try: {invite_link}"
|
|
).format(invite_link=invite.url)
|
|
)
|
|
|
|
@staticmethod
|
|
async def get_invite_for_reinvite(ctx: commands.Context, max_age: int = 86400):
|
|
"""Handles the reinvite logic for getting an invite
|
|
to send the newly unbanned user
|
|
:returns: :class:`Invite`"""
|
|
guild = ctx.guild
|
|
my_perms: discord.Permissions = guild.me.guild_permissions
|
|
if my_perms.manage_guild or my_perms.administrator:
|
|
if "VANITY_URL" in guild.features:
|
|
# guild has a vanity url so use it as the one to send
|
|
return await guild.vanity_invite()
|
|
invites = await guild.invites()
|
|
else:
|
|
invites = []
|
|
for inv in invites: # Loop through the invites for the guild
|
|
if not (inv.max_uses or inv.max_age or inv.temporary):
|
|
# Invite is for the guild's default channel,
|
|
# has unlimited uses, doesn't expire, and
|
|
# doesn't grant temporary membership
|
|
# (i.e. they won't be kicked on disconnect)
|
|
return inv
|
|
else: # No existing invite found that is valid
|
|
channels_and_perms = zip(
|
|
guild.text_channels, map(guild.me.permissions_in, guild.text_channels)
|
|
)
|
|
channel = next(
|
|
(channel for channel, perms in channels_and_perms if perms.create_instant_invite),
|
|
None,
|
|
)
|
|
if channel is None:
|
|
return
|
|
try:
|
|
# Create invite that expires after max_age
|
|
return await channel.create_invite(max_age=max_age)
|
|
except discord.HTTPException:
|
|
return
|
|
|
|
@staticmethod
|
|
async def _voice_perm_check(
|
|
ctx: commands.Context, user_voice_state: Optional[discord.VoiceState], **perms: bool
|
|
) -> bool:
|
|
"""Check if the bot and user have sufficient permissions for voicebans.
|
|
|
|
This also verifies that the user's voice state and connected
|
|
channel are not ``None``.
|
|
|
|
Returns
|
|
-------
|
|
bool
|
|
``True`` if the permissions are sufficient and the user has
|
|
a valid voice state.
|
|
|
|
"""
|
|
if user_voice_state is None or user_voice_state.channel is None:
|
|
await ctx.send(_("That user is not in a voice channel."))
|
|
return False
|
|
voice_channel: discord.VoiceChannel = user_voice_state.channel
|
|
required_perms = discord.Permissions()
|
|
required_perms.update(**perms)
|
|
if not voice_channel.permissions_for(ctx.me) >= required_perms:
|
|
await ctx.send(
|
|
_("I require the {perms} permission(s) in that user's channel to do that.").format(
|
|
perms=format_perms_list(required_perms)
|
|
)
|
|
)
|
|
return False
|
|
if (
|
|
ctx.permission_state is commands.PermState.NORMAL
|
|
and not voice_channel.permissions_for(ctx.author) >= required_perms
|
|
):
|
|
await ctx.send(
|
|
_(
|
|
"You must have the {perms} permission(s) in that user's channel to use this "
|
|
"command."
|
|
).format(perms=format_perms_list(required_perms))
|
|
)
|
|
return False
|
|
return True
|
|
|
|
@commands.command()
|
|
@commands.guild_only()
|
|
@checks.admin_or_permissions(mute_members=True, deafen_members=True)
|
|
async def voiceban(self, ctx: commands.Context, user: discord.Member, *, reason: str = None):
|
|
"""Ban a user from speaking and listening in the server's voice channels."""
|
|
user_voice_state: discord.VoiceState = user.voice
|
|
if (
|
|
await self._voice_perm_check(
|
|
ctx, user_voice_state, deafen_members=True, mute_members=True
|
|
)
|
|
is False
|
|
):
|
|
return
|
|
needs_mute = True if user_voice_state.mute is False else False
|
|
needs_deafen = True if user_voice_state.deaf is False else False
|
|
audit_reason = get_audit_reason(ctx.author, reason)
|
|
author = ctx.author
|
|
guild = ctx.guild
|
|
if needs_mute and needs_deafen:
|
|
await user.edit(mute=True, deafen=True, reason=audit_reason)
|
|
elif needs_mute:
|
|
await user.edit(mute=True, reason=audit_reason)
|
|
elif needs_deafen:
|
|
await user.edit(deafen=True, reason=audit_reason)
|
|
else:
|
|
await ctx.send(_("That user is already muted and deafened server-wide!"))
|
|
return
|
|
await ctx.send(_("User has been banned from speaking or listening in voice channels"))
|
|
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"voiceban",
|
|
user,
|
|
author,
|
|
reason,
|
|
until=None,
|
|
channel=None,
|
|
)
|
|
except RuntimeError as e:
|
|
await ctx.send(e)
|
|
|
|
@commands.command()
|
|
@commands.guild_only()
|
|
async def voiceunban(self, ctx: commands.Context, user: discord.Member, *, reason: str = None):
|
|
"""Unban a user from speaking and listening in the server's voice channels."""
|
|
user_voice_state = user.voice
|
|
if (
|
|
await self._voice_perm_check(
|
|
ctx, user_voice_state, deafen_members=True, mute_members=True
|
|
)
|
|
is False
|
|
):
|
|
return
|
|
needs_unmute = True if user_voice_state.mute else False
|
|
needs_undeafen = True if user_voice_state.deaf else False
|
|
audit_reason = get_audit_reason(ctx.author, reason)
|
|
if needs_unmute and needs_undeafen:
|
|
await user.edit(mute=False, deafen=False, reason=audit_reason)
|
|
elif needs_unmute:
|
|
await user.edit(mute=False, reason=audit_reason)
|
|
elif needs_undeafen:
|
|
await user.edit(deafen=False, reason=audit_reason)
|
|
else:
|
|
await ctx.send(_("That user isn't muted or deafened by the server!"))
|
|
return
|
|
await ctx.send(_("User is now allowed to speak and listen in voice channels"))
|
|
guild = ctx.guild
|
|
author = ctx.author
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"voiceunban",
|
|
user,
|
|
author,
|
|
reason,
|
|
until=None,
|
|
channel=None,
|
|
)
|
|
except RuntimeError as e:
|
|
await ctx.send(e)
|
|
|
|
@commands.command()
|
|
@commands.guild_only()
|
|
@commands.bot_has_permissions(manage_nicknames=True)
|
|
@checks.admin_or_permissions(manage_nicknames=True)
|
|
async def rename(self, ctx: commands.Context, user: discord.Member, *, nickname: str = ""):
|
|
"""Change a user's nickname.
|
|
|
|
Leaving the nickname empty will remove it.
|
|
"""
|
|
nickname = nickname.strip()
|
|
me = cast(discord.Member, ctx.me)
|
|
if not nickname:
|
|
nickname = None
|
|
elif not 2 <= len(nickname) <= 32:
|
|
await ctx.send(_("Nicknames must be between 2 and 32 characters long."))
|
|
return
|
|
if not (
|
|
(me.guild_permissions.manage_nicknames or me.guild_permissions.administrator)
|
|
and me.top_role > user.top_role
|
|
and user != ctx.guild.owner
|
|
):
|
|
await ctx.send(
|
|
_(
|
|
"I do not have permission to rename that member. They may be higher than or "
|
|
"equal to me in the role hierarchy."
|
|
)
|
|
)
|
|
else:
|
|
try:
|
|
await user.edit(reason=get_audit_reason(ctx.author, None), nick=nickname)
|
|
except discord.Forbidden:
|
|
# Just in case we missed something in the permissions check above
|
|
await ctx.send(_("I do not have permission to rename that member."))
|
|
except discord.HTTPException as exc:
|
|
if exc.status == 400: # BAD REQUEST
|
|
await ctx.send(_("That nickname is invalid."))
|
|
else:
|
|
await ctx.send(_("An unexpected error has occured."))
|
|
else:
|
|
await ctx.send(_("Done."))
|
|
|
|
@commands.group()
|
|
@commands.guild_only()
|
|
@checks.mod_or_permissions(manage_channels=True)
|
|
async def mute(self, ctx: commands.Context):
|
|
"""Mute users."""
|
|
pass
|
|
|
|
@mute.command(name="voice")
|
|
@commands.guild_only()
|
|
async def voice_mute(self, ctx: commands.Context, user: discord.Member, *, reason: str = None):
|
|
"""Mute a user in their current voice channel."""
|
|
user_voice_state = user.voice
|
|
if (
|
|
await self._voice_perm_check(
|
|
ctx, user_voice_state, mute_members=True, manage_channels=True
|
|
)
|
|
is False
|
|
):
|
|
return
|
|
guild = ctx.guild
|
|
author = ctx.author
|
|
channel = user_voice_state.channel
|
|
audit_reason = get_audit_reason(author, reason)
|
|
|
|
success, issue = await self.mute_user(guild, channel, author, user, audit_reason)
|
|
|
|
if success:
|
|
await ctx.send(
|
|
_("Muted {user} in channel {channel.name}").format(user=user, channel=channel)
|
|
)
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"vmute",
|
|
user,
|
|
author,
|
|
reason,
|
|
until=None,
|
|
channel=channel,
|
|
)
|
|
except RuntimeError as e:
|
|
await ctx.send(e)
|
|
else:
|
|
await ctx.send(issue)
|
|
|
|
@mute.command(name="channel")
|
|
@commands.guild_only()
|
|
@commands.bot_has_permissions(manage_roles=True)
|
|
@checks.mod_or_permissions(administrator=True)
|
|
async def channel_mute(
|
|
self, ctx: commands.Context, user: discord.Member, *, reason: str = None
|
|
):
|
|
"""Mute a user in the current text channel."""
|
|
author = ctx.message.author
|
|
channel = ctx.message.channel
|
|
guild = ctx.guild
|
|
audit_reason = get_audit_reason(author, reason)
|
|
|
|
success, issue = await self.mute_user(guild, channel, author, user, audit_reason)
|
|
|
|
if success:
|
|
await channel.send(_("User has been muted in this channel."))
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"cmute",
|
|
user,
|
|
author,
|
|
reason,
|
|
until=None,
|
|
channel=channel,
|
|
)
|
|
except RuntimeError as e:
|
|
await ctx.send(e)
|
|
else:
|
|
await channel.send(issue)
|
|
|
|
@mute.command(name="server", aliases=["guild"])
|
|
@commands.guild_only()
|
|
@commands.bot_has_permissions(manage_roles=True)
|
|
@checks.mod_or_permissions(administrator=True)
|
|
async def guild_mute(self, ctx: commands.Context, user: discord.Member, *, reason: str = None):
|
|
"""Mutes user in the server"""
|
|
author = ctx.message.author
|
|
guild = ctx.guild
|
|
audit_reason = get_audit_reason(author, reason)
|
|
|
|
mute_success = []
|
|
for channel in guild.channels:
|
|
success, issue = await self.mute_user(guild, channel, author, user, audit_reason)
|
|
mute_success.append((success, issue))
|
|
await asyncio.sleep(0.1)
|
|
await ctx.send(_("User has been muted in this server."))
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"smute",
|
|
user,
|
|
author,
|
|
reason,
|
|
until=None,
|
|
channel=None,
|
|
)
|
|
except RuntimeError as e:
|
|
await ctx.send(e)
|
|
|
|
async def mute_user(
|
|
self,
|
|
guild: discord.Guild,
|
|
channel: discord.abc.GuildChannel,
|
|
author: discord.Member,
|
|
user: discord.Member,
|
|
reason: str,
|
|
) -> (bool, str):
|
|
"""Mutes the specified user in the specified channel"""
|
|
overwrites = channel.overwrites_for(user)
|
|
permissions = channel.permissions_for(user)
|
|
|
|
if permissions.administrator:
|
|
return False, T_(mute_unmute_issues["is_admin"])
|
|
|
|
new_overs = {}
|
|
if not isinstance(channel, discord.TextChannel):
|
|
new_overs.update(speak=False)
|
|
if not isinstance(channel, discord.VoiceChannel):
|
|
new_overs.update(send_messages=False, add_reactions=False)
|
|
|
|
if all(getattr(permissions, p) is False for p in new_overs.keys()):
|
|
return False, T_(mute_unmute_issues["already_muted"])
|
|
|
|
elif not await is_allowed_by_hierarchy(self.bot, self.settings, guild, author, user):
|
|
return False, T_(mute_unmute_issues["hierarchy_problem"])
|
|
|
|
old_overs = {k: getattr(overwrites, k) for k in new_overs}
|
|
overwrites.update(**new_overs)
|
|
try:
|
|
await channel.set_permissions(user, overwrite=overwrites, reason=reason)
|
|
except discord.Forbidden:
|
|
return False, T_(mute_unmute_issues["permissions_issue"])
|
|
else:
|
|
await self.settings.member(user).set_raw(
|
|
"perms_cache", str(channel.id), value=old_overs
|
|
)
|
|
return True, None
|
|
|
|
@commands.group()
|
|
@commands.guild_only()
|
|
@commands.bot_has_permissions(manage_roles=True)
|
|
@checks.mod_or_permissions(manage_channels=True)
|
|
async def unmute(self, ctx: commands.Context):
|
|
"""Unmute users."""
|
|
pass
|
|
|
|
@unmute.command(name="voice")
|
|
@commands.guild_only()
|
|
async def unmute_voice(
|
|
self, ctx: commands.Context, user: discord.Member, *, reason: str = None
|
|
):
|
|
"""Unmute a user in their current voice channel."""
|
|
user_voice_state = user.voice
|
|
if (
|
|
await self._voice_perm_check(
|
|
ctx, user_voice_state, mute_members=True, manage_channels=True
|
|
)
|
|
is False
|
|
):
|
|
return
|
|
guild = ctx.guild
|
|
author = ctx.author
|
|
channel = user_voice_state.channel
|
|
audit_reason = get_audit_reason(author, reason)
|
|
|
|
success, message = await self.unmute_user(guild, channel, author, user, audit_reason)
|
|
|
|
if success:
|
|
await ctx.send(
|
|
_("Unmuted {user} in channel {channel.name}").format(user=user, channel=channel)
|
|
)
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"vunmute",
|
|
user,
|
|
author,
|
|
reason,
|
|
until=None,
|
|
channel=channel,
|
|
)
|
|
except RuntimeError as e:
|
|
await ctx.send(e)
|
|
else:
|
|
await ctx.send(_("Unmute failed. Reason: {}").format(message))
|
|
|
|
@checks.mod_or_permissions(administrator=True)
|
|
@unmute.command(name="channel")
|
|
@commands.bot_has_permissions(manage_roles=True)
|
|
@commands.guild_only()
|
|
async def unmute_channel(
|
|
self, ctx: commands.Context, user: discord.Member, *, reason: str = None
|
|
):
|
|
"""Unmute a user in this channel."""
|
|
channel = ctx.channel
|
|
author = ctx.author
|
|
guild = ctx.guild
|
|
audit_reason = get_audit_reason(author, reason)
|
|
|
|
success, message = await self.unmute_user(guild, channel, author, user, audit_reason)
|
|
|
|
if success:
|
|
await ctx.send(_("User unmuted in this channel."))
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"cunmute",
|
|
user,
|
|
author,
|
|
reason,
|
|
until=None,
|
|
channel=channel,
|
|
)
|
|
except RuntimeError as e:
|
|
await ctx.send(e)
|
|
else:
|
|
await ctx.send(_("Unmute failed. Reason: {}").format(message))
|
|
|
|
@checks.mod_or_permissions(administrator=True)
|
|
@unmute.command(name="server", aliases=["guild"])
|
|
@commands.bot_has_permissions(manage_roles=True)
|
|
@commands.guild_only()
|
|
async def unmute_guild(
|
|
self, ctx: commands.Context, user: discord.Member, *, reason: str = None
|
|
):
|
|
"""Unmute a user in this server."""
|
|
guild = ctx.guild
|
|
author = ctx.author
|
|
audit_reason = get_audit_reason(author, reason)
|
|
|
|
unmute_success = []
|
|
for channel in guild.channels:
|
|
success, message = await self.unmute_user(guild, channel, author, user, audit_reason)
|
|
unmute_success.append((success, message))
|
|
await asyncio.sleep(0.1)
|
|
await ctx.send(_("User has been unmuted in this server."))
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"sunmute",
|
|
user,
|
|
author,
|
|
reason,
|
|
until=None,
|
|
)
|
|
except RuntimeError as e:
|
|
await ctx.send(e)
|
|
|
|
async def unmute_user(
|
|
self,
|
|
guild: discord.Guild,
|
|
channel: discord.abc.GuildChannel,
|
|
author: discord.Member,
|
|
user: discord.Member,
|
|
reason: str,
|
|
) -> (bool, str):
|
|
overwrites = channel.overwrites_for(user)
|
|
perms_cache = await self.settings.member(user).perms_cache()
|
|
|
|
if channel.id in perms_cache:
|
|
old_values = perms_cache[channel.id]
|
|
else:
|
|
old_values = {"send_messages": None, "add_reactions": None, "speak": None}
|
|
|
|
if all(getattr(overwrites, k) == v for k, v in old_values.items()):
|
|
return False, T_(mute_unmute_issues["already_unmuted"])
|
|
|
|
elif not await is_allowed_by_hierarchy(self.bot, self.settings, guild, author, user):
|
|
return False, T_(mute_unmute_issues["hierarchy_problem"])
|
|
|
|
overwrites.update(**old_values)
|
|
try:
|
|
if overwrites.is_empty():
|
|
await channel.set_permissions(
|
|
user, overwrite=cast(discord.PermissionOverwrite, None), reason=reason
|
|
)
|
|
else:
|
|
await channel.set_permissions(user, overwrite=overwrites, reason=reason)
|
|
except discord.Forbidden:
|
|
return False, T_(mute_unmute_issues["permissions_issue"])
|
|
else:
|
|
await self.settings.member(user).clear_raw("perms_cache", str(channel.id))
|
|
return True, None
|
|
|
|
@commands.group()
|
|
@commands.guild_only()
|
|
@checks.admin_or_permissions(manage_channels=True)
|
|
async def ignore(self, ctx: commands.Context):
|
|
"""Add servers or channels to the ignore list."""
|
|
if ctx.invoked_subcommand is None:
|
|
await ctx.send(await self.count_ignored())
|
|
|
|
@ignore.command(name="channel")
|
|
async def ignore_channel(self, ctx: commands.Context, channel: discord.TextChannel = None):
|
|
"""Ignore commands in the channel.
|
|
|
|
Defaults to the current channel.
|
|
"""
|
|
if not channel:
|
|
channel = ctx.channel
|
|
if not await self.settings.channel(channel).ignored():
|
|
await self.settings.channel(channel).ignored.set(True)
|
|
await ctx.send(_("Channel added to ignore list."))
|
|
else:
|
|
await ctx.send(_("Channel already in ignore list."))
|
|
|
|
@ignore.command(name="server", aliases=["guild"])
|
|
@checks.admin_or_permissions(manage_guild=True)
|
|
async def ignore_guild(self, ctx: commands.Context):
|
|
"""Ignore commands in this server."""
|
|
guild = ctx.guild
|
|
if not await self.settings.guild(guild).ignored():
|
|
await self.settings.guild(guild).ignored.set(True)
|
|
await ctx.send(_("This server has been added to the ignore list."))
|
|
else:
|
|
await ctx.send(_("This server is already being ignored."))
|
|
|
|
@commands.group()
|
|
@commands.guild_only()
|
|
@checks.admin_or_permissions(manage_channels=True)
|
|
async def unignore(self, ctx: commands.Context):
|
|
"""Remove servers or channels from the ignore list."""
|
|
if ctx.invoked_subcommand is None:
|
|
await ctx.send(await self.count_ignored())
|
|
|
|
@unignore.command(name="channel")
|
|
async def unignore_channel(self, ctx: commands.Context, channel: discord.TextChannel = None):
|
|
"""Remove a channel from ignore the list.
|
|
|
|
Defaults to the current channel.
|
|
"""
|
|
if not channel:
|
|
channel = ctx.channel
|
|
|
|
if await self.settings.channel(channel).ignored():
|
|
await self.settings.channel(channel).ignored.set(False)
|
|
await ctx.send(_("Channel removed from ignore list."))
|
|
else:
|
|
await ctx.send(_("That channel is not in the ignore list."))
|
|
|
|
@unignore.command(name="server", aliases=["guild"])
|
|
@checks.admin_or_permissions(manage_guild=True)
|
|
async def unignore_guild(self, ctx: commands.Context):
|
|
"""Remove this server from the ignore list."""
|
|
guild = ctx.message.guild
|
|
if await self.settings.guild(guild).ignored():
|
|
await self.settings.guild(guild).ignored.set(False)
|
|
await ctx.send(_("This server has been removed from the ignore list."))
|
|
else:
|
|
await ctx.send(_("This server is not in the ignore list."))
|
|
|
|
async def count_ignored(self):
|
|
ch_count = 0
|
|
svr_count = 0
|
|
for guild in self.bot.guilds:
|
|
if not await self.settings.guild(guild).ignored():
|
|
for channel in guild.text_channels:
|
|
if await self.settings.channel(channel).ignored():
|
|
ch_count += 1
|
|
else:
|
|
svr_count += 1
|
|
msg = _("Currently ignoring:\n{} channels\n{} guilds\n").format(ch_count, svr_count)
|
|
return box(msg)
|
|
|
|
async def __global_check(self, ctx):
|
|
"""Global check to see if a channel or server is ignored.
|
|
|
|
Any users who have permission to use the `ignore` or `unignore` commands
|
|
surpass the check.
|
|
"""
|
|
perms = ctx.channel.permissions_for(ctx.author)
|
|
surpass_ignore = (
|
|
isinstance(ctx.channel, discord.abc.PrivateChannel)
|
|
or perms.manage_guild
|
|
or await ctx.bot.is_owner(ctx.author)
|
|
or await ctx.bot.is_admin(ctx.author)
|
|
)
|
|
if surpass_ignore:
|
|
return True
|
|
guild_ignored = await self.settings.guild(ctx.guild).ignored()
|
|
chann_ignored = await self.settings.channel(ctx.channel).ignored()
|
|
return not (guild_ignored or chann_ignored and not perms.manage_channels)
|
|
|
|
@commands.command()
|
|
@commands.guild_only()
|
|
@commands.bot_has_permissions(embed_links=True)
|
|
async def userinfo(self, ctx, *, user: discord.Member = None):
|
|
"""Show information about a user.
|
|
|
|
This includes fields for status, discord join date, server
|
|
join date, voice state and previous names/nicknames.
|
|
|
|
If the user has no roles, previous names or previous nicknames,
|
|
these fields will be omitted.
|
|
"""
|
|
author = ctx.author
|
|
guild = ctx.guild
|
|
|
|
if not user:
|
|
user = author
|
|
|
|
# A special case for a special someone :^)
|
|
special_date = datetime(2016, 1, 10, 6, 8, 4, 443000)
|
|
is_special = user.id == 96130341705637888 and guild.id == 133049272517001216
|
|
|
|
roles = sorted(user.roles)[1:]
|
|
names, nicks = await self.get_names_and_nicks(user)
|
|
|
|
joined_at = user.joined_at if not is_special else special_date
|
|
since_created = (ctx.message.created_at - user.created_at).days
|
|
if joined_at is not None:
|
|
since_joined = (ctx.message.created_at - joined_at).days
|
|
user_joined = joined_at.strftime("%d %b %Y %H:%M")
|
|
else:
|
|
since_joined = "?"
|
|
user_joined = "Unknown"
|
|
user_created = user.created_at.strftime("%d %b %Y %H:%M")
|
|
voice_state = user.voice
|
|
member_number = (
|
|
sorted(guild.members, key=lambda m: m.joined_at or ctx.message.created_at).index(user)
|
|
+ 1
|
|
)
|
|
|
|
created_on = _("{}\n({} days ago)").format(user_created, since_created)
|
|
joined_on = _("{}\n({} days ago)").format(user_joined, since_joined)
|
|
|
|
activity = _("Chilling in {} status").format(user.status)
|
|
if user.activity is None: # Default status
|
|
pass
|
|
elif user.activity.type == discord.ActivityType.playing:
|
|
activity = _("Playing {}").format(user.activity.name)
|
|
elif user.activity.type == discord.ActivityType.streaming:
|
|
activity = _("Streaming [{}]({})").format(user.activity.name, user.activity.url)
|
|
elif user.activity.type == discord.ActivityType.listening:
|
|
activity = _("Listening to {}").format(user.activity.name)
|
|
elif user.activity.type == discord.ActivityType.watching:
|
|
activity = _("Watching {}").format(user.activity.name)
|
|
|
|
if roles:
|
|
roles = ", ".join([x.name for x in roles])
|
|
else:
|
|
roles = None
|
|
|
|
data = discord.Embed(description=activity, colour=user.colour)
|
|
data.add_field(name=_("Joined Discord on"), value=created_on)
|
|
data.add_field(name=_("Joined this server on"), value=joined_on)
|
|
if roles is not None:
|
|
data.add_field(name=_("Roles"), value=roles, inline=False)
|
|
if names:
|
|
# May need sanitizing later, but mentions do not ping in embeds currently
|
|
val = filter_invites(", ".join(names))
|
|
data.add_field(name=_("Previous Names"), value=val, inline=False)
|
|
if nicks:
|
|
# May need sanitizing later, but mentions do not ping in embeds currently
|
|
val = filter_invites(", ".join(nicks))
|
|
data.add_field(name=_("Previous Nicknames"), value=val, inline=False)
|
|
if voice_state and voice_state.channel:
|
|
data.add_field(
|
|
name=_("Current voice channel"),
|
|
value="{0.name} (ID {0.id})".format(voice_state.channel),
|
|
inline=False,
|
|
)
|
|
data.set_footer(text=_("Member #{} | User ID: {}").format(member_number, user.id))
|
|
|
|
name = str(user)
|
|
name = " ~ ".join((name, user.nick)) if user.nick else name
|
|
name = filter_invites(name)
|
|
|
|
if user.avatar:
|
|
avatar = user.avatar_url_as(static_format="png")
|
|
data.set_author(name=name, url=avatar)
|
|
data.set_thumbnail(url=avatar)
|
|
else:
|
|
data.set_author(name=name)
|
|
|
|
await ctx.send(embed=data)
|
|
|
|
@commands.command()
|
|
async def names(self, ctx: commands.Context, user: discord.Member):
|
|
"""Show previous names and nicknames of a user."""
|
|
names, nicks = await self.get_names_and_nicks(user)
|
|
msg = ""
|
|
if names:
|
|
msg += _("**Past 20 names**:")
|
|
msg += "\n"
|
|
msg += ", ".join(names)
|
|
if nicks:
|
|
if msg:
|
|
msg += "\n\n"
|
|
msg += _("**Past 20 nicknames**:")
|
|
msg += "\n"
|
|
msg += ", ".join(nicks)
|
|
if msg:
|
|
msg = filter_various_mentions(msg)
|
|
await ctx.send(msg)
|
|
else:
|
|
await ctx.send(_("That user doesn't have any recorded name or nickname change."))
|
|
|
|
async def ban_user(
|
|
self,
|
|
user: discord.Member,
|
|
ctx: commands.Context,
|
|
days: Optional[int] = 0,
|
|
reason: str = None,
|
|
create_modlog_case=False,
|
|
) -> Union[str, bool]:
|
|
author = ctx.author
|
|
guild = ctx.guild
|
|
|
|
if author == user:
|
|
return _("I cannot let you do that. Self-harm is bad {}").format("\N{PENSIVE FACE}")
|
|
elif not await is_allowed_by_hierarchy(self.bot, self.settings, guild, author, user):
|
|
return _(
|
|
"I cannot let you do that. You are "
|
|
"not higher than the user in the role "
|
|
"hierarchy."
|
|
)
|
|
elif guild.me.top_role <= user.top_role or user == guild.owner:
|
|
return _("I cannot do that due to discord hierarchy rules")
|
|
elif not (days >= 0 and days <= 7):
|
|
return _("Invalid days. Must be between 0 and 7.")
|
|
|
|
audit_reason = get_audit_reason(author, reason)
|
|
|
|
queue_entry = (guild.id, user.id)
|
|
self.ban_queue.append(queue_entry)
|
|
try:
|
|
await guild.ban(user, reason=audit_reason, delete_message_days=days)
|
|
log.info(
|
|
"{}({}) banned {}({}), deleting {} days worth of messages".format(
|
|
author.name, author.id, user.name, user.id, str(days)
|
|
)
|
|
)
|
|
except discord.Forbidden:
|
|
self.ban_queue.remove(queue_entry)
|
|
return _("I'm not allowed to do that.")
|
|
except Exception as e:
|
|
self.ban_queue.remove(queue_entry)
|
|
return e
|
|
|
|
if create_modlog_case:
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
ctx.message.created_at,
|
|
"ban",
|
|
user,
|
|
author,
|
|
reason,
|
|
until=None,
|
|
channel=None,
|
|
)
|
|
except RuntimeError as e:
|
|
return _(
|
|
"The user was banned but an error occurred when trying to "
|
|
"create the modlog entry: {}".format(e)
|
|
)
|
|
|
|
return True
|
|
|
|
async def get_names_and_nicks(self, user):
|
|
names = await self.settings.user(user).past_names()
|
|
nicks = await self.settings.member(user).past_nicks()
|
|
if names:
|
|
names = [escape_spoilers(escape(name, mass_mentions=True)) for name in names if name]
|
|
if nicks:
|
|
nicks = [escape_spoilers(escape(nick, mass_mentions=True)) for nick in nicks if nick]
|
|
return names, nicks
|
|
|
|
async def check_tempban_expirations(self):
|
|
member = namedtuple("Member", "id guild")
|
|
while self == self.bot.get_cog("Mod"):
|
|
for guild in self.bot.guilds:
|
|
async with self.settings.guild(guild).current_tempbans() as guild_tempbans:
|
|
for uid in guild_tempbans.copy():
|
|
unban_time = datetime.utcfromtimestamp(
|
|
await self.settings.member(member(uid, guild)).banned_until()
|
|
)
|
|
now = datetime.utcnow()
|
|
if now > unban_time: # Time to unban the user
|
|
user = await self.bot.get_user_info(uid)
|
|
queue_entry = (guild.id, user.id)
|
|
self.unban_queue.append(queue_entry)
|
|
try:
|
|
await guild.unban(user, reason=_("Tempban finished"))
|
|
guild_tempbans.remove(uid)
|
|
except discord.Forbidden:
|
|
self.unban_queue.remove(queue_entry)
|
|
log.info("Failed to unban member due to permissions")
|
|
except discord.HTTPException:
|
|
self.unban_queue.remove(queue_entry)
|
|
await asyncio.sleep(60)
|
|
|
|
async def check_duplicates(self, message):
|
|
guild = message.guild
|
|
author = message.author
|
|
|
|
if await self.settings.guild(guild).delete_repeats():
|
|
if not message.content:
|
|
return False
|
|
self.cache[author].append(message)
|
|
msgs = self.cache[author]
|
|
if len(msgs) == 3 and msgs[0].content == msgs[1].content == msgs[2].content:
|
|
try:
|
|
await message.delete()
|
|
return True
|
|
except discord.HTTPException:
|
|
pass
|
|
return False
|
|
|
|
async def check_mention_spam(self, message):
|
|
guild = message.guild
|
|
author = message.author
|
|
|
|
max_mentions = await self.settings.guild(guild).ban_mention_spam()
|
|
if max_mentions:
|
|
mentions = set(message.mentions)
|
|
if len(mentions) >= max_mentions:
|
|
try:
|
|
await guild.ban(author, reason=_("Mention spam (Autoban)"))
|
|
except discord.HTTPException:
|
|
log.info(
|
|
"Failed to ban member for mention spam in server {}.".format(guild.id)
|
|
)
|
|
else:
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot,
|
|
guild,
|
|
message.created_at,
|
|
"ban",
|
|
author,
|
|
guild.me,
|
|
_("Mention spam (Autoban)"),
|
|
until=None,
|
|
channel=None,
|
|
)
|
|
except RuntimeError as e:
|
|
print(e)
|
|
return False
|
|
return True
|
|
return False
|
|
|
|
async def on_command_completion(self, ctx: commands.Context):
|
|
await self._delete_delay(ctx)
|
|
|
|
# noinspection PyUnusedLocal
|
|
async def on_command_error(self, ctx: commands.Context, error):
|
|
await self._delete_delay(ctx)
|
|
|
|
async def _delete_delay(self, ctx: commands.Context):
|
|
"""Currently used for:
|
|
* delete delay"""
|
|
guild = ctx.guild
|
|
if guild is None:
|
|
return
|
|
message = ctx.message
|
|
delay = await self.settings.guild(guild).delete_delay()
|
|
|
|
if delay == -1:
|
|
return
|
|
|
|
async def _delete_helper(m):
|
|
with contextlib.suppress(discord.HTTPException):
|
|
await m.delete()
|
|
log.debug("Deleted command msg {}".format(m.id))
|
|
|
|
await asyncio.sleep(delay)
|
|
await _delete_helper(message)
|
|
|
|
async def on_message(self, message):
|
|
author = message.author
|
|
if message.guild is None or self.bot.user == author:
|
|
return
|
|
valid_user = isinstance(author, discord.Member) and not author.bot
|
|
if not valid_user:
|
|
return
|
|
|
|
# Bots and mods or superior are ignored from the filter
|
|
mod_or_superior = await is_mod_or_superior(self.bot, obj=author)
|
|
if mod_or_superior:
|
|
return
|
|
# As are anyone configured to be
|
|
if await self.bot.is_automod_immune(message):
|
|
return
|
|
deleted = await self.check_duplicates(message)
|
|
if not deleted:
|
|
await self.check_mention_spam(message)
|
|
|
|
async def on_member_ban(self, guild: discord.Guild, member: discord.Member):
|
|
if (guild.id, member.id) in self.ban_queue:
|
|
self.ban_queue.remove((guild.id, member.id))
|
|
return
|
|
try:
|
|
await modlog.get_modlog_channel(guild)
|
|
except RuntimeError:
|
|
return # No modlog channel so no point in continuing
|
|
mod, reason, date = await self.get_audit_entry_info(
|
|
guild, discord.AuditLogAction.ban, member
|
|
)
|
|
if date is None:
|
|
date = datetime.now()
|
|
try:
|
|
await modlog.create_case(
|
|
self.bot, guild, date, "ban", member, mod, reason if reason else None
|
|
)
|
|
except RuntimeError as e:
|
|
print(e)
|
|
|
|
async def on_member_unban(self, guild: discord.Guild, user: discord.User):
|
|
if (guild.id, user.id) in self.unban_queue:
|
|
self.unban_queue.remove((guild.id, user.id))
|
|
return
|
|
try:
|
|
await modlog.get_modlog_channel(guild)
|
|
except RuntimeError:
|
|
return # No modlog channel so no point in continuing
|
|
mod, reason, date = await self.get_audit_entry_info(
|
|
guild, discord.AuditLogAction.unban, user
|
|
)
|
|
if date is None:
|
|
date = datetime.now()
|
|
try:
|
|
await modlog.create_case(self.bot, guild, date, "unban", user, mod, reason)
|
|
except RuntimeError as e:
|
|
print(e)
|
|
|
|
@staticmethod
|
|
async def on_modlog_case_create(case: modlog.Case):
|
|
"""
|
|
An event for modlog case creation
|
|
"""
|
|
try:
|
|
mod_channel = await modlog.get_modlog_channel(case.guild)
|
|
except RuntimeError:
|
|
return
|
|
use_embeds = await case.bot.embed_requested(mod_channel, case.guild.me)
|
|
case_content = await case.message_content(use_embeds)
|
|
if use_embeds:
|
|
msg = await mod_channel.send(embed=case_content)
|
|
else:
|
|
msg = await mod_channel.send(case_content)
|
|
await case.edit({"message": msg})
|
|
|
|
@staticmethod
|
|
async def on_modlog_case_edit(case: modlog.Case):
|
|
"""
|
|
Event for modlog case edits
|
|
"""
|
|
if not case.message:
|
|
return
|
|
use_embed = await case.bot.embed_requested(case.message.channel, case.guild.me)
|
|
case_content = await case.message_content(use_embed)
|
|
if use_embed:
|
|
await case.message.edit(embed=case_content)
|
|
else:
|
|
await case.message.edit(content=case_content)
|
|
|
|
@classmethod
|
|
async def get_audit_entry_info(
|
|
cls, guild: discord.Guild, action: discord.AuditLogAction, target
|
|
):
|
|
"""Get info about an audit log entry.
|
|
|
|
Parameters
|
|
----------
|
|
guild : discord.Guild
|
|
Same as ``guild`` in `get_audit_log_entry`.
|
|
action : int
|
|
Same as ``action`` in `get_audit_log_entry`.
|
|
target : `discord.User` or `discord.Member`
|
|
Same as ``target`` in `get_audit_log_entry`.
|
|
|
|
Returns
|
|
-------
|
|
tuple
|
|
A tuple in the form``(mod: discord.Member, reason: str,
|
|
date_created: datetime.datetime)``. Returns ``(None, None, None)``
|
|
if the audit log entry could not be found.
|
|
"""
|
|
try:
|
|
entry = await cls.get_audit_log_entry(guild, action=action, target=target)
|
|
except discord.HTTPException:
|
|
entry = None
|
|
if entry is None:
|
|
return None, None, None
|
|
return entry.user, entry.reason, entry.created_at
|
|
|
|
@staticmethod
|
|
async def get_audit_log_entry(guild: discord.Guild, action: discord.AuditLogAction, target):
|
|
"""Get an audit log entry.
|
|
|
|
Any exceptions encountered when looking through the audit log will be
|
|
propogated out of this function.
|
|
|
|
Parameters
|
|
----------
|
|
guild : discord.Guild
|
|
The guild for the audit log.
|
|
action : int
|
|
The audit log action (see `discord.AuditLogAction`).
|
|
target : `discord.Member` or `discord.User`
|
|
The target of the audit log action.
|
|
|
|
Returns
|
|
-------
|
|
discord.AuditLogEntry
|
|
The audit log entry. Returns ``None`` if not found.
|
|
|
|
"""
|
|
async for entry in guild.audit_logs(action=action):
|
|
if entry.target == target:
|
|
return entry
|
|
|
|
async def on_member_update(self, before: discord.Member, after: discord.Member):
|
|
if before.name != after.name:
|
|
async with self.settings.user(before).past_names() as name_list:
|
|
while None in name_list: # clean out null entries from a bug
|
|
name_list.remove(None)
|
|
if after.name in name_list:
|
|
# Ensure order is maintained without duplicates occuring
|
|
name_list.remove(after.name)
|
|
name_list.append(after.name)
|
|
while len(name_list) > 20:
|
|
name_list.pop(0)
|
|
|
|
if before.nick != after.nick and after.nick is not None:
|
|
async with self.settings.member(before).past_nicks() as nick_list:
|
|
while None in nick_list: # clean out null entries from a bug
|
|
nick_list.remove(None)
|
|
if after.nick in nick_list:
|
|
nick_list.remove(after.nick)
|
|
nick_list.append(after.nick)
|
|
while len(nick_list) > 20:
|
|
nick_list.pop(0)
|
|
|
|
|
|
_ = lambda s: s
|
|
mute_unmute_issues = {
|
|
"already_muted": _("That user can't send messages in this channel."),
|
|
"already_unmuted": _("That user isn't muted in this channel."),
|
|
"hierarchy_problem": _(
|
|
"I cannot let you do that. You are not higher than the user in the role hierarchy."
|
|
),
|
|
"is_admin": _("That user cannot be muted, as they have the Administrator permission."),
|
|
"permissions_issue": _(
|
|
"Failed to mute user. I need the manage roles "
|
|
"permission and the user I'm muting must be "
|
|
"lower than myself in the role hierarchy."
|
|
),
|
|
}
|