44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from common.constants import mods
|
|
from constants import clientPackets
|
|
from constants import matchModModes
|
|
from objects import glob
|
|
|
|
|
|
def handle(userToken, packetData):
|
|
# Get token data
|
|
userID = userToken.userID
|
|
|
|
# Get packet data
|
|
packetData = clientPackets.changeMods(packetData)
|
|
|
|
# Make sure the match exists
|
|
matchID = userToken.matchID
|
|
if matchID not in glob.matches.matches:
|
|
return
|
|
|
|
# Set slot or match mods according to modType
|
|
with glob.matches.matches[matchID] as match:
|
|
if match.matchModMode == matchModModes.FREE_MOD:
|
|
# Freemod
|
|
# Host can set global DT/HT
|
|
if userID == match.hostUserID:
|
|
# If host has selected DT/HT and Freemod is enabled, set DT/HT as match mod
|
|
if (packetData["mods"] & mods.DOUBLETIME) > 0:
|
|
match.changeMods(mods.DOUBLETIME)
|
|
# Nightcore
|
|
if (packetData["mods"] & mods.NIGHTCORE) > 0:
|
|
match.changeMods(match.mods + mods.NIGHTCORE)
|
|
elif (packetData["mods"] & mods.HALFTIME) > 0:
|
|
match.changeMods(mods.HALFTIME)
|
|
else:
|
|
# No DT/HT, set global mods to 0 (we are in freemod mode)
|
|
match.changeMods(0)
|
|
|
|
# Set slot mods
|
|
slotID = match.getUserSlotID(userID)
|
|
if slotID is not None:
|
|
match.setSlotMods(slotID, packetData["mods"])
|
|
else:
|
|
# Not freemod, set match mods
|
|
match.changeMods(packetData["mods"])
|