aswi2021jmsd-gitlab4/app.py @ 0cf35f70

import os

from flask import Flask, redirect
from injector import Injector
from flask_injector import FlaskInjector

from src.config import configuration
from src.config.connection_provider import ConnectionProvider
from src.controllers.certificates_controller import CertController
from src.services.cryptography import CryptographyService, CryptographyException


class X509ManagementApp(Flask):

    def run(self, host=None, port=None, debug=None, load_dotenv=True, **options):
        with self.app_context():
            if not initialize_app():
                return
        super(X509ManagementApp, self).run(host=host, port=port, debug=debug, load_dotenv=load_dotenv, **options)


app = X509ManagementApp(__name__)


@app.route('/')
def index():
    return redirect("/static/index.html")


@app.route('/api/certificates', methods=["POST"])
def create_certificate(certificate_controller: CertController):
    return certificate_controller.create_certificate()


@app.route('/api/certificates', methods=["GET"])
def get_cert_list(certificate_controller: CertController):
    return certificate_controller.get_certificate_list()


@app.route('/api/certificates/<id>', methods=["GET"])
def get_cert(id, certificate_controller: CertController):
    return certificate_controller.get_certificate_by_id(id)


@app.route('/api/certificates/<id>/details', methods=["GET"])
def get_cert_details(id, certificate_controller: CertController):
    return certificate_controller.get_certificate_details_by_id(id)


@app.route('/api/certificates/<id>/root', methods=["GET"])
def get_cert_root(id, certificate_controller: CertController):
    return certificate_controller.get_certificate_root_by_id(id)


@app.route('/api/certificates/<id>/chain', methods=["GET"])
def get_cert_chain(id, certificate_controller: CertController):
    print("Get cert chain via flask injector")
    return certificate_controller.get_certificate_trust_chain_by_id(id)


def initialize_app() -> bool:
    """
    Initializes the application
        -   configure dependency injection
        -   check whether OpenSSL is on the system
    :return: boolean flag indicating whether initialization was successfull or not
    """

    modules = [configuration.configure_env_variable, ConnectionProvider]
    injector = Injector(modules)
    FlaskInjector(app=app, modules=modules)

    # There's a little dependency on the CryptoService, which is not a pretty thing from
    # architectural point of view. However it is only a minimal piece of code and
    # it makes sense to do it in this way instead of trying to run openssl via subprocess here
    cryptography_service = injector.get(CryptographyService)
    try:
        # if version string is returned, OpenSSL is present on the system
        print(f"Using {cryptography_service.get_openssl_version()}")
        # TODO log the version instead of prining it out
        return True
    except CryptographyException:
        # If getting the version string throws an exception the OpenSSL is not available
        print("OpenSSL was not located on the system. Application will now exit.")
        # TODO add logging here
        return False


if __name__ == '__main__':
    app_host = "0.0.0.0"
    app_port = 5000

    # TODO better load this from config.py
    if "FLASK_HOST" in os.environ:
        app_host = os.environ["FLASK_HOST"]

    if "FLASK_PORT" in os.environ:
        app_host = os.environ["FLASK_PORT"]

    app.run(host=app_host, port=app_port)