detsys_ids_client/

builder.rs

use std::time::Duration;

use reqwest::Certificate;
use url::Url;

use crate::transport::TransportsError;
use crate::{system_snapshot::SystemSnapshotter, DeviceId, DistinctId, Map};
use crate::{Recorder, Worker};

#[derive(Default)]
pub struct Builder {
    device_id: Option<DeviceId>,
    distinct_id: Option<DistinctId>,
    enable_reporting: bool,
    endpoint: Option<String>,
    facts: Option<Map>,
    groups: Option<Map>,
    proxy: Option<Url>,
    certificate: Option<Certificate>,
    timeout: Option<Duration>,
}

impl Builder {
    pub fn new() -> Self {
        Builder {
            device_id: None,
            distinct_id: None,
            enable_reporting: true,
            endpoint: None,
            facts: None,
            groups: None,
            proxy: None,
            certificate: None,
            timeout: None,
        }
    }

    pub fn set_distinct_id(mut self, distinct_id: Option<DistinctId>) -> Self {
        self.distinct_id = distinct_id;
        self
    }

    pub fn set_device_id(mut self, device_id: Option<DeviceId>) -> Self {
        self.device_id = device_id;
        self
    }

    pub fn set_facts(mut self, facts: Option<Map>) -> Self {
        self.facts = facts;
        self
    }

    pub fn set_groups(mut self, groups: Option<Map>) -> Self {
        self.groups = groups;
        self
    }

    pub fn add_fact(
        mut self,
        key: impl Into<String> + std::fmt::Debug,
        value: impl Into<serde_json::Value>,
    ) -> Self {
        self.facts
            .get_or_insert_with(Default::default)
            .insert(key.into(), value.into());
        self
    }

    pub fn set_endpoint(mut self, endpoint: Option<String>) -> Self {
        self.endpoint = endpoint;
        self
    }

    /// Set whether reporting is enabled or disabled.
    /// Reporting is enabled by default, but this function can be used in a pipeline for easy configuration:
    ///
    /// ```rust
    /// use detsys_ids_client::builder;
    ///
    /// struct Cli {
    ///   no_telemetry: bool,
    /// }
    ///
    ///
    /// # tokio_test::block_on(async {
    ///
    /// let cli = Cli { no_telemetry: false, };
    ///
    /// let (recorder, worker) = builder!()
    ///   .set_enable_reporting(!cli.no_telemetry)
    ///   .build_or_default()
    ///   .await;
    /// # })
    /// ```
    pub fn set_enable_reporting(mut self, enable_reporting: bool) -> Self {
        self.enable_reporting = enable_reporting;
        self
    }

    pub fn set_timeout(mut self, duration: Option<Duration>) -> Self {
        self.timeout = duration;
        self
    }

    pub fn set_certificate(mut self, certificate: Option<Certificate>) -> Self {
        self.certificate = certificate;
        self
    }

    pub fn set_proxy(mut self, proxy: Option<Url>) -> Self {
        self.proxy = proxy;
        self
    }

    #[tracing::instrument(skip(self))]
    pub async fn try_build(mut self) -> Result<(Recorder, Worker), TransportsError> {
        let transport = self.transport().await?;

        Ok(self
            .build_with_transport_snapshotter(transport, crate::system_snapshot::Generic::default())
            .await)
    }

    #[tracing::instrument(skip(self))]
    pub async fn build_or_default(mut self) -> (Recorder, Worker) {
        let transport = self.transport_or_default().await;

        self.build_with_transport_snapshotter(transport, crate::system_snapshot::Generic::default())
            .await
    }

    #[tracing::instrument(skip(self, snapshotter))]
    pub async fn try_build_with_snapshotter<S: SystemSnapshotter>(
        mut self,
        snapshotter: S,
    ) -> Result<(Recorder, Worker), TransportsError> {
        let transport = self.transport().await?;

        Ok(self
            .build_with_transport_snapshotter(transport, snapshotter)
            .await)
    }

    #[tracing::instrument(skip(self, snapshotter))]
    pub async fn build_or_default_with_snapshotter<S: SystemSnapshotter>(
        mut self,
        snapshotter: S,
    ) -> (Recorder, Worker) {
        let transport = self.transport_or_default().await;

        self.build_with_transport_snapshotter(transport, snapshotter)
            .await
    }

    #[tracing::instrument(skip(self, transport, snapshotter))]
    async fn build_with_transport_snapshotter<S: SystemSnapshotter>(
        &mut self,
        transport: crate::transport::Transports,
        snapshotter: S,
    ) -> (Recorder, Worker) {
        Worker::new(
            self.distinct_id.take(),
            self.device_id.take(),
            self.facts.take(),
            self.groups.take(),
            snapshotter,
            transport,
        )
        .await
    }

    async fn transport_or_default(&mut self) -> crate::transport::Transports {
        match self.transport().await {
            Ok(t) => {
                return t;
            }
            Err(e) => {
                tracing::warn!(%e, "Failed to construct the transport as configured, falling back to the default");
            }
        }

        match crate::transport::Transports::try_new(
            None,
            self.timeout
                .take()
                .unwrap_or_else(|| Duration::from_secs(3)),
            None,
            None,
        )
        .await
        {
            Ok(t) => {
                return t;
            }
            Err(e) => {
                tracing::warn!(%e, "Failed to construct the default transport, falling back to none");
            }
        }

        crate::transport::Transports::none()
    }

    async fn transport(&mut self) -> Result<crate::transport::Transports, TransportsError> {
        if self.enable_reporting {
            crate::transport::Transports::try_new(
                self.endpoint.take(),
                self.timeout.unwrap_or_else(|| Duration::from_secs(3)),
                self.certificate.take(),
                self.proxy.take(),
            )
            .await
        } else {
            Ok(crate::transport::Transports::none())
        }
    }
}