import os
import sys
from pymxs import runtime as rt

def install_plugin():
    # Caminhos de instalação local
    user_scripts_dir = rt.getDir(rt.name("userScripts"))
    plugin_dir = os.path.join(user_scripts_dir, "VR4Life_Plugin").replace("\\", "/")
    
    # Adiciona ao path para evitar o erro 'ModuleNotFoundError' que apareceu nos logs
    if plugin_dir not in sys.path:
        sys.path.append(plugin_dir)
    
    run_script = os.path.join(plugin_dir, "run_vr4life.py").replace("\\", "/")
    
    # Nomes solicitados para aparecer no topo
    product_name = "1-VR4Life"
    category = "Immerse Games"
    macro_name = "VR4Life_Launcher"

    # 1. Registro da Macro (Base para todas as versões)
    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:
        cui_mgr = rt.cui.getContentManager()
        main_menu_bar = cui_mgr.mainMenuBar
        
        # Verifica se o menu já existe
        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:
            # Cria o menu customizado moderno
            new_menu = cui_mgr.createCustomMenu(product_name)
            new_menu.addActionItem(macro_name, category)
            
            # Posição 1 coloca logo após o menu 'File'
            main_menu_bar.addItem(new_menu, 1)
            print(f"[LOG] {product_name} instalado via CUI.")
            
    except AttributeError:
        # 3. Lógica para 3ds Max 2024 e anteriores (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 via menuMan.")

    # Abre a ferramenta automaticamente após instalar
    if os.path.exists(run_script):
        rt.python.ExecuteFile(run_script)

    rt.messageBox(f"Instalação do '{product_name}' finalizada!", title=product_name)

if __name__ == "__main__":
    install_plugin()