"""
ComfyUI Manager install script.
Runs build_ext.bat to compile the CUDA extension after clone.
Requires: CUDA Toolkit 12.x, Visual Studio 2022 (Windows) or gcc/nvcc (Linux).
"""
import subprocess
import sys
import os

node_dir = os.path.dirname(os.path.abspath(__file__))

if sys.platform == "win32":
    build_script = os.path.join(node_dir, "build_ext.bat")
    if os.path.exists(build_script):
        print("[FLUXNATION] Building CUDA extension...")
        result = subprocess.run([build_script], cwd=node_dir, shell=True)
        if result.returncode == 0:
            print("[FLUXNATION] Build succeeded.")
        else:
            print("[FLUXNATION] Build failed. Install CUDA Toolkit 12.x and Visual Studio 2022, then run build_ext.bat manually.")
    else:
        print("[FLUXNATION] build_ext.bat not found.")
else:
    print("[FLUXNATION] Linux build: run 'python setup_ext.py install' from the node directory.")