import logging
from collections.abc import Mapping
from typing import Any

from flask_restx import Resource
from pydantic import Field, RootModel

from controllers.common.schema import register_response_schema_models
from controllers.console.wraps import (
    account_initialization_required,
    setup_required,
)
from core.schemas.schema_manager import SchemaManager
from fields.base import ResponseModel
from libs.login import login_required

from . import console_ns

logger = logging.getLogger(__name__)


class SchemaDefinitionItemResponse(ResponseModel):
    name: str
    label: str
    schema_: Mapping[str, Any] = Field(alias="schema")


class SchemaDefinitionsResponse(RootModel[list[SchemaDefinitionItemResponse]]):
    pass


register_response_schema_models(console_ns, SchemaDefinitionItemResponse, SchemaDefinitionsResponse)


@console_ns.route("/spec/schema-definitions")
class SpecSchemaDefinitionsApi(Resource):
    @console_ns.response(200, "Success", console_ns.models[SchemaDefinitionsResponse.__name__])
    @setup_required
    @login_required
    @account_initialization_required
    def get(self):
        """
        Get system JSON Schema definitions specification
        Used for frontend component type mapping
        """
        try:
            schema_manager = SchemaManager()
            schema_definitions = schema_manager.get_all_schema_definitions()
            return schema_definitions, 200
        except Exception:
            logger.exception("Failed to get schema definitions from local registry")
            # Return empty array as fallback
            return [], 200