def _define_methods(self):
d = self._define_proxy_method
r = self._control.server
d(frostbite.commands.Version, r.version, filter=lambda r: r.split(" "))
d(frostbite.commands.ServerInfo, r.info, filter=lambda i: ServerState.from_dict(i).to_packet_array())
d(frostbite.commands.AdminListPlayers, r.list_all_players, filter=lambda p: PlayerCollection.from_dict(p).to_packet_array())
d(frostbite.commands.FrostbiteVariable, r.get_variable, filter=lambda v: [v], extract_args=lambda m:[m.words[0]])
d(frostbite.commands.MapListList, r.list_maps, filter=lambda m: MapList.from_dict(m).to_packet_array())
d(frostbite.commands.MapListGetMapIndices, r.get_map_indices)
d(frostbite.commands.MapListRestartRound, r.restart_round)
d(frostbite.commands.MapListRunNextRound, r.next_round)
d(frostbite.commands.MapListEndRound, r.end_round, extract_args=lambda m:[m.winning_team])
d(frostbite.commands.AdminSay, r.say, extract_args=lambda m:[m.message])
for var in frostbite.commands.variable_types:
type = frostbite.commands.variable_types[var]
self._command_to_method[type] = self._command_to_method[frostbite.commands.FrostbiteVariable]
def _parse_players_list(self, packet, client):
return PlayerCollection.from_packet_array(packet.words[1:])
def _parse_players_list(self, packet, client): return PlayerCollection.from_packet_array(packet.words[1:])
def _handle_command(self, server, seq, command):
if isinstance(command, frostbite.commands.LoginHashed):
if command.password is None:
server.send(seq, frostbite.commands.ResponsePacket("OK", "1234"))
else:
server.send(seq, frostbite.commands.ResponsePacket("OK"))
elif isinstance(command, frostbite.commands.Version):
server.send(seq, frostbite.commands.ResponsePacket("OK", "TestServer 1"))
elif isinstance(command, frostbite.commands.ServerInfo):
state = ServerState("testServer", 1, 16, "testmode0", "MP_Test", 1, 2, 2, [200, 200], 200, "", True, True, False, "1234", "200", "", "1.0", False, "US", "SF", "US")
server.send(seq, frostbite.commands.ResponsePacket("OK", *state.to_packet_array()))
elif isinstance(command, frostbite.commands.MapListRunNextRound):
self.last_command = "next_round"
server.send(seq, frostbite.commands.ResponsePacket("OK"))
elif isinstance(command, frostbite.commands.MapListRestartRound):
self.last_command = "restart_round"
server.send(seq, frostbite.commands.ResponsePacket("OK"))
elif isinstance(command, frostbite.commands.MapListEndRound):
self.last_command = "end_round %s" % command.winning_team
server.send(seq, frostbite.commands.ResponsePacket("OK"))
elif isinstance(command, frostbite.commands.MapListList):
maps = MapList([Map("MP_Test1", "TestMode0", 2), Map("MP_Test2", "TestMode1", 5)])
server.send(seq, frostbite.commands.ResponsePacket("OK", *maps.to_packet_array()))
elif isinstance(command, frostbite.commands.MapListGetMapIndices):
server.send(seq, frostbite.commands.ResponsePacket("OK", "0", "1"))
elif isinstance(command, frostbite.commands.MapListAdd):
self.last_command = "add_map %s %s %s" % (command.map_name, command.gamemode, command.rounds)
server.send(seq, frostbite.commands.ResponsePacket("OK"))
elif isinstance(command, frostbite.commands.MapListSetNextMapIndex):
self.last_command = "set_next_map %s" % command.index
server.send(seq, frostbite.commands.ResponsePacket("OK"))
elif isinstance(command, frostbite.commands.MapListClear):
self.last_command = "clear_map_list"
server.send(seq, frostbite.commands.ResponsePacket("OK"))
elif isinstance(command, frostbite.commands.MapListSave):
self.last_command = "save_map_list"
server.send(seq, frostbite.commands.ResponsePacket("OK"))
elif isinstance(command, frostbite.commands.MapListRemove):
self.last_command = "remove_map %s" % command.index
server.send(seq, frostbite.commands.ResponsePacket("OK"))
elif isinstance(command, frostbite.commands.AdminListPlayers):
list = PlayerCollection(["field1", "field2", "field3"], [["what", "the", "crap"]])
server.send(seq, frostbite.commands.ResponsePacket("OK", *list.to_packet_array()))
elif isinstance(command, frostbite.commands.AdminKickPlayer):
self.last_command = "kick_player %s %s" % (command.player_name, command.reason)
server.send(seq, frostbite.commands.ResponsePacket("OK"))
elif isinstance(command, frostbite.commands.AdminKillPlayer):
self.last_command = "kill_player %s" % (command.player_name)
server.send(seq, frostbite.commands.ResponsePacket("OK"))
elif isinstance(command, frostbite.commands.BanListAdd):
self.last_command = "add_ban %s %s %s %s" % (command.id_type, command.id, command.timeout, command.reason)
server.send(seq, frostbite.commands.ResponsePacket("OK"))
elif isinstance(command, frostbite.commands.BanListList):
list = [Ban("guid", "random-id", "forever", "100", "no reason")]
flat = []
for b in list:
flat.extend(b.to_packet_array())
server.send(seq, frostbite.commands.ResponsePacket("OK", len(list), *flat))
elif isinstance(command, frostbite.commands.AdminSay):
self.last_command = "say %s %s" % (command.scope, command.message)
server.send(seq, frostbite.commands.ResponsePacket("OK"))
else:
server.send(seq, frostbite.commands.ResponsePacket("OK"))
