Ejemplo n.º 1
0
    def __init__(self, ip, port, sock = None):
        ClientHandler.__init__(self)

        self.message_handler = Handler("\r\n")
        self.action_handlers = { "connection_response" : self.handle_connection_response , 
                                 'session_logon' : self.handle_session_logon, 
                                 'ping' : self.handle_ping, 
                                 'lobby_session_logon' : self.session_logon,
                                 'lobby_info' : self.handle_lobby_info,
                                 'logon_response' : self.handle_session_logon,
                                 'lobby_create_game' : self.handle_game_create,
                                 'wargame_lobby_join' : self.handle_join_game,
                                 'wargame_startup_info' : self.handle_startup_info,
                                 'wargame_handshake' : self.handle_wargame_handshake }

        lobby_actions = [ ('say something' , { 'action' : 'lobby_chat', 'fields' : [ ('message', str, True) ] }),
                          ('create game' , { 'action' : 'lobby_create_game', 'fields' : [('game_type', str, True), ('room_name', str, True) ] } ),
                          ('join game', { 'action' : 'lobby_join_game' , 'fields' : [('room_id', str, True) ] } )  ]
        
        game_lobby_actions  = [ ('say something' , { 'action' : 'wargame_lobby_chat', 'fields' : [ ('message', str, True) ] }),
                                ('change color', { 'action' : 'wargame_lobby_set_self_color', 'fields' : [ ('color', str, True) ] } ),
                                ('set ready' , { 'action' : 'wargame_lobby_set_self_ready', 'fields' : [ ('ready', str, True) ] }),
                                ('start game', { 'action' : 'wargame_lobby_start_game' , 'fields' : [] } )]
        
        in_game_actions = [ ('remove a piece', {'action' : 'wargame_remove_piece', 'fields' : [ ('from', str, True) , ('quantity', int, True ) ] } ),
                            ('add a piece', {'action' : 'wargame_add_piece', 'fields' : [ ('to', str, True) , ('quantity', int, True ) ] } ),
                            ('say something' , { 'action' : 'wargame_chat', 'fields' : [ ('message', str, True ) ] } ),
                            ('attack a land', {'action' : 'wargame_attack_land', 'fields' : [ ('quantity', str, True), ('from', str, True), ('to', str, True) ] } ),
                            ('end turn' , { 'action' : 'wargame_end_turn', 'fields' : [] }  ) ]                                               
        
        self.menu = [ lobby_actions, game_lobby_actions, in_game_actions ]
        
        self.current_state = self.LOBBY
        
        self.read_handler = self.handle_message

        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
        self.connect((ip , port))
        
        self.showing_menu = False
