aboutsummaryrefslogtreecommitdiff
path: root/mcwrapper/__init__.py
blob: dec7194dfaaf12b62df1d3d161106c74a9e140f0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# vim: expandtab ft=python ts=4 sw=4 sts=4:
import signal
import atexit
import argparse

from . import prints
from . import alarm
from .wrapper import MCWrapper
from .mod import MoD

mcserver_wrapper = None
mcserver_mod = None


def __wrapper_atexit__():
    "This is called when wrapper is exiting"
    if mcserver_wrapper is not None:
        mcserver_wrapper.clean()


def __wrapper_toexit__():
    "This function is called when system signalizes that mcwrapper should exit"
    if mcserver_wrapper is not None:
        mcserver_wrapper.stop()


def __signal_term__(_signo, _stack_frame):
    __wrapper_toexit__()


__HELP_DESC__ = """
    This script is executing Minecraft server and reads its output. From output
    is extracted server status and list of online players. And standard input
    can be accessed by fifo file.
    """


def reload():
    "Reloads input files. Currently applicable only on mod."
    if mcserver_mod is not None:
        mcserver_mod.load_mods()


def main():
    "Main function"
    global verbose_level
    parser = argparse.ArgumentParser(description=__HELP_DESC__)
    parser.add_argument('--verbose', '-v', action='count', default=0,
                        help="Increase verbose level of output")
    parser.add_argument('--quiet', '-q', action='count', default=0,
                        help="Decrease verbose level of output")
    parser.add_argument('--status-file', '-s', action='store_true',
                        help="Outputs server status to file \"status\"")
    parser.add_argument('--players-file', '-p', action='store_true',
                        help="""Outputs list of online players to file
                        \"players\" """)
    parser.add_argument('--mod-file', '-m', type=str,
                        help="""Prints periodically random line from
                        given file as message of the day.""")
    parser.add_argument('--mod-time', type=int,
                        help="""Period used for message of the day in
                        seconds. In default 900 (15 minutes).""")
    parser.add_argument('command', nargs=argparse.REMAINDER,
                        help="""Command to be executed to start Minecraft
                        server.""")
    args = parser.parse_args()

    prints.verbose_level = args.verbose - args.quiet
    command = args.command
    sfile = args.status_file
    pfile = args.players_file
    mod_file = args.mod_file
    mod_time = args.mod_time

    if not command:
        parser.print_help()
        return
    # Just small hack to not open minecraft server gui
    if 'nogui' not in command:
        command.append('nogui')

    alarm.init()
    signal.signal(signal.SIGUSR1, reload)
    signal.signal(signal.SIGUSR2, reload)  # probably can be used for something else in future

    global mcserver_wrapper
    mcserver_wrapper = MCWrapper(command, pfile, sfile)
    signal.signal(signal.SIGTERM, __signal_term__)
    signal.signal(signal.SIGINT, __signal_term__)
    atexit.register(__wrapper_atexit__)

    mcserver_wrapper.start()
    if mod_file is not None:
        global mcserver_mod
        mcserver_mod = MoD(mcserver_wrapper, mod_file, mod_time or 900)
    mcserver_wrapper.process.wait()

if __name__ == '__main__':
    main()