aswi2021jmsd-gitlab/app.py @ 3195e946

import os

from flask import Flask, redirect
from injector import Injector

from src.config import configuration
from src.config.connection_provider import ConnectionProvider
from src.controllers.certificates_controller import CertController


class X509ManagementApp(Flask):

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


app = X509ManagementApp(__name__)
certificate_controller: CertController = None


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


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


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


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


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


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


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


def initialize_app():
    global certificate_controller

    injector = Injector([configuration.configure_default, ConnectionProvider])
    certificate_controller = injector.get(CertController)


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)