import os
import sys
import folder_paths as comfy_paths

# Set up paths
ROOT_PATH = os.path.join(comfy_paths.get_folder_paths("custom_nodes")[0], "ComfyUI-GG")
NODES_PATH = os.path.join(ROOT_PATH, "nodes")
sys.path.append(ROOT_PATH)
sys.path.append(NODES_PATH)

# Import node classes
from .nodes.add_padding_to_image import AddPaddingToImage
from .nodes.metadata_extract_by_seed import MetadataExtractorBySeed
from .nodes.resize_image_proportionally import ResizeImageProportionally
from .nodes.image_metadata_extractor import ImageMetadataExtractor
from .nodes.workspace_node import WorkspaceNode
from .nodes.input_node import InputNode
from .nodes.output_node import OutputNode
from .nodes.for_loop_node import ForLoopNode

# Node mappings
NODE_CLASS_MAPPINGS = {
    "AddPaddingToImage": AddPaddingToImage,
    "MetadataExtractBySeed": MetadataExtractorBySeed,
    "ResizeImageProportionally": ResizeImageProportionally,
    "ImageMetadataExtractor": ImageMetadataExtractor,
    "WorkspaceNode": WorkspaceNode,
    "InputNode": InputNode,
    "OutputNode": OutputNode,
    "ForLoopNode": ForLoopNode
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "AddPaddingToImage": "Add Padding to Image",
    "MetadataExtractBySeed": "Metadata Extract by Seed",
    "ResizeImageProportionally": "Resize Image Proportionally",
    "ImageMetadataExtractor": "Image Metadata Extractor",
    "WorkspaceNode": "Workspace Node",
    "InputNode": "Input Node",
    "OutputNode": "Output Node",
    "ForLoopNode": "For Loop"
}

WEB_DIRECTORY = "./js"
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', "WEB_DIRECTORY"]