home · contact · privacy
4cf8700b1f5b93473fabb70c3acedf471167667d
[plomrogue2] / plomrogue / tasks.py
1 from plomrogue.errors import PlayError, GameError
2 from plomrogue.misc import quote
3
4
5
6 class Task:
7     argtypes = ''
8     todo = 1
9
10     def __init__(self, thing, args=()):
11         self.thing = thing
12         self.args = args
13
14     def _get_move_target(self):
15         if self.args[0] == 'HERE':
16             return self.thing.position
17         return self.thing.game.map_geometry.move_yxyx(self.thing.position,
18                                                       self.args[0])
19
20     def check(self):
21         pass
22
23
24
25 class Task_WAIT(Task):
26
27     def do(self):
28         pass
29
30
31
32 class Task_MOVE(Task):
33     argtypes = 'string:direction'
34
35     def check(self):
36         test_yxyx = self._get_move_target()
37         move_blockers = self.thing.game.get_movement_blockers()
38         if test_yxyx in [t.position for t in self.thing.game.things
39                          if t.blocks_movement]:
40             raise PlayError('blocked by other thing')
41         elif self.thing.game.maps[test_yxyx[0]][test_yxyx[1]] in move_blockers:
42             raise PlayError('blocked by impassable tile')
43
44     def do(self):
45         if self.thing.type_ == 'Player' and not self.thing.standing:
46             self.thing.standing = True
47             self.thing.send_msg('CHAT "You get up."')
48         self.thing.game.record_change(self.thing.position, 'other')
49         if self.thing.blocks_light:
50             self.thing.game.record_change(self.thing.position, 'fov')
51         self.thing.position = self._get_move_target()
52         for t in [t for t in self.thing.game.things
53                   if t.type_ == 'Player' and not t == self.thing
54                   and t.position == self.thing.position]:
55             self.thing.send_msg('CHAT %s' %
56                                 quote('You get awkwardly close to %s.' % t.name))
57             for c_id in self.thing.game.sessions:
58                 if self.thing.game.sessions[c_id]['thing_id'] == t.id_:
59                     t.send_msg('CHAT %s' %
60                                quote('%s gets awkwardly close to you.' %
61                                      self.thing.name))
62                     break
63         self.thing.game.record_change(self.thing.position, 'other')
64         terrain = \
65             self.thing.game.maps[self.thing.position[0]][self.thing.position[1]]
66         if terrain in self.thing.game.terrains:
67             terrain_type = self.thing.game.terrains[terrain]
68             if 'sittable' in terrain_type.tags:
69                 self.thing.standing = False
70                 self.thing.send_msg('CHAT "You sink into the %s."'
71                                     % terrain_type.description)
72         self.thing.invalidate('fov')
73         if self.thing.blocks_light:
74             self.thing.game.record_change(self.thing.position, 'fov')
75         if self.thing.carrying:
76             self.thing.carrying.position = self.thing.position
77
78
79
80 class Task_WRITE(Task):
81     argtypes = 'string:char string'
82
83     def check(self):
84         if not self.thing.game.can_do_tile_with_pw(*self.thing.position,
85                                                    self.args[1]):
86             raise GameError('wrong password for tile')
87
88     def do(self):
89         big_yx = self.thing.position[0]
90         little_yx = self.thing.position[1]
91         self.thing.game.maps[big_yx][little_yx] = self.args[0]
92         self.thing.game.record_change((big_yx, little_yx), 'fov')
93
94
95
96 class Task_FLATTEN_SURROUNDINGS(Task):
97     argtypes = 'string'
98
99     def check(self):
100         pass
101
102     def do(self):
103         for yxyx in [self.thing.position] + \
104                 list(self.thing.game.map_geometry.get_neighbors_yxyx(
105                     self.thing.position).values()):
106             if not self.thing.game.can_do_tile_with_pw(*yxyx, self.args[0]):
107                 continue
108             self.thing.game.maps[yxyx[0]][yxyx[1]] = self.thing.game.get_flatland()
109             self.thing.game.record_change(yxyx, 'fov')
110
111
112
113 class Task_PICK_UP(Task):
114     argtypes = 'int:pos'
115
116     def check(self):
117         if self.thing.carrying:
118             raise PlayError('already carrying something')
119         to_pick_up = self.thing.game.get_thing(self.args[0])
120         neighbors = self.thing.game.map_geometry.\
121             get_neighbors_yxyx(self.thing.position).values()
122         reach = [self.thing.position] + list(neighbors)
123         if to_pick_up is None:
124             raise PlayError('no such thing exists')
125         elif to_pick_up == self.thing:
126             raise PlayError('cannot pick up oneself')
127         elif to_pick_up.type_ == 'Player':
128             raise PlayError('cannot pick up player')
129         elif to_pick_up.carried:
130             raise PlayError('thing already carried by a player')
131         elif to_pick_up.position not in reach:
132             raise PlayError('thing not in reach')
133         elif not to_pick_up.portable:
134             raise PlayError('thing not portable')
135
136     def do(self):
137         to_pick_up = self.thing.game.get_thing(self.args[0])
138         to_pick_up.position = self.thing.position[:]
139         self.thing.carrying = to_pick_up
140         to_pick_up.carried = True
141         self.thing.game.record_change(self.thing.position, 'other')
142
143
144
145 class Task_DROP(Task):
146     argtypes = 'string:direction+here'
147
148     def check(self):
149         if not self.thing.carrying:
150             raise PlayError('nothing to drop')
151         target_position = self._get_move_target()
152         if self.thing.carrying.type_ == 'Bottle' and self.thing.carrying.full:
153             for t in [t for t in self.thing.game.things
154                       if t.type_ == 'BottleDeposit'
155                       and t.position == target_position]:
156                 raise PlayError('cannot drop full bottle into bottle deposit')
157
158     def do(self):
159         target_position = self._get_move_target()
160         dropped = self.thing.uncarry()
161         dropped.position = target_position
162         over_cookie_spawner = None
163         for t in [t for t in self.thing.game.things
164                   if t.type_ == 'CookieSpawner'
165                   and t.position == dropped.position]:
166             over_cookie_spawner = t
167             break
168         if over_cookie_spawner:
169             over_cookie_spawner.accept(dropped)
170             self.thing.game.remove_thing(dropped)
171         elif dropped.type_ == 'Bottle' and not dropped.full:
172             for t in [t for t in self.thing.game.things
173                       if t.type_ == 'BottleDeposit'
174                       and t.position == dropped.position]:
175                 t.accept()
176                 self.thing.game.remove_thing(dropped)
177                 break
178         elif dropped.type_ == 'Hat':
179             for t in [t for t in self.thing.game.things
180                       if t.type_ == 'HatRemixer'
181                       and t.position == dropped.position]:
182                 t.accept(dropped)
183                 break
184         self.thing.game.record_change(self.thing.position, 'other')
185
186
187
188 class Task_DOOR(Task):
189
190     def check(self):
191         action_radius = list(self.thing.game.map_geometry.
192                              get_neighbors_yxyx(self.thing.position).values())
193         if len([t for t in self.thing.game.things if
194                   t.type_ == 'Door' and t.position in action_radius]) == 0:
195             raise PlayError('not standing next to a door to open/close')
196
197     def do(self):
198         action_radius = list(self.thing.game.map_geometry.
199                              get_neighbors_yxyx(self.thing.position).values())
200         for t in [t for t in self.thing.game.things if
201                   t.type_ == 'Door' and t.position in action_radius]:
202             if t.blocks_movement:
203                 t.open()
204             else:
205                 t.close()
206             self.thing.game.record_change(t.position, 'other')
207             self.thing.game.record_change(t.position, 'fov')
208
209
210
211 class Task_INTOXICATE(Task):
212
213     def check(self):
214         if self.thing.carrying is None:
215             raise PlayError('carrying nothing to consume')
216         if self.thing.carrying.type_ not in {'Bottle', 'Cookie', 'Psychedelic'}:
217             raise PlayError('cannot consume this kind of thing')
218         if self.thing.carrying.type_ == 'Bottle' and\
219            not self.thing.carrying.full:
220             raise PlayError('bottle is empty')
221
222     def do(self):
223         if self.thing.carrying.type_ == 'Bottle':
224             self.thing.carrying.full = False
225             self.thing.carrying.empty()
226             self.thing.send_msg('CHAT "You are drunk now."')
227             self.thing.need_for_toilet += 10000
228             self.thing.drunk = 10000
229             self.thing.invalidate('fov')
230             self.thing.game.record_change(self.thing.position, 'other')
231         elif self.thing.carrying.type_ == 'Psychedelic':
232             self.thing.tripping = 10000
233             self.thing.send_msg('CHAT "You start tripping."')
234             self.thing.send_msg('RANDOM_COLORS')
235             eaten = self.thing.uncarry()
236             self.thing.game.remove_thing(eaten)
237         elif self.thing.carrying.type_ == 'Cookie':
238             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))
239             self.thing.add_cookie_char(self.thing.carrying.thing_char)
240             eaten = self.thing.uncarry()
241             self.thing.game.remove_thing(eaten)
242
243
244
245 class Task_COMMAND(Task):
246     argtypes = 'string'
247
248     def check(self):
249         if self.thing.carrying is None:
250             raise PlayError('nothing to command')
251         if not self.thing.carrying.commandable:
252             raise PlayError('cannot command this item type')
253
254     def do(self):
255         reply_lines = self.thing.carrying.interpret(self.args[0])
256         for line in reply_lines:
257             self.thing.send_msg('REPLY ' + quote(line))
258
259
260
261 class Task_INSTALL(Task):
262     argtypes = 'string'
263
264     def _get_uninstallables(self):
265         return [t for t in self.thing.game.things
266                 if t != self.thing
267                 and hasattr(t, 'installable') and t.installable
268                 and (not t.portable)
269                 and t.position == self.thing.position]
270
271     def check(self):
272         if not self.thing.game.can_do_tile_with_pw(*self.thing.position,
273                                                    self.args[0]):
274             raise GameError('wrong password for tile')
275         if self.thing.carrying:
276             if not hasattr(self.thing.carrying, 'installable')\
277                or not self.thing.carrying.installable:
278                 raise PlayError('carried thing not installable')
279         elif len(self._get_uninstallables()) == 0:
280             raise PlayError('nothing to uninstall here')
281
282     def do(self):
283         if self.thing.carrying:
284             t = self.thing.uncarry()
285             t.install()
286             self.thing.send_msg('CHAT "You install the thing you carry."')
287         else:
288             self._get_uninstallables()[0].uninstall()
289             self.thing.send_msg('CHAT "You uninstall the thing here."')
290         self.thing.game.record_change(self.thing.position, 'other')
291
292
293
294 class Task_WEAR(Task):
295
296     def check(self):
297         if self.thing.name in self.thing.game.hats:
298             return
299         if not self.thing.carrying:
300             raise PlayError('carrying nothing to wear')
301         if self.thing.name in self.thing.game.hats:
302             raise PlayError('already wearing a hat')
303         if self.thing.carrying.type_ not in {'Hat', 'Bottle'}:
304             raise PlayError('can not wear the kind of thing you hold')
305
306     def do(self):
307         if self.thing.name in self.thing.game.hats:
308             t = self.thing.game.add_thing('Hat', self.thing.position)
309             t.design = self.thing.game.hats[self.thing.name]
310             del self.thing.game.hats[self.thing.name]
311             self.thing.send_msg('CHAT "You drop your hat."')
312             for remixer in [t for t in self.thing.game.things
313                             if t.type_ == 'HatRemixer'
314                             and t.position == self.thing.position]:
315                 remixer.accept(t)
316                 break
317         else:
318             if self.thing.carrying.type_ == 'Bottle':
319                 self.thing.send_msg('CHAT "Your attempt to wear a bottle on '
320                                     'your head fails."')
321                 self.thing.carrying.sound('BOTTLE', 'SMASH')
322             elif self.thing.carrying.type_ == 'Hat':
323                 self.thing.game.hats[self.thing.name] =\
324                     self.thing.carrying.design
325                 self.thing.send_msg('CHAT "You put on a hat."')
326             dropped = self.thing.uncarry()
327             self.thing.game.remove_thing(dropped)
328         self.thing.game.record_change(self.thing.position, 'other')
329
330
331
332 class Task_SPIN(Task):
333
334     def check(self):
335         if not self.thing.carrying:
336             raise PlayError('holding nothing to spin')
337         if not hasattr(self.thing.carrying, 'spinnable'):
338             raise PlayError('held object not spinnable')
339
340     def do(self):
341         self.thing.carrying.spin()
342         self.thing.send_msg('CHAT "You spin this object."')