Split LOGIN into NICK and LOGIN.

[plomrogue2] / plomrogue / commands.py

diff --git a/plomrogue/commands.py b/plomrogue/commands.py
index a80731da48336af7db5dbd9359a00ad56a4ad05f..c2484bf5d228ea1407e63caab79be1000554d3e0 100644 (file)
--- a/plomrogue/commands.py
+++ b/plomrogue/commands.py
@@ -11,28 +11,35 @@ def cmd_ALL(game, msg, connection_id):
     game.io.send('CHAT ' + quote(t.nickname + ': ' + msg))
 cmd_ALL.argtypes = 'string'
 
-# TOOD split into two commands
 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')
     if connection_id in game.sessions:
-        t_id = game.sessions[connection_id]
-        t = game.get_thing(t_id, False)
-        old_nick = t.nickname
-        t.nickname = nick
-        game.io.send('CHAT ' + quote(old_nick + ' renamed themselves to ' + nick))
-    else:
-        t = game.thing_types['player'](game)
-        t.position = YX(game.map.size.y // 2, game.map.size.x // 2)
-        game.things += [t]  # TODO refactor into Thing.__init__?
-        game.sessions[connection_id] = t.id_
-        game.io.send('LOGIN_OK', connection_id)
-        t.nickname = nick
-        game.io.send('CHAT ' + quote(t.nickname + ' entered the map.'))
-        game.io.send('PLAYER_ID %s' % t.id_, connection_id)
+        raise GameError('cannot log in twice')
+    t = game.thing_types['player'](game)
+    t.position = YX(game.map.size.y // 2, game.map.size.x // 2)
+    game.things += [t]  # TODO refactor into Thing.__init__?
+    game.sessions[connection_id] = t.id_
+    game.io.send('LOGIN_OK', connection_id)
+    t.nickname = nick
+    game.io.send('CHAT ' + quote(t.nickname + ' entered the map.'))
+    game.io.send('PLAYER_ID %s' % t.id_, connection_id)
     game.changed = True
 cmd_LOGIN.argtypes = 'string'
 
+def cmd_NICK(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')
+    if not connection_id in game.sessions:
+        raise GameError('can only rename when already logged in')
+    t_id = game.sessions[connection_id]
+    t = game.get_thing(t_id, False)
+    old_nick = t.nickname
+    t.nickname = nick
+    game.io.send('CHAT ' + quote(old_nick + ' renamed themselves to ' + nick))
+    game.changed = True
+cmd_NICK.argtypes = 'string'
+
 def cmd_GET_GAMESTATE(game, connection_id):
     game.send_gamestate(connection_id)
 cmd_GET_GAMESTATE.argtypes = ''