home · contact · privacy
Add psychedelics, move random colouring from drunkenness to tripping.
[plomrogue2] / plomrogue / tasks.py
index e73b174a1fed351c03cf8c22bd791900e4f76cb9..d42aa4deb232de0a17dcaebaa838dc20e10a01c9 100644 (file)
 from plomrogue.errors import PlayError, GameError
-from plomrogue.mapping import YX
+from plomrogue.misc import quote
 
 
 
 class Task:
     argtypes = ''
-    todo = 3
+    todo = 1
 
     def __init__(self, thing, args=()):
         self.thing = thing
         self.args = args
 
+    def _get_move_target(self):
+        if self.args[0] == 'HERE':
+            return self.thing.position
+        return self.thing.game.map_geometry.move_yxyx(self.thing.position,
+                                                      self.args[0])
+
     def check(self):
         pass
 
 
 
 class Task_WAIT(Task):
-    todo = 1
 
     def do(self):
-        return 'success'
+        pass
 
 
 
 class Task_MOVE(Task):
-    todo = 1
     argtypes = 'string:direction'
 
-    def get_move_target(self):
-        return self.thing.game.map_geometry.move(self.thing.position,
-                                                 self.args[0])
-
     def check(self):
-        test_pos = self.get_move_target()
-        if test_pos is None:
-            raise PlayError('would move out of map')
-        elif test_pos in [t.position for t in self.thing.game.things
-                          if t.blocking]:
+        test_yxyx = self._get_move_target()
+        move_blockers = self.thing.game.get_movement_blockers()
+        if test_yxyx in [t.position for t in self.thing.game.things
+                         if t.blocks_movement]:
             raise PlayError('blocked by other thing')
-        elif self.thing.game.map[test_pos] != '.':
-            raise PlayError('would move into illegal territory')
+        elif self.thing.game.maps[test_yxyx[0]][test_yxyx[1]] in move_blockers:
+            raise PlayError('blocked by impassable tile')
 
     def do(self):
-        self.thing.position = self.get_move_target()
+        self.thing.game.record_change(self.thing.position, 'other')
+        if self.thing.blocks_light:
+            self.thing.game.record_change(self.thing.position, 'fov')
+        self.thing.position = self._get_move_target()
+        self.thing.game.record_change(self.thing.position, 'other')
+        self.thing.invalidate('fov')
+        if self.thing.blocks_light:
+            self.thing.game.record_change(self.thing.position, 'fov')
+        if self.thing.carrying:
+            self.thing.carrying.position = self.thing.position
 
 
 
 class Task_WRITE(Task):
-    todo = 1
     argtypes = 'string:char string'
 
     def check(self):
