import os
import json
import sys
import re

def generate_game_lists(roms_path):
    def collect_game_data(system_dir, extensions):
        game_data = []
        for root, _, files in os.walk(system_dir):
            for file in files:
                filename = os.path.basename(file)
                extension = filename.split('.')[-1]
                name = '.'.join(filename.split('.')[:-1])
                if extension in extensions:

                    name_cleaned = re.sub(r'\(.*?\)', '', name)
                    name_cleaned = re.sub(r'\[.*?\]', '', name_cleaned)
                    name_cleaned = name_cleaned.strip()

                    name_cleaned = name_cleaned.replace(' ', '_')
                    name_cleaned = name_cleaned.replace('-', '_')
                    name_cleaned = re.sub(r'_+', '_', name_cleaned)
                    name_cleaned = name_cleaned.replace('+', '')
                    name_cleaned = name_cleaned.replace('&', '')
                    name_cleaned = name_cleaned.replace('!', '')
                    name_cleaned = name_cleaned.replace("'", '')
                    name_cleaned = name_cleaned.replace('.', '')
                    name_cleaned_pegasus = name.replace(',_', ',')

                    clean_name = name_cleaned
                    game_img = f"/customimages/emudeck/{clean_name}.jpg"
                    game_info = {
                        "name": clean_name,
                        "filename": os.path.join(root, file),
                        "file": name_cleaned_pegasus,
                        "img": game_img
                    }
                    game_data.append(game_info)
        game_data_sorted = sorted(game_data, key=lambda x: x['name'])
        return game_data_sorted

    roms_dir = roms_path
    valid_system_dirs = []

    for system_dir in os.listdir(roms_dir):
        if system_dir == "wiiu":
            system_dir = "wiiu/roms"
        if system_dir == "xbox360":
            system_dir = "xbox360/roms"
        full_path = os.path.join(roms_dir, system_dir)
        if os.path.isdir(full_path) and not os.path.islink(full_path) and os.path.isfile(os.path.join(full_path, 'metadata.txt')):
            file_count = sum([len(files) for r, d, files in os.walk(full_path)])
            if file_count > 2:
                valid_system_dirs.append(full_path)

    game_list = []

    for system_dir in valid_system_dirs:
        if any(x in system_dir for x in ["/ps3", "/xbox360", "/model2", "/genesiswide", "/mame", "/emulators", "/desktop"]):
            continue

        with open(os.path.join(system_dir, 'metadata.txt')) as f:
            metadata = f.read()
        collection = next((line.split(':')[1].strip() for line in metadata.splitlines() if line.startswith('collection:')), '')
        shortname = next((line.split(':')[1].strip() for line in metadata.splitlines() if line.startswith('shortname:')), '')
        launcher = next((line.split(':')[1].strip() for line in metadata.splitlines() if line.startswith('launch:')), '').replace('"', '\\"')
        extensions = next((line.split(':')[1].strip().replace(',', ' ') for line in metadata.splitlines() if line.startswith('extensions:')), '').split()

        games = collect_game_data(system_dir, extensions)
        system_info = {
            "title": collection,
            "id": shortname,
            "launcher": launcher,
            "games": games
        }
        game_list.append(system_info)
        game_list_sorted = sorted(game_list, key=lambda x: x['title'])


    json_output = json.dumps(game_list_sorted, indent=4)
    home_directory = os.path.expanduser("~")
    output_file = os.path.join(home_directory, 'emudeck', 'cache', 'roms_games.json')
    with open(output_file, 'w') as f:
        f.write(json_output)

roms_path = sys.argv[1]

generate_game_lists(f"{roms_path}")