home · contact · privacy
Fix game state change overwriting input fields.
[plomrogue2] / rogue_chat_curses.py
index 2362d99f8dd29a5030f44d62777c78637e8d4a27..2adbe3bb8c3e60cd7b08da5bca2ebbe948f78c8f 100755 (executable)
@@ -9,35 +9,39 @@ 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 plomrogue.errors import BrokenSocketConnection, ArgError
 
 mode_helps = {
     'play': {
         'short': 'play',
-        'long': 'This mode allows you to interact with the map.'
+        'long': 'This mode allows you to interact with the map in various ways.'
     },
     'study': {
         'short': 'study',
-        'long': 'This mode allows you to study the map and its tiles in detail.  Move the question mark over a tile, and the right half of the screen will show detailed information on it.'},
+        'long': 'This mode allows you to study the map and its tiles in detail.  Move the question mark over a tile, and the right half of the screen will show detailed information on it.  Toggle the map view to show or hide different information layers.'},
     'edit': {
-        'short': 'terrain edit',
+        'short': 'map edit',
+        'long': 'This mode allows you to change the map in various ways.  Individual map tiles can be protected by "protection characters", which you can see by toggling into the protections map view.  You can edit a tile if you set the map edit password that matches its protection character.  The character "." marks the absence of protection:  Such tiles can always be edited.'
+    },
+    'write': {
+        'short': 'change terrain',
         'long': 'This mode allows you to change the map tile you currently stand on (if your map editing password authorizes you so).  Just enter any printable ASCII character to imprint it on the ground below you.'
     },
     'control_pw_type': {
-        'short': 'change tiles control password',
-        'long': 'This mode is the first of two steps to change the password for a tile control character.  First enter the tile control character for which you want to change the password!'
+        'short': 'change protection character password',
+        'long': 'This mode is the first of two steps to change the password for a tile protection character.  First enter the tile protection character for which you want to change the password.'
     },
     'control_pw_pw': {
-        'short': 'change tiles control password',
-        'long': 'This mode is the second of two steps to change the password for a tile control character.  Enter the new password for the tile control character you chose.'
+        'short': 'change protection character password',
+        'long': 'This mode is the second of two steps to change the password for a tile protection character.  Enter the new password for the tile protection character you chose.'
     },
     'control_tile_type': {
-        'short': 'change tiles control',
-        'long': 'This mode is the first of two steps to change tile control areas on the map.  First enter the tile control character you want to write.'
+        'short': 'change tiles protection',
+        'long': 'This mode is the first of two steps to change tile protection areas on the map.  First enter the tile tile protection character you want to write.'
     },
     'control_tile_draw': {
-        'short': 'change tiles control',
-        'long': 'This mode is the second of two steps to change tile control areas on the map.  Move cursor around the map to draw selected tile control character'
+        'short': 'change tiles protection',
+        'long': 'This mode is the second of two steps to change tile protection areas on the map.  Toggle tile protection drawing on/off and move the ?? cursor around the map to draw the selected tile protection character.'
     },
     'annotate': {
         'short': 'annotate tile',
@@ -53,7 +57,7 @@ mode_helps = {
     },
     'login': {
         'short': 'login',
-        'long': 'Pick your player name.'
+        'long': 'Enter your player name.'
     },
     'waiting_for_server': {
         'short': 'waiting for server response',
@@ -64,12 +68,16 @@ mode_helps = {
         'long': 'Waiting for a server response.'
     },
     'password': {
-        'short': 'map edit password',
+        'short': 'set map edit password',
         'long': 'This mode allows you to change the password that you send to authorize yourself for editing password-protected map tiles.  Hit return to confirm and leave.'
     },
-    'admin': {
+    'admin_enter': {
         'short': 'become admin',
         'long': 'This mode allows you to become admin if you know an admin password.'
+    },
+    'admin': {
+        'short': 'admin',
+        'long': 'This mode allows you access to actions limited to administrators.'
     }
 }
 
@@ -128,6 +136,13 @@ def cmd_LOGIN_OK(game):
     game.tui.log_msg('@ welcome')
 cmd_LOGIN_OK.argtypes = ''
 
+def cmd_ADMIN_OK(game):
+    game.tui.is_admin = True
+    game.tui.log_msg('@ you now have admin rights')
+    game.tui.switch_mode('admin')
+    game.tui.do_refresh = True
+cmd_ADMIN_OK.argtypes = ''
+
 def cmd_CHAT(game, msg):
     game.tui.log_msg('# ' + msg)
     game.tui.do_refresh = True
@@ -223,14 +238,13 @@ cmd_ANNOTATION_HINT.argtypes = 'yx_tuple:nonneg'
 
 def cmd_ANNOTATION(game, position, msg):
     game.info_db[position] = msg
-    game.tui.restore_input_values()
     if game.tui.mode.shows_info:
         game.tui.do_refresh = True
 cmd_ANNOTATION.argtypes = 'yx_tuple:nonneg string'
 
 def cmd_TASKS(game, tasks_comma_separated):
     game.tasks = tasks_comma_separated.split(',')
-    game.tui.mode_edit.legal = 'WRITE' in game.tasks
+    game.tui.mode_write.legal = 'WRITE' in game.tasks
 cmd_TASKS.argtypes = 'string'
 
 def cmd_THING_TYPE(game, thing_type, symbol_hint):
@@ -253,6 +267,7 @@ class Game(GameBase):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.register_command(cmd_LOGIN_OK)
+        self.register_command(cmd_ADMIN_OK)
         self.register_command(cmd_PONG)
         self.register_command(cmd_CHAT)
         self.register_command(cmd_PLAYER_ID)
@@ -331,13 +346,15 @@ class Mode:
         return False
 
 class TUI:
-    mode_admin = Mode('admin', has_input_prompt=True)
+    mode_admin_enter = Mode('admin_enter', has_input_prompt=True)
+    mode_admin = Mode('admin')
     mode_play = Mode('play')
     mode_study = Mode('study', shows_info=True)
-    mode_edit = Mode('edit', is_single_char_entry=True)
-    mode_control_pw_type = Mode('control_pw_type', is_single_char_entry=True)
+    mode_write = Mode('write', is_single_char_entry=True)
+    mode_edit = Mode('edit')
+    mode_control_pw_type = Mode('control_pw_type', has_input_prompt=True)
     mode_control_pw_pw = Mode('control_pw_pw', has_input_prompt=True)
-    mode_control_tile_type = Mode('control_tile_type', is_single_char_entry=True)
+    mode_control_tile_type = Mode('control_tile_type', has_input_prompt=True)
     mode_control_tile_draw = Mode('control_tile_draw')
     mode_annotate = Mode('annotate', has_input_prompt=True, shows_info=True)
     mode_portal = Mode('portal', has_input_prompt=True, shows_info=True)
@@ -346,17 +363,22 @@ class TUI:
     mode_login = Mode('login', has_input_prompt=True, is_intro=True)
     mode_post_login_wait = Mode('post_login_wait', is_intro=True)
     mode_password = Mode('password', has_input_prompt=True)
+    is_admin = False
+    tile_draw = False
 
     def __init__(self, host):
         import os
         import json
-        self.mode_play.available_modes = ["chat", "study", "edit",
-                                          "annotate", "portal",
-                                          "password", "admin",
-                                          "control_pw_type",
-                                          "control_tile_type"]
-        self.mode_study.available_modes = ["chat", "play"]
-        self.mode_control_tile_draw.available_modes = ["play"]
+        self.mode_play.available_modes = ["chat", "study", "edit", "admin_enter"]
+        self.mode_study.available_modes = ["chat", "play", "admin_enter", "edit"]
+        self.mode_admin.available_modes = ["control_pw_type",
+                                           "control_tile_type", "chat",
+                                           "study", "play", "edit"]
+        self.mode_control_tile_draw.available_modes = ["admin_enter"]
+        self.mode_edit.available_modes = ["write", "annotate", "portal",
+                                          "password", "chat", "study", "play",
+                                          "admin_enter"]
+        self.mode = None
         self.host = host
         self.game = Game()
         self.game.tui = self
@@ -365,7 +387,7 @@ class TUI:
         self.do_refresh = True
         self.queue = queue.Queue()
         self.login_name = None
-        self.map_mode = 'terrain'
+        self.map_mode = 'terrain + things'
         self.password = 'foo'
         self.switch_mode('waiting_for_server')
         self.keys = {
@@ -375,8 +397,9 @@ class TUI:
             'switch_to_annotate': 'M',
             'switch_to_portal': 'T',
             'switch_to_study': '?',
-            'switch_to_edit': 'm',
-            'switch_to_admin': 'A',
+            'switch_to_edit': 'E',
+            'switch_to_write': 'm',
+            'switch_to_admin_enter': 'A',
             'switch_to_control_pw_type': 'C',
             'switch_to_control_tile_type': 'Q',
             'flatten': 'F',
@@ -384,7 +407,8 @@ class TUI:
             'drop_thing': 'u',
             'teleport': 'p',
             'help': 'h',
-            'toggle_map_mode': 'M',
+            'toggle_map_mode': 'L',
+            'toggle_tile_draw': 'm',
             'hex_move_upleft': 'w',
             'hex_move_upright': 'e',
             'hex_move_right': 'd',
@@ -428,6 +452,7 @@ class TUI:
             self.disconnected = False
             self.game.thing_types = {}
             self.game.terrains = {}
+            time.sleep(0.1)  # give potential SSL negotation some time …
             self.socket.send('TASKS')
             self.socket.send('TERRAINS')
             self.socket.send('THING_TYPES')
@@ -441,8 +466,8 @@ class TUI:
     def reconnect(self):
         self.log_msg('@ attempting reconnect')
         self.send('QUIT')
-        time.sleep(0.1)  # FIXME necessitated by some some strange SSL race
-                         # conditions with ws4py, find out what exactly
+        # necessitated by some strange SSL race conditions with ws4py
+        time.sleep(0.1)  # FIXME find out why exactly necessary
         self.switch_mode('waiting_for_server')
         self.connect()
 
@@ -479,17 +504,33 @@ class TUI:
         self.send('SET_TILE_CONTROL %s %s' %
                   (self.explorer, quote(self.tile_control_char)))
 
+    def toggle_map_mode(self):
+        if self.map_mode == 'terrain only':
+            self.map_mode = 'terrain + annotations'
+        elif self.map_mode == 'terrain + annotations':
+            self.map_mode = 'terrain + things'
+        elif self.map_mode == 'terrain + things':
+            self.map_mode = 'protections'
+        elif self.map_mode == 'protections':
+            self.map_mode = 'terrain only'
+
     def switch_mode(self, mode_name):
-        self.map_mode = 'terrain'
+        self.tile_draw = False
+        if mode_name == 'admin_enter' and self.is_admin:
+            mode_name = 'admin'
         self.mode = getattr(self, 'mode_' + mode_name)
+        if self.mode.name == 'control_tile_draw':
+            self.log_msg('@ finished tile protection drawing.')
+        if self.mode.name in {'control_tile_draw', 'control_tile_type',
+                              'control_pw_type'}:
+            self.map_mode = 'protections'
+        elif self.mode.name != 'edit':
+            self.map_mode = 'terrain + things'
         if self.mode.shows_info or self.mode.name == 'control_tile_draw':
             player = self.game.get_thing(self.game.player_id)
             self.explorer = YX(player.position.y, player.position.x)
             if self.mode.shows_info:
                 self.query_info()
-            elif self.mode.name == 'control_tile_draw':
-                self.send_tile_control_command()
-                self.map_mode = 'control'
         if self.mode.is_single_char_entry:
             self.show_help = True
         if self.mode.name == 'waiting_for_server':
@@ -499,10 +540,17 @@ class TUI:
                 self.send('LOGIN ' + quote(self.login_name))
             else:
                 self.log_msg('@ enter username')
-        elif self.mode.name == 'admin':
+        elif self.mode.name == 'admin_enter':
             self.log_msg('@ enter admin password:')
+        elif self.mode.name == 'control_pw_type':
+            self.log_msg('@ enter tile protection character for which you want to change the password:')
+        elif self.mode.name == 'control_tile_type':
+            self.log_msg('@ enter tile protection character which you want to draw:')
         elif self.mode.name == 'control_pw_pw':
-            self.log_msg('@ enter tile control password for "%s":' % self.tile_control_char)
+            self.log_msg('@ enter tile protection password for "%s":' % self.tile_control_char)
+        elif self.mode.name == 'control_tile_draw':
+            self.log_msg('@ can draw tile protection character "%s", turn drawing on/off with [%s], finish with [%s].' % (self.tile_control_char, self.keys['toggle_tile_draw'], self.keys['switch_to_admin_enter']))
+        self.input_ = ""
         self.restore_input_values()
 
     def loop(self, stdscr):
@@ -559,7 +607,7 @@ class TUI:
                 self.explorer = target
                 if self.mode.shows_info:
                     self.query_info()
-                elif self.mode.name == 'control_tile_draw':
+                if self.tile_draw:
                     self.send_tile_control_command()
             else:
                 self.flash = True
@@ -580,17 +628,19 @@ class TUI:
             if not self.game.turn_complete:
                 return
             pos_i = self.explorer.y * self.game.map_geometry.size.x + self.explorer.x
-            info = 'outside field of view'
-            if self.game.fov[pos_i] == '.':
+            info = 'MAP VIEW: %s\n' % self.map_mode
+            if self.game.fov[pos_i] != '.':
+                info += 'outside field of view'
+            else:
                 terrain_char = self.game.map_content[pos_i]
                 terrain_desc = '?'
                 if terrain_char in self.game.terrains:
                     terrain_desc = self.game.terrains[terrain_char]
-                info = 'TERRAIN: "%s" / %s\n' % (terrain_char, terrain_desc)
+                info += 'TERRAIN: "%s" / %s\n' % (terrain_char, terrain_desc)
                 protection = self.game.map_control_content[pos_i]
                 if protection == '.':
                     protection = 'unprotected'
-                info = 'PROTECTION: %s\n' % protection
+                info += 'PROTECTION: %s\n' % protection
                 for t in self.game.things:
                     if t.position == self.explorer:
                         info += 'THING: %s / %s' % (t.type_,
@@ -638,19 +688,22 @@ class TUI:
             if not self.game.turn_complete:
                 return
             map_lines_split = []
-            map_content = self.game.map_content
-            if self.map_mode == 'control':
-                map_content = self.game.map_control_content
             for y in range(self.game.map_geometry.size.y):
                 start = self.game.map_geometry.size.x * y
                 end = start + self.game.map_geometry.size.x
-                map_lines_split += [[c + ' ' for c in map_content[start:end]]]
-            if self.map_mode == 'annotations':
+                if self.map_mode == 'protections':
+                    map_lines_split += [[c + ' ' for c
+                                         in self.game.map_control_content[start:end]]]
+                else:
+                    map_lines_split += [[c + ' ' for c
+                                         in self.game.map_content[start:end]]]
+            if self.map_mode == 'terrain + annotations':
                 for p in self.game.info_hints:
                     map_lines_split[p.y][p.x] = 'A '
-            elif self.map_mode == 'terrain':
+            elif self.map_mode == 'terrain + things':
                 for p in self.game.portals.keys():
-                    map_lines_split[p.y][p.x] = 'P '
+                    original = map_lines_split[p.y][p.x]
+                    map_lines_split[p.y][p.x] = original[0] + 'P'
                 used_positions = []
                 for t in self.game.things:
                     symbol = self.game.thing_types[t.type_]
@@ -661,22 +714,24 @@ class TUI:
                         meta_char = '+'
                     map_lines_split[t.position.y][t.position.x] = symbol + meta_char
                     used_positions += [t.position]
+            player = self.game.get_thing(self.game.player_id)
             if self.mode.shows_info or self.mode.name == 'control_tile_draw':
                 map_lines_split[self.explorer.y][self.explorer.x] = '??'
+            elif self.map_mode != 'terrain + things':
+                map_lines_split[player.position.y][player.position.x] = '??'
             map_lines = []
             if type(self.game.map_geometry) == MapGeometryHex:
                 indent = 0
                 for line in map_lines_split:
-                    map_lines += [indent*' ' + ''.join(line)]
+                    map_lines += [indent * ' ' + ''.join(line)]
                     indent = 0 if indent else 1
             else:
                 for line in map_lines_split:
                     map_lines += [''.join(line)]
             window_center = YX(int(self.size.y / 2),
                                int(self.window_width / 2))
-            player = self.game.get_thing(self.game.player_id)
             center = player.position
-            if self.mode.shows_info:
+            if self.mode.shows_info or self.mode.name == 'control_tile_draw':
                 center = self.explorer
             center = YX(center.y, center.x * 2)
             offset = center - window_center
@@ -700,27 +755,39 @@ class TUI:
                 if 'MOVE' in self.game.tasks:
                     content += "[%s] – move player\n" % ','.join(self.movement_keys)
                 if 'PICK_UP' in self.game.tasks:
-                    content += "[%s] – take thing under player\n" % self.keys['take_thing']
+                    content += "[%s] – pick up thing\n" % self.keys['take_thing']
                 if 'DROP' in self.game.tasks:
-                    content += "[%s] – drop carried thing\n" % self.keys['drop_thing']
-                if 'FLATTEN_SURROUNDINGS' in self.game.tasks:
-                    content += "[%s] – flatten player's surroundings\n" % self.keys['flatten']
-                content += '[%s] – teleport to other space\n' % self.keys['teleport']
+                    content += "[%s] – drop thing\n" % self.keys['drop_thing']
+                content += '[%s] – teleport\n' % self.keys['teleport']
                 content += '\n'
             elif self.mode.name == 'study':
                 content += 'Available actions:\n'
                 content += '[%s] – move question mark\n' % ','.join(self.movement_keys)
-                content += '[%s] – toggle view between terrain, annotations, and password protection areas\n' % self.keys['toggle_map_mode']
+                content += '[%s] – toggle map view\n' % self.keys['toggle_map_mode']
+                content += '\n'
+            elif self.mode.name == 'edit':
+                content += "Available actions:\n"
+                if 'MOVE' in self.game.tasks:
+                    content += "[%s] – move player\n" % ','.join(self.movement_keys)
+                if 'FLATTEN_SURROUNDINGS' in self.game.tasks:
+                    content += "[%s] – flatten surroundings\n" % self.keys['flatten']
+                content += '[%s] – toggle map view\n' % self.keys['toggle_map_mode']
+                content += '\n'
+            elif self.mode.name == 'control_tile_draw':
+                content += "Available actions:\n"
+                content += "[%s] – toggle tile protection drawing\n" % self.keys['toggle_tile_draw']
                 content += '\n'
             elif self.mode.name == 'chat':
                 content += '/nick NAME – re-name yourself to NAME\n'
                 content += '/%s or /play – switch to play mode\n' % self.keys['switch_to_play']
                 content += '/%s or /study – switch to study mode\n' % self.keys['switch_to_study']
+                content += '/%s or /edit – switch to map edit mode\n' % self.keys['switch_to_edit']
+                content += '/%s or /admin – switch to admin mode\n' % self.keys['switch_to_admin_enter']
             content += self.mode.list_available_modes(self)
             for i in range(self.size.y):
                 safe_addstr(i,
                             self.window_width * (not self.mode.has_input_prompt),
-                            ' '*self.window_width)
+                            ' ' * self.window_width)
             lines = []
             for line in content.split('\n'):
                 lines += msg_into_lines_of_width(line, self.window_width)
@@ -748,7 +815,7 @@ class TUI:
                 draw_help()
 
         curses.curs_set(False)  # hide cursor
-        curses.use_default_colors();
+        curses.use_default_colors()
         stdscr.timeout(10)
         reset_screen_size()
         self.explorer = YX(0, 0)
@@ -809,18 +876,30 @@ class TUI:
                     self.log_msg('@ aborted')
                 else:
                     self.send('SET_MAP_CONTROL_PASSWORD ' + quote(self.tile_control_char) + ' ' + quote(self.input_))
-                    self.input_ = ""
-                self.switch_mode('play')
+                    self.log_msg('@ sent new password for protection character "%s"' % self.tile_control_char)
+                self.switch_mode('admin')
             elif self.mode.name == 'password' and key == '\n':
                 if self.input_ == '':
                     self.input_ = ' '
                 self.password = self.input_
-                self.input_ = ""
-                self.switch_mode('play')
-            elif self.mode.name == 'admin' and key == '\n':
+                self.switch_mode('edit')
+            elif self.mode.name == 'admin_enter' and key == '\n':
                 self.send('BECOME_ADMIN ' + quote(self.input_))
-                self.input_ = ""
                 self.switch_mode('play')
+            elif self.mode.name == 'control_pw_type' and key == '\n':
+                if len(self.input_) != 1:
+                    self.log_msg('@ entered non-single-char, therefore aborted')
+                    self.switch_mode('admin')
+                else:
+                    self.tile_control_char = self.input_
+                    self.switch_mode('control_pw_pw')
+            elif self.mode.name == 'control_tile_type' and key == '\n':
+                if len(self.input_) != 1:
+                    self.log_msg('@ entered non-single-char, therefore aborted')
+                    self.switch_mode('admin')
+                else:
+                    self.tile_control_char = self.input_
+                    self.switch_mode('control_tile_draw')
             elif self.mode.name == 'chat' and key == '\n':
                 if self.input_ == '':
                     continue
@@ -829,6 +908,10 @@ class TUI:
                         self.switch_mode('play')
                     elif self.input_ in {'/' + self.keys['switch_to_study'], '/study'}:
                         self.switch_mode('study')
+                    elif self.input_ in {'/' + self.keys['switch_to_edit'], '/edit'}:
+                        self.switch_mode('edit')
+                    elif self.input_ in {'/' + self.keys['switch_to_admin_enter'], '/admin'}:
+                        self.switch_mode('admin_enter')
                     elif self.input_.startswith('/nick'):
                         tokens = self.input_.split(maxsplit=1)
                         if len(tokens) == 2:
@@ -845,33 +928,23 @@ class TUI:
                     self.input_ = ' '
                 self.send('ANNOTATE %s %s %s' % (self.explorer, quote(self.input_),
                                                  quote(self.password)))
-                self.input_ = ""
-                self.switch_mode('play')
+                self.switch_mode('edit')
             elif self.mode.name == 'portal' and key == '\n':
                 if self.input_ == '':
                     self.input_ = ' '
                 self.send('PORTAL %s %s %s' % (self.explorer, quote(self.input_),
                                                quote(self.password)))
-                self.input_ = ""
-                self.switch_mode('play')
+                self.switch_mode('edit')
             elif self.mode.name == 'study':
                 if self.mode.mode_switch_on_key(self, key):
                     continue
                 elif key == self.keys['toggle_map_mode']:
-                    if self.map_mode == 'terrain':
-                        self.map_mode = 'annotations'
-                    elif self.map_mode == 'annotations':
-                        self.map_mode = 'control'
-                    else:
-                        self.map_mode = 'terrain'
+                    self.toggle_map_mode()
                 elif key in self.movement_keys:
                     move_explorer(self.movement_keys[key])
             elif self.mode.name == 'play':
                 if self.mode.mode_switch_on_key(self, key):
                     continue
-                if key == self.keys['flatten'] and\
-                     'FLATTEN_SURROUNDINGS' in self.game.tasks:
-                    self.send('TASK:FLATTEN_SURROUNDINGS ' + quote(self.password))
                 elif key == self.keys['take_thing'] and 'PICK_UP' in self.game.tasks:
                     self.send('TASK:PICK_UP')
                 elif key == self.keys['drop_thing'] and 'DROP' in self.game.tasks:
@@ -886,20 +959,29 @@ class TUI:
                         self.log_msg('? not standing on portal')
                 elif key in self.movement_keys and 'MOVE' in self.game.tasks:
                     self.send('TASK:MOVE ' + self.movement_keys[key])
-            elif self.mode.name == 'edit':
+            elif self.mode.name == 'write':
                 self.send('TASK:WRITE %s %s' % (key, quote(self.password)))
-                self.switch_mode('play')
-            elif self.mode.name == 'control_pw_type':
-                self.tile_control_char = key
-                self.switch_mode('control_pw_pw')
-            elif self.mode.name == 'control_tile_type':
-                self.tile_control_char = key
-                self.switch_mode('control_tile_draw')
+                self.switch_mode('edit')
             elif self.mode.name == 'control_tile_draw':
                 if self.mode.mode_switch_on_key(self, key):
                     continue
                 elif key in self.movement_keys:
                     move_explorer(self.movement_keys[key])
+                elif key == self.keys['toggle_tile_draw']:
+                    self.tile_draw = False if self.tile_draw else True
+            elif self.mode.name == 'admin':
+                if self.mode.mode_switch_on_key(self, key):
+                    continue
+            elif self.mode.name == 'edit':
+                if self.mode.mode_switch_on_key(self, key):
+                    continue
+                elif key == self.keys['flatten'] and\
+                    'FLATTEN_SURROUNDINGS' in self.game.tasks:
+                    self.send('TASK:FLATTEN_SURROUNDINGS ' + quote(self.password))
+                elif key == self.keys['toggle_map_mode']:
+                    self.toggle_map_mode()
+                elif key in self.movement_keys and 'MOVE' in self.game.tasks:
+                    self.send('TASK:MOVE ' + self.movement_keys[key])
 
 if len(sys.argv) != 2:
     raise ArgError('wrong number of arguments, need game host')