Introduce god mode to protect against destructive commands.

[plomrogue2-experiments] / new2 / plomrogue / commands.py

diff --git a/new2/plomrogue/commands.py b/new2/plomrogue/commands.py
index df347289f64459ce4718df6aefb6a87a6ec920d2..28b76006fe8088ffcc3d93bc64a455f71db8348b 100644 (file)
--- a/new2/plomrogue/commands.py
+++ b/new2/plomrogue/commands.py
@@ -7,17 +7,23 @@ def cmd_ALL(game, msg, connection_id):
     if not connection_id in game.sessions:
         raise GameError('need to be logged in for this')
     t = game.get_thing(game.sessions[connection_id], False)
-    game.io.send('LOG ' + quote(t.nickname + ': ' + msg))
+    game.io.send('CHAT ' + quote(t.nickname + ': ' + msg))
 cmd_ALL.argtypes = 'string'
 
 def cmd_LOGIN(game, nick, connection_id):
     for t in [t for t in game.things if t.type_ == 'player' and t.nickname == nick]:
         raise GameError('name already in use')
-    t = game.thing_types['player'](game)
+    if connection_id in game.sessions:
+        t_id = game.sessions[connection_id]
+        t = game.get_thing(t_id, False)
+        game.io.send('META ' + quote('you rename yourself to: ' + nick), connection_id)
+    else:
+        t = game.thing_types['player'](game)
+        game.things += [t]  # TODO refactor into Thing.__init__?
+        game.sessions[connection_id] = t.id_
+        game.io.send('META ' + quote('you are now: ' + nick), connection_id)
     t.nickname = nick
-    game.things += [t]  # TODO refactor into Thing.__init__?
-    game.sessions[connection_id] = t.id_ 
-    game.io.send('META ' + quote('you are now: ' + nick), connection_id)
+    game.io.send('PLAYER_ID %s' % t.id_, connection_id)
 cmd_LOGIN.argtypes = 'string'
 
 def cmd_QUERY(game, target_nick, msg, connection_id):
@@ -28,8 +34,8 @@ def cmd_QUERY(game, target_nick, msg, connection_id):
     for t in [t for t in game.things if t.type_ == 'player' and t.nickname == target_nick]:
         for c_id in game.sessions:
             if game.sessions[c_id] == t.id_:
-                game.io.send('LOG ' + quote(source_nick+ '->' + target_nick + ': ' + msg), c_id)
-                game.io.send('LOG ' + quote(source_nick+ '->' + target_nick + ': ' + msg), connection_id)
+                game.io.send('CHAT ' + quote(source_nick+ '->' + target_nick + ': ' + msg), c_id)
+                game.io.send('CHAT ' + quote(source_nick+ '->' + target_nick + ': ' + msg), connection_id)
                 return
         raise GameError('target user offline')
     raise GameError('can only query with registered nicknames')
@@ -39,21 +45,30 @@ def cmd_PING(game, connection_id):
     game.io.send('PONG')
 cmd_PING.argtypes = ''
 
-def cmd_SAVE(game):
-
-    def write(f, msg):
-        f.write(msg + '\n')
-
-    with open(game.io.save_file, 'w') as f:
-        write(f, 'TURN %s' % game.turn)
-        for y, line in game.map.lines():
-            write(f, 'MAP_LINE %5s %s' % (y, quote(line)))
-cmd_SAVE.argtypes = ''
-
 def cmd_TURN(game, n):
     game.turn = n
 cmd_TURN.argtypes = 'int:nonneg'
 
+def cmd_ANNOTATE(game, yx, msg, connection_id):
+    if msg == ' ':
+        if yx in game.annotations:
+            del game.annotations[yx]
+    else:
+        game.annotations[yx] = msg
+    game.changed = True
+cmd_ANNOTATE.argtypes = 'yx_tuple:nonneg string'
+
+def cmd_GET_ANNOTATION(game, yx, connection_id):
+    annotation = '(none)';
+    if yx in game.annotations:
+        annotation = game.annotations[yx]
+    game.io.send('ANNOTATION %s %s' % (yx, quote(annotation)))
+cmd_GET_ANNOTATION.argtypes = 'yx_tuple:nonneg'
+
 def cmd_MAP_LINE(game, y, line):
     game.map.set_line(y, line)
 cmd_MAP_LINE.argtypes = 'int:nonneg string'
+
+def cmd_MAP(game, size):
+    game.new_world(size)
+cmd_MAP.argtypes = 'yx_tuple:pos'