import discord
import asyncio
from discord.ext import commands
from config import log_channel, staff_role_ids, named_roles, community_channels, general_channels


class Lockdown:
    "Lockdown Commands"

    def __init__(self, bot):
        self.bot = bot

    @commands.has_permissions(manage_messages=True)
    @commands.command(pass_context=True)
    async def lock(self, ctx):
        """Locks the channel"""
        log_chan = self.bot.get_channel(log_channel)
        if ctx.message.channel in community_channels:
            roles = (named_roles["hacker"], named_roles["community"])
        else:
            roles = named_roles["participant"]
        overwrites = ctx.message.channel.overwrites_for(roles[0])
        if not overwrites.send_message:
            await ctx.send("This channel is already locked!")
            return
        overwrites.send_message = False
        overwrites.add_reactions = False
        await asyncio.gather(*(self.bot.edit_channel_permissions(ctx.channel, role, overwrites) for role in roles))
        await ctx.send("🔒 Channel locked down. Only staff members may speak. Do not bring the topic to other channels or risk disciplinary actions.")
        await log_chan.send(f"Channel Lockdown for {ctx.channel.mention} by {ctx.author.mention}")


def setup(bot):
    bot.add_cog(Lockdown(bot))