home · contact · privacy
c3600aa558cc9eaa844feded252a16b62cac53b4
[plomrogue] / plomrogue-server.py
1 import argparse
2 import errno
3 import os
4 import shlex
5 import shutil
6 import time
7
8
9 def strong_write(file, string):
10     """Apply write(string), flush(), and os.fsync() to file."""
11     file.write(string)
12     file.flush()
13     os.fsync(file)
14
15
16 def setup_server_io():
17     """Fill IO files DB with proper file( path)s. Write process IO test string.
18
19     Ensure IO files directory at server/. Remove any old input file if found.
20     Set up new input file for reading, and new output file for writing. Start
21     output file with process hash line of format PID + " " + floated UNIX time
22     (io_db["teststring"]). Raise SystemExit if file is found at path of either
23     record or save file plus io_db["tmp_suffix"].
24     """
25     def detect_atomic_leftover(path, tmp_suffix):
26         path_tmp = path + tmp_suffix
27         msg = "Found file '" + path_tmp + "' that may be a leftover from an " \
28               "aborted previous attempt to write '" + path + "'. Aborting " \
29              "until matter is resolved by removing it from its current path."
30         if os.access(path_tmp, os.F_OK):
31             raise SystemExit(msg)
32     io_db["teststring"] = str(os.getpid()) + " " + str(time.time())
33     os.makedirs(io_db["path_server"], exist_ok=True)
34     io_db["file_out"] = open(io_db["path_out"], "w")
35     strong_write(io_db["file_out"], io_db["teststring"] + "\n")
36     if os.access(io_db["path_in"], os.F_OK):
37         os.remove(io_db["path_in"])
38     io_db["file_in"] = open(io_db["path_in"], "w")
39     io_db["file_in"].close()
40     io_db["file_in"] = open(io_db["path_in"], "r")
41     detect_atomic_leftover(io_db["path_save"], io_db["tmp_suffix"])
42     detect_atomic_leftover(io_db["path_record"], io_db["tmp_suffix"])
43
44
45 def cleanup_server_io():
46     """Close and (if io_db["kicked_by_rival"] false) remove files in io_db."""
47     def helper(file_key, path_key):
48         if file_key in io_db:
49             io_db[file_key].close()
50             if not io_db["kicked_by_rival"] \
51                and os.access(io_db[path_key], os.F_OK):
52                 os.remove(io_db[path_key])
53     helper("file_out", "path_out")
54     helper("file_in", "path_in")
55     helper("file_worldstate", "path_worldstate")
56     if "file_record" in io_db:
57         io_db["file_record"].close()
58
59
60 def obey(command, prefix, replay=False, do_record=False):
61     """Call function from commands_db mapped to command's first token.
62
63     Tokenize command string with shlex.split(comments=True). If replay is set,
64     a non-meta command from the commands_db merely triggers obey() on the next
65     command from the records file. If not, non-meta commands set
66     io_db["worldstate_updateable"] to world_db["WORLD_EXISTS"], and, if
67     do_record is set, are recorded via record(), and save_world() is called.
68     The prefix string is inserted into the server's input message between its
69     beginning 'input ' & ':'. All activity is preceded by a server_test() call.
70     """
71     server_test()
72     print("input " + prefix + ": " + command)
73     try:
74         tokens = shlex.split(command, comments=True)
75     except ValueError as err:
76         print("Can't tokenize command string: " + str(err) + ".")
77         return
78     if len(tokens) > 0 and tokens[0] in commands_db \
79        and len(tokens) == commands_db[tokens[0]][0] + 1:
80         if commands_db[tokens[0]][1]:
81             commands_db[tokens[0]][2](*tokens[1:])
82         elif replay:
83             print("Due to replay mode, reading command as 'go on in record'.")
84             line = io_db["file_record"].readline()
85             if len(line) > 0:
86                 obey(line.rstrip(), io_db["file_record"].prefix
87                      + str(io_db["file_record"].line_n))
88                 io_db["file_record"].line_n = io_db["file_record"].line_n + 1
89             else:
90                 print("Reached end of record file.")
91         else:
92             commands_db[tokens[0]][2](*tokens[1:])
93             if do_record:
94                 record(command)
95                 save_world()
96             io_db["worldstate_updateable"] = world_db["WORLD_ACTIVE"]
97     elif 0 != len(tokens):
98         print("Invalid command/argument, or bad number of tokens.")
99
100
101 def atomic_write(path, text, do_append=False):
102     """Atomic write of text to file at path, appended if do_append is set."""
103     path_tmp = path + io_db["tmp_suffix"]
104     mode = "w"
105     if do_append:
106         mode = "a"
107         if os.access(path, os.F_OK):
108             shutil.copyfile(path, path_tmp)
109     file = open(path_tmp, mode)
110     strong_write(file, text)
111     file.close()
112     if os.access(path, os.F_OK):
113         os.remove(path)
114     os.rename(path_tmp, path)
115
116
117 def record(command):
118     """Append command string plus newline to record file. (Atomic.)"""
119     # This misses some optimizations from the original record(), namely only
120     # finishing the atomic write with expensive flush() and fsync() every 15
121     # seconds unless explicitely forced. Implement as needed.
122     atomic_write(io_db["path_record"], command + "\n", do_append=True)
123
124
125 def save_world():
126     """Save all commands needed to reconstruct current world state."""
127     # TODO: Misses same optimizations as record() from the original record().
128
129     def quote(string):
130         string = string.replace("\u005C", '\u005C\u005C')
131         return '"' + string.replace('"', '\u005C"') + '"'
132
133     def mapsetter(key):
134         def helper(id):
135             string = ""
136             if world_db["Things"][id][key]:
137                 map = world_db["Things"][id][key]
138                 length = world_db["MAP_LENGTH"]
139                 for i in range(length):
140                     line = map[i * length:(i * length) + length].decode()
141                     string = string + key + " " + str(i) + " " + quote(line) \
142                              + "\n"
143             return string
144         return helper
145
146     def memthing(id):
147         string = ""
148         for memthing in world_db["Things"][id]["T_MEMTHING"]:
149             string = string + "T_MEMTHING " + str(memthing[0]) + " " + \
150                      str(memthing[1]) + " " + str(memthing[2]) + "\n"
151         return string
152
153     def helper(category, id_string, special_keys={}):
154         string = ""
155         for id in world_db[category]:
156             string = string + id_string + " " + str(id) + "\n"
157             for key in world_db[category][id]:
158                 if not key in special_keys:
159                     x = world_db[category][id][key]
160                     argument = quote(x) if str == type(x) else str(x)
161                     string = string + key + " " + argument + "\n"
162                 elif special_keys[key]:
163                     string = string + special_keys[key](id)
164         return string
165
166     string = ""
167     for key in world_db:
168         if dict != type(world_db[key]) and key != "MAP":
169             string = string + key + " " + str(world_db[key]) + "\n"
170     string = string + helper("ThingActions", "TA_ID")
171     string = string + helper("ThingTypes", "TT_ID", {"TT_CORPSE_ID": False})
172     for id in world_db["ThingTypes"]:
173         string = string + "TT_ID " + str(id) + "\n" + "TT_CORPSE_ID " + \
174                  str(world_db["ThingTypes"][id]["TT_CORPSE_ID"]) + "\n"
175     string = string + helper("Things", "T_ID",
176                              {"T_CARRIES": False, "carried": False,
177                               "T_MEMMAP": mapsetter("T_MEMMAP"),
178                               "T_MEMTHING": memthing, "fovmap": False,
179                               "T_MEMDEPTHMAP": mapsetter("T_MEMDEPTHMAP")})
180     for id in world_db["Things"]:
181         if [] != world_db["Things"][id]["T_CARRIES"]:
182             string = string + "T_ID " + str(id) + "\n"
183             for carried_id in world_db["Things"][id]["T_CARRIES"]:
184                 string = string + "T_CARRIES " + str(carried_id) + "\n"
185     string = string + "WORLD_ACTIVE " + str(world_db["WORLD_ACTIVE"])
186     atomic_write(io_db["path_save"], string)
187
188
189 def obey_lines_in_file(path, name, do_record=False):
190     """Call obey() on each line of path's file, use name in input prefix."""
191     file = open(path, "r")
192     line_n = 1
193     for line in file.readlines():
194         obey(line.rstrip(), name + "file line " + str(line_n),
195              do_record=do_record)
196         line_n = line_n + 1
197     file.close()
198
199
200 def parse_command_line_arguments():
201     """Return settings values read from command line arguments."""
202     parser = argparse.ArgumentParser()
203     parser.add_argument('-s', nargs='?', type=int, dest='replay', const=1,
204                         action='store')
205     opts, unknown = parser.parse_known_args()
206     return opts
207
208
209 def server_test():
210     """Ensure valid server out file belonging to current process.
211
212     This is done by comparing io_db["teststring"] to what's found at the start
213     of the current file at io_db["path_out"]. On failure, set
214     io_db["kicked_by_rival"] and raise SystemExit.
215     """
216     if not os.access(io_db["path_out"], os.F_OK):
217         raise SystemExit("Server output file has disappeared.")
218     file = open(io_db["path_out"], "r")
219     test = file.readline().rstrip("\n")
220     file.close()
221     if test != io_db["teststring"]:
222         io_db["kicked_by_rival"] = True
223         msg = "Server test string in server output file does not match. This" \
224               " indicates that the current server process has been " \
225               "superseded by another one."
226         raise SystemExit(msg)
227
228
229 def read_command():
230     """Return next newline-delimited command from server in file.
231
232     Keep building return string until a newline is encountered. Pause between
233     unsuccessful reads, and after too much waiting, run server_test().
234     """
235     wait_on_fail = 1
236     max_wait = 5
237     now = time.time()
238     command = ""
239     while True:
240         add = io_db["file_in"].readline()
241         if len(add) > 0:
242             command = command + add
243             if len(command) > 0 and "\n" == command[-1]:
244                 command = command[:-1]
245                 break
246         else:
247             time.sleep(wait_on_fail)
248             if now + max_wait < time.time():
249                 server_test()
250                 now = time.time()
251     return command
252
253
254 def try_worldstate_update():
255     """Write worldstate file if io_db["worldstate_updateable"] is set."""
256     if io_db["worldstate_updateable"]:
257
258         def draw_visible_Things(map, run):
259             for id in world_db["Things"]:
260                 type = world_db["Things"][id]["T_TYPE"]
261                 consumable = world_db["ThingTypes"][type]["TT_CONSUMABLE"]
262                 alive = world_db["ThingTypes"][type]["TT_LIFEPOINTS"]
263                 if (0 == run and not consumable and not alive) \
264                    or (1 == run and consumable and not alive) \
265                    or (2 == run and alive):
266                     y = world_db["Things"][id]["T_POSY"]
267                     x = world_db["Things"][id]["T_POSX"]
268                     fovflag = world_db["Things"][0]["fovmap"][(y * length) + x]
269                     if 'v' == chr(fovflag):
270                         c = world_db["ThingTypes"][type]["TT_SYMBOL"]
271                         map[(y * length) + x] = ord(c)
272
273         def write_map(string, map):
274             for i in range(length):
275                 line = map[i * length:(i * length) + length].decode()
276                 string = string + line + "\n"
277             return string
278
279         inventory = ""
280         if [] == world_db["Things"][0]["T_CARRIES"]:
281             inventory = "(none)\n"
282         else:
283             for id in world_db["Things"][0]["T_CARRIES"]:
284                 type_id = world_db["Things"][id]["T_TYPE"]
285                 name = world_db["ThingTypes"][type_id]["TT_NAME"]
286                 inventory = inventory + name + "\n"
287         string = str(world_db["TURN"]) + "\n" + \
288                  str(world_db["Things"][0]["T_LIFEPOINTS"]) + "\n" + \
289                  str(world_db["Things"][0]["T_SATIATION"]) + "\n" + \
290                  inventory + "%\n" + \
291                  str(world_db["Things"][0]["T_POSY"]) + "\n" + \
292                  str(world_db["Things"][0]["T_POSX"]) + "\n" + \
293                  str(world_db["MAP_LENGTH"]) + "\n"
294         length = world_db["MAP_LENGTH"]
295         fov = bytearray(b' ' * (length ** 2))
296         for pos in range(length ** 2):
297             if 'v' == chr(world_db["Things"][0]["fovmap"][pos]):
298                 fov[pos] = world_db["MAP"][pos]
299         for i in range(3):
300             draw_visible_Things(fov, i)
301         string = write_map(string, fov)
302         mem = world_db["Things"][0]["T_MEMMAP"][:]
303         for i in range(2):
304             for memthing in world_db["Things"][0]["T_MEMTHING"]:
305                 type = world_db["Things"][memthing[0]]["T_TYPE"]
306                 consumable = world_db["ThingTypes"][type]["TT_CONSUMABLE"]
307                 if (i == 0 and not consumable) or (i == 1 and consumable):
308                     c = world_db["ThingTypes"][type]["TT_SYMBOL"]
309                     mem[(memthing[1] * length) + memthing[2]] = ord(c)
310         string = write_map(string, mem)
311         atomic_write(io_db["path_worldstate"], string)
312         strong_write(io_db["file_out"], "WORLD_UPDATED\n")
313         io_db["worldstate_updateable"] = False
314
315
316 def replay_game():
317     """Replay game from record file.
318
319     Use opts.replay as breakpoint turn to which to replay automatically before
320     switching to manual input by non-meta commands in server input file
321     triggering further reads of record file. Ensure opts.replay is at least 1.
322     Run try_worldstate_update() before each interactive obey()/read_command().
323     """
324     if opts.replay < 1:
325         opts.replay = 1
326     print("Replay mode. Auto-replaying up to turn " + str(opts.replay) +
327           " (if so late a turn is to be found).")
328     if not os.access(io_db["path_record"], os.F_OK):
329         raise SystemExit("No record file found to replay.")
330     io_db["file_record"] = open(io_db["path_record"], "r")
331     io_db["file_record"].prefix = "record file line "
332     io_db["file_record"].line_n = 1
333     while world_db["TURN"] < opts.replay:
334         line = io_db["file_record"].readline()
335         if "" == line:
336             break
337         obey(line.rstrip(), io_db["file_record"].prefix
338              + str(io_db["file_record"].line_n))
339         io_db["file_record"].line_n = io_db["file_record"].line_n + 1
340     while True:
341         try_worldstate_update()
342         obey(read_command(), "in file", replay=True)
343
344
345 def play_game():
346     """Play game by server input file commands. Before, load save file found.
347
348     If no save file is found, a new world is generated from the commands in the
349     world config plus a 'MAKE WORLD [current Unix timestamp]'. Record this
350     command and all that follow via the server input file. Run
351     try_worldstate_update() before each interactive obey()/read_command().
352     """
353     if os.access(io_db["path_save"], os.F_OK):
354         obey_lines_in_file(io_db["path_save"], "save")
355     else:
356         if not os.access(io_db["path_worldconf"], os.F_OK):
357             msg = "No world config file from which to start a new world."
358             raise SystemExit(msg)
359         obey_lines_in_file(io_db["path_worldconf"], "world config ",
360                            do_record=True)
361         obey("MAKE_WORLD " + str(int(time.time())), "in file", do_record=True)
362     while True:
363         try_worldstate_update()
364         obey(read_command(), "in file", do_record=True)
365
366
367 def remake_map():
368     # DUMMY map creator.
369     world_db["MAP"] = bytearray(b'.' * (world_db["MAP_LENGTH"] ** 2))
370
371
372 def update_map_memory(t):
373     """Update t's T_MEMMAP with what's in its FOV now,age its T_MEMMEPTHMAP."""
374     if not t["T_MEMMAP"]:
375         t["T_MEMMAP"] = bytearray(b' ' * (world_db["MAP_LENGTH"] ** 2))
376     if not t["T_MEMDEPTHMAP"]:
377         t["T_MEMDEPTHMAP"] = bytearray(b' ' * (world_db["MAP_LENGTH"] ** 2))
378     for pos in range(world_db["MAP_LENGTH"] ** 2):
379         if "v" == chr(t["fovmap"][pos]):
380             t["T_MEMDEPTHMAP"][pos] = ord("0")
381             if " " == chr(t["T_MEMMAP"][pos]):
382                 t["T_MEMMAP"][pos] = world_db["MAP"][pos]
383             continue
384         # TODO: Aging of MEMDEPTHMAP.
385     for memthing in t["T_MEMTHING"]:
386         y = world_db["Things"][memthing[0]]["T_POSY"]
387         x = world_db["Things"][memthing[1]]["T_POSY"]
388         if "v" == chr(t["fovmap"][(y * world_db["MAP_LENGTH"]) + x]):
389             t["T_MEMTHING"].remove(memthing)
390     for id in world_db["Things"]:
391         type = world_db["Things"][id]["T_TYPE"]
392         if not world_db["ThingTypes"][type]["TT_LIFEPOINTS"]:
393             y = world_db["Things"][id]["T_POSY"]
394             x = world_db["Things"][id]["T_POSY"]
395             if "v" == chr(t["fovmap"][(y * world_db["MAP_LENGTH"]) + x]):
396                 t["T_MEMTHING"].append((type, y, x))
397
398
399 def set_world_inactive():
400     """Set world_db["WORLD_ACTIVE"] to 0 and remove worldstate file."""
401     server_test()
402     if os.access(io_db["path_worldstate"], os.F_OK):
403         os.remove(io_db["path_worldstate"])
404     world_db["WORLD_ACTIVE"] = 0
405
406
407 def integer_test(val_string, min, max):
408     """Return val_string if possible integer >= min and <= max, else None."""
409     try:
410         val = int(val_string)
411         if val < min or val > max:
412             raise ValueError
413         return val
414     except ValueError:
415         print("Ignoring: Please use integer >= " + str(min) + " and <= " +
416               str(max) + ".")
417         return None
418
419
420 def setter(category, key, min, max):
421     """Build setter for world_db([category + "s"][id])[key] to >=min/<=max."""
422     if category is None:
423         def f(val_string):
424             val = integer_test(val_string, min, max)
425             if None != val:
426                 world_db[key] = val
427     else:
428         if category == "Thing":
429             id_store = command_tid
430             decorator = test_Thing_id
431         elif category == "ThingType":
432             id_store = command_ttid
433             decorator = test_ThingType_id
434         elif category == "ThingAction":
435             id_store = command_taid
436             decorator = test_ThingAction_id
437
438         @decorator
439         def f(val_string):
440             val = integer_test(val_string, min, max)
441             if None != val:
442                 world_db[category + "s"][id_store.id][key] = val
443     return f
444
445
446 def build_fov_map(t):
447     """Build Thing's FOV map."""
448     t["fovmap"] = bytearray(b'v' * (world_db["MAP_LENGTH"] ** 2))
449     # DUMMY so far. Just builds an all-visible map.
450
451
452 def actor_wait(Thing):
453     """Do nothing (but loudly, if by player)."""
454     if Thing == world_db["Things"][0]:
455         strong_write(io_db["file_out"], "LOG You wait.\n")
456
457
458 def actor_move(Thing):
459     pass
460
461
462 def actor_pick_up(Thing):
463     pass
464
465
466 def actor_drop(Thing):
467     pass
468
469
470 def actor_use(Thing):
471     pass
472
473
474 def turn_over():
475     """Run game world and its inhabitants until new player input expected."""
476     id = 0
477     whilebreaker = False
478     while world_db["Things"][0]["T_LIFEPOINTS"]:
479         for id in [id for id in world_db["Things"]
480                       if not world_db["Things"][id]["carried"]
481                       if world_db["Things"][id]["T_LIFEPOINTS"]]:
482             Thing = world_db["Things"][id]
483             if Thing["T_LIFEPOINTS"]:
484                 if not Thing["T_COMMAND"]:
485                     update_map_memory(Thing)
486                     if 0 == id:
487                         whilebreaker = True
488                         break
489                     # DUMMY: ai(thing)
490                     Thing["T_COMMAND"] = 1
491                 # DUMMY: try_healing
492                 Thing["T_PROGRESS"] += 1
493                 taid = [a for a in world_db["ThingActions"]
494                           if a == Thing["T_COMMAND"]][0]
495                 ThingAction = world_db["ThingActions"][taid]
496                 if Thing["T_PROGRESS"] == ThingAction["TA_EFFORT"]:
497                     eval("actor_" + ThingAction["TA_NAME"])(Thing)
498                     Thing["T_COMMAND"] = 0
499                     Thing["T_PROGRESS"] = 0
500                 # DUMMY: hunger
501             # DUMMY: thingproliferation
502         if whilebreaker:
503             break
504         world_db["TURN"] += 1
505
506
507 def new_Thing(type):
508     """Return Thing of type T_TYPE, with fovmap if alive and world active."""
509     thing = {
510         "T_LIFEPOINTS": world_db["ThingTypes"][type]["TT_LIFEPOINTS"],
511         "T_ARGUMENT": 0,
512         "T_PROGRESS": 0,
513         "T_SATIATION": 0,
514         "T_COMMAND": 0,
515         "T_TYPE": type,
516         "T_POSY": 0,
517         "T_POSX": 0,
518         "T_CARRIES": [],
519         "carried": False,
520         "T_MEMTHING": [],
521         "T_MEMMAP": False,
522         "T_MEMDEPTHMAP": False,
523         "fovmap": False
524     }
525     if world_db["WORLD_ACTIVE"] and thing["T_LIFEPOINTS"]:
526         build_fov_map(thing)
527     return thing
528
529
530 def id_setter(id, category, id_store=False, start_at_1=False):
531     """Set ID of object of category to manipulate ID unused? Create new one.
532
533     The ID is stored as id_store.id (if id_store is set). If the integer of the
534     input is valid (if start_at_1, >= 0 and <= 255, else >= -32768 and <=
535     32767), but <0 or (if start_at_1) <1, calculate new ID: lowest unused ID
536     >=0 or (if start_at_1) >= 1, and <= 255. None is always returned when no
537     new object is created, otherwise the new object's ID.
538     """
539     min = 0 if start_at_1 else -32768
540     max = 255 if start_at_1 else 32767
541     if str == type(id):
542         id = integer_test(id, min, max)
543     if None != id:
544         if id in world_db[category]:
545             if id_store:
546                 id_store.id = id
547             return None
548         else:
549             if (start_at_1 and 0 == id) \
550                or ((not start_at_1) and (id < 0 or id > 255)):
551                 id = -1
552                 while 1:
553                     id = id + 1
554                     if id not in world_db[category]:
555                         break
556                 if id > 255:
557                     print("Ignoring: "
558                           "No unused ID available to add to ID list.")
559                     return None
560             if id_store:
561                 id_store.id = id
562     return id
563
564
565 def command_ping():
566     """Send PONG line to server output file."""
567     strong_write(io_db["file_out"], "PONG\n")
568
569
570 def command_quit():
571     """Abort server process."""
572     raise SystemExit("received QUIT command")
573
574
575 def command_thingshere(str_y, str_x):
576     """Write to out file list of Things known to player at coordinate y, x."""
577     def write_thing_if_here():
578         if y == world_db["Things"][id]["T_POSY"] \
579            and x == world_db["Things"][id]["T_POSX"]:
580             type = world_db["Things"][id]["T_TYPE"]
581             name = world_db["ThingTypes"][type]["TT_NAME"]
582             strong_write(io_db["file_out"], name + "\n")
583     if world_db["WORLD_ACTIVE"]:
584         y = integer_test(str_y, 0, 255)
585         x = integer_test(str_x, 0, 255)
586         length = world_db["MAP_LENGTH"]
587         if None != y and None != x and y < length and x < length:
588             pos = (y * world_db["MAP_LENGTH"]) + x
589             strong_write(io_db["file_out"], "THINGS_HERE START\n")
590             if "v" == chr(world_db["Things"][0]["fovmap"][pos]):
591                 for id in world_db["Things"]:
592                     write_thing_if_here()
593             else:
594                 for id in world_db["Things"]["T_MEMTHING"]:
595                     write_thing_if_here()
596             strong_write(io_db["file_out"], "THINGS_HERE END\n")
597         else:
598             print("Ignoring: Invalid map coordinates.")
599     else:
600         print("Ignoring: Command only works on existing worlds.")
601
602
603 def play_commander(action, args=False):
604     """Setter for player's T_COMMAND and T_ARGUMENT, then calling turn_over().
605
606     T_ARGUMENT is set to direction char if action=="wait",or 8-bit int if args.
607     """
608
609     def set_command():
610         id = [x for x in world_db["ThingActions"]
611                 if world_db["ThingActions"][x]["TA_NAME"] == action][0]
612         world_db["Things"][0]["T_COMMAND"] = id
613         turn_over()
614         # TODO: call turn_over()
615
616     def set_command_and_argument_int(str_arg):
617         val = integer_test(str_arg, 0, 255)
618         if None != val:
619             world_db["Things"][0]["T_ARGUMENT"] = val
620             set_command()
621         else:
622             print("Ignoring: Argument must be integer >= 0 <=255.")
623
624     def set_command_and_argument_movestring(str_arg):
625         dirs = {"east": "d", "south-east": "c", "south-west": "x",
626                 "west": "s", "north-west": "w", "north-east": "e"}
627         if str_arg in dirs:
628             world_db["Things"][0]["T_ARGUMENT"] = dirs[str_arg]
629             set_command()
630         else:
631             print("Ignoring: Argument must be valid direction string.")
632
633     if action == "move":
634         return set_command_and_argument_movestring
635     elif args:
636         return set_command_and_argument_int
637     else:
638         return set_command
639
640
641 def command_seedmap(seed_string):
642     """Set world_db["SEED_MAP"] to int(seed_string), then (re-)make map."""
643     setter(None, "SEED_MAP", 0, 4294967295)(seed_string)
644     remake_map()
645
646
647 def command_makeworld(seed_string):
648     """(Re-)build game world, i.e. map, things, to a new turn 1 from seed.
649
650     Make seed world_db["SEED_RANDOMNESS"] and world_db["SEED_MAP"]. Do more
651     only with a "wait" ThingAction and world["PLAYER_TYPE"] matching ThingType
652     of TT_START_NUMBER > 0. Then, world_db["Things"] emptied, call remake_map()
653     and set world_db["WORLD_ACTIVE"], world_db["TURN"] to 1. Build new Things
654     according to ThingTypes' TT_START_NUMBERS, with Thing of ID 0 to ThingType
655     of ID = world["PLAYER_TYPE"]. Place Things randomly, and actors not on each
656     other. Init player's memory map. Write "NEW_WORLD" line to out file.
657     """
658     setter(None, "SEED_RANDOMNESS", 0, 4294967295)(seed_string)
659     setter(None, "SEED_MAP", 0, 4294967295)(seed_string)
660     player_will_be_generated = False
661     playertype = world_db["PLAYER_TYPE"]
662     for ThingType in world_db["ThingTypes"]:
663         if playertype == ThingType:
664             if 0 < world_db["ThingTypes"][ThingType]["TT_START_NUMBER"]:
665                 player_will_be_generated = True
666             break
667     if not player_will_be_generated:
668         print("Ignoring beyond SEED_MAP: " +
669               "No player type with start number >0 defined.")
670         return
671     wait_action = False
672     for ThingAction in world_db["ThingActions"]:
673         if "wait" == world_db["ThingActions"][ThingAction]["TA_NAME"]:
674             wait_action = True
675     if not wait_action:
676         print("Ignoring beyond SEED_MAP: " +
677               "No thing action with name 'wait' defined.")
678         return
679     world_db["Things"] = {}
680     remake_map()
681     world_db["WORLD_ACTIVE"] = 1
682     world_db["TURN"] = 1
683     for i in range(world_db["ThingTypes"][playertype]["TT_START_NUMBER"]):
684         id = id_setter(-1, "Things")
685         world_db["Things"][id] = new_Thing(playertype)
686     # TODO: Positioning.
687     update_map_memory(world_db["Things"][0])
688     for type in world_db["ThingTypes"]:
689         for i in range(world_db["ThingTypes"][type]["TT_START_NUMBER"]):
690             if type != playertype:
691                 id = id_setter(-1, "Things")
692                 world_db["Things"][id] = new_Thing(type)
693     # TODO: Positioning.
694     strong_write(io_db["file_out"], "NEW_WORLD\n")
695
696
697 def command_maplength(maplength_string):
698     """Redefine map length. Invalidate map, therefore lose all things on it."""
699     set_world_inactive()
700     world_db["Things"] = {}
701     setter(None, "MAP_LENGTH", 1, 256)(maplength_string)
702
703
704 def command_worldactive(worldactive_string):
705     """Toggle world_db["WORLD_ACTIVE"] if possible.
706
707     An active world can always be set inactive. An inactive world can only be
708     set active with a "wait" ThingAction, and a player Thing (of ID 0). On
709     activation, rebuild all Things' FOVs, and the player's map memory.
710     """
711     # In original version, map existence was also tested (unnecessarily?).
712     val = integer_test(worldactive_string, 0, 1)
713     if val:
714         if 0 != world_db["WORLD_ACTIVE"]:
715             if 0 == val:
716                 set_world_inactive()
717             else:
718                 print("World already active.")
719         elif 0 == world_db["WORLD_ACTIVE"]:
720             wait_exists = False
721             for ThingAction in world_db["ThingActions"]:
722                 if "wait" == world_db["ThingActions"][ThingAction]["TA_NAME"]:
723                     wait_exists = True
724                     break
725             player_exists = False
726             for Thing in world_db["Things"]:
727                 if 0 == Thing:
728                     player_exists = True
729                     break
730             if wait_exists and player_exists:
731                 for id in world_db["Things"]:
732                     if world_db["Things"][id]["T_LIFEPOINTS"]:
733                         build_fov_map(world_db["Things"][id])
734                         if 0 == id:
735                             update_map_memory(world_db["Things"][id])
736                 world_db["WORLD_ACTIVE"] = 1
737
738
739 def test_for_id_maker(object, category):
740     """Return decorator testing for object having "id" attribute."""
741     def decorator(f):
742         def helper(*args):
743             if hasattr(object, "id"):
744                 f(*args)
745             else:
746                 print("Ignoring: No " + category +
747                       " defined to manipulate yet.")
748         return helper
749     return decorator
750
751
752 def command_tid(id_string):
753     """Set ID of Thing to manipulate. ID unused? Create new one.
754
755     Default new Thing's type to the first available ThingType, others: zero.
756     """
757     id = id_setter(id_string, "Things", command_tid)
758     if None != id:
759         if world_db["ThingTypes"] == {}:
760             print("Ignoring: No ThingType to settle new Thing in.")
761             return
762         type = list(world_db["ThingTypes"].keys())[0]
763         world_db["Things"][id] = new_Thing(type)
764
765
766 test_Thing_id = test_for_id_maker(command_tid, "Thing")
767
768
769 @test_Thing_id
770 def command_tcommand(str_int):
771     """Set T_COMMAND of selected Thing."""
772     val = integer_test(str_int, 0, 255)
773     if None != val:
774         if 0 == val or val in world_db["ThingActions"]:
775             world_db["Things"][command_tid.id]["T_COMMAND"] = val
776         else:
777             print("Ignoring: ThingAction ID belongs to no known ThingAction.")
778
779
780 @test_Thing_id
781 def command_ttype(str_int):
782     """Set T_TYPE of selected Thing."""
783     val = integer_test(str_int, 0, 255)
784     if None != val:
785         if val in world_db["ThingTypes"]:
786             world_db["Things"][command_tid.id]["T_TYPE"] = val
787         else:
788             print("Ignoring: ThingType ID belongs to no known ThingType.")
789
790
791 @test_Thing_id
792 def command_tcarries(str_int):
793     """Append int(str_int) to T_CARRIES of selected Thing.
794
795     The ID int(str_int) must not be of the selected Thing, and must belong to a
796     Thing with unset "carried" flag. Its "carried" flag will be set on owning.
797     """
798     val = integer_test(str_int, 0, 255)
799     if None != val:
800         if val == command_tid.id:
801             print("Ignoring: Thing cannot carry itself.")
802         elif val in world_db["Things"] \
803              and not world_db["Things"][val]["carried"]:
804             world_db["Things"][command_tid.id]["T_CARRIES"].append(val)
805             world_db["Things"][val]["carried"] = True
806         else:
807             print("Ignoring: Thing not available for carrying.")
808     # Note that the whole carrying structure is different from the C version:
809     # Carried-ness is marked by a "carried" flag, not by Things containing
810     # Things internally.
811
812
813 @test_Thing_id
814 def command_tmemthing(str_t, str_y, str_x):
815     """Add (int(str_t), int(str_y), int(str_x)) to selected Thing's T_MEMTHING.
816
817     The type must fit to an existing ThingType, and the position into the map.
818     """
819     type = integer_test(str_t, 0, 255)
820     posy = integer_test(str_y, 0, 255)
821     posx = integer_test(str_x, 0, 255)
822     if None != type and None != posy and None != posx:
823         if type not in world_db["ThingTypes"] \
824            or posy >= world_db["MAP_LENGTH"] or posx >= world_db["MAP_LENGTH"]:
825             print("Ignoring: Illegal value for thing type or position.")
826         else:
827             memthing = (type, posy, posx)
828             world_db["Things"][command_tid.id]["T_MEMTHING"].append(memthing)
829
830
831 def setter_map(maptype):
832     """Set selected Thing's map of maptype's int(str_int)-th line to mapline.
833
834     If Thing has no map of maptype yet, initialize it with ' ' bytes first.
835     """
836     @test_Thing_id
837     def helper(str_int, mapline):
838         val = integer_test(str_int, 0, 255)
839         if None != val:
840             if val >= world_db["MAP_LENGTH"]:
841                 print("Illegal value for map line number.")
842             elif len(mapline) != world_db["MAP_LENGTH"]:
843                 print("Map line length is unequal map width.")
844             else:
845                 length = world_db["MAP_LENGTH"]
846                 map = None
847                 if not world_db["Things"][command_tid.id][maptype]:
848                     map = bytearray(b' ' * (length ** 2))
849                 else:
850                     map = world_db["Things"][command_tid.id][maptype]
851                 map[val * length:(val * length) + length] = mapline.encode()
852                 world_db["Things"][command_tid.id][maptype] = map
853     return helper
854
855
856 def setter_tpos(axis):
857     """Generate setter for T_POSX or  T_POSY of selected Thing.
858
859     If world is active, rebuilds animate things' fovmap, player's memory map.
860     """
861     @test_Thing_id
862     def helper(str_int):
863         val = integer_test(str_int, 0, 255)
864         if None != val:
865             if val < world_db["MAP_LENGTH"]:
866                 world_db["Things"][command_tid.id]["T_POS" + axis] = val
867                 if world_db["WORLD_ACTIVE"] \
868                    and world_db["Things"][command_tid.id]["T_LIFEPOINTS"]:
869                     build_fov_map(world_db["Things"][command_tid.id])
870                     if 0 == command_tid.id:
871                         update_map_memory(world_db["Things"][command_tid.id])
872             else:
873                 print("Ignoring: Position is outside of map.")
874     return helper
875
876
877 def command_ttid(id_string):
878     """Set ID of ThingType to manipulate. ID unused? Create new one.
879
880     Default new ThingType's TT_SYMBOL to "?", TT_CORPSE_ID to self, others: 0.
881     """
882     id = id_setter(id_string, "ThingTypes", command_ttid)
883     if None != id:
884         world_db["ThingTypes"][id] = {
885             "TT_NAME": "(none)",
886             "TT_CONSUMABLE": 0,
887             "TT_LIFEPOINTS": 0,
888             "TT_PROLIFERATE": 0,
889             "TT_START_NUMBER": 0,
890             "TT_SYMBOL": "?",
891             "TT_CORPSE_ID": id
892         }
893
894
895 test_ThingType_id = test_for_id_maker(command_ttid, "ThingType")
896
897
898 @test_ThingType_id
899 def command_ttname(name):
900     """Set TT_NAME of selected ThingType."""
901     world_db["ThingTypes"][command_ttid.id]["TT_NAME"] = name
902
903
904 @test_ThingType_id
905 def command_ttsymbol(char):
906     """Set TT_SYMBOL of selected ThingType. """
907     if 1 == len(char):
908         world_db["ThingTypes"][command_ttid.id]["TT_SYMBOL"] = char
909     else:
910         print("Ignoring: Argument must be single character.")
911
912
913 @test_ThingType_id
914 def command_ttcorpseid(str_int):
915     """Set TT_CORPSE_ID of selected ThingType."""
916     val = integer_test(str_int, 0, 255)
917     if None != val:
918         if val in world_db["ThingTypes"]:
919             world_db["ThingTypes"][command_ttid.id]["TT_CORPSE_ID"] = val
920         else:
921             print("Ignoring: Corpse ID belongs to no known ThignType.")
922
923
924 def command_taid(id_string):
925     """Set ID of ThingAction to manipulate. ID unused? Create new one.
926
927     Default new ThingAction's TA_EFFORT to 1, its TA_NAME to "wait".
928     """
929     id = id_setter(id_string, "ThingActions", command_taid, True)
930     if None != id:
931         world_db["ThingActions"][id] = {
932             "TA_EFFORT": 1,
933             "TA_NAME": "wait"
934         }
935
936
937 test_ThingAction_id = test_for_id_maker(command_taid, "ThingAction")
938
939
940 @test_ThingAction_id
941 def command_taname(name):
942     """Set TA_NAME of selected ThingAction.
943
944     The name must match a valid thing action function. If after the name
945     setting no ThingAction with name "wait" remains, call set_world_inactive().
946     """
947     if name == "wait" or name == "move" or name == "use" or name == "drop" \
948        or name == "pick_up":
949         world_db["ThingActions"][command_taid.id]["TA_NAME"] = name
950         if 1 == world_db["WORLD_ACTIVE"]:
951             wait_defined = False
952             for id in world_db["ThingActions"]:
953                 if "wait" == world_db["ThingActions"][id]["TA_NAME"]:
954                     wait_defined = True
955                     break
956             if not wait_defined:
957                 set_world_inactive()
958     else:
959         print("Ignoring: Invalid action name.")
960     # In contrast to the original,naming won't map a function to a ThingAction.
961
962
963 """Commands database.
964
965 Map command start tokens to ([0]) number of expected command arguments, ([1])
966 the command's meta-ness (i.e. is it to be written to the record file, is it to
967 be ignored in replay mode if read from server input file), and ([2]) a function
968 to be called on it.
969 """
970 commands_db = {
971     "QUIT": (0, True, command_quit),
972     "PING": (0, True, command_ping),
973     "THINGS_HERE": (2, True, command_thingshere),
974     "MAKE_WORLD": (1, False, command_makeworld),
975     "SEED_MAP": (1, False, command_seedmap),
976     "SEED_RANDOMNESS": (1, False, setter(None, "SEED_RANDOMNESS",
977                                          0, 4294967295)),
978     "TURN": (1, False, setter(None, "TURN", 0, 65535)),
979     "PLAYER_TYPE": (1, False, setter(None, "PLAYER_TYPE", 0, 255)),
980     "MAP_LENGTH": (1, False, command_maplength),
981     "WORLD_ACTIVE": (1, False, command_worldactive),
982     "TA_ID": (1, False, command_taid),
983     "TA_EFFORT": (1, False, setter("ThingAction", "TA_EFFORT", 0, 255)),
984     "TA_NAME": (1, False, command_taname),
985     "TT_ID": (1, False, command_ttid),
986     "TT_NAME": (1, False, command_ttname),
987     "TT_SYMBOL": (1, False, command_ttsymbol),
988     "TT_CORPSE_ID": (1, False, command_ttcorpseid),
989     "TT_CONSUMABLE": (1, False, setter("ThingType", "TT_CONSUMABLE",
990                                        0, 65535)),
991     "TT_START_NUMBER": (1, False, setter("ThingType", "TT_START_NUMBER",
992                                          0, 255)),
993     "TT_PROLIFERATE": (1, False, setter("ThingType", "TT_PROLIFERATE",
994                                         0, 255)),
995     "TT_LIFEPOINTS": (1, False, setter("ThingType", "TT_LIFEPOINTS", 0, 255)),
996     "T_ID": (1, False, command_tid),
997     "T_ARGUMENT": (1, False, setter("Thing", "T_ARGUMENT", 0, 255)),
998     "T_PROGRESS": (1, False, setter("Thing", "T_PROGRESS", 0, 255)),
999     "T_LIFEPOINTS": (1, False, setter("Thing", "T_LIFEPOINTS", 0, 255)),
1000     "T_SATIATION": (1, False, setter("Thing", "T_SATIATION", -32768, 32767)),
1001     "T_COMMAND": (1, False, command_tcommand),
1002     "T_TYPE": (1, False, command_ttype),
1003     "T_CARRIES": (1, False, command_tcarries),
1004     "T_MEMMAP": (2, False, setter_map("T_MEMMAP")),
1005     "T_MEMDEPTHMAP": (2, False, setter_map("T_MEMDEPTHMAP")),
1006     "T_MEMTHING": (3, False, command_tmemthing),
1007     "T_POSY": (1, False, setter_tpos("Y")),
1008     "T_POSX": (1, False, setter_tpos("X")),
1009     "wait": (0, False, play_commander("wait")),
1010     "move": (1, False, play_commander("move")),
1011     "pick_up": (0, False, play_commander("pick_up")),
1012     "drop": (1, False, play_commander("drop", True)),
1013     "use": (1, False, play_commander("use", True)),
1014 }
1015
1016
1017 """World state database. With sane default values."""
1018 world_db = {
1019     "TURN": 0,
1020     "SEED_MAP": 0,
1021     "SEED_RANDOMNESS": 0,
1022     "PLAYER_TYPE": 0,
1023     "MAP_LENGTH": 64,
1024     "WORLD_ACTIVE": 0,
1025     "ThingActions": {},
1026     "ThingTypes": {},
1027     "Things": {}
1028 }
1029
1030
1031 """File IO database."""
1032 io_db = {
1033     "path_save": "save",
1034     "path_record": "record",
1035     "path_worldconf": "confserver/world",
1036     "path_server": "server/",
1037     "path_in": "server/in",
1038     "path_out": "server/out",
1039     "path_worldstate": "server/worldstate",
1040     "tmp_suffix": "_tmp",
1041     "kicked_by_rival": False,
1042     "worldstate_updateable": False
1043 }
1044
1045
1046 try:
1047     opts = parse_command_line_arguments()
1048     setup_server_io()
1049     if None != opts.replay:
1050         replay_game()
1051     else:
1052         play_game()
1053 except SystemExit as exit:
1054     print("ABORTING: " + exit.args[0])
1055 except:
1056     print("SOMETHING WENT WRONG IN UNEXPECTED WAYS")
1057     raise
1058 finally:
1059     cleanup_server_io()