import os
import sys
from pymxs import runtime as rt

def install_plugin():
    # Configurações de Caminho
    user_scripts_dir = rt.getDir(rt.name("userScripts"))
    plugin_dir = os.path.join(user_scripts_dir, "VR4Life_Plugin").replace("\\", "/")
    
    if plugin_dir not in sys.path:
        sys.path.append(plugin_dir)
    
    run_script = os.path.join(plugin_dir, "run_vr4life.py").replace("\\", "/")
    
    # Nome solicitado: "1-VR4Life" para o topo
    product_name = "1-VR4Life"
    category = "Immerse Games"
    macro_name = "VR4Life_Launcher"

    # 1. Registo da Macro (Essencial para o 2026 "enxergar" o comando)
    macro_code = f'''
    macroScript {macro_name}
    category:"{category}"
    buttonText:"{product_name}"
    tooltip:"Abrir {product_name}"
    (
        on execute do ( python.ExecuteFile @"{run_script}" )
    )
    '''
    rt.execute(macro_code)

    # 2. Lógica para 3ds Max 2025 e 2026 (Sistema CUI Moderno)
    try:
        # Tenta aceder ao Gestor de Conteúdo Moderno
        cui_mgr = rt.cui.getContentManager()
        main_menu_bar = cui_mgr.mainMenuBar
        
        # Verifica se já existe para não duplicar no 2026
        menu_exists = False
        for i in range(main_menu_bar.numItems):
            if main_menu_bar.getItem(i).displayText == product_name:
                menu_exists = True
                break
        
        if not menu_exists:
            # No 2026, criamos um menu customizado e forçamos a atualização do ficheiro MNX
            new_menu = cui_mgr.createCustomMenu(product_name)
            new_menu.addActionItem(macro_name, category)
            
            # Posição 1 coloca-o logo após o menu "File"
            main_menu_bar.addItem(new_menu, 1)
            
            print(f"[LOG] {product_name} instalado com sucesso via CUI (2025/2026).")
            
    except AttributeError:
        # 3. Lógica para 3ds Max 2024 e anteriores (Sistema menuMan)
        main_menu = rt.menuMan.getMainMenuBar()
        existing_menu = rt.menuMan.findMenu(product_name)
        
        if existing_menu:
            rt.menuMan.unRegisterMenu(existing_menu)
            
        new_menu = rt.menuMan.createMenu(product_name)
        menu_item = rt.menuMan.createActionItem(macro_name, category)
        new_menu.addItem(menu_item, -1)
        
        sub_menu_item = rt.menuMan.createSubMenuItem(product_name, new_menu)
        main_menu.addItem(sub_menu_item, 1)
        rt.menuMan.updateMenuBar()
        print(f"[LOG] {product_name} instalado com sucesso via menuMan (Legacy).")

    # Força a abertura da ferramenta na primeira instalação para o artista já começar a trabalhar no Zombisco
    if os.path.exists(run_script):
        rt.python.ExecuteFile(run_script)

    rt.messageBox(f"O menu '{product_name}' foi adicionado com sucesso!", title=product_name)

if __name__ == "__main__":
    install_plugin()