# Copyright 2024 VyOS maintainers and contributors <maintainers@vyos.io>
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this library.  If not, see <http://www.gnu.org/licenses/>.

# pylint: disable=import-outside-toplevel


import typing

from ariadne.asgi import GraphQL
from starlette.middleware.cors import CORSMiddleware


if typing.TYPE_CHECKING:
    from fastapi import FastAPI


def graphql_init(app: 'FastAPI'):
    from ..session import SessionState
    from .libs.token_auth import get_user_context

    state = SessionState()

    # import after initializaion of state
    from .bindings import generate_schema

    schema = generate_schema()

    in_spec = state.introspection

    # remove route and reinstall below, for any changes; alternatively, test
    # for config_diff before proceeding
    graphql_clear(app)

    if state.origins:
        origins = state.origins
        app.add_route(
            '/graphql',
            CORSMiddleware(
                GraphQL(
                    schema,
                    context_value=get_user_context,
                    debug=True,
                    introspection=in_spec,
                ),
                allow_origins=origins,
                allow_methods=('GET', 'POST', 'OPTIONS'),
                allow_headers=('Authorization',),
            ),
        )
    else:
        app.add_route(
            '/graphql',
            GraphQL(
                schema,
                context_value=get_user_context,
                debug=True,
                introspection=in_spec,
            ),
        )


def graphql_clear(app: 'FastAPI'):
    for r in app.routes:
        if r.path == '/graphql':
            app.routes.remove(r)