Module supertokens_python.framework.django.django_middleware

# Copyright (c) 2021, VRAI Labs and/or its affiliates. All rights reserved.
#
# This software is licensed under the Apache License, Version 2.0 (the
# "License") as published by the Apache Software Foundation.
#
# You may not use this file except in compliance with the License. You may
# obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
from __future__ import annotations

import asyncio
from asgiref.sync import async_to_sync
from supertokens_python.async_to_sync_wrapper import sync


def middleware():
    from django.utils.decorators import sync_and_async_middleware

    @sync_and_async_middleware
    def _middleware(get_response):
        from supertokens_python import Supertokens
        from supertokens_python.exceptions import SuperTokensError
        from supertokens_python.framework.django.django_request import DjangoRequest
        from supertokens_python.framework.django.django_response import DjangoResponse
        from supertokens_python.recipe.session import Session
        from supertokens_python.supertokens import manage_cookies_post_response

        if asyncio.iscoroutinefunction(get_response):
            async def __middleware(request):
                st = Supertokens.get_instance()
                custom_request = DjangoRequest(request)
                from django.http import HttpResponse
                response = DjangoResponse(HttpResponse())
                try:
                    result = await st.middleware(custom_request, response)
                    if result is None:
                        result = await get_response(request)
                        result = DjangoResponse(result)
                    if hasattr(request, "state") and isinstance(
                            request.state, Session):
                        manage_cookies_post_response(request.state, result)

                    return result.response

                except SuperTokensError as e:
                    response = DjangoResponse(HttpResponse())
                    result = await st.handle_supertokens_error(DjangoRequest(request), e, response)

                    return result.response
        else:
            def __middleware(request):
                st = Supertokens.get_instance()
                custom_request = DjangoRequest(request)
                from django.http import HttpResponse
                response = DjangoResponse(HttpResponse())
                try:
                    result = async_to_sync(st.middleware)(custom_request, response)

                    if result is None:
                        result = get_response(request)
                        result = DjangoResponse(result)

                    if hasattr(request, "state") and isinstance(
                            request.state, Session):
                        manage_cookies_post_response(request.state, result)

                    return result.response

                except SuperTokensError as e:
                    response = DjangoResponse(HttpResponse())
                    result = sync(
                        st.handle_supertokens_error)(
                        DjangoRequest(request), e, response)

                return result.response

        return __middleware

    return _middleware