plomlompom.com Git - plomrogue/blob - roguelike-client

   1 #!/usr/bin/python3
   2
   3 # This file is part of PlomRogue. PlomRogue is licensed under the GPL version 3
   4 # or any later version. For details on its copyright, license, and warranties,
   5 # see the file NOTICE in the root directory of the PlomRogue source package.
   6
   7
   8 import curses
   9 import os
  10 import signal
  11 import time
  12
  13 from client.config.world_data import world_data
  14 from client.config.io import io
  15 from client.config.commands import commands
  16 from client.window_management import redraw_windows, set_windows, draw_screen, \
  17                                      stdscr
  18 from client.query_mapcell import query_mapcell
  19
  20
  21 message_queue = {
  22     "open_end": False,
  23     "messages": []
  24 }
  25
  26
  27 def read_worldstate():
  28     global redraw_windows
  29     if not os.access(io["path_worldstate"], os.F_OK):
  30         msg = "No world state file found at " + io["path_worldstate"] + "."
  31         raise SystemExit(msg)
  32     read_anew = False
  33     worldstate_file = open(io["path_worldstate"], "r")
  34     turn_string = worldstate_file.readline()
  35     if int(turn_string) != world_data["turn"]:
  36         read_anew = True
  37     if not read_anew: # In rare cases, world may change, but not turn number.
  38         mtime = os.stat(io["path_worldstate"])
  39         if mtime != read_worldstate.last_checked_mtime:
  40             read_worldstate.last_checked_mtime = mtime
  41             read_anew = True
  42     if read_anew:
  43         # TODO: Hardcode order of necessary fields, ensure order dependencies.
  44         redraw_windows = True
  45         world_data["turn"] = int(turn_string)
  46         for entry in io["worldstate_read_order"]:
  47             if entry[1] == "int":
  48                 if 2 == len(entry):
  49                     world_data[entry[0]] = int(worldstate_file.readline())
  50                 elif 3 == len(entry):
  51                     world_data[entry[0]][entry[2]] = \
  52                             int(worldstate_file.readline())
  53             elif entry[1] == "lines":
  54                 world_data[entry[0]] = []
  55                 while True:
  56                     line = worldstate_file.readline().replace("\n", "")
  57                     if line == '%':
  58                         break
  59                     world_data[entry[0]] += [line]
  60             elif entry[1] == "map":
  61                 world_data[entry[0]] = ""
  62                 for i in range(world_data["map_size"]):
  63                     line = worldstate_file.readline().replace("\n", "")
  64                     world_data[entry[0]] += line
  65         if not world_data["look_mode"]:
  66             world_data["map_center"] = world_data["avatar_position"][:]
  67     worldstate_file.close()
  68 read_worldstate.last_checked_mtime = -1
  69
  70
  71 def read_message_queue():
  72     global redraw_windows
  73     while (len(message_queue["messages"]) > 1
  74         or (len(message_queue["messages"]) == 1
  75             and not message_queue["open_end"])):
  76         message = message_queue["messages"].pop(0)
  77         if message == "THINGS_HERE START":
  78             read_message_queue.parse_thingshere = True
  79             world_data["look"] = []
  80         elif message == "THINGS_HERE END":
  81             read_message_queue.parse_thingshere = False
  82             if world_data["look"] == []:
  83                 world_data["look"] = ["(none known)"]
  84             redraw_windows = True
  85         elif read_message_queue.parse_thingshere:
  86             world_data["look"] += [message]
  87         elif message[0:4] == "LOG ":
  88             world_data["log"] += [message[4:]]
  89             redraw_windows = True
  90         elif message == "WORLD_UPDATED":
  91             query_mapcell()
  92         elif message == "PLUGIN":
  93             str_plugin = message[7:]
  94             if (str_plugin.replace("_", "").isalnum()
  95                 and os.access("plugins/client/" + str_plugin, os.F_OK)):
  96                 exec(open("plugins/client/" + str_plugin).read())
  97                 return
  98             raise SystemExit("Invalid plugin load path in message: " + message)
  99 read_message_queue.parse_thingshere = False
 100
 101
 102 def cursed_main(stdscr):
 103     global redraw_windows
 104
 105     def ping_test():
 106         half_wait_time = 5
 107         if len(new_data_from_server) > 0:
 108             ping_test.sent = False
 109         elif ping_test.wait_start + half_wait_time < time.time():
 110             if not ping_test.sent:
 111                 io["file_out"].write("PING\n")
 112                 io["file_out"].flush()
 113                 ping_test.sent = True
 114                 ping_test.wait_start = time.time()
 115             elif ping_test.sent:
 116                 raise SystemExit("Server not answering anymore.")
 117     ping_test.wait_start = 0
 118
 119     def read_into_message_queue():
 120         if new_data_from_server == "":
 121             return
 122         new_open_end = False
 123         if new_data_from_server[-1] is not "\n":
 124             new_open_end = True
 125         new_messages = new_data_from_server.splitlines()
 126         if message_queue["open_end"]:
 127             message_queue["messages"][-1] += new_messages[0]
 128             del new_messages[0]
 129         message_queue["messages"] += new_messages
 130         if new_open_end:
 131             message_queue["open_end"] = True
 132
 133     curses.noecho()
 134     curses.curs_set(False)
 135     signal.signal(signal.SIGWINCH,
 136         lambda ignore_1, ignore_2: set_windows())
 137     set_windows()
 138     delay = 1
 139     while True:
 140         stdscr.timeout(int(delay))
 141         if delay < 1000:
 142             delay = delay * 1.1
 143         if redraw_windows:
 144             delay = 1
 145             draw_screen()
 146             redraw_windows = False
 147         char = stdscr.getch()
 148         if char >= 0:
 149             char = chr(char)
 150             if char in commands:
 151                 if len(commands[char]) == 1 or not world_data["look_mode"]:
 152                     commands[char][0]()
 153                 else:
 154                     commands[char][1]()
 155                 redraw_windows = True
 156         new_data_from_server = io["file_in"].read()
 157         ping_test()
 158         read_into_message_queue()
 159         read_worldstate()
 160         read_message_queue()
 161
 162
 163 try:
 164     if (not os.access(io["path_out"], os.F_OK)):
 165         msg = "No server input file found at " + io["path_out"] + "."
 166         raise SystemExit(msg)
 167     io["file_out"] = open(io["path_out"], "a")
 168     io["file_in"] = open(io["path_in"], "r")
 169     curses.wrapper(cursed_main)
 170 except SystemExit as exit:
 171     print("ABORTING: " + exit.args[0])
 172 except:
 173     print("SOMETHING WENT WRONG IN UNEXPECTED WAYS")
 174     raise
 175 finally:
 176     if "file_out" in io:
 177         io["file_out"].close()
 178     if "file_in" in io:
 179         io["file_in"].close()