import os
from pymxs import runtime as rt

def install_plugin():
    # Define o caminho do run_vr4life.py que foi acabado de transferir
    user_scripts_dir = rt.getDir(rt.name("userScripts"))
    plugin_dir = os.path.join(user_scripts_dir, "VR4Life_Plugin").replace("\\", "/")
    run_script = os.path.join(plugin_dir, "run_vr4life.py").replace("\\", "/")
    
    # 1. Cria a Action (MacroScript) que funciona em TODAS as versões.
    # Isto regista a ferramenta no sistema para ser usada em toolbars ou atalhos.
    macro_name = "Zombisco_Engine"
    category = "Immerse Games"
    tooltip = "Abrir Motor do Zombisco"
    
    macro_code = f'''
    macroScript {macro_name}
    category:"{category}"
    buttonText:"Zombisco"
    tooltip:"{tooltip}"
    (
        on execute do (
            python.ExecuteFile @"{run_script}"
        )
    )
    '''
    rt.execute(macro_code)
    print(f"[LOG] MacroScript registado com sucesso para todas as versões.")

    # 2. Tenta criar o Menu Superior (Para 3ds Max 2024 e anteriores)
    try:
        main_menu = rt.menuMan.getMainMenuBar()
        menu_name = "Immerse Games"
        
        # Remove o menu se já existir para evitar duplicações
        existing_menu = rt.menuMan.findMenu(menu_name)
        if existing_menu:
            rt.menuMan.unRegisterMenu(existing_menu)
        
        # Cria o novo menu
        new_menu = rt.menuMan.createMenu(menu_name)
        menu_item = rt.menuMan.createActionItem(macro_name, category)
        new_menu.addItem(menu_item, -1)
        
        sub_menu_item = rt.menuMan.createSubMenuItem(menu_name, new_menu)
        main_menu.addItem(sub_menu_item, -1)
        rt.menuMan.updateMenuBar()
        
        print("[LOG] Menu superior criado com sucesso (Método Clássico).")
        rt.messageBox("Instalação do motor do Zombisco concluída com sucesso!\nVerifique o menu 'Immerse Games' no topo do seu 3ds Max.", title="Instalação Concluída")
        
    except AttributeError:
        # Falhou propositadamente porque é o 3ds Max 2025 ou 2026
        print("[LOG] 3ds Max 2025/2026 detetado. O 'menuMan' foi descontinuado.")
        
        # Abre a UI de imediato para entregar as ferramentas à equipa sem atrasos
        if os.path.exists(run_script):
            rt.python.ExecuteFile(run_script)
        
        msg = ("Instalação Concluída!\n\n"
               "No 3ds Max 2025/2026, a Autodesk alterou a forma de criar menus no topo.\n"
               "Para não interromper o seu trabalho, o motor foi aberto de imediato no seu ecrã.\n\n"
               "Para adicionar um botão permanente à sua área de trabalho:\n"
               "1. Vá ao menu Customize > Hotkey Editor (ou Toolbars).\n"
               "2. Procure pela categoria 'Immerse Games'.\n"
               "3. Arraste a ferramenta 'Zombisco' para a sua barra de ferramentas preferida.")
        rt.messageBox(msg, title="Immerse Games")

if __name__ == "__main__":
    install_plugin()