class SocketHandler(WebSocketHandler):
def __init__(self, application, request, **kwargs):
super().__init__(application, request, **kwargs)
self.db: torndb.Connection = self.application.db
self.player: Player = None
def data_received(self, chunk):
logger.info('socket data_received')
def get_current_user(self):
return json_decode(self.get_secure_cookie("user"))
@property
def uid(self):
return self.player.uid
@property
def room(self):
return self.player.room
@authenticated
def open(self):
user = self.current_user
self.player = Player(user['uid'], user['username'], self)
logger.info('SOCKET[%s] OPEN', self.player.uid)
def on_close(self):
self.player.leave_table()
logger.info('SOCKET[%s] CLOSE', self.player.uid)
def on_message(self, message):
packet = msgpack.unpackb(message, use_list=False)
# packet = json.loads(message)
logger.info('REQ[%d]: %s', self.uid, packet)
code = packet[0]
if code == Pt.REQ_LOGIN:
response = [Pt.RSP_LOGIN, self.player.uid, self.player.name]
self.write_message(response)
elif code == Pt.REQ_ROOM_LIST:
self.write_message([Pt.RSP_ROOM_LIST])
elif code == Pt.REQ_TABLE_LIST:
self.write_message([Pt.RSP_TABLE_LIST, self.room.rsp_tables()])
elif code == Pt.REQ_JOIN_ROOM:
self.player.room = RoomManager.find_room(packet[1])
self.write_message([Pt.RSP_JOIN_ROOM, self.room.rsp_tables()])
elif code == Pt.REQ_NEW_TABLE:
# TODO: check player was already in table.
table = self.room.new_table()
self.player.join_table(table)
logger.info('PLAYER[%s] NEW TABLE[%d]', self.uid, table.uid)
self.write_message([Pt.RSP_NEW_TABLE, table.uid])
elif code == Pt.REQ_JOIN_TABLE:
table = self.room.find_waiting_table(packet[1])
if not table:
self.write_message([Pt.RSP_TABLE_LIST, self.room.rsp_tables()])
logger.info('PLAYER[%d] JOIN TABLE[%d] NOT FOUND', self.uid,
packet[1])
return
self.player.join_table(table)
logger.info('PLAYER[%s] JOIN TABLE[%d]', self.uid, table.uid)
if table.is_full():
table.deal_poker()
self.room.on_table_changed(table)
logger.info('TABLE[%s] GAME BEGIN[%s]', table.uid,
table.players)
elif code == Pt.REQ_CALL_SCORE:
self.handle_call_score(packet)
elif code == Pt.REQ_DEAL_POKER:
if self.player.table.state == 2:
self.player.ready = True
self.player.table.ready()
elif code == Pt.REQ_SHOT_POKER:
self.handle_shot_poker(packet)
elif code == Pt.REQ_CHAT:
self.handle_chat(packet)
elif code == Pt.REQ_CHEAT:
self.handle_cheat(packet[1])
else:
logger.info('UNKNOWN PACKET: %s', code)
@shot_turn
def handle_call_score(self, packet):
score = packet[1]
self.player.handle_call_score(score)
@shot_turn
def handle_shot_poker(self, packet):
pokers = packet[1]
self.player.handle_shot_poker(pokers)
def handle_chat(self, packet):
if self.player.table:
self.player.table.handle_chat(self.player, packet[1])
def handle_cheat(self, uid):
for p in self.player.table.players:
if p.uid == uid:
self.player.send([Pt.RSP_CHEAT, p.uid, p.hand_pokers])
def write_message(self, message, binary=False):
if self.ws_connection is None:
raise WebSocketClosedError()
# packet = msgpack.packb(message)
logger.info('RSP[%d]: %s', self.uid, message)
packet = json.dumps(message)
return self.ws_connection.write_message(packet, binary=binary)
def send_updates(cls, chat):
logger.info('sending message to %d waiters', len(cls.waiters))
for waiter in cls.waiters:
waiter.write_message('tornado:' + chat)
class SocketHandler(WebSocketHandler):
def __init__(self, application, request, **kwargs):
super().__init__(application, request, **kwargs)
self.db: torndb.Connection = self.application.db
self.player: Player = None
def data_received(self, chunk):
logger.info('socket data_received')
@property
def uid(self):
return self.player.uid
@property
def room(self):
return self.player.room
def open(self):
self.player = Player(10, 'Jerry', self)
self.player.room = RoomManager.find_room(0, True)
logger.info('SOCKET[%s] OPEN', self.player.uid)
def on_message(self, message):
packet = msgpack.unpackb(message, use_list=False)
# packet = json.loads(message)
logger.info('REQ[%d]: %s', self.uid, packet)
code = packet[0]
if code == Pt.REQ_LOGIN:
response = [Pt.RSP_LOGIN, self.player.uid, self.player.name]
self.write_message(response)
elif code == Pt.REQ_ROOM_LIST:
response = [Pt.RSP_ROOM_LIST]
self.write_message(response)
elif code == Pt.REQ_TABLE_LIST:
response = [Pt.RSP_TABLE_LIST, self.room.tables()]
self.write_message(response)
elif code == Pt.REQ_JOIN_TABLE:
table_id = packet[1]
table = self.find_table(table_id)
if not table:
response = [Pt.RSP_TABLE_LIST, self.room.tables()]
self.write_message(response)
logger.info('PLAYER[%d] JOIN FULL TABLE[%d]', self.uid, table.uid)
self.player.join_table(table)
logger.info('PLAYER[%s] JOIN TABLE[%d]', self.uid, table.uid)
table.sync_table()
if table.size() == 3:
table.deal_poker()
logger.info('TABLE[%s] GAME BEGIN[%s]', table.uid, table.players)
del self.room.waiting_tables[table.uid]
self.room.playing_tables[table.uid] = table
elif code == Pt.REQ_CALL_SCORE:
self.handle_call_score(packet)
elif code == Pt.REQ_DEAL_POKER:
if self.player.table.state == 2:
self.player.ready = True
self.player.table.ready()
elif code == Pt.REQ_SHOT_POKER:
self.handle_shot_poker(packet)
elif code == Pt.REQ_CHAT:
self.handle_chat(packet)
else:
logger.info('UNKNOWN PACKET: %s', code)
@shot_turn
def handle_call_score(self, packet):
score = packet[1]
self.player.handle_call_score(score)
@shot_turn
def handle_shot_poker(self, packet):
pokers = packet[1]
self.player.handle_shot_poker(pokers)
def find_table(self, table_id):
if table_id == -1: # fast join
return self.room.first_waiting_table()
return self.room.find_waiting_table(table_id)
def handle_chat(self, packet):
if self.player and self.player.table:
self.player.table.handle_chat(self.player, packet[1])
def write_message(self, message, binary=False):
if self.ws_connection is None:
raise WebSocketClosedError()
# packet = msgpack.packb(message)
logger.info('RSP[%d]: %s', self.uid, message)
packet = json.dumps(message)
return self.ws_connection.write_message(packet, binary=binary)
def on_close(self):
# self.session.get(self.uid).socket = None
logger.info('socket[%s] close', self.player.uid)
if self.player:
self.player.leave_table()
def send_updates(cls, chat):
logger.info('sending message to %d waiters', len(cls.waiters))
for waiter in cls.waiters:
waiter.write_message('tornado:' + chat)
class SocketHandler(WebSocketHandler):
def __init__(self, application, request, **kwargs):
super().__init__(application, request, **kwargs)
self.db: torndb.Connection = self.application.db
self.player: Player = None
def data_received(self, chunk):
logger.info('socket data_received')
@property
def uid(self):
return self.player.uid
@property
def room(self):
return self.player.room
def open(self):
self.player = Player(10, 'Jerry', self)
self.player.room = RoomManager.find_room(0, True)
logger.info('SOCKET[%s] OPEN', self.player.uid)
def on_message(self, message):
packet = msgpack.unpackb(message, use_list=False)
# packet = json.loads(message)
logger.info('REQ[%d]: %s', self.uid, packet)
code = packet[0]
if code == Pt.REQ_LOGIN:
response = [Pt.RSP_LOGIN, self.player.uid, self.player.name]
self.write_message(response)
elif code == Pt.REQ_ROOM_LIST:
response = [Pt.RSP_ROOM_LIST]
self.write_message(response)
elif code == Pt.REQ_TABLE_LIST:
response = [Pt.RSP_TABLE_LIST, self.room.tables()]
self.write_message(response)
elif code == Pt.REQ_JOIN_TABLE:
table_id = packet[1]
table = self.find_table(table_id)
if not table:
response = [Pt.RSP_TABLE_LIST, self.room.tables()]
self.write_message(response)
logger.info('PLAYER[%d] JOIN FULL TABLE[%d]', self.uid,
table.uid)
self.player.join_table(table)
logger.info('PLAYER[%s] JOIN TABLE[%d]', self.uid, table.uid)
table.sync_table()
if table.size() == 3:
table.deal_poker()
logger.info('TABLE[%s] GAME BEGIN[%s]', table.uid,
table.players)
del self.room.waiting_tables[table.uid]
self.room.playing_tables[table.uid] = table
elif code == Pt.REQ_CALL_SCORE:
self.handle_call_score(packet)
elif code == Pt.REQ_DEAL_POKER:
if self.player.table.state == 2:
self.player.ready = True
self.player.table.ready()
elif code == Pt.REQ_SHOT_POKER:
self.handle_shot_poker(packet)
elif code == Pt.REQ_CHAT:
self.handle_chat(packet)
else:
logger.info('UNKNOWN PACKET: %s', code)
@shot_turn
def handle_call_score(self, packet):
score = packet[1]
self.player.handle_call_score(score)
@shot_turn
def handle_shot_poker(self, packet):
pokers = packet[1]
self.player.handle_shot_poker(pokers)
def find_table(self, table_id):
if table_id == -1: # fast join
return self.room.first_waiting_table()
return self.room.find_waiting_table(table_id)
def handle_chat(self, packet):
if self.player and self.player.table:
self.player.table.handle_chat(self.player, packet[1])
def write_message(self, message, binary=False):
if self.ws_connection is None:
raise WebSocketClosedError()
# packet = msgpack.packb(message)
logger.info('RSP[%d]: %s', self.uid, message)
packet = json.dumps(message)
return self.ws_connection.write_message(packet, binary=binary)
def on_close(self):
# self.session.get(self.uid).socket = None
logger.info('socket[%s] close', self.player.uid)
if self.player:
self.player.leave_table()
def send_updates(cls, chat):
logger.info('sending message to %d waiters', len(cls.waiters))
for waiter in cls.waiters:
waiter.write_message('tornado:' + chat)
