EGameTools/_IDAScripts/ExportClassH/UI.py

import os
import json
import ida_kernwin

from ExportClassH import Config, JSONGen

def SetConfigVars(settings):
    Config.PROJECT_INCLUDES_PATH = settings["PROJECT_INCLUDES_PATH"]
    Config.OUTPUT_PATH = settings["OUTPUT_PATH"]
    Config.LAST_CLICKED_RADIO = settings["LAST_CLICKED_RADIO"]

    Config.HEADER_OUTPUT_PATH = os.path.join(Config.OUTPUT_PATH, "generated")
    Config.CACHE_OUTPUT_PATH = os.path.join(Config.OUTPUT_PATH, "cache")
    Config.PARSED_VARS_CACHE_FILENAME = os.path.join(Config.CACHE_OUTPUT_PATH, "parsedClassVarsByClass.cache")
    Config.PARSED_FUNCS_CACHE_FILENAME = os.path.join(Config.CACHE_OUTPUT_PATH, "parsedFuncsByClass.cache")

# Load settings from file
def LoadConfig():
    if os.path.exists(Config.CONFIG_FILE):
        with open(Config.CONFIG_FILE, "r") as f:
            loadedJson = json.load(f)
            SetConfigVars(loadedJson)
            return loadedJson
    return Config.DEFAULT_CONFIG

# Save settings to file
def SaveConfig(settings = {}):
    if not settings:
        settings = {
            "PROJECT_INCLUDES_PATH": Config.PROJECT_INCLUDES_PATH,
            "OUTPUT_PATH": Config.OUTPUT_PATH,
            "LAST_CLICKED_RADIO": Config.LAST_CLICKED_RADIO
        }

    with open(Config.CONFIG_FILE, "w") as f:
        json.dump(settings, f, indent=4)
    
    SetConfigVars(settings)

class SettingsDialog(ida_kernwin.Form):
    def __init__(self, current_config):
        self.config = current_config
        ida_kernwin.Form.__init__(self, r"""STARTITEM 0
BUTTON YES* Save
BUTTON CANCEL Cancel
Settings

Modify the paths below:

<##Project Includes Path:{i_project_path}>
<##Output Path:{i_output_path}>
""", {
            'i_project_path': ida_kernwin.Form.StringInput(swidth=50, value=self.config["PROJECT_INCLUDES_PATH"]),
            'i_output_path': ida_kernwin.Form.StringInput(swidth=50, value=self.config["OUTPUT_PATH"]),
        })

    def GetValues(self):
        return {
            "PROJECT_INCLUDES_PATH": self.i_project_path.value,
            "OUTPUT_PATH": self.i_output_path.value,
            "LAST_CLICKED_RADIO": Config.LAST_CLICKED_RADIO
        }
    
    def OnFormChange(self, fid):
        return 1  # Required for form functionality

class MainDialog(ida_kernwin.Form):
    def __init__(self):
        ida_kernwin.Form.__init__(self, r"""STARTITEM 0
BUTTON YES* OK
BUTTON CANCEL Cancel
Export Class to C++ Header

{FormChangeCb}

<##Update Project Code:{r_update}>
<##Generate Class Code:{r_generate}>
<##Settings:{r_settings}>{radioGroup}>
""", {
            'radioGroup': ida_kernwin.Form.RadGroupControl(("r_update", "r_generate", "r_settings")),
            'FormChangeCb': ida_kernwin.Form.FormChangeCb(self.OnFormChange),
        })

    def OnFormChange(self, fid):
        return 1  # Required for form functionality

def OpenSettingsDlg():
    """Opens the Settings Dialog and returns to Main Dialog on Cancel"""
    settings = LoadConfig()
    settingsDlg = SettingsDialog(settings)
    settingsDlg.Compile()
    result = settingsDlg.Execute()

    if result == 1:  # Save clicked
        newSettings = settingsDlg.GetValues()
        SaveConfig(newSettings)  # Save new settings to file
        print("[INFO] Settings saved successfully!")

    settingsDlg.Free()
    
    # Always return to the main dialog after closing settings
    OpenMainDlg()

def OpenMainDlg():
    """Reopens the Main Dialog"""
    LoadConfig()

    mainDlg = MainDialog()
    mainDlg.Compile()
    mainDlg.radioGroup.value = Config.LAST_CLICKED_RADIO
    result = mainDlg.Execute()

    if result == 1:  # OK clicked
        selectedOption = mainDlg.radioGroup.value
        Config.LAST_CLICKED_RADIO = selectedOption
        SaveConfig()

        if selectedOption == 0:
            print("[INFO] Update Project Code selected!")
            #ProjectManager.ProcessExistingHeaders()
        elif selectedOption == 1:
            print("[INFO] Generate Class Code selected!")
            # targetClassName = ida_kernwin.ask_str("", 0, "Enter target class name:")
            # if not targetClassName:
            #     print("No target class specified. Aborting.")
            #     return
            #HeaderGen.ExportClassHeader(ClassName(targetClassName))
            JSONGen.GetAllParsedClasses()
        elif selectedOption == 2:
            print("[INFO] Settings selected!")
            OpenSettingsDlg()  # Open settings when selected
    else:
        print("[INFO] User clicked Cancel. No action taken.")

    mainDlg.Free()