trz-gateway-common 0.2.3

use std::convert::Infallible;
use std::sync::Arc;

use nameth::nameth;
use openssl::x509::X509;

use super::CertificateConfig;
use super::X509CertificateInfo;
use crate::security_configuration::common::get_or_init;

/// A [CertificateConfig] that computes the certificate and intermediates once,
/// and then memoizes the results.
#[nameth]
pub struct MemoizedCertificate<C> {
    base: C,
    intermediates: std::sync::Mutex<Option<Arc<Vec<X509>>>>,
    certificate: std::sync::Mutex<Option<Arc<X509CertificateInfo>>>,
}

impl<C> MemoizedCertificate<C> {
    /// Creates a [MemoizedCertificate] based on a certificate.
    pub fn new(base: C) -> Self {
        Self {
            base,
            intermediates: Default::default(),
            certificate: Default::default(),
        }
    }
}

impl<C: CertificateConfig> CertificateConfig for MemoizedCertificate<C> {
    type Error = C::Error;

    fn intermediates(&self) -> Result<Arc<Vec<X509>>, Self::Error> {
        get_or_init(&self.intermediates, || self.base.intermediates())
    }

    fn certificate(&self) -> Result<Arc<X509CertificateInfo>, Self::Error> {
        get_or_init(&self.certificate, || self.base.certificate())
    }
}

/// A [CertificateConfig] that contains the pre-computed X509 objects.
///
/// Computing certificate and intermediates from a [CachedCertificate] is thus an infallible operation.
#[nameth]
#[derive(Clone)]
pub struct CachedCertificate {
    intermediates: Arc<Vec<X509>>,
    certificate: Arc<X509CertificateInfo>,
}

impl CachedCertificate {
    /// Creates a [CachedCertificate] based on a certificate.
    pub fn new<C: CertificateConfig>(base: C) -> Result<Self, C::Error> {
        Ok(Self {
            intermediates: base.intermediates()?,
            certificate: base.certificate()?,
        })
    }
}

impl CertificateConfig for CachedCertificate {
    type Error = Infallible;

    fn intermediates(&self) -> Result<Arc<Vec<X509>>, Self::Error> {
        Ok(self.intermediates.clone())
    }

    fn certificate(&self) -> Result<Arc<X509CertificateInfo>, Self::Error> {
        Ok(self.certificate.clone())
    }
}

mod debug {

    use super::CACHED_CERTIFICATE;
    use super::CachedCertificate;
    use super::MEMOIZED_CERTIFICATE;
    use super::MemoizedCertificate;

    impl<C: std::fmt::Debug> std::fmt::Debug for MemoizedCertificate<C> {
        fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
            f.debug_struct(MEMOIZED_CERTIFICATE)
                .field("base", &self.base)
                .finish()
        }
    }

    impl std::fmt::Debug for CachedCertificate {
        fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
            f.debug_struct(CACHED_CERTIFICATE)
                .field("intermediates", &self.intermediates)
                .field("certificate", &self.certificate)
                .finish()
        }
    }
}