tls.rs

use bytes::Bytes;
use pyo3::{
    pymethods,
    types::PyByteArray,
    PyRefMut,
};

use crate::listener_builder::TlsListenerBuilder;

#[pymethods]
#[allow(dead_code)]
impl TlsListenerBuilder {
    /// The domain to request for this edge, any valid domain or hostname that you have
    /// previously registered with ngrok. If using a custom domain, this requires
    /// registering in the `ngrok dashboard`_ and setting a DNS CNAME value.
    ///
    /// .. _ngrok dashboard: https://dashboard.ngrok.com/cloud-edge/domains    
    pub fn domain(self_: PyRefMut<Self>, domain: String) -> PyRefMut<Self> {
        self_.set(|b| {
            b.domain(domain);
        });
        self_
    }
    /// Certificates to use for client authentication at the ngrok edge.
    /// See `Mutual TLS`_ in the ngrok docs for additional details.
    ///
    /// .. _Mutual TLS: https://ngrok.com/docs/cloud-edge/modules/mutual-tls/
    pub fn mutual_tlsca<'a>(
        self_: PyRefMut<'a, Self>,
        mutual_tlsca: &PyByteArray,
    ) -> PyRefMut<'a, Self> {
        self_.set(|b| {
            b.mutual_tlsca(Bytes::from(mutual_tlsca.to_vec()));
        });
        self_
    }

    /// Enable endpoint pooling for this listener.
    pub fn pooling_enabled(self_: PyRefMut<Self>, endpoint_pooling: bool) -> PyRefMut<Self> {
        self_.set(|b| {
            b.pooling_enabled(endpoint_pooling);
        });
        self_
    }

    /// The key to use for TLS termination at the ngrok edge in PEM format.
    /// See `TLS Termination`_ in the ngrok docs for additional details.
    ///
    /// .. _TLS Termination: https://ngrok.com/docs/cloud-edge/modules/tls-termination/
    pub fn termination<'a>(
        self_: PyRefMut<'a, Self>,
        cert_pem: &PyByteArray,
        key_pem: &PyByteArray,
    ) -> PyRefMut<'a, Self> {
        self_.set(|b| {
            b.termination(
                Bytes::from(cert_pem.to_vec()),
                Bytes::from(key_pem.to_vec()),
            );
        });
        self_
    }
}