From a91fb0e0988c0bb8db8cced00a095ad69c1c75bd Mon Sep 17 00:00:00 2001
From: cyber-dream <gregory_brzezinski@protonmail.com>
Date: Tue, 2 Aug 2022 06:10:24 +0300
Subject: [PATCH] Initial script

---
 main.py | 121 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 121 insertions(+)
 create mode 100644 main.py

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..57847e9
--- /dev/null
+++ b/main.py
@@ -0,0 +1,121 @@
+import os
+import sys
+import hashlib
+import json
+from pprint import pprint
+from configparser import ConfigParser
+
+
+def calculate_md5(file_path):
+    hash_md5 = hashlib.md5()
+    with open(file_path, "rb") as f:
+        for chunk in iter(lambda: f.read(4096), b""):
+            hash_md5.update(chunk)
+    return hash_md5.hexdigest()
+
+
+def read_config(config_path):
+    config = ConfigParser()
+    config.read(config_path)
+    return config
+
+
+def temp_forge_json():
+    with open('temp_forge.json') as json_file:
+        data = json.load(json_file)
+    return data
+
+
+def parse_servers_config(config: ConfigParser):
+    servers = []
+    for section in config.sections():
+        if section.split(' ')[0] == 'Server' or section.split(' ')[0] == 'server':
+            new_server = {'id': (config.get(section, 'id')),
+                          'name': (config.get(section, 'name')),
+                          'description': (config.get(section, 'description')),
+                          'icon': (config.get(section, 'icon')),
+                          'version': (config.get(section, 'version')),
+                          'address': (config.get(section, 'address')),
+                          'minecraftVersion': (config.get(section, 'minecraftVersion')),
+                          'discord_shortId': (config.get(section, 'discord_shortId')),
+                          'discord_largeImageText': (config.get(section, 'discord_largeImageText')),
+                          'discord_largeImageKey': (config.get(section, 'discord_largeImageKey')),
+                          'mainServer': (config.get(section, 'mainServer')),
+                          'autoconnect': (config.get(section, 'autoconnect')),
+                          '_path': (config.get(section, 'path')),
+                          '_download_url': (config.get(section, 'download_url')),
+                          'modules': generate_mod_list(config.get(section, 'path'),
+                                                       config.get(section, 'download_url'))}
+            servers.append(new_server)
+    return servers
+
+
+def scan_mods(path):
+    f = []
+    for (dirpath, dirnames, filenames) in os.walk(path):
+        f.extend(filenames)
+        break
+    return f
+
+
+def generate_mod_list(server_path, download_url):
+    mods_file_list = scan_mods(os.path.join(server_path, 'mods'))
+
+    mod_list = []
+    mod_list.append(temp_forge_json())
+    for mod in mods_file_list:
+        path_to_mod = os.path.join(server_path, 'mods', mod)
+        mod = mod[:-4]
+        parsed_mod = mod.split('_')
+        mod_id = parsed_mod[0]
+        parsed_mod[0] = parsed_mod[0].split(':')
+        parsed_mod[1] = parsed_mod[1].replace('-', ' ')
+
+        is_mod_required = True
+        if parsed_mod[-1] is not None:
+            if parsed_mod[-1] == 'ServerSide':
+                continue
+            if parsed_mod[-1] == 'Optional':
+                is_mod_required = False
+        new_mod = {
+            'id': mod_id,
+            'name': parsed_mod[1],
+            'type': 'ForgeMod',
+            'required': is_mod_required,
+            'artifact': {'size': sys.getsizeof(path_to_mod),
+                         'MD5': calculate_md5(path_to_mod),
+                         'url': download_url + mod + '.jar'
+                         }
+        }
+        mod_list.append(new_mod)
+    return mod_list
+
+
+def generate_json(config: ConfigParser):
+    new_distribution = {
+        'version': (config.get('Basic', 'version')),
+        'discord': {
+            'clientId': (config.get('Basic', 'version')),
+            'smallImageText': (config.get('Basic', 'discord_smallImageText')),
+            'smallImageKey': (config.get('Basic', 'dicord_smallImageKey'))
+        },
+        'java': {
+            'oracle': (config.get('Basic', 'java'))
+        },
+        'rss': (config.get('Basic', 'rss')),
+        'servers': parse_servers_config(config)
+    }
+    return new_distribution
+
+
+if __name__ == '__main__':
+    temp_forge_json()
+    config = read_config(sys.argv[1])
+    # servers_list = parse_servers_config(config)
+    # mod_list = generate_mod_list(servers_list)
+    distribution = generate_json(config)
+
+    pprint(distribution)
+
+    with open('distribution.json', 'w', encoding='utf-8') as f:
+        json.dump(distribution, f, ensure_ascii=False, indent=4)