From 47b2bb35d53f1e11e62343d70c839b646cb6b600 Mon Sep 17 00:00:00 2001
From: Anonymus Raccoon <tr.tristan-roux@outlook.fr>
Date: Wed, 15 Jul 2020 02:41:15 +0200
Subject: [PATCH] Adding the kick & colab command

---
 config.py |  2 ++
 main.py   | 34 ++++++++++++++++++++++++++++++++++
 2 files changed, 36 insertions(+)

diff --git a/config.py b/config.py
index 2e6ac3a..4466c67 100644
--- a/config.py
+++ b/config.py
@@ -82,3 +82,5 @@ LEAVE_MSG = "@User vient de quitter l'event."
 INVITE_MSG = """@User vient d'inviter @Invited.
 Vous pouvez le quitter en tappant !leave.
 """
+KICK_MSG = """@User vient de kick @Invited."""
+COLAB_MSG = """@User vient d'ajouter @Invited en temps que collaborateur."""
diff --git a/main.py b/main.py
index 7ce7cf0..11668df 100644
--- a/main.py
+++ b/main.py
@@ -228,5 +228,39 @@ async def invite(*,
 	await channel.send(config.INVITE_MSG.replace("@User", f"<@{member.id}>").replace("@Invited", users))
 
 
+@nemo.command("!kick")
+@helper.event_command
+@helper.auto_delete
+async def kick(*,
+				 channel: discord.TextChannel,
+				 member: discord.Member,
+				 message: discord.Message,
+				 guild: discord.Guild,
+				 event: int,
+				 **_):
+	role = discord.utils.get(guild.roles, name=f"{config.PARTICIPANT_PREFIX}{event}")
+	for user in message.mentions:
+		await user.remove_roles(role)
+	users = ", ".join([f"<@{user.id}>" for user in message.mentions])
+	await channel.send(config.KICK_MSG.replace("@User", f"<@{member.id}>").replace("@Invited", users))
+
+
+@nemo.command("!colab")
+@helper.event_command
+@helper.auto_delete
+async def colab(*,
+				 channel: discord.TextChannel,
+				 member: discord.Member,
+				 message: discord.Message,
+				 guild: discord.Guild,
+				 event: int,
+				 **_):
+	role = discord.utils.get(guild.roles, name=f"{config.ORGANIZER_PREFIX}{event}")
+	for user in message.mentions:
+		await user.add_roles(role)
+	users = ", ".join([f"<@{user.id}>" for user in message.mentions])
+	await channel.send(config.COLAB_MSG.replace("@User", f"<@{member.id}>").replace("@Invited", users))
+	
+
 if __name__ == "__main__":
 	nemo.run(config.TOKEN)