From ce889e608aa81108f3fc1f0e58151db5b2d1971c Mon Sep 17 00:00:00 2001
From: Giuseppe Guerra <to@nyo.zz.mu>
Date: Mon, 7 Aug 2017 22:30:44 +0200
Subject: [PATCH] Add !mp mods

---
 constants/fokabotCommands.py | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/constants/fokabotCommands.py b/constants/fokabotCommands.py
index eafff08..2e2f360 100644
--- a/constants/fokabotCommands.py
+++ b/constants/fokabotCommands.py
@@ -10,7 +10,7 @@ from common import generalUtils
 from common.constants import mods
 from common.log import logUtils as log
 from common.ripple import userUtils
-from constants import exceptions, slotStatuses
+from constants import exceptions, slotStatuses, matchModModes
 from common.constants import gameModes
 from common.constants import privileges
 from constants import serverPackets
@@ -968,6 +968,33 @@ def multiplayer(fro, chan, message):
 		_match.changePassword(password)
 		return "Match password has been changed to a random one"
 
+	def mpMods():
+		if len(message) < 2:
+			raise exceptions.invalidArgumentsException("Wrong syntax: !mp <mod1> [<mod2>] ...")
+		_match = glob.matches.matches[getMatchIDFromChannel(chan)]
+		newMods = 0
+		freeMod = False
+		for _mod in message[1:]:
+			if _mod.lower().strip() == "hd":
+				newMods |= mods.HIDDEN
+			elif _mod.lower().strip() == "hr":
+				newMods |= mods.HARDROCK
+			elif _mod.lower().strip() == "dt":
+				newMods |= mods.DOUBLETIME
+			elif _mod.lower().strip() == "fl":
+				newMods |= mods.FLASHLIGHT
+			elif _mod.lower().strip() == "fi":
+				newMods |= mods.FADEIN
+			if _mod.lower().strip() == "none":
+				newMods = 0
+
+			if _mod.lower().strip() == "freemod":
+				freeMod = True
+
+		_match.matchModMode = matchModModes.FREE_MOD if freeMod else matchModModes.NORMAL
+		_match.changeMods(newMods)
+
+
 	try:
 		subcommands = {
 			"make": mpMake,
@@ -987,6 +1014,7 @@ def multiplayer(fro, chan, message):
 			"kick": mpKick,
 			"password": mpPassword,
 			"randompassword": mpRandomPassword,
+			"mods": mpMods,
 		}
 		requestedSubcommand = message[0].lower().strip()
 		if requestedSubcommand not in subcommands: