"""Functions which rely on gradio components io and change the interface"""

import json

from jsonschema import validate, ValidationError


def validate_mcp_file(mcp_file_content: str, token: str) -> str:
    """Validates the user uploaded MCP file

    The JSON file should follow this structure (we only allow SSE type for the demo,
    but if you run locally you can add stdio to your json file):

    ```json
    {
        "mcpServers": {
            "app-name-1": {
                "type": "sse",
                "url": "https://api.example.com/mcp",
                "headers": {
                    "Authorization": "Bearer your-token-here"
                }
            }
        }
    }
    ```
    """
    # Define the JSON schema
    mcp_schema = {
        "$schema": "https://json-schema.org/draft/2020-12/json-schema-core.html",
        "type": "object",
        "properties": {
            "mcpServers": {
                "type": "object",
                "patternProperties": {
                    "^[\w-]+$": {
                        "type": "object",
                        "properties": {
                            "type": {"type": "string", "enum": ["sse"]},
                            "url": {"type": "string", "format": "uri"},
                            "headers": {
                                "type": "object",
                                "patternProperties": {"^[\w-]+$": {"type": "string"}},
                                "additionalProperties": False,
                            },
                        },
                        "required": ["type", "url"],
                        "additionalProperties": False,
                    }
                },
                "additionalProperties": False,
                "minProperties": 1,
            }
        },
        "required": ["mcpServers"],
        "additionalProperties": False,
    }

    try:
        mcp_data = json.loads(mcp_file_content)
        validate(instance=mcp_data, schema=mcp_schema)

        for server in mcp_data.get("mcpServers", {}).values():
            try:
                if "HF_TOKEN" in server["headers"]["Authorization"]:
                    server["headers"]["Authorization"] = server["headers"][
                        "Authorization"
                    ].replace("HF_TOKEN", token or "")
            except KeyError:
                continue

        return mcp_data

    except json.JSONDecodeError as e:
        raise ValueError(f"Invalid JSON format: {str(e)}")
    except ValidationError as e:
        # Provide more user-friendly error messages
        error_path = " -> ".join(str(p) for p in e.path) if e.path else "root"
        raise ValueError(f"Invalid MCP file structure at {error_path}: {e.message}")
    except Exception as e:
        raise ValueError(f"Error validating MCP file: {str(e)}")