Module supertokens_python.framework.flask.flask_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.

import json


from asgiref.sync import async_to_sync


class Middleware:
    def __init__(self, app):
        self.app = app

    def __call__(self, environ, start_response):

        from supertokens_python.framework.flask.flask_request import FlaskRequest
        from supertokens_python.framework.flask.flask_response import FlaskResponse
        from supertokens_python import Supertokens
        from supertokens_python.supertokens import manage_cookies_post_response
        from flask import Response, Request

        st = Supertokens.get_instance()
        request = FlaskRequest(Request(environ))
        response = FlaskResponse(Response())
        result = async_to_sync(st.middleware)(request, response)

        if result is None:
            def injecting_start_response(status, headers, exc_info=None):
                headers = None
                return start_response(status, headers, exc_info)
            self.app(environ, injecting_start_response)

        response = FlaskResponse()

        if 'additional_storage' in environ:
            manage_cookies_post_response(
                environ['additional_storage'], response)

        def injecting_start_response(status, headers, exc_info=None):
            headers.extend(response.get_headers())
            return start_response(status, headers, exc_info)

        return self.app(environ, injecting_start_response)


def error_handler(error):
    from werkzeug import Response
    from supertokens_python import Supertokens
    from supertokens_python.framework.flask.flask_response import FlaskResponse

    st = Supertokens.get_instance()
    response = Response(json.dumps({}),
                        mimetype='application/json',
                        status=200)
    result = async_to_sync(
        st.handle_supertokens_error)(
        None, error, FlaskResponse(response))
    return result.response