Solidify imbusy blocking pms, tpas, etc. #42

Merged
Dico200 merged 8 commits from imbusy-reply-patch into dev 2016-05-28 20:22:01 +00:00
Showing only changes of commit a2a43ed464 - Show all commits

View File

@ -144,6 +144,7 @@ class CommandWrapper(Command):
self.checker = checker self.checker = checker
def execute(self, sender, label, args): def execute(self, sender, label, args):
info("/" + self.getLabel() + " executed by " + sender.getName())
try: try:
if self.checker(sender, args): if self.checker(sender, args):
return self.wrapped.execute(sender, label, args) return self.wrapped.execute(sender, label, args)
@ -184,39 +185,40 @@ def on_player_command_preprocess(event):
def replace_ess_commands(): def replace_ess_commands():
try: try:
mapField = server.getPluginManager().getClass().getDeclaredField("commandMap") info("[imbusy] Wrapping ess commands")
mapField.setAccessible(True) map_field = server.getPluginManager().getClass().getDeclaredField("commandMap")
commandMap = mapField.get(server.getPluginManager()) map_field.setAccessible(True)
command_map = map_field.get(server.getPluginManager())
commandsField = commandMap.getClass().getDeclaredField("knownCommands") commands_field = command_map.getClass().getDeclaredField("knownCommands")
commandsField.setAccessible(True) commands_field.setAccessible(True)
map = commandsField.get(commandMap) map = commands_field.get(command_map)
essMsgCmd = map.get("essentials:msg") ess_msg_cmd = map.get("essentials:msg")
essReplyCmd = map.get("essentials:reply") ess_reply_cmd = map.get("essentials:reply")
essTpaCmd = map.get("essentials:tpa") ess_tpa_cmd = map.get("essentials:tpa")
essTpahereCmd = map.get("essentials:tpahere") ess_tpahere_cmd = map.get("essentials:tpahere")
msgCmdWrapper = CommandWrapper(essMsgCmd, msg_command_checker) msg_cmd_wrapper = CommandWrapper(ess_msg_cmd, msg_command_checker)
replyCmdWrapper = CommandWrapper(essReplyCmd, reply_command_checker) reply_cmd_wrapper = CommandWrapper(ess_reply_cmd, reply_command_checker)
tpaCmdWrapper = CommandWrapper(essTpaCmd, tpa_command_checker) tpa_cmd_wrapper = CommandWrapper(ess_tpa_cmd, tpa_command_checker)
tpahereCmdWrapper = CommandWrapper(essTpahereCmd, tpahere_command_checker) tpahere_cmd_wrapper = CommandWrapper(ess_tpahere_cmd, tpahere_command_checker)
iterator = map.entrySet().iterator() iterator = map.entrySet().iterator()
while iterator.hasNext(): while iterator.hasNext():
entry = iterator.next() entry = iterator.next()
value = entry.getValue() value = entry.getValue()
if value is essMsgCmd: if value is ess_msg_cmd:
entry.setValue(msgCmdWrapper) entry.setValue(msg_cmd_wrapper)
info("[imbusy] wrapped /" + entry.getKey()) info("[imbusy] wrapped /" + entry.getKey())
elif value is essReplyCmd: elif value is ess_reply_cmd:
entry.setValue(replyCmdWrapper) entry.setValue(reply_cmd_wrapper)
info("[imbusy] wrapped /" + entry.getKey()) info("[imbusy] wrapped /" + entry.getKey())
elif value is essTpaCmd: elif value is ess_tpa_cmd:
entry.setValue(tpaCmdWrapper) entry.setValue(tpa_cmd_wrapper)
info("[imbusy] wrapped /" + entry.getKey()) info("[imbusy] wrapped /" + entry.getKey())
elif value is essTpahereCmd: elif value is ess_tpahere_cmd:
entry.setValue(tpahereCmdWrapper) entry.setValue(tpahere_cmd_wrapper)
info("[imbusy] wrapped /" + entry.getKey()) info("[imbusy] wrapped /" + entry.getKey())
except: except: