home · contact · privacy
Add turn widget to client.
[plomrogue2-experiments] / client.py
index 7e9f18e9f50cb2f74c37ac1f6a8dd92473c22522..308a15f1c3ef4df8b619668f11f362baecefa0b1 100755 (executable)
--- a/client.py
+++ b/client.py
@@ -39,6 +39,7 @@ class UrwidSetup:
         self.server_output = []
         input_handler = getattr(self.InputHandler(self.reply_widget,
                                                   self.map_widget,
+                                                  self.turn_widget,
                                                   self.server_output),
                                 'handle_input')
         self.urwid_pipe_write_fd = self.main_loop.watch_pipe(input_handler)
@@ -54,10 +55,12 @@ class UrwidSetup:
         - self.reply_widget, a urwid.Text widget printing self.socket replies
         """
         edit_widget = self.EditToSocketWidget(self.socket, 'SEND: ')
-        self.reply_widget = urwid.Text('')
         self.map_widget = self.MapWidget('', wrap='clip')
+        self.turn_widget = self.TurnWidget('')
+        self.reply_widget = self.LogWidget('')
         map_box = urwid.Padding(self.map_widget, width=50)
-        widget_pile = urwid.Pile([edit_widget, map_box, self.reply_widget])
+        widget_pile = urwid.Pile([edit_widget, map_box, self.turn_widget,
+                                  self.reply_widget])
         return urwid.Filler(widget_pile, valign='top')
 
     class EditToSocketWidget(urwid.Edit):
@@ -74,19 +77,42 @@ class UrwidSetup:
             plom_socket_io.send(self.socket, self.edit_text)
             self.edit_text = ''
 
+    class TurnWidget(urwid.Text):
+        """Displays turn number."""
+
+        def set_turn(self, turn_string):
+            turn_string = turn_string.strip()
+            if not turn_string.isdigit():
+                raise ArgumentError('Argument must be non-negative integer.')
+            self.set_text('TURN: ' + turn_string)
+
+    class LogWidget(urwid.Text):
+        """Displays client log, newest message on top."""
+
+        def add(self, text):
+            """Add text plus newline to (top of) log."""
+            self.set_text(text + '\n' + self.text)
+
     class MapWidget(urwid.Text):
         """Stores/updates/draws game map."""
         map_size = (5, 5)
         terrain_map = ' ' * 25
         position = (0, 0)
+        things = []
+
+        class Thing:
+            def __init__(self, position, symbol):
+                self.position = position
+                self.symbol = symbol
 
         def draw_map(self):
             """Draw map view from .map_size, .terrain_map, .position."""
             whole_map = []
             for c in self.terrain_map:
                 whole_map += [c]
-            pos_i = self.position[0] * (self.map_size[1] + 1) + self.position[1]
-            whole_map[pos_i] = '@'
+            for t in self.things:
+                pos_i = t.position[0] * (self.map_size[1] + 1) + t.position[1]
+                whole_map[pos_i] = t.symbol
             self.set_text(''.join(whole_map))
 
         def get_yx(self, yx_string):
@@ -126,22 +152,30 @@ class UrwidSetup:
             self.terrain_map = terrain_map
             self.draw_map()
 
-        def update_position(self, position_string):
-            """Update self.position, ensure it's within map bounds."""
-
-            def get_axis_position_from_argument(axis, token):
-                if len(token) < 3 or token[:2] != axis + ':' or \
-                        not token[2:].isdigit():
-                    raise ArgumentError('Bad arg for ' + axis + ' position.')
-                return int(token[2:])
-
-            new_position = self.get_yx(position_string)
-            if new_position[0] >= self.map_size[0] or \
-                    new_position[1] >= self.map_size[1]:
+        def update_things(self, thing_description):
+            """Append thing of thing_description to self.things."""
+            thing_types = {'human': '@', 'monster': 'M'}
+            tokens = thing_description.split()
+            if len(tokens) != 2:
+                raise ArgumentError('Wrong number of tokens.')
+            yx = self.get_yx(tokens[1])
+            if yx[0] >= self.map_size[0] or yx[1] >= self.map_size[1]:
                 raise ArgumentError('Position outside of map size bounds.')
-            self.position = new_position
+            type_token = tokens[0]
+            prefix = 'TYPE:'
+            type_ = '?'
+            if len(type_token) <= len(prefix) or \
+                    type_token[:len(prefix)] != prefix:
+                raise ArgumentError('Invalid type token.')
+            type_ = type_token[len(prefix):]
+            if type_ not in thing_types:
+                raise ArgumentError('Unknown thing type.')
+            self.things += [self.Thing(yx, thing_types[type_])]
             self.draw_map()
 
+        def clear_things(self, _):
+            self.things = []
+
     class InputHandler:
         """Delivers data from other thread to widget via message_container.
 
@@ -152,9 +186,11 @@ class UrwidSetup:
         urwid-external thread.
         """
 
-        def __init__(self, widget1, widget2, message_container):
-            self.widget1 = widget1
-            self.widget2 = widget2
+        def __init__(self, log_widget, map_widget, turn_widget,
+                     message_container):
+            self.log_widget = log_widget
+            self.map_widget = map_widget
+            self.turn_widget = turn_widget
             self.message_container = message_container
 
         def handle_input(self, trigger):
@@ -174,7 +210,15 @@ class UrwidSetup:
             def mapdraw_command(prefix, func):
                 n = len(prefix)
                 if len(msg) > n and msg[:n] == prefix:
-                    m = getattr(self.widget2, func)
+                    m = getattr(self.map_widget, func)
+                    m(msg[n:])
+                    return True
+                return False
+
+            def turndraw_command(prefix, func):
+                n = len(prefix)
+                if len(msg) > n and msg[:n] == prefix:
+                    m = getattr(self.turn_widget, func)
                     m(msg[n:])
                     return True
                 return False
@@ -185,16 +229,16 @@ class UrwidSetup:
                 return
             found_command = False
             try:
-                found_command = (
+                found_command = turndraw_command('NEW_TURN ', 'set_turn') or (
+                    mapdraw_command('NEW_TURN ', 'clear_things') or
                     mapdraw_command('TERRAIN\n', 'update_terrain') or
-                    mapdraw_command('POSITION ', 'update_position') or
+                    mapdraw_command('THING ', 'update_things') or
                     mapdraw_command('MAP_SIZE ', 'update_map_size'))
             except ArgumentError as e:
-                self.widget1.set_text('BAD ARGUMENT: ' + msg + '\n' +
-                                      str(e))
+                self.log_widget.add('ARGUMENT ERROR: ' + msg + '\n' + str(e))
             else:
                 if not found_command:
-                    self.widget1.set_text('UNKNOWN COMMAND: ' + msg)
+                    self.log_widget.add('UNHANDLED INPUT: ' + msg)
             del self.message_container[0]
 
     def recv_loop(self):