-        if not self.thing.game.can_do_tile_with_pw(self.thing.position,
+        if not self.thing.game.can_do_tile_with_pw(*self.thing.position,
                                                    self.args[1]):
             raise GameError('wrong password for tile')
 
     def do(self):
-        self.thing.game.map[self.thing.position] = self.args[0]
+        big_yx = self.thing.position[0]
+        little_yx = self.thing.position[1]
+        self.thing.game.maps[big_yx][little_yx] = self.args[0]
+        self.thing.game.record_change((big_yx, little_yx), 'fov')
 
 
 
 class Task_FLATTEN_SURROUNDINGS(Task):
-    todo = 10
     argtypes = 'string'
 
     def check(self):
         pass
 
     def do(self):
-        for yx in[self.thing.position] + \
-            list(self.thing.game.map_geometry.get_neighbors(self.thing.position).values()):
-            if yx is not None:
-                if not self.thing.game.can_do_tile_with_pw(yx, self.args[0]):
-                    continue
-                self.thing.game.map[yx] = '.'
+        for yxyx in [self.thing.position] + \
+                list(self.thing.game.map_geometry.get_neighbors_yxyx(
+                    self.thing.position).values()):
+            if not self.thing.game.can_do_tile_with_pw(*yxyx, self.args[0]):
+                continue
+            self.thing.game.maps[yxyx[0]][yxyx[1]] = self.thing.game.get_flatland()
+            self.thing.game.record_change(yxyx, 'fov')
+
+
+
+class Task_PICK_UP(Task):
+    argtypes = 'int:pos'
+
+    def check(self):
+        if self.thing.carrying:
+            raise PlayError('already carrying something')
+        to_pick_up = self.thing.game.get_thing(self.args[0])
+        neighbors = self.thing.game.map_geometry.\
+            get_neighbors_yxyx(self.thing.position).values()
+        reach = [self.thing.position] + list(neighbors)
+        if to_pick_up is None:
+            raise PlayError('no such thing exists')
+        elif to_pick_up == self.thing:
+            raise PlayError('cannot pick up oneself')
+        elif to_pick_up.type_ == 'Player':
+            raise PlayError('cannot pick up player')
+        elif to_pick_up.carried:
+            raise PlayError('thing already carried by a player')
+        elif to_pick_up.position not in reach:
+            raise PlayError('thing not in reach')
+        elif not to_pick_up.portable:
+            raise PlayError('thing not portable')
+
+    def do(self):
+        to_pick_up = self.thing.game.get_thing(self.args[0])
+        to_pick_up.position = self.thing.position[:]
+        self.thing.carrying = to_pick_up
+        to_pick_up.carried = True
+        self.thing.game.record_change(self.thing.position, 'other')
+
+
+
+class Task_DROP(Task):
+    argtypes = 'string:direction+here'
+
+    def check(self):
+        if not self.thing.carrying:
+            raise PlayError('nothing to drop')
+        target_position = self._get_move_target()
+        if self.thing.carrying.type_ == 'Bottle' and self.thing.carrying.full:
+            for t in [t for t in self.thing.game.things
+                      if t.type_ == 'BottleDeposit'
+                      and t.position == target_position]:
+                raise PlayError('cannot drop full bottle into bottle deposit')
+
+    def do(self):
+        target_position = self._get_move_target()
+        dropped = self.thing.uncarry()
+        dropped.position = target_position
+        over_cookie_spawner = None
+        for t in [t for t in self.thing.game.things
+                  if t.type_ == 'CookieSpawner'
+                  and t.position == dropped.position]:
+            over_cookie_spawner = t
+            break
+        if over_cookie_spawner:
+            over_cookie_spawner.accept(dropped)
+            self.thing.game.remove_thing(dropped)
+        elif dropped.type_ == 'Bottle' and not dropped.full:
+            for t in [t for t in self.thing.game.things
+                      if t.type_ == 'BottleDeposit'
+                      and t.position == dropped.position]:
+                t.accept()
+                self.thing.game.remove_thing(dropped)
+                break
+        elif dropped.type_ == 'Hat':
+            for t in [t for t in self.thing.game.things
+                      if t.type_ == 'HatRemixer'
+                      and t.position == dropped.position]:
+                t.accept(dropped)
+                break
+        self.thing.game.record_change(self.thing.position, 'other')
+
+
+
+class Task_DOOR(Task):
+
+    def do(self):
+        action_radius = list(self.thing.game.map_geometry.
+                             get_neighbors_yxyx(self.thing.position).values())
+        for t in [t for t in self.thing.game.things if
+                  t.type_ == 'Door' and t.position in action_radius]:
+            if t.blocks_movement:
+                t.open()
+            else:
+                t.close()
+            self.thing.game.record_change(t.position, 'other')
+            self.thing.game.record_change(t.position, 'fov')
+
+
+
+class Task_INTOXICATE(Task):
+
+    def check(self):
+        if self.thing.carrying is None:
+            raise PlayError('carrying nothing to consume')
+        if self.thing.carrying.type_ not in {'Bottle', 'Cookie', 'Psychedelic'}:
+            raise PlayError('cannot consume this kind of thing')
+        if self.thing.carrying.type_ == 'Bottle' and\
+           not self.thing.carrying.full:
+            raise PlayError('bottle is empty')
+
+    def do(self):
+        if self.thing.carrying.type_ == 'Bottle':
+            self.thing.carrying.full = False
+            self.thing.carrying.empty()
+            self.thing.send_msg('CHAT "You are drunk now."')
+            self.thing.drunk = 10000
+            self.thing.invalidate('fov')
+            self.thing.game.record_change(self.thing.position, 'other')
+        elif self.thing.carrying.type_ == 'Psychedelic':
+            self.thing.tripping = 10000
+            self.thing.send_msg('CHAT "You start tripping."')
+            self.thing.send_msg('RANDOM_COLORS')
+            eaten = self.thing.uncarry()
+            self.thing.game.remove_thing(eaten)
+        elif self.thing.carrying.type_ == 'Cookie':
+            self.thing.send_msg('CHAT ' + quote('You eat a cookie and gain the ability to draw the following character: "%s"' % self.thing.carrying.thing_char))
+            self.thing.add_cookie_char(self.thing.carrying.thing_char)
+            eaten = self.thing.uncarry()
+            self.thing.game.remove_thing(eaten)
+
+
+
+class Task_COMMAND(Task):
+    argtypes = 'string'
+
+    def check(self):
+        if self.thing.carrying is None:
+            raise PlayError('nothing to command')
+        if not self.thing.carrying.commandable:
+            raise PlayError('cannot command this item type')
+
+    def do(self):
+        reply_lines = self.thing.carrying.interpret(self.args[0])
+        for line in reply_lines:
+            self.thing.send_msg('REPLY ' + quote(line))
+
+
+
+class Task_INSTALL(Task):
+    argtypes = 'string'
+
+    def _get_uninstallables(self):
+        return [t for t in self.thing.game.things
+                if t != self.thing
+                and hasattr(t, 'installable') and t.installable
+                and (not t.portable)
+                and t.position == self.thing.position]
+
+    def check(self):
+        if not self.thing.game.can_do_tile_with_pw(*self.thing.position,
+                                                   self.args[0]):
+            raise GameError('wrong password for tile')
+        if self.thing.carrying:
+            if not hasattr(self.thing.carrying, 'installable')\
+               or not self.thing.carrying.installable:
+                raise PlayError('carried thing not installable')
+        elif len(self._get_uninstallables()) == 0:
+            raise PlayError('nothing to uninstall here')
+
+    def do(self):
+        if self.thing.carrying:
+            t = self.thing.uncarry()
+            t.install()
+            self.thing.send_msg('CHAT "You install the thing you carry."')
+        else:
+            self._get_uninstallables()[0].uninstall()
+            self.thing.send_msg('CHAT "You uninstall the thing here."')
+        self.thing.game.record_change(self.thing.position, 'other')
+
+
+
+class Task_WEAR(Task):
+
+    def check(self):
+        if self.thing.name in self.thing.game.hats:
+            return
+        if not self.thing.carrying:
+            raise PlayError('carrying nothing to wear')
+        if self.thing.name in self.thing.game.hats:
+            raise PlayError('already wearing a hat')
+        if self.thing.carrying.type_ not in {'Hat', 'Bottle'}:
+            raise PlayError('can not wear the kind of thing you hold')
+
+    def do(self):
+        if self.thing.name in self.thing.game.hats:
+            t = self.thing.game.add_thing('Hat', self.thing.position)
+            t.design = self.thing.game.hats[self.thing.name]
+            del self.thing.game.hats[self.thing.name]
+            self.thing.send_msg('CHAT "You drop your hat."')
+            for remixer in [t for t in self.thing.game.things
+                            if t.type_ == 'HatRemixer'
+                            and t.position == self.thing.position]:
+                remixer.accept(t)
+                break
+        else:
+            if self.thing.carrying.type_ == 'Bottle':
+                self.thing.send_msg('CHAT "Your attempt to wear a bottle on '
+                                    'your head fails."')
+                self.thing.carrying.sound('BOTTLE', 'SMASH')
+            elif self.thing.carrying.type_ == 'Hat':
+                self.thing.game.hats[self.thing.name] =\
+                    self.thing.carrying.design
+                self.thing.send_msg('CHAT "You put on a hat."')
+            dropped = self.thing.uncarry()
+            self.thing.game.remove_thing(dropped)
+        self.thing.game.record_change(self.thing.position, 'other')
+
+
+
+class Task_SPIN(Task):
+
+    def check(self):
+        if not self.thing.carrying:
+            raise PlayError('holding nothing to spin')
+        if not hasattr(self.thing.carrying, 'spinnable'):
+            raise PlayError('held object not spinnable')
+
+    def do(self):
+        self.thing.carrying.spin()
+        self.thing.send_msg('CHAT "You spin this object."')