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
|
from argparse import ArgumentParser
import ardsmm
import os.path
import platform
import json
import sys
PROG_NAME = os.path.basename(os.path.dirname(__file__))
IS_WINDOWS = platform.platform().startswith("Windows")
INPUT_CONT_MSG = "CTRL-D"
if IS_WINDOWS:
INPUT_CONT_MSG = "enter"
def parse_args():
parser = ArgumentParser(prog=PROG_NAME)
parser.add_argument("-i", "--in-place", action="store_true", help="modify JSON config file in-place")
parser.add_argument("-o", "--output-file", type=str, help="write JSON output to file (default: stdout)")
parser.add_argument("-m", "--mod-file", type=str, help="read mods from file (default: stdin)")
parser.add_argument("-I", "--indent", type=int, default=4, help=f"set JSON indentation level (default: {ardsmm.ArmaConfig.DEFAULT_INDENT})")
parser.add_argument("-V", "--version", action="store_true", help="display version number and exit")
parser.add_argument("configfile", type=str, nargs="?", help="An Arma Reforger dedicated server JSON config")
return parser.parse_args()
def main():
args = parse_args()
input_data = ""
if args.version:
print(ardsmm._version.__version__)
return 0
if not args.configfile:
print("Missing required positional argument: configfile", file=sys.stderr)
return 1
if args.in_place and args.output_file:
print("-i/--in-place and -o/--output-file are mutually exclusive options", file=sys.stderr)
return 1
if args.mod_file:
print(f"Reading Arma Reforger mod list from {args.mod_file}", file=sys.stderr)
if os.path.exists(args.mod_file):
input_data = open(args.mod_file).read()
else:
print(f"Input file not found: {args.mod_file}", file=sys.stderr)
return 1
else:
if sys.stdin.isatty():
print(f"\nPaste Arma Reforger mod list and hit {INPUT_CONT_MSG} to continue...\n",
file=sys.stderr)
if IS_WINDOWS:
while True:
line = sys.stdin.readline()
if not line.rstrip():
break
input_data += line
else:
input_data = sys.stdin.read()
if not input_data:
print("Warning: No mods consumed!", file=sys.stderr)
config = ardsmm.ArmaConfig(args.configfile)
try:
data = json.loads("[" + input_data + "]")
for x in data:
config.append_mod(x)
config.update()
except json.JSONDecodeError as e:
print("Invalid JSON", file=sys.stderr)
print(f"Reason: {e}", file=sys.stderr)
return 1
filename = ""
if args.output_file:
filename = args.output_file
elif args.in_place:
filename = args.configfile
if filename:
print(f"Writing to {filename}", file=sys.stderr)
with open(filename, "w+") as fp:
fp.write(config.to_string(args.indent))
else:
print(config.to_string(args.indent))
return 0
if __name__ == "__main__":
sys.exit(main())
|
