Initial commit

2 files changed, 167 insertions, 0 deletions

diff --git a/ardsmm/__init__.py b/ardsmm/__init__.py
new file mode 100644
index 0000000..cb0b90e
--- /dev/null
+++ b/ardsmm/__init__.py
@@ -0,0 +1,72 @@
+import json
+import sys
+from typing import Any
+from ardsmm._version import __version__, __version_tuple__
+
+
+class ArmaConfigMod:
+    data: dict[Any, Any]
+    MOD_SCHEMA = {
+        "modId": str,
+        "name": str,
+        "version": str,
+    }
+
+    def __init__(self, mod_dict):
+        if isinstance(mod_dict, str):
+            if mod_dict.endswith(","):
+                mod_dict = mod_dict[:-1]
+            self.data = json.loads(mod_dict)
+        else:
+            self.data = mod_dict
+        self.check()
+
+    def check(self):
+        for key, expected_type in self.MOD_SCHEMA.items():
+            if key not in self.data.keys():
+                raise KeyError(f"Mod key '{key}' is missing")
+            elif expected_type is not type(self.data[key]):
+                raise TypeError(f"Mod '{key}' value should be {expected_type}, but got {type(self.data[key])}")
+
+
+class ArmaConfig:
+    mods: list[Any]
+    data: dict[Any, Any]
+    file: str
+    DEFAULT_INDENT = 4
+
+    def __init__(self, configfile):
+        self.file = configfile
+        self.data = {}
+        self.mods = []
+        self.read()
+
+    def read(self):
+        with open(self.file, "r") as fp:
+            self.data = json.load(fp)
+
+        if not self.data.get("game"):
+            self.data["game"] = {}
+            if not self.data["game"].get("mods"):
+                self.data["game"]["mods"]: []
+
+        for mod in self.data["game"]["mods"]:
+            self.mods.append(ArmaConfigMod(mod))
+
+    def to_string(self, indent=DEFAULT_INDENT):
+        return json.dumps(self.data, indent=indent)
+
+    def append_mod(self, s):
+        mod = ArmaConfigMod(s)
+
+        if mod.data["name"] in [x.data["name"] for x in self.mods]:
+            print(f"[Skip  ] {mod.data['name']} exists", file=sys.stderr)
+            return
+        print(f"[Append] {mod.data['name']}", file=sys.stderr)
+        self.mods.append(mod)
+
+    def update(self):
+        self.data["game"]["mods"] = sorted(
+            [x.data for x in self.mods],
+            key=lambda y: y["name"]
+        )
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())