Initial commit

1 files changed, 95 insertions, 0 deletions

diff --git a/ardsmm/__main__.py b/ardsmm/__main__.py
new file mode 100644
index 0000000..cb24c18
--- /dev/null
+++ b/ardsmm/__main__.py
@@ -0,0 +1,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())