import importlib.util
import subprocess
import os
import re


def install_package(full_name, package_name):
    try:
        spec = importlib.util.find_spec(package_name)
        if spec is None:
            raise ImportError(f"{package_name} 未安装")
    except ImportError:
        print(f"{package_name} 未安装，正在安装...")
        subprocess.check_call(["python", "-m", "pip", "install", full_name])
        print(f"{package_name} 安装完成。")


def check_and_install_packages():
    packages = ["pygtrans", "fake_useragent", "lxml", "openai"]
    for package in packages:
        package_name = re.match(r"^([^\s=<>!]+)", package.strip())
        if package_name:
            install_package(package, package_name.group(1))


# 检查 requirements.txt 文件中的包
check_and_install_packages()


from .nodes.Translate import TranslateNode
from .nodes.JoyTag.JoyTag import JoyTagNode
from .nodes.JoyCaption.JoyCaption import JoyCaptionNode
from .nodes.JoyCaption.JoyCaption import (
    JoyCaptionAlpha2OnlineNode,
    JoyCaptionAlpha1OnlineNode,
)
from .nodes.civitai_prompt import CivitaiPromptNode
from .nodes.image_tools import CustomLatentImageNode, CustomLatentImageSimpleNode
from .nodes.number_tools import FloatSliderNode

NODE_CLASS_MAPPINGS = {
    "Translate": TranslateNode,
    "JoyTag": JoyTagNode,
    "JoyCaption": JoyCaptionNode,
    "JoyCaptionAlpha2Online": JoyCaptionAlpha2OnlineNode,
    "JoyCaptionAlpha1Online": JoyCaptionAlpha1OnlineNode,
    "CivitaiPrompt": CivitaiPromptNode,
    "CustomLatentImage-NYJY": CustomLatentImageNode,
    "CustomLatentImageSimple": CustomLatentImageSimpleNode,
    "FloatSlider-NYJY": FloatSliderNode,
}
NODE_DISPLAY_NAME_MAPPINGS = {
    "Translate": "Translate (NYJY)",
    "JoyTag": "JoyTag (NYJY)",
    "JoyCaption": "JoyCaption (NYJY)",
    "JoyCaptionAlpha2Online": "JoyCaptionAlpha2Online (NYJY)",
    "JoyCaptionAlpha1Online": "JoyCaptionAlphaOneOnline (NYJY)",
    "CivitaiPrompt": "CivitaiPrompt（NYJY）",
    "CustomLatentImage-NYJY": "CustomLatentImage（NYJY）",
    "CustomLatentImageSimple": "CustomLatentImageSimple（NYJY）",
    "FloatSlider-NYJY": "FloatSlider（NYJY）",
}

WEB_DIRECTORY = "./web"
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]