from apispec.ext.marshmallow import MarshmallowPlugin
from apispec_webframeworks.flask import FlaskPlugin

from flasgger import APISpec, Schema, Swagger, fields

from eligibility_api.elig_api_exceptions import SwaggerValidationException


def _swagger_validation_error_handler(err, data, _):
    print("API swagger error 400")
    raise SwaggerValidationException(err, data)


def start_swagger(flask_app):
    """
    Creates swagger /apidocs endpoint and adds it to flask app
    """

    flask_app.config["SWAGGER"] = {
        "uiversion": 3,
        "ui_params": {
            "apisSorter": "alpha",
            "operationsSorter": "alpha",
            "tagsSorter": "alpha",
        },
    }

    # Create an APISpec
    spec = APISpec(
        title="API éligibilité",
        version="0.1.0",
        openapi_version="2.0",
        plugins=(
            FlaskPlugin(),
            MarshmallowPlugin(),
        ),
    )

    template = spec.to_flasgger(
        flask_app,
        definitions=[
            # ContributionSchema,
            # PredictConfigSchema,
            # VersionSchema,
            # PredictInputsSchema,
            # ImplementedModelSchema,
            # ApiErrorSchema,
            # LoggingHealthCheckSchema,
        ],
    )
    Swagger(
        flask_app,
        template=template,
        parse=True,
        validation_error_handler=_swagger_validation_error_handler,
    )