Ejemplo n.º 2
0
class Client(ClientHandler):
    LOBBY = 0
    GAME_LOBBY = 1
    IN_GAME = 2
    
    def __init__(self, ip, port, sock = None):
        ClientHandler.__init__(self)

        self.message_handler = Handler("\r\n")
        self.action_handlers = { "connection_response" : self.handle_connection_response , 
                                 'session_logon' : self.handle_session_logon, 
                                 'ping' : self.handle_ping, 
                                 'lobby_session_logon' : self.session_logon,
                                 'lobby_info' : self.handle_lobby_info,
                                 'logon_response' : self.handle_session_logon,
                                 'lobby_create_game' : self.handle_game_create,
                                 'wargame_lobby_join' : self.handle_join_game,
                                 'wargame_startup_info' : self.handle_startup_info,
                                 'wargame_handshake' : self.handle_wargame_handshake }

        lobby_actions = [ ('say something' , { 'action' : 'lobby_chat', 'fields' : [ ('message', str, True) ] }),
                          ('create game' , { 'action' : 'lobby_create_game', 'fields' : [('game_type', str, True), ('room_name', str, True) ] } ),
                          ('join game', { 'action' : 'lobby_join_game' , 'fields' : [('room_id', str, True) ] } )  ]
        
        game_lobby_actions  = [ ('say something' , { 'action' : 'wargame_lobby_chat', 'fields' : [ ('message', str, True) ] }),
                                ('change color', { 'action' : 'wargame_lobby_set_self_color', 'fields' : [ ('color', str, True) ] } ),
                                ('set ready' , { 'action' : 'wargame_lobby_set_self_ready', 'fields' : [ ('ready', str, True) ] }),
                                ('start game', { 'action' : 'wargame_lobby_start_game' , 'fields' : [] } )]
        
        in_game_actions = [ ('remove a piece', {'action' : 'wargame_remove_piece', 'fields' : [ ('from', str, True) , ('quantity', int, True ) ] } ),
                            ('add a piece', {'action' : 'wargame_add_piece', 'fields' : [ ('to', str, True) , ('quantity', int, True ) ] } ),
                            ('say something' , { 'action' : 'wargame_chat', 'fields' : [ ('message', str, True ) ] } ),
                            ('attack a land', {'action' : 'wargame_attack_land', 'fields' : [ ('quantity', str, True), ('from', str, True), ('to', str, True) ] } ),
                            ('end turn' , { 'action' : 'wargame_end_turn', 'fields' : [] }  ) ]                                               
        
        self.menu = [ lobby_actions, game_lobby_actions, in_game_actions ]
        
        self.current_state = self.LOBBY
        
        self.read_handler = self.handle_message

        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
        self.connect((ip , port))
        
        self.showing_menu = False
                
    def print_menu(self):
      def respawn():
        threading.Timer(0.01, self.print_menu).start()
      
      if self.showing_menu:
        return
      
      self.showing_menu = True
      
      i = 0
      for item in self.menu[self.current_state]:
        print "{0} - {1}".format(i, item[0])
        i += 1
      
      value = raw_input("select a value: ")
      params = {}
      
      try:
        value = int(value)
      except:
        print 'Error: invalid format for menu value'
        respawn()
        return
      
      if value > len(self.menu[self.current_state]):
        print ' invalid value {0}.'.format(value)
        respawn()
        return
      
      reqd = { True : ' (required) ', False : ' (enter to proceed) ' }
      
      for param in self.menu[self.current_state][value][1]['fields']:
        item_value = raw_input('{0}{1}: '.format(param[0], reqd[param[2]]))        
        params[param[0]] = param[1](item_value)
      
      message = params
      message['action'] = self.menu[self.current_state][value][1]['action']
      
      self.write(message)
      
      self.showing_menu = False
   
    def handle_startup_info(self, message):
        print message
        self.current_state = self.IN_GAME
        
    def handle_join_game(self, message):
        if message['status'] == 'successfull':
            self.current_state = self.GAME_LOBBY
      
    def session_logon(self, message):
      print "session {0} logged.".format(message['username'])
    
    def handle_lobby_info(self, message):
      print 'received lobby info. :', message
      
    def handle_ping(self, message):
        if 'time' in message:
            before = float(message['time'])
            print 'response delay:', time.time() - before

    def handle_game_create(self, message):
      if message['status'] == 'sucessfull':
        self.current_state = self.GAME_LOBBY
      else:
        print 'error creating game: ', message
      
    def handle_wargame_handshake(self, message):
        self.current_state = self.IN_GAME
        
    def handle_session_logon(self, message):
        print 'received seession logon response: ', message
        if message['authenticated'] == 'yes':
            print 'authenticated'
            
    def handle_connection_response(self, message): 
        print 'received connection response: ', message
        self.write({"action" : "login", "username" : "user", "password" : "123456"})
    
    def write(self, message):
        ClientHandler.write(self, self.message_handler.to_string(message))
        
    def handle_connect(self):
        logging.info("Connected!")
        self.send("json\r\n")
            
    def handle_message(self, message):
        logging.debug("handle_message: {0}".format(message))
        if message['action'] in self.action_handlers:            
            self.action_handlers[message['action']](message)
        else:
            print "Unhandled message: {0}".format(message)
        
        if not self.showing_menu:
          threading.Timer(0.01, self.print_menu).start()