X-Git-Url: https://plomlompom.com/repos/berlin_corona.txt?a=blobdiff_plain;f=plomrogue%2Fcommands.py;h=9c34a1c98058043027e9cf9766c7ac9eeaedacfe;hb=af6c471be94c1a15cdd0df1d4add917b7a686eaa;hp=9cdab97a20963109c70460b6135c610b8cf94aaf;hpb=7c03e8e8e4ed15f01e29dd331596719058af9f07;p=plomrogue2

diff --git a/plomrogue/commands.py b/plomrogue/commands.py
index 9cdab97..9c34a1c 100644
--- a/plomrogue/commands.py
+++ b/plomrogue/commands.py
@@ -337,25 +337,6 @@ def cmd_PLAYER_FACE(game, face, connection_id):
     game.record_change(t.position, 'other')
 cmd_PLAYER_FACE.argtypes = 'string'
 
-def cmd_PLAYER_HAT(game, hat, connection_id):
-    t = game.get_player(connection_id)
-    if not t:
-        raise GameError('can only edit hat when already logged in')
-    if not t.name in game.hats:
-        raise GameError('not currently wearing an editable hat')
-    if len(hat) != 18:
-        raise GameError('wrong hat string length')
-    legal_chars = t.get_cookie_chars()
-    for c in hat:
-        if c not in legal_chars:
-            raise GameError('used illegal character: "%s" – '
-                            'allowed characters: %s'
-                            % (c, legal_chars))
-    game.hats[t.name] = hat
-    game.changed = True
-    game.record_change(t.position, 'other')
-cmd_PLAYER_HAT.argtypes = 'string'
-
 def cmd_GOD_PLAYER_FACE(game, name, face):
     if len(face) != 18:
         raise GameError('wrong face string length')
@@ -372,16 +353,43 @@ def cmd_GOD_PLAYERS_HAT_CHARS(game, name, hat_chars):
     game.players_hat_chars[name] = hat_chars
 cmd_GOD_PLAYERS_HAT_CHARS.argtypes = 'string string'
 
-def cmd_THING_HAT_DESIGN(game, thing_id, design):
-    if len(design) != 18:
-        raise GameError('hat design of wrong length')
+def cmd_THING_DESIGN(game, design, pw, connection_id):
+    player = game.get_player(connection_id)
+    if not player:
+        raise GameError('need to be logged in for this')
+    if not player.carrying:
+        raise GameError('need to carry a thing to re-draw it')
+    if not game.can_do_thing_with_pw(player.carrying, pw):
+        raise GameError('wrong password for thing')
+    if not hasattr(player.carrying, 'design'):
+        raise GameError('carried thing not designable')
+    if player.carrying.type_ == 'Hat':
+        legal_chars = player.get_cookie_chars()
+        for c in design:
+            if c not in legal_chars:
+                raise GameError('used illegal character: "%s" – '
+                                'allowed characters: %s'
+                                % (c, legal_chars))
+    size = player.carrying.design_size
+    if len(design) != size.y * size.x:
+        raise GameError('design for carried thing of wrong length')
+    player.carrying.design = design
+    game.changed = True
+    game.record_change(player.carrying.position, 'other')
+cmd_THING_DESIGN.argtypes = 'string string'
+
+def cmd_GOD_THING_DESIGN(game, thing_id, design):
     t = game.get_thing(thing_id)
     if not t:
         raise GameError('thing of ID %s not found' % thing_id)
-    if t.type_ != 'Hat':
-        raise GameError('thing of ID %s not a hat' % thing_id)
+    if not hasattr(t, 'design'):
+        raise GameError('thing of ID %s not designable' % thing_id)
+    if len(design) != t.design_size.y * t.design_size.x:
+        raise GameError('design for thing of ID %s of wrong length' % thing_id)
     t.design = design
-cmd_THING_HAT_DESIGN.argtypes = 'int:pos string'
+cmd_GOD_THING_DESIGN.argtypes = 'int:pos string'
+
+# TODO: refactor similar god and player commands
 
 def cmd_THING_DOOR_KEY(game, key_id, door_id):
     key = game.get_thing(key_id)