aswi2021jmsd-gitlab4/src/services/certificate_service.py @ fa72c969

from src.constants import ROOT_CA_ID
from src.dao.certificate_repository import CertificateRepository
from src.model.certificate import Certificate
from src.model.private_key import PrivateKey
from src.model.subject import Subject
from src.services.cryptography import CryptographyService

import time

DATE_FORMAT = "%d.%m.%Y %H:%M:%S"


class CertificateService:

    def __init__(self, cryptography_service: CryptographyService, certificate_repository: CertificateRepository):
        self.cryptography_service = cryptography_service
        self.certificate_repository = certificate_repository

    def create_root_ca(self, key: PrivateKey, subject: Subject, extensions: str = "", config: str = ""):
        # create a new self signed  certificate
        cert_pem = self.cryptography_service.create_sscrt(subject, key.private_key, key_pass=key.password,
                                                          extensions=extensions, config=config)

        # parse the generated pem for subject and notBefore/notAfter fields
        subj, not_before, not_after = self.cryptography_service.parse_cert_pem(cert_pem)

        # format the parsed date
        not_before_formatted = time.strftime(DATE_FORMAT, not_before)
        not_after_formatted = time.strftime(DATE_FORMAT, not_after)

        # create a certificate wrapper
        certificate = Certificate(-1, subject.common_name, not_before_formatted, not_after_formatted, cert_pem,
                                  key.private_key_id, ROOT_CA_ID, 0, {})

        # store the wrapper into the repository
        created_id = self.certificate_repository.create(certificate)

        # assign the generated ID to the inserted certificate
        certificate.certificate_id = created_id

        return certificate

    def get_certificate(self, unique_id: int) -> Certificate:
        return self.certificate_repository.read(unique_id)