#!/usr/bin/env python3
import curses
-import socket
import queue
import threading
-from plomrogue.io_tcp import PlomSocket
from plomrogue.game import GameBase
from plomrogue.parser import Parser
from plomrogue.mapping import YX, MapGeometrySquare, MapGeometryHex
from plomrogue.things import ThingBase
from plomrogue.misc import quote
+from plomrogue.errors import BrokenSocketConnection
+
+from ws4py.client import WebSocketBaseClient
+class WebSocketClient(WebSocketBaseClient):
+
+ def __init__(self, recv_handler, *args, **kwargs):
+ super().__init__(*args, **kwargs)
+ self.recv_handler = recv_handler
+ self.connect()
+
+ def received_message(self, message):
+ if message.is_text:
+ message = str(message)
+ self.recv_handler(message)
+
+ @property
+ def plom_closed(self):
+ return self.client_terminated
+
+from plomrogue.io_tcp import PlomSocket
+class PlomSocketClient(PlomSocket):
+
+ def __init__(self, recv_handler, url):
+ import socket
+ self.recv_handler = recv_handler
+ host, port = url.split(':')
+ super().__init__(socket.create_connection((host, port)))
+
+ def close(self):
+ self.socket.close()
+
+ def run(self):
+ import ssl
+ try:
+ for msg in self.recv():
+ if msg == 'NEED_SSL':
+ self.socket = ssl.wrap_socket(self.socket)
+ continue
+ self.recv_handler(msg)
+ except BrokenSocketConnection:
+ pass # we assume socket will be known as dead by now
def cmd_TURN(game, n):
game.turn = n
game.tui.query_info()
player = game.get_thing(game.player_id, False)
if player.position in game.portals:
- #host, port = game.portals[player.position].split(':')
game.tui.teleport_target_host = game.portals[player.position]
- game.tui.teleport_target_port = 5000
game.tui.switch_mode('teleport')
game.turn_complete = True
game.tui.do_refresh = True
game.tui.do_refresh = True
cmd_ANNOTATION.argtypes = 'yx_tuple:nonneg string'
+def cmd_PONG(game):
+ pass
+cmd_PONG.argtypes = ''
+
class Game(GameBase):
- commands = {'LOGIN_OK': cmd_LOGIN_OK,
- 'CHAT': cmd_CHAT,
- 'PLAYER_ID': cmd_PLAYER_ID,
- 'TURN': cmd_TURN,
- 'THING_POS': cmd_THING_POS,
- 'THING_NAME': cmd_THING_NAME,
- 'MAP': cmd_MAP,
- 'PORTAL': cmd_PORTAL,
- 'ANNOTATION': cmd_ANNOTATION,
- 'GAME_STATE_COMPLETE': cmd_GAME_STATE_COMPLETE,
- 'ARGUMENT_ERROR': cmd_ARGUMENT_ERROR,
- 'GAME_ERROR': cmd_GAME_ERROR,
- 'PLAY_ERROR': cmd_PLAY_ERROR}
thing_type = ThingBase
turn_complete = False
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
+ self.register_command(cmd_LOGIN_OK)
+ self.register_command(cmd_PONG)
+ self.register_command(cmd_CHAT)
+ self.register_command(cmd_PLAYER_ID)
+ self.register_command(cmd_TURN)
+ self.register_command(cmd_THING_POS)
+ self.register_command(cmd_THING_NAME)
+ self.register_command(cmd_MAP)
+ self.register_command(cmd_PORTAL)
+ self.register_command(cmd_ANNOTATION)
+ self.register_command(cmd_GAME_STATE_COMPLETE)
+ self.register_command(cmd_ARGUMENT_ERROR)
+ self.register_command(cmd_GAME_ERROR)
+ self.register_command(cmd_PLAY_ERROR)
self.map_content = ''
self.player_id = -1
self.info_db = {}
self.shows_info = shows_info
self.is_intro = is_intro
- def __init__(self, host, port):
+ def __init__(self, host):
import os
import json
self.host = host
- self.port = port
self.mode_play = self.Mode('play')
self.mode_study = self.Mode('study', shows_info=True)
self.mode_edit = self.Mode('edit')
self.login_name = None
self.switch_mode('waiting_for_server')
self.keys = {
- 'switch_to_chat': 'C',
- 'switch_to_play': 'P',
- 'switch_to_annotate': 'E',
- 'switch_to_portal': 'p',
+ 'switch_to_chat': 't',
+ 'switch_to_play': 'p',
+ 'switch_to_annotate': 'm',
+ 'switch_to_portal': 'P',
'switch_to_study': '?',
- 'switch_to_edit': 'E',
- 'flatten': 'f',
+ 'switch_to_edit': 'm',
+ 'flatten': 'F',
'hex_move_upleft': 'w',
'hex_move_upright': 'e',
'hex_move_right': 'd',
- 'hex_move_downright': 'c',
- 'hex_move_downleft': 'x',
- 'hex_move_left': 's',
+ 'hex_move_downright': 'x',
+ 'hex_move_downleft': 'y',
+ 'hex_move_left': 'a',
'square_move_up': 'w',
'square_move_left': 'a',
'square_move_down': 's',
def send(self, msg):
try:
+ if hasattr(self.socket, 'plom_closed') and self.socket.plom_closed:
+ raise BrokenSocketConnection
self.socket.send(msg)
- except BrokenPipeError:
+ except (BrokenPipeError, BrokenSocketConnection):
self.log_msg('@ server disconnected :(')
self.do_refresh = True
else:
self.log_msg('@ enter username')
elif self.mode.name == 'teleport':
- self.log_msg("@ May teleport to %s:%s" % (self.teleport_target_host,
- self.teleport_target_port));
+ self.log_msg("@ May teleport to %s" % (self.teleport_target_host)),
self.log_msg("@ Enter 'YES!' to enthusiastically affirm.");
elif self.mode.name == 'annotate' and self.explorer in self.game.info_db:
info = self.game.info_db[self.explorer]
self.log_msg(" /nick NAME - re-name yourself to NAME");
self.log_msg(" /msg USER TEXT - send TEXT to USER");
self.log_msg(" /help - show this help");
- self.log_msg(" /P or /play - switch to play mode");
- self.log_msg(" /? or /study - switch to study mode");
+ self.log_msg(" /%s or /play - switch to play mode" % self.keys['switch_to_play']);
+ self.log_msg(" /%s or /study - switch to study mode" % self.keys['switch_to_study']);
self.log_msg("commands common to study and play mode:");
self.log_msg(" %s - move" % ','.join(self.movement_keys));
self.log_msg(" %s - switch to chat mode" % self.keys['switch_to_chat']);
self.log_msg(" %s - switch to play mode" % self.keys['switch_to_play']);
def loop(self, stdscr):
+ import time
+ import datetime
def safe_addstr(y, x, line):
if y < self.size.y - 1 or x + len(line) < self.size.x:
stdscr.addstr(y, x, cut)
def connect():
- import time
- def recv_loop():
- for msg in self.socket.recv():
- if msg == 'BYE':
- break
+ def handle_recv(msg):
+ if msg == 'BYE':
+ self.socket.close()
+ else:
self.queue.put(msg)
+ socket_client_class = PlomSocketClient
+ if self.host.startswith('ws://') or self.host.startswith('wss://'):
+ socket_client_class = WebSocketClient
while True:
try:
- s = socket.create_connection((self.host, self.port))
- self.socket = PlomSocket(s)
- self.socket_thread = threading.Thread(target=recv_loop)
+ self.socket = socket_client_class(handle_recv, self.host)
+ self.socket_thread = threading.Thread(target=self.socket.run)
self.socket_thread.start()
self.switch_mode('login')
return
def reconnect():
self.send('QUIT')
+ time.sleep(0.1) # FIXME necessitated by some some strange SSL race
+ # conditions with ws4py, find out what exactly
self.switch_mode('waiting_for_server')
connect()
def draw_info():
if not self.game.turn_complete:
return
+ pos_i = self.explorer.y * self.game.map_geometry.size.x + self.explorer.x
+ info = 'TERRAIN: %s\n' % self.game.map_content[pos_i]
+ for t in self.game.things:
+ if t.position == self.explorer:
+ info += 'PLAYER @: %s\n' % t.name
if self.explorer in self.game.portals:
- info = 'PORTAL: ' + self.game.portals[self.explorer] + '\n'
+ info += 'PORTAL: ' + self.game.portals[self.explorer] + '\n'
else:
- info = 'PORTAL: (none)\n'
+ info += 'PORTAL: (none)\n'
if self.explorer in self.game.info_db:
info += 'ANNOTATION: ' + self.game.info_db[self.explorer]
else:
draw_map()
curses.curs_set(False) # hide cursor
+ curses.use_default_colors();
stdscr.timeout(10)
reset_screen_size()
self.explorer = YX(0, 0)
self.input_ = ''
input_prompt = '> '
connect()
+ last_ping = datetime.datetime.now()
+ interval = datetime.timedelta(seconds=30)
while True:
+ now = datetime.datetime.now()
+ if now - last_ping > interval:
+ self.send('PING')
+ last_ping = now
if self.do_refresh:
draw_screen()
self.do_refresh = False
self.input_ = ""
elif self.mode == self.mode_chat and key == '\n':
if self.input_[0] == '/':
- if self.input_ in {'/P', '/play'}:
+ if self.input_ in {'/' + self.keys['switch_to_play'], '/play'}:
self.switch_mode('play')
- elif self.input_ in {'/?', '/study'}:
+ elif self.input_ in {'/' + self.keys['switch_to_study'], '/study'}:
self.switch_mode('study')
elif self.input_ == '/help':
self.help()
elif self.mode == self.mode_teleport and key == '\n':
if self.input_ == 'YES!':
self.host = self.teleport_target_host
- self.port = self.teleport_target_port
reconnect()
else:
self.log_msg('@ teleport aborted')
self.send('TASK:WRITE ' + key)
self.switch_mode('play')
-TUI('127.0.0.1', 5000)
+TUI('localhost:5000')
projects / plomrogue2-experiments / blobdiff