diff options
| -rwxr-xr-x | mcwrapper | 227 |
1 files changed, 227 insertions, 0 deletions
diff --git a/mcwrapper b/mcwrapper new file mode 100755 index 0000000..b3cf223 --- /dev/null +++ b/mcwrapper @@ -0,0 +1,227 @@ +#!/usr/bin/env python3 +import os +import sys +import re +import subprocess +import datetime +import signal +import time +from threading import Thread + +__log_prefix__ = '/home/minecraft/log/' +__folderprefix__ = '/dev/shm/' +__folder__ = __folderprefix__ +__players__ = 'players' +__status__ = 'status' +__pipe__ = 'input_pipe' +__log__ = 'none.log' + +__STATUSSTRINGS__ = { 0: "Not running", + 1: "Starting", + 2: "Running", + 3: "Stopping" + } + +def __setfiles__(identifier): + global __folder__ + global __players__ + global __status__ + global __pipe__ + global __log__ + __folder__ = __folderprefix__ + 'mcwrapper_' + identifier + '/' + __players__ = __folder__ + 'players' + __status__ = __folder__ + 'status' + __pipe__ = __folder__ + 'input_pipe' + __log__ = __log_prefix__ + identifier + '/' + \ + datetime.datetime.now().strftime('%y-%m-%d-%H-%M-%S') + +def __user_join__(username): + print("User '" + username + "' joined server.") + with open(__players__, 'a') as f: + f.write(username + '\n') + +def __user_leave__(username): + print("User '" + username + "' left server.") + players = set() + with open(__players__) as f: + for line in f: + pl = line.rstrip() + if pl != username: + players.add(line.rstrip()) + with open(__players__, 'w') as f: + f.writelines(players) + if players: + f.write('\n') + + +def __server_init__(identifier): + __setfiles__(identifier) + __server_clean__() # Clean before execute + print("Wrapper initializing with identifier: " + identifier) + try: + os.mkdir(__folder__) + except Exception: + pass + if os.path.isfile(__status__): + sys.exit("Server status file already exists. Is another wrapper running?") + with open(__status__, 'w') as f: + f.write(__STATUSSTRINGS__[1]) + os.mkfifo(__pipe__, 0o640) + +def __server_start__(): + print("Server start.") + with open(__status__, 'w') as f: + f.write(__STATUSSTRINGS__[2]) + pass + +def __server_stop__(): + print("Server stop.") + with open(__status__, 'w') as f: + f.write(__STATUSSTRINGS__[3]) + pass + +def __server_clean__(): + print("Wrapper clean.") + try: + os.remove(__players__) + except Exception: + pass + try: + os.remove(__status__) + except Exception: + pass + try: + os.remove(__pipe__) + except Exception: + pass + +def __parse_line__(line): + if ': Done' in line: + __server_start__() + elif ': Stopping the server' in line: + __server_stop__() + elif 'logged in with entity id' in line: + name = line[len('[00:00:00] [Server thread/INFO]: '):] + name = name[:name.index('[')] + __user_join__(name) + elif 'left the game' in line: + name = line[len('[00:00:00] [Server thread/INFO]: '):] + name = name[:name.index(' ')] + __user_leave__(name) + +################################################################################# + +class __InputThread__(Thread): + def __init__(self, pipein, pipeprocess): + Thread.__init__(self, name='InputThread') + self.pipein = pipein + self.pipeprocess = pipeprocess + self.stopread = False + def stopexec(self): + self.stopread = True + def wake(self): + with open(__pipe__, 'w') as f: + f.write("\n") + f.flush() + def run(self): + with open(self.pipein, 'r') as p: + while not self.stopread: + ln = p.readline() + if ln: + print("Input: " + ln, end="") + self.pipeprocess.write(bytes(ln, sys.getdefaultencoding())) + self.pipeprocess.flush() + else: + time.sleep(1) + try: + os.remove(__pipe__) + except Exception: + pass + +def __server_send_stop__(): + global prc + prc.stdin.write(bytes("/stop\n", sys.getdefaultencoding())) + prc.stdin.flush() + +def mcexec(identifier, cmd): + """Executes cmd and parses output for server status changes. + + cmd - List of program and arguments to be executed. + logfile - Path in string. To entered file will be printed out full process output. + """ + global prc + __server_init__(identifier) + try: + os.makedirs(__log_prefix__ + identifier) + except Exception: + pass + prc = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, + stderr=subprocess.STDOUT) + inputThread = __InputThread__(__pipe__, prc.stdin) + inputThread.start() + inputThread.wake() # Input thread is stack in waiting for first line + for linen in prc.stdout: + line = linen.decode(sys.getdefaultencoding()) + with open(__log__, 'a') as flg: + flg.write(line) + __parse_line__(line.rstrip()) + inputThread.stopexec() + __server_clean__() + +################################################################################# + +def __get_status__(identifier): + if not os.path.isfile(__status__): + return 0 + else: + with open(__status__) as f: + status = f.readline().rstrip() + for key, val in __STATUSSTRINGS__.items(): + if val == status: + return key + +def server_stop(identifier): + __setfiles__(identifier) + status = __get_status__(identifier) + if status == 0: + sys.exit("Such server is not running") + elif status == 1: + print("Server is starting. Waiting...") + while __get_status__(identifier) != 2: + time.sleep(1) + elif status == 3: + sys.exit("Server already stops") + with open(__pipe__, 'w') as f: + f.write("/stop\n") + f.flush() + while os.path.isfile(__status__): + pass + +def server_say(identifier, message): + __setfiles__(identifier) + if __get_status__(identifier) != 2: + sys.exit("Server is not running") + with open(__pipe__, 'w') as f: + f.write("/say " + ' '.join(map(str, message))) + f.flush() + +################################################################################# + +def __signal_term__(_signo, _stack_frame): + __server_send_stop__() + +if __name__ == '__main__': + todo = sys.argv[1] + identifier = sys.argv[2] + if todo == 'start': + signal.signal(signal.SIGTERM, __signal_term__) + cmd = sys.argv[3:] + mcexec(identifier, cmd) + __server_clean__() + elif todo == 'stop': + server_stop(identifier) + elif todo == 'say': + message = sys.argv[3:] + server_say(identifier, message) + else: + print("unknown action: " + todo) |