from .py.image_output_nodes import *
from .py.prompting_nodes import *
from .py.selector_nodes import *
from .py.media_nodes import *
from .py.misc import *
from .py.parameter_list import *
from .py.server_backend import *

from .py.deprecated_nodes import *

from aiohttp import web
import server

@server.PromptServer.instance.routes.get('/jnodes_list_comfyui_subdirectories')
async def list_comfyui_subdirectories_wrapper(request):
    return list_comfyui_subdirectories_request(request)

@server.PromptServer.instance.routes.get('/jnodes_get_comfyui_subdirectory_images')
async def get_comfyui_subdirectory_images_wrapper(request):
    return get_comfyui_subdirectory_images_request(request)

@server.PromptServer.instance.routes.get('/jnodes_model_items')
async def get_model_items_wrapper(request):
    return get_model_items(request)

@server.PromptServer.instance.routes.get('/jnodes_save_model_config')
async def save_model_config_wrapper(request):
    return save_model_config(request)

@server.PromptServer.instance.routes.get("/jnodes_view_image")
async def view_image_wrapper(request):
    return view_image(request)

@server.PromptServer.instance.routes.get("/jnodes_load_info")
async def load_info_wrapper(request):
    return load_info(request)

@server.PromptServer.instance.routes.post('/jnodes_request_task_cancellation')
async def request_task_cancellation_wrapper(request):
    return request_task_cancellation(request)

@server.PromptServer.instance.routes.post('/jnodes_save_text')
async def save_text_wrapper(request):
    return await save_text(request)

@server.PromptServer.instance.routes.post('/jnodes_load_text')
async def load_text_wrapper(request):
    return await load_text(request)

@server.PromptServer.instance.routes.post("/jnodes_upload_image")
async def upload_image_wrapper(request):
    return await upload_image(request)

@server.PromptServer.instance.routes.delete("/jnodes_delete_item")
async def delete_item_wrapper(request):
    return await delete_item(request)
    
WEB_DIRECTORY = "./web"

# A dictionary that contains all nodes you want to export with their names
# NOTE: names should be globally unique
NODE_CLASS_MAPPINGS = {
    
    # image_output_nodes
    "JNodes_SaveVideo": SaveVideo,
    "JNodes_SaveImageWithOutput": SaveImageWithOutput,
    
    # parameter_list
    # Globals aren't ready yet
    #"JNodes_ParseParametersToGlobalList": ParseParametersToGlobalList,
    #"JNodes_GetParameterGlobal": GetParameterGlobal,
    "JNodes_GetParameterFromList": GetParameterFromList,
    
    # prompting_nodes
    "JNodes_SyncedStringLiteral": SyncedStringLiteral,
    "JNodes_ParseDynamicPrompts": ParseDynamicPrompts,
    "JNodes_RemoveCommentedText": RemoveCommentedText,
    "JNodes_SplitAndJoin": SplitAndJoin,
    "JNodes_TrimAndStrip": TrimAndStrip,
    "JNodes_ParseWildcards": ParseWildcards,
    "JNodes_LoraExtractor": LoraExtractor,
    "JNodes_RemoveParseableDataForInference": RemoveParseableDataForInference,
    "JNodes_PromptBuilderSingleSubject": PromptBuilderSingleSubject,
    "JNodes_SearchAndReplaceFromList": SearchAndReplaceFromList,
    "JNodes_SearchAndReplaceFromFile": SearchAndReplaceFromFile,
    "JNodes_SearchAndReplace": SearchAndReplace,
    "JNodes_AddOrSetMetaDataKey" : AddOrSetMetaDataKey,
    "JNodes_SetPositivePromptInMetaData": SetPositivePromptInMetaData,
    "JNodes_SetNegativePromptInMetaData": SetNegativePromptInMetaData,
    "JNodes_RemoveMetaDataKey" : RemoveMetaDataKey,
    "JNodes_TokenCounter": TokenCounter,
    
    # selector_nodes
    "JNodes_BooleanSelectorWithString": BooleanSelector,
    "JNodes_ImageSizeSelector": ImageSizeSelector,
    "JNodes_CheckpointSelectorWithString": CheckpointSelector,
    "JNodes_VaeSelectorWithString": VaeSelector,
    "JNodes_SamplerSelectorWithString": SamplerSelector,
    "JNodes_SchedulerSelectorWithString": SchedulerSelector,
    "JNodes_ImageFormatSelector": ImageFormatSelector,
    
    # video_nodes
    "JNodes_MediaInfoToString": MediaInfoToString,
    "JNodes_BreakMediaInfo": BreakMediaInfo,
    "JNodes_AppendReversedFrames": AppendReversedFrames,
    "JNodes_LoadVisualMediaFromPath": LoadVisualMediaFromPath,
    "JNodes_UploadVisualMedia": UploadVisualMedia,
    
    # misc
    "JNodes_GetTempDirectory": GetTempDirectory,
    "JNodes_GetOutputDirectory": GetOutputDirectory,
    "JNodes_StringLiteral" : StringLiteral,
    "JNodes_AnyToString" : AnyToString,

    # deprecated_nodes
    "JNodes_OutVideoInfo": OutVideoInfo,
    "JNodes_LoadVideo": LoadVideo,
    "JNodes_UploadVideo": UploadVideo,
}

# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {
    
    # image_output_nodes
    "JNodes_SaveVideo": "Save Video",
    "JNodes_SaveImageWithOutput": "Save Image With Output",
    
    # parameter_list
    "JNodes_ParseParametersToGlobalList": "Parse Parameters To Global List",
    "JNodes_GetParameterGlobal": "Get Parameter Global",
    "JNodes_GetParameterFromList": "Get Parameter From List",
    
    # prompting_nodes
    "JNodes_SyncedStringLiteral": "Synced String Literal",
    "JNodes_ParseDynamicPrompts": "Parse Dynamic Prompts",
    "JNodes_RemoveCommentedText": "Remove Commented Text",
    "JNodes_SplitAndJoin": "Split And Join",
    "JNodes_TrimAndStrip": "Trim And Strip",
    "JNodes_ParseWildcards": "Parse Wildcards",
    "JNodes_LoraExtractor": "Lora Extractor",
    "JNodes_RemoveParseableDataForInference": "Remove Parseable Data For Inference",
    "JNodes_PromptBuilderSingleSubject": "Prompt Builder Single Subject",
    "JNodes_SearchAndReplaceFromList": "Search And Replace From List",
    "JNodes_SearchAndReplaceFromFile": "Search And Replace From File",
    "JNodes_SearchAndReplace": "Search And Replace",
    "JNodes_AddOrSetPngInfoKey" : "Add Or Set Png Info Key",
    "JNodes_SetPositivePromptInMetaData": "Set Positive Prompt In MetaData",
    "JNodes_SetNegativePromptInMetaData": "Set Negative Prompt In MetaData",
    "JNodes_RemovePngInfoKey" : "Remove Png Info Key",
    "JNodes_TokenCounter": "Token Counter",
    
    # selector_nodes
    "JNodes_BooleanSelectorWithString": "Boolean Selector + String",
    "JNodes_ImageSizeSelector": "Image Size Selector",
    "JNodes_CheckpointSelectorWithString": "Checkpoint Selector + String",
    "JNodes_VaeSelectorWithString": "Vae Selector + String",
    "JNodes_SamplerSelectorWithString": "Sampler Selector + String",
    "JNodes_SchedulerSelectorWithString": "Scheduler Selector + String",
    "JNodes_ImageFormatSelector": "Image Format Selector",
    
    # video_nodes
    "JNodes_MediaInfoToString": "Media Info To String",
    "JNodes_BreakMediaInfo": "Break Media Info",
    "JNodes_AppendReversedFrames": "Append Reversed Frames",
    "JNodes_LoadVisualMediaFromPath": "Load Visual Media From Path",
    "JNodes_UploadVisualMedia": "Upload Visual Media",
    
    # misc
    "JNodes_GetTempDirectory": "Get Temp Directory",
    "JNodes_GetOutputDirectory": "Get Output Directory",
    "JNodes_StringLiteral" : "String Literal",
    "JNodes_AnyToString" : "Anything To String",

    # deprecated_nodes
    "JNodes_OutVideoInfo": "Out Video Info (DEPRECATED, USE JNodes_BreakMediaInfo)",
    "JNodes_LoadVideo": "Load Video (DEPRECATED, USE JNodes_LoadVisualMediaFromPath)",
    "JNodes_UploadVideo": "Upload Video (DEPRECATED, USE JNodes_UploadVisualMedia)",
}


__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]