aswi2021jmsd-gitlab4/src/controllers/crl_ocsp_controller.py @ dd6b4c4c

from flask import Response
from injector import inject

from src.exceptions.certificate_not_found_exception import CertificateNotFoundException
from src.exceptions.private_key_not_found_exception import PrivateKeyNotFoundException
from src.services.crl.crl_service import CrlService
from src.controllers.return_codes import *

E_WRONG_PARAMETERS = {"success": False, "data": "Invalid request, wrong parameters."}
E_NO_CERTIFICATES_FOUND = {"success": False, "data": "No such certificate found."}


class CrlOcspController:

    @inject
    def __init__(self, crl_service: CrlService):
        self.crl_service = crl_service

    def get_crl(self, ca_id: str):
        """
        Generate and download a CRL for a selected certificate authority
        :param ca_id: certificate authority whose CRL is requested
        :return: download response / error
        """
        # convert id from string to int
        try:
            identifier = int(ca_id)
        except ValueError:
            return E_WRONG_PARAMETERS, C_BAD_REQUEST

        # generate the CRL
        try:
            crl = self.crl_service.generate_crl_response(identifier)
        except (CertificateNotFoundException, PrivateKeyNotFoundException):
            return E_NO_CERTIFICATES_FOUND, C_NOT_FOUND

        return Response(crl, mimetype="application/x-x509-ca-cert",
                        headers={"Content-Disposition": f"attachment;filename={ca_id}_crl.pem"})