setup.py

#!/usr/bin/env python3
"""
Setup script for Nuke Nodes for ComfyUI
Helps verify installation and dependencies
"""

import importlib
import os
import sys


def check_python_version():
    """Check Python version compatibility"""
    if sys.version_info < (3, 8):
        print("❌ Python 3.8 or higher is required")
        print(f"Current version: {sys.version}")
        return False
    else:
        print(f"✅ Python version: {sys.version}")
        return True


def check_dependencies():
    """Check if required dependencies are installed"""
    dependencies = {"torch": "PyTorch", "numpy": "NumPy", "cv2": "OpenCV"}

    missing = []
    pytorch_missing = False

    for module, name in dependencies.items():
        try:
            mod = importlib.import_module(module)
            version = getattr(mod, "__version__", "unknown")
            print(f"✅ {name}: {version}")
        except ImportError:
            if module == "torch":
                print(f"❌ {name}: Not installed (CRITICAL - ComfyUI requires PyTorch)")
                pytorch_missing = True
            else:
                print(f"❌ {name}: Not installed")
                missing.append(module)

    if pytorch_missing:
        print(
            "⚠️  PyTorch is missing. This usually means ComfyUI is not properly installed."
        )
        print("   Please ensure ComfyUI is working before installing these nodes.")

    return missing


def install_dependencies(missing):
    """Report missing dependencies to the user.

    Dependencies are managed via requirements.txt and installed automatically
    by the ComfyUI Manager. Manual pip install via subprocess is not used.
    """
    if not missing:
        return True

    print(f"\n⚠️  Missing dependencies: {', '.join(missing)}")
    print("   Dependencies are listed in requirements.txt and will be installed")
    print("   automatically by the ComfyUI Manager.")
    print("   If installing manually, run: pip install -r requirements.txt")
    return False


def check_comfyui_structure():
    """Check if we're in the correct ComfyUI structure"""
    current_dir = os.getcwd()

    # Check if we're in a custom_nodes directory
    if "custom_nodes" in current_dir:
        print("✅ Located in ComfyUI custom_nodes directory")
        return True
    else:
        print("⚠️  Not in ComfyUI custom_nodes directory")
        print("This package should be installed in ComfyUI/custom_nodes/")
        return False


def verify_node_files():
    """Verify all required node files exist"""
    required_files = [
        "__init__.py",
        "utils.py",
        "merge_nodes.py",
        "grade_nodes.py",
        "transform_nodes.py",
        "blur_nodes.py",
    ]

    missing_files = []

    for file in required_files:
        if os.path.exists(file):
            print(f"✅ {file}")
        else:
            print(f"❌ {file}")
            missing_files.append(file)

    return len(missing_files) == 0


def test_imports():
    """Test importing the node modules"""
    modules = ["merge_nodes", "grade_nodes", "transform_nodes", "blur_nodes"]

    success = True

    for module in modules:
        try:
            importlib.import_module(module)
            print(f"✅ Import {module}")
        except Exception as e:
            print(f"❌ Import {module}: {e}")
            success = False

    return success


def show_version_info():
    """Show version information"""
    try:
        from version import get_full_version

        version_info = get_full_version()
        print(f"📦 Package: {version_info['title']} v{version_info['version']}")
        print(f"📝 Description: {version_info['description']}")
        print(f"👨‍💻 Author: {version_info['author']}")
        print(f"📅 Release Date: {version_info['release_date']}")
        print(f"📄 License: {version_info['license']}")
        if version_info["is_stable"]:
            print("✅ Status: Stable Release")
        elif version_info["is_beta"]:
            print("🧪 Status: Beta Release")
        else:
            print("🚧 Status: Development Version")
        print(f"🏷️  Node Categories: {', '.join(version_info['node_categories'])}")
    except Exception as e:
        print(f"⚠️  Could not load version info: {e}")
        print("📦 Package: Nuke Nodes for ComfyUI")


def main():
    """Main setup function"""
    print("🔧 Nuke Nodes for ComfyUI - Setup & Verification")
    print("=" * 50)

    # Show version information
    print()
    show_version_info()
    print()

    # Check Python version
    if not check_python_version():
        return False

    print("\n📋 Checking dependencies...")
    missing = check_dependencies()

    if missing:
        install_dependencies(missing)

    print("\n📁 Checking file structure...")
    check_comfyui_structure()

    print("\n📄 Verifying node files...")
    if not verify_node_files():
        print("❌ Some required files are missing")
        return False

    print("\n🔍 Testing imports...")
    if not test_imports():
        print("❌ Some modules failed to import")
        return False

    print("\n✅ Setup complete! Restart ComfyUI to use the nodes.")
    print("\nNodes will appear in the menu under 'Nuke' category:")
    print("- Nuke/Merge")
    print("- Nuke/Color")
    print("- Nuke/Transform")
    print("- Nuke/Filter")

    return True


if __name__ == "__main__":
    success = main()
    sys.exit(0 if success else 1)