105 lines
3.5 KiB
Python
105 lines
3.5 KiB
Python
import discord
|
|
from discord.ext import commands
|
|
from discord.ext.commands import Cog
|
|
|
|
from robocop_ng.helpers.checks import check_if_staff
|
|
|
|
|
|
class Lockdown(Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
async def set_sendmessage(
|
|
self, channel: discord.TextChannel, role, allow_send, issuer
|
|
):
|
|
try:
|
|
roleobj = channel.guild.get_role(role)
|
|
overrides = channel.overwrites_for(roleobj)
|
|
overrides.send_messages = allow_send
|
|
await channel.set_permissions(
|
|
roleobj, overwrite=overrides, reason=str(issuer)
|
|
)
|
|
except:
|
|
pass
|
|
|
|
async def unlock_for_staff(self, channel: discord.TextChannel, issuer):
|
|
for role in self.bot.config.staff_role_ids:
|
|
await self.set_sendmessage(channel, role, True, issuer)
|
|
|
|
@commands.guild_only()
|
|
@commands.check(check_if_staff)
|
|
@commands.command()
|
|
async def lock(self, ctx, channel: discord.TextChannel = None, soft: bool = False):
|
|
"""Prevents people from speaking in a channel, staff only.
|
|
|
|
Defaults to current channel."""
|
|
if not channel:
|
|
channel = ctx.channel
|
|
log_channel = self.bot.get_channel(self.bot.config.modlog_channel)
|
|
|
|
roles = None
|
|
for key, lockdown_conf in self.bot.config.lockdown_configs.items():
|
|
if channel.id in lockdown_conf["channels"]:
|
|
roles = lockdown_conf["roles"]
|
|
|
|
if roles is None:
|
|
roles = self.bot.config.lockdown_configs["default"]["roles"]
|
|
|
|
for role in roles:
|
|
await self.set_sendmessage(channel, role, False, ctx.author)
|
|
|
|
await self.unlock_for_staff(channel, ctx.author)
|
|
|
|
public_msg = "🔒 Channel locked down. "
|
|
if not soft:
|
|
public_msg += (
|
|
"Only staff members may speak. "
|
|
"Do not bring the topic to other channels or risk "
|
|
"disciplinary actions."
|
|
)
|
|
|
|
await ctx.send(public_msg)
|
|
safe_name = await commands.clean_content(escape_markdown=True).convert(
|
|
ctx, str(ctx.author)
|
|
)
|
|
msg = (
|
|
f"🔒 **Lockdown**: {ctx.channel.mention} by {ctx.author.mention} "
|
|
f"| {safe_name}"
|
|
)
|
|
await log_channel.send(msg)
|
|
|
|
@commands.guild_only()
|
|
@commands.check(check_if_staff)
|
|
@commands.command()
|
|
async def unlock(self, ctx, channel: discord.TextChannel = None):
|
|
"""Unlocks speaking in current channel, staff only."""
|
|
if not channel:
|
|
channel = ctx.channel
|
|
log_channel = self.bot.get_channel(self.bot.config.modlog_channel)
|
|
|
|
roles = None
|
|
for key, lockdown_conf in self.bot.config.lockdown_configs.items():
|
|
if channel.id in lockdown_conf["channels"]:
|
|
roles = lockdown_conf["roles"]
|
|
|
|
if roles is None:
|
|
roles = self.bot.config.lockdown_configs["default"]["roles"]
|
|
|
|
await self.unlock_for_staff(channel, ctx.author)
|
|
|
|
for role in roles:
|
|
await self.set_sendmessage(channel, role, True, ctx.author)
|
|
|
|
safe_name = await commands.clean_content(escape_markdown=True).convert(
|
|
ctx, str(ctx.author)
|
|
)
|
|
await ctx.send("🔓 Channel unlocked.")
|
|
msg = (
|
|
f"🔓 **Unlock**: {ctx.channel.mention} by {ctx.author.mention} "
|
|
f"| {safe_name}"
|
|
)
|
|
await log_channel.send(msg)
|
|
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(Lockdown(bot))
|