From c1bc6084427331535e9084db3ee1770e79fc8f66 Mon Sep 17 00:00:00 2001
From: ppkantorski <ppkantorski@berkeley.edu>
Date: Sun, 7 Jul 2024 20:22:33 -0700
Subject: [PATCH] Create format_repo4.py

---
 scripts/theboy181/format_repo4.py | 118 ++++++++++++++++++++++++++++++
 1 file changed, 118 insertions(+)
 create mode 100644 scripts/theboy181/format_repo4.py

diff --git a/scripts/theboy181/format_repo4.py b/scripts/theboy181/format_repo4.py
new file mode 100644
index 0000000..59e5b50
--- /dev/null
+++ b/scripts/theboy181/format_repo4.py
@@ -0,0 +1,118 @@
+import os
+import shutil
+import re
+import sys
+import rarfile
+
+def transform_game_name(game_name):
+    # Move 'The' to the front if it exists
+    if ', The' in game_name:
+        parts = game_name.split(', The')
+        game_name = f"The {parts[0]}{parts[1]}"
+    
+    # Remove ' - ' from the game name
+    game_name = game_name.replace(' - ', ' ')
+    
+    # Remove any '/'
+    game_name = game_name.replace(':', '')
+
+    print("Game Name "+game_name)
+    return game_name
+
+def extract_rar_files(folder_path):
+    for root, dirs, files in os.walk(folder_path):
+        for file in files:
+            if file.endswith('.rar'):
+                rar_path = os.path.join(root, file)
+                with rarfile.RarFile(rar_path) as rf:
+                    rf.extractall(root)
+                print(f"Extracted {rar_path}")
+
+def get_game_name_and_mod_name(path, root_dir):
+    relative_path = os.path.relpath(path, root_dir)
+    parts = relative_path.split(os.sep)
+    
+    # The first part is the game name
+    game_name = parts[0]
+    
+    # Remove any parts within square brackets
+    game_name = re.sub(r'\[.*?\]', '', game_name).strip()
+    
+    # Transform the game name
+    game_name = transform_game_name(game_name)
+    
+    # Check for country-specific folders and adjust game name accordingly
+    country = None
+    for part in parts[1:]:
+        if re.search(r'\[.*?\]', part):
+            country = re.sub(r'\[.*?\]', '', part).strip()
+            parts.remove(part)
+            break
+
+    if country:
+        game_name = f"{game_name} ({country})"
+    else:
+        game_name = game_name.strip()
+
+    # Handle Aspect Ratio mods specifically
+    if 'Aspect Ratio' in relative_path:
+        aspect_ratio = os.path.basename(os.path.dirname(path)).replace("'", ".").replace("`", ".")
+        mod_name = f'Aspect Ratio {aspect_ratio}'
+    else:
+        # Handle versioned mod names like "Disable Fog v1"
+        if re.search(r' v\d+', parts[-1]):
+            mod_name = parts[-2] + " " + parts[-1]
+        else:
+            # For other cases, use the immediate parent directory name
+            mod_name = parts[-2]
+    
+    # Replace ` and ' with . in the mod name
+    mod_name = mod_name.replace("'", ".").replace("`", ".").replace("21-9", "21.9")
+    # Remove any '/' from mod name
+    mod_name = mod_name.replace(':', '')
+
+    # For handling the outlier mod (missing name) on the repo
+    mod_name = mod_name.replace("Trailblazers", "4K")
+
+    return game_name, mod_name
+
+def create_formatted_structure(folder_path):
+    extract_rar_files(folder_path)
+    
+    formatted_path = os.path.join(folder_path, 'formatted')
+    if not os.path.exists(formatted_path):
+        os.makedirs(formatted_path)
+    #print(f"Creating formatted structure at: {formatted_path}")
+
+    for root, dirs, files in os.walk(folder_path):
+        if 'formatted' in root:
+            continue
+        for file in files:
+            if file.endswith('.pchtxt'):
+                game_name, mod_name = get_game_name_and_mod_name(root, folder_path)
+                
+                version = file.replace('.pchtxt', '').strip()
+                
+                new_dir = os.path.join(formatted_path, f"{game_name} - {mod_name}")
+                
+                if not os.path.exists(new_dir):
+                    os.makedirs(new_dir)
+                    #print(f"Created directory: {new_dir}")
+                
+                source_file = os.path.join(root, file)
+                destination_file = os.path.join(new_dir, f"{version}.pchtxt")
+                
+                shutil.copy(source_file, destination_file)
+                #print(f"Copied {source_file} to {destination_file}")
+
+def main(folder_path):
+    create_formatted_structure(folder_path)
+    print("Files have been organized successfully.")
+
+if __name__ == "__main__":
+    if len(sys.argv) != 2:
+        print("Usage: python format_repo_4.py /path/to/folder/")
+        sys.exit(1)
+    
+    folder_path = sys.argv[1]
+    main(folder_path)