import os
import sys
from pymxs import runtime as rt

def log(msg):
    """Função auxiliar para imprimir logs claros no Listener (F11)"""
    print(f"[VR4Life Install Log] {msg}")

def install_plugin():
    rt.clearListener()
    log("Iniciando fase final de instalação e configuração de interface...")

    # 1. Configuração de Caminhos
    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)
        log(f"Pasta adicionada ao Path do Python: {plugin_dir}")
    
    run_script = os.path.join(plugin_dir, "run_vr4life.py").replace("\\", "/")
    log(f"Caminho do script de execução: {run_script}")

    # Nome solicitado para o topo da lista
    product_name = "1-VR4Life"
    category = "Immerse Games"
    macro_name = "VR4Life_Launcher"

    # 2. Registo da Macro (MacroScript)
    log(f"A registar MacroScript '{macro_name}' na categoria '{category}'...")
    macro_code = f'''
    macroScript {macro_name}
    category:"{category}"
    buttonText:"{product_name}"
    tooltip:"Abrir {product_name}"
    (
        on execute do ( python.ExecuteFile @"{run_script}" )
    )
    '''
    try:
        rt.execute(macro_code)
        log("MacroScript registado com sucesso.")
    except Exception as e:
        log(f"ERRO ao registar MacroScript: {str(e)}")

    # 3. Lógica de Menu para 3ds Max 2025/2026 (Sistema CUI)
    try:
        log("A tentar detetar sistema CUI (3ds Max 2025+)...")
        cui_mgr = rt.cui.getContentManager()
        main_menu_bar = cui_mgr.mainMenuBar
        log("Sistema CUI detetado.")
        
        # Verificação de duplicados
        menu_exists = False
        for i in range(main_menu_bar.numItems):
            if main_menu_bar.getItem(i).displayText == product_name:
                menu_exists = True
                log("O menu já existe na barra CUI. A ignorar criação para evitar duplicados.")
                break
        
        if not menu_exists:
            log(f"A criar novo menu '{product_name}' no 3ds Max 2026...")
            new_menu = cui_mgr.createCustomMenu(product_name)
            new_menu.addActionItem(macro_name, category)
            
            # Insere logo após o menu 'File'
            main_menu_bar.addItem(new_menu, 1)
            log("Menu CUI criado e inserido na posição 1.")
            
    except AttributeError:
        # 4. Lógica Legada para 3ds Max 2024 e anteriores
        log("Sistema CUI não encontrado. A mudar para modo legado (menuMan)...")
        try:
            main_menu = rt.menuMan.getMainMenuBar()
            existing_menu = rt.menuMan.findMenu(product_name)
            
            if existing_menu:
                rt.menuMan.unRegisterMenu(existing_menu)
                log("Menu antigo removido para atualização.")
                
            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()
            log("Menu legado criado com sucesso via menuMan.")
        except Exception as e:
            log(f"ERRO FATAL ao criar menu legado: {str(e)}")

    # 5. Execução Automática
    if os.path.exists(run_script):
        log("A abrir a interface do VR4Life pela primeira vez...")
        rt.python.ExecuteFile(run_script)
    else:
        log(f"AVISO: O ficheiro {run_script} não foi encontrado para execução imediata.")

    log("PROCESSO DE INSTALAÇÃO FINALIZADO.")
    rt.messageBox(f"Instalação do '{product_name}' concluída!\nConsulte o log (F11) em caso de dúvidas.", title=product_name)

if __name__ == "__main__":
    install_plugin()