Nemo/main.py

#!/usr/bin/env python3
import asyncio

import nemo
import helper
import config
import discord
from discord.ext.commands import has_permissions

number_emojis = ["1⃣", "2⃣", "3⃣", "4⃣", "5⃣", "6⃣", "7⃣", "8⃣", "9⃣"]

nemo = nemo.Nemo()


@nemo.command("!")
@has_permissions(administrator=True)
async def setup(*, message: discord.Message, guild: discord.Guild, **_):
	if any(x.name == config.CATEGORY_NAME and x.type == discord.ChannelType.category for x in guild.channels):
		return
	category = await guild.create_category(config.CATEGORY_NAME)
	organization = await guild.create_text_channel(config.ORGANIZATION_NAME, overwrites={
		guild.default_role: discord.PermissionOverwrite(send_messages=False),
		guild.me: discord.PermissionOverwrite(send_messages=True)
	}, category=category)
	await help_msg(organization)
	await message.delete()


@nemo.command("!help")
@has_permissions(administrator=True)
@helper.auto_delete
async def help_command(*, message: discord.Message, **_):
	await help_msg(message.channel)


async def help_msg(channel: discord.TextChannel):
	await channel.send(config.HELP_MSG.replace("@Me", f"<@{nemo.user.id}>"))
	await channel.send(config.EVENT_LIST)
	msg = await channel.send(config.CREATE_MSG)
	await msg.add_reaction("✅")


@nemo.reaction(lambda reaction, member: reaction.message.author.bot and config.LIST_KEY in reaction.message.content)
async def join_event(*,
					 reaction: discord.Reaction,
					 member: discord.Member,
					 guild: discord.Guild,
					 channel: discord.TextChannel,
					 **_):
	event: int = number_emojis.index(reaction.emoji) + 1
	role: str = discord.utils.get(guild.roles, name=f"{config.PARTICIPANT_PREFIX}{event}")
	if role is None or role in member.roles:
		return
	
	list_msg: discord.Message = [x async for x in channel.history() if config.LIST_KEY in x.content][0]
	try:
		status = list_msg.content.split('\n')[event + 1]
	except IndexError:
		return
	if config.EMPTY_SLOT in status or config.EVENT_CONFIGURING_KEY in status or config.PRIVATE_EVENT in status:
		return
	
	await member.add_roles(role)
	event_channel: discord.TextChannel = discord.utils.get(guild.channels, name=f"{event}{config.EVENT_SUFFIX}")
	await event_channel.send(config.EVENT_JOIN.replace("@User", f"<@{member.id}>"))
	

@nemo.reaction(lambda reaction, member: reaction.message.author.bot and config.CREATE_KEY in reaction.message.content)
async def create_event(reaction: discord.Reaction, 
					   message: discord.Message, 
					   member: discord.Member,
					   guild: discord.Guild, **_):
	list_msg: discord.Message = [x async for x in message.channel.history() if config.LIST_KEY in x.content][0]
	index = get_new_event_index(list_msg)
	if index == -1:
		await reaction.remove(member)
		return

	await edit_event_status(index, config.CONFIGURING_EVENT.replace("@User", f"<@{member.id}>"), list_msg)
	await reaction.remove(member)

	org_role = await guild.create_role(name=f"{config.ORGANIZER_PREFIX}{index}", color=discord.Color.purple())
	await member.add_roles(org_role)
	user_role = await guild.create_role(name=f"{config.PARTICIPANT_PREFIX}{index}")
	await member.add_roles(user_role)
	channel = await guild.create_text_channel(f"{index}{config.EVENT_SUFFIX}", overwrites={
		guild.default_role: discord.PermissionOverwrite(read_messages=False),
		guild.me: discord.PermissionOverwrite(read_messages=True),
		org_role: discord.PermissionOverwrite(read_messages=True, manage_messages=True),
		user_role: discord.PermissionOverwrite(read_messages=True)
	}, category=message.channel.category)
	await list_msg.add_reaction(number_emojis[index - 1])
	await channel.send(config.SETUP_MSG)


def get_new_event_index(msg: discord.Message):
	if config.EMPTY_KEY in msg.content:
		return 1
	for x in msg.content.split('\n'):
		if config.EMPTY_SLOT in x:
			return int(x[2:x.index(':')])
	last_slot = msg.content.split('\n')[-1]
	slot = int(last_slot[2:last_slot.index(':')]) + 1
	return slot if slot < 10 else -1


async def edit_event_status(index: int, text: str, list_msg: discord.Message):
	content = list_msg.content.split('\n')
	try:
		content[index + 1] = f"**{index}:** {text}"
	except IndexError:
		content.append(f"**{index}:** {text}")
	await list_msg.edit(content="\n".join(content))


@nemo.command("!close")
@nemo.command("!stop")
@helper.event_command
@helper.organizer_only
async def stop(*,
			   channel: discord.TextChannel,
			   member: discord.Member,
			   message: discord.Message,
			   guild: discord.Guild,
			   event: int,
			   **_):
	await message.delete()
	msg = await channel.send(config.STOP_MSG.replace("@User", f"<@{member.id}>"))
	await msg.add_reaction("✅")
	try:
		await nemo.wait_for("reaction_add", timeout=60, check=lambda reaction, user: user == message.author)
	except asyncio.TimeoutError:
		await msg.delete()
		return
	await discord.utils.get(guild.roles, name=f"{config.ORGANIZER_PREFIX}{event}").delete()
	await discord.utils.get(guild.roles, name=f"{config.PARTICIPANT_PREFIX}{event}").delete()
	await channel.delete()
	org_channel: discord.TextChannel = discord.utils.get(guild.channels, name=config.ORGANIZATION_NAME)
	list_msg: discord.Message = [x async for x in org_channel.history() if config.LIST_KEY in x.content][0]
	if f"**{event + 1}:**" in list_msg.content:
		await edit_event_status(event, config.EMPTY_SLOT, list_msg)
	elif event == 1:
		await list_msg.edit(content=config.EVENT_LIST)
	else:
		await list_msg.edit(content='\n'.join(list_msg.content.split('\n')[:-1]))
	await list_msg.clear_reaction(number_emojis[event - 1])


@nemo.command("!open")
@helper.event_command
@helper.organizer_only
@helper.auto_delete
async def open_cmd(*,
				   message: discord.Message,
				   guild: discord.Guild,
				   channel: discord.TextChannel,
				   event: int,
				   **_):
	org_channel: discord.TextChannel = discord.utils.get(guild.channels, name=config.ORGANIZATION_NAME)
	list_msg: discord.Message = [x async for x in org_channel.history() if config.LIST_KEY in x.content][0]
	status = message.content[len("!open "):]
	skip_broadcast = config.EVENT_CONFIGURING_KEY not in list_msg.content
	await edit_event_status(event, status, list_msg)
	await channel.send(config.SET_EVENT_MSG.replace("@Status", status).replace("@User", f"<@{message.author.id}>"))
	if skip_broadcast:
		return
	async for x in org_channel.history():
		if config.NEW_EVENT_KEY in x.content:
			await x.delete()
	await org_channel.send(config.NEW_EVENT.replace("@everyone", f"{guild.default_role}").replace("@id", str(event)))


if __name__ == "__main__":
	nemo.run(config.TOKEN)