[plomrogue2] / plomrogue / tasks.py

from plomrogue.errors import PlayError, GameError



class Task:
    argtypes = ''
    todo = 1

    def __init__(self, thing, args=()):
        self.thing = thing
        self.args = args

    def check(self):
        pass



class Task_WAIT(Task):

    def do(self):
        pass



class Task_MOVE(Task):
    argtypes = 'string:direction'

    def get_move_target(self):
        return self.thing.game.map_geometry.move_yxyx(self.thing.position,
                                                      self.args[0])

    def check(self):
        test_yxyx = self.get_move_target()
        if test_yxyx in [t.position for t in self.thing.game.things
                         if t.blocking]:
            raise PlayError('blocked by other thing')
        elif self.thing.game.maps[test_yxyx[0]][test_yxyx[1]] != '.':
            raise PlayError('blocked by impassable tile')

    def do(self):
        self.thing.position = self.get_move_target()
        if self.thing.carrying:
            self.thing.carrying.position = self.thing.position



class Task_WRITE(Task):
    argtypes = 'string:char string'

    def check(self):
        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):
        big_yx = self.thing.position[0]
        little_yx = self.thing.position[1]
        self.thing.game.maps[big_yx][little_yx] = self.args[0]



class Task_FLATTEN_SURROUNDINGS(Task):
    argtypes = 'string'

    def check(self):
        pass

    def do(self):
        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]] = '.'



class Task_PICK_UP(Task):
    argtypes = 'int:nonneg'

    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 thing of ID %s exists %s' % self.args[0])
        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.position not in reach:
            raise PlayError('thing of ID %s not in reach' % self.args[0])
        elif not to_pick_up.portable:
            raise PlayError('thing of ID %s not portable' % self.args[0])

    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



class Task_DROP(Task):

    def check(self):
        if not self.thing.carrying:
            raise PlayError('nothing to drop')
        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 == self.thing.position]:
                raise PlayError('cannot drop full bottle into bottle deposit')

    def do(self):
        if self.thing.carrying.type_ == 'Bottle' and not self.thing.carrying.full:
            for t in [t for t in self.thing.game.things
                      if t.type_ == 'BottleDeposit'
                      and t.position == self.thing.position]:
                t.accept()
                self.thing.game.things.remove(self.thing.carrying)
                break
        self.thing.carrying = None



class Task_DOOR(Task):

    def do(self):
        self.thing.carrying = None
        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.blocking:
                t.open()
            else:
                t.close()



class Task_INTOXICATE(Task):

    def check(self):
        if self.thing.carrying is None:
            raise PlayError('carrying nothing to drink from')
        if self.thing.carrying.type_ != 'Bottle':
            raise PlayError('cannot drink from non-bottle')
        if not self.thing.carrying.full:
            raise PlayError('bottle is empty')

    def do(self):
        self.thing.carrying.full = False
        self.thing.carrying.empty()
        for c_id in self.thing.game.sessions:
            if self.thing.game.sessions[c_id]['thing_id'] == self.thing.id_:
                self.thing.game.io.send('RANDOM_COLORS', c_id)
                self.thing.game.io.send('CHAT "You are drunk now."', c_id)
                break
        self.thing.drunk = 10000


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):
        from plomrogue.misc import quote
        reply = self.thing.carrying.interpret(self.args[0])
        for c_id in self.thing.game.sessions:
            if self.thing.game.sessions[c_id]['thing_id'] == self.thing.id_:
                self.thing.game.io.send('REPLY ' + quote(reply), c_id)