import server
from aiohttp import web
import os
import json
import folder_paths
from app.user_manager import UserManager


def get_workflow_data(workflow_file):
    nodes_input = {}
    nodes_output = {}


    # Extraire les nœuds WorkflowInput et WorkflowOutput
    for node in workflow_file["nodes"]:
        if node.get("type") == "WorkflowInput":
            # Convertir au format compatible pour le client
            node_id = str(node.get("id", "unknown"))
            nodes_input[node_id] = {
                "class_type": "WorkflowInput",
                "inputs": node.get("widgets_values", {}),
                "position": node.get("pos", {})[1]
            }
        elif node.get("type") == "WorkflowOutput":
            node_id = str(node.get("id", "unknown"))
            nodes_output[node_id] = {
                "class_type": "WorkflowOutput",
                "inputs": node.get("widgets_values", {}),
                "position": node.get("pos", {})[1]
            }
        # sort by position
        nodes_input = dict(sorted(nodes_input.items(), key=lambda item: item[1]["position"]))
        nodes_output = dict(sorted(nodes_output.items(), key=lambda item: item[1]["position"]))

    return {"inputs": nodes_input,
            "outputs": nodes_output,
            'workflow': workflow_file}


@server.PromptServer.instance.routes.get("/flowchain/workflows")
async def workflows(request):
    user = UserManager().get_request_user_id(request)
    json_path = os.path.join(folder_paths.user_directory, user, "workflows")
    result = {}

    # Vérifier si le répertoire principal existe
    if os.path.exists(json_path):
        # Utiliser os.walk pour parcourir récursivement tous les sous-répertoires
        for root, dirs, files in os.walk(str(json_path)):
            for file in files:
                # Ne traiter que les fichiers JSON
                if file.lower().endswith('.json'):
                    file_path = os.path.join(root, file)
                    try:
                        with open(file_path, "r", encoding="utf-8") as f:
                            json_content = json.load(f)
                        relative_path = os.path.relpath(file_path, str(json_path))
                        if "nodes" in json_content:
                            file_conf = get_workflow_data(json_content)
                            if file_conf["inputs"] or file_conf["outputs"]:
                                result[relative_path] = file_conf

                    except json.JSONDecodeError:
                        # Ignorer les fichiers JSON mal formés
                        print(f"Ignoring malformed JSON file: {file_path}")
                    except Exception as e:
                        print(f"Error processing {file_path}: {str(e)}")
    else:
        # Créer le répertoire s'il n'existe pas
        os.makedirs(json_path)
        result["No file in worflows folder"] = {"inputs": {}, "outputs": {}}

    # Si aucun fichier valide n'a été trouvé
    if not result:
        result["No compatible workflow files found"] = {"inputs": {}, "outputs": {}}

    return web.json_response(result, content_type='application/json')


@server.PromptServer.instance.routes.get("/flowchain/workflow")
async def workflow(request):
    user = UserManager().get_request_user_id(request)
    result = {}
    original_path = request.query.get("workflow_path")
    unversal_path = original_path.replace("\\", "/")
    json_path = unversal_path.split("/")
    if ".json" in json_path[-1]:
        # file_name = json_path[-1]
        json_path = folder_paths.user_directory + "/" + user + "/workflows/" + unversal_path

    if os.path.exists(json_path):
        with open(json_path, "r", encoding="utf-8") as f:
            json_content = json.load(f)
        if "nodes" in json_content:
            result = get_workflow_data(json_content)
        else:
            result = {"error": "File not found"}
    else:
        result = {"error": "File not found"}

    return web.json_response(result, content_type='application/json')