app.py

from flask import Flask
import logging
import time
from tests.openid_configuration import OpenIdConfiguration
from tests.auth_code_flow import AuthCodeFlow
from tests.token_refresh_flow import TokenRefreshFlow
from tests.access_token_available import AccessTokenAvailable
from shared.auth_provider import AuthProvider

logging.basicConfig(
    level=logging.DEBUG, format="%(asctime)s - %(levelname)s - %(message)s"
)

app = Flask(__name__)

auth_provider = AuthProvider()


@app.route("/auth/openid-configuration")
def serve_openid_configuration():
    logging.info("Serving OpenID configuration test")
    openid_configuration = OpenIdConfiguration(auth_provider)
    start_time = time.time()
    result = openid_configuration.run_test()
    end_time = time.time()
    logging.info(
        f"OpenID configuration test completed in {end_time - start_time:.2f} seconds"
    )
    return result


@app.route("/auth/auth-code")
def serve_auth_code():
    logging.info("Serving auth code flow test")
    auth_code_flow = AuthCodeFlow(auth_provider)
    start_time = time.time()
    result = auth_code_flow.run_test()
    end_time = time.time()
    logging.info(
        f"Auth code flow test completed in {end_time - start_time:.2f} seconds"
    )
    return result


@app.route("/auth/refresh-token")
def serve_token_refresh():
    logging.info("Serving token refresh flow test")
    token_refresh_flow = TokenRefreshFlow(auth_provider)
    start_time = time.time()
    result = token_refresh_flow.run_test()
    end_time = time.time()
    logging.info(
        f"Token refresh flow test completed in {end_time - start_time:.2f} seconds"
    )
    return result


@app.route("/auth/access-token")
def serve_access_token():
    logging.info("Serving access token availability test")
    access_token_available = AccessTokenAvailable(auth_provider)
    start_time = time.time()
    result = access_token_available.run_test()
    end_time = time.time()
    logging.info(
        f"Access token availability test completed in {end_time - start_time:.2f} seconds"
    )
    return result


@app.route("/auth/callback")
def serve_callback():
    logging.info("Received callback request")
    return "Callback received"


if __name__ == "__main__":
    app.run(host="0.0.0.0", port=5000, debug=True)