import sys
import json
import inspect
import difflib
from slpp import slpp as lua

SEARCH_FOLDER = "D:\\Eagle Dynamics\\DCS World OpenBeta"

sys.path.append("..\..\..\dcs-master\dcs-master")

from dcs.weapons_data import Weapons
from dcs.planes import *
from dcs.helicopters import *
from dcs.liveries.liveryscanner import LiveryScanner

livery_scanner = LiveryScanner()
livery_scanner.scan_dcs_installation(SEARCH_FOLDER)

#print("Liveries found for units:")
#[print(key) for key in livery_scanner.map.keys()]

# Known id mismatches (because reasons, ask ED)
mismatched_ids = {
    "A-10CII": "A-10C_2"
}

# Get the weapons ids from pydcs
weapon_ids = [a for a in dir(Weapons) if not a.startswith('__') and not callable(getattr(Weapons, a))]

# This function get the human readable name of a weapon from its CLSID
def find_weapon_name(clsid):
    for weapon_id in weapon_ids:
        if getattr(Weapons, weapon_id)["clsid"] == clsid:
            return getattr(Weapons, weapon_id)["name"]

# The database file on which to operate is the first standard argument of the call
if len(sys.argv) > 1:
    # Loads the database 
    with open(sys.argv[1]) as f:
        database = json.load(f)
        for unit_name in database:
            database[unit_name]["enabled"] = True

    # Loads the loadout names
    with open('../unitPayloads.lua') as f:
        lines = f.readlines()
        unit_payloads = lua.decode("".join(lines).replace("Olympus.unitPayloads = ", "").replace("\n", "")) 

    # Loads the loadout roles
    with open('payloadRoles.json') as f:
        payloads_roles = json.load(f)
    
    # Loop on all the units in the database
    for unit_name in database:
        try:
            # Get the pydcs Python class for the unit
            cls = getattr(sys.modules[__name__], unit_name.replace("-", "_").replace(" ", "_"))

            # Add the liveries
            liveries = []
            if unit_name in livery_scanner.map:
                liveries = livery_scanner.map[unit_name]
            else:
                if (unit_name in mismatched_ids):
                    found_name = mismatched_ids[unit_name]
                else:
                    lowercase_keys = [key.lower() for key in livery_scanner.map.keys()]
                    res = difflib.get_close_matches(unit_name.lower(), lowercase_keys)
                    found_name = list(livery_scanner.map.keys())[lowercase_keys.index(res[0])]
                print(f"Warning, could not find {unit_name} in liveries list. Best match is {found_name}. Manual check required!")
                liveries = livery_scanner.map[found_name]

            database[unit_name]["liveries"] = {}
            for livery in liveries:
                database[unit_name]["liveries"][livery.id] =  livery.name

            # Create the loadouts table and add the empty loadout for the default task
            database[unit_name]["loadouts"] = []
            empty_loadout = {
                "items": [],
                "enabled": True,
                "code": "",
                "name": "Empty loadout",
                "roles": [cls.task_default.name]
            }
            database[unit_name]["loadouts"].append(empty_loadout)

            # Loop on all the loadouts for that unit
            for payload_name in unit_payloads[unit_name]:
                payload_weapons = {}
                # Get the names of all the weapons in the loadout and count how many there are for each
                for payload_idx in unit_payloads[unit_name][payload_name]:
                    payload_clsid = unit_payloads[unit_name][payload_name][payload_idx]["CLSID"]
                    weapon_name = find_weapon_name(payload_clsid)
                    if weapon_name in payload_weapons:
                        payload_weapons[weapon_name] += 1
                    else:
                        payload_weapons[weapon_name] = 1

                # Get the roles of this loadout. Some are numeric, some are string. Convert them to be all string
                payload_roles = []
                for role in payloads_roles[unit_name][payload_name].values():
                    if isinstance(role, int):
                        for name, obj in inspect.getmembers(task):
                            if inspect.isclass(obj) and issubclass(obj, task.MainTask):
                                if (obj.id == role):
                                    payload_roles.append(obj.name)
                    else:
                        for name, obj in inspect.getmembers(task):
                            if inspect.isclass(obj) and issubclass(obj, task.MainTask):
                                if (name == role):
                                    payload_roles.append(obj.name)

                # Create the loadout structure and append it to the table
                new_payload = {
                    "items": [{"name": weapon_name, "quantity": payload_weapons[weapon_name]} for weapon_name in payload_weapons],
                    "enabled": True,
                    "code": payload_name,
                    "name": payload_name,
                    "roles": payload_roles
                }
                database[unit_name]["loadouts"].append(new_payload)
        except Exception as e:
            print(f"Could not find data for aircraft of type {unit_name}: {e}, skipping...")

    # Dump everything in the database
    with open(sys.argv[1], "w") as f:
        json.dump(database, f, indent=2)

    # Done!
    print("Done!")