Skip to main content

babelforce_manager_sdk/resources/
logs.rs

1use crate::error::{map_manager_err, ManagerError};
2use crate::gen::manager::apis::configuration::Configuration;
3use crate::gen::manager::apis::manager_api as api;
4use crate::gen::manager::models;
5use crate::http::collect_all;
6use crate::retry::{with_retry, RetryPolicy};
7use crate::token::SharedCfg;
8
9/// Logs — the request audit log (`/api/v2/audit/request`) and the live log (`/api/v2/logs`).
10pub struct LogsResource {
11    pub(crate) cfg: SharedCfg<Configuration>,
12    pub(crate) retry: RetryPolicy,
13}
14
15impl LogsResource {
16    /// Collect every audit-log entry across pages.
17    pub async fn audit_all(&self) -> Result<Vec<models::AuditLog>, ManagerError> {
18        let cfg = self.cfg.get().await?;
19        let cfg = cfg.as_ref();
20        collect_all(&self.retry, map_manager_err, |page| async move {
21            let r = api::list_audit_logs(cfg, None, None, None, Some(page), None, None, None, None)
22                .await?;
23            Ok((
24                r.items,
25                r.pagination.pages.unwrap_or(1),
26                r.pagination.current.unwrap_or(1),
27            ))
28        })
29        .await
30    }
31
32    /// Read the live log.
33    pub async fn live(&self) -> Result<Vec<models::LiveLog>, ManagerError> {
34        let cfg = self.cfg.get().await?;
35        let cfg = cfg.as_ref();
36        let resp = with_retry(&self.retry, true, || api::list_live_logs(cfg, None))
37            .await
38            .map_err(map_manager_err)?;
39        Ok(resp.items)
40    }
41
42    /// Enable live logging.
43    pub async fn enable_live(&self) -> Result<models::DefaultV2MessageResponse, ManagerError> {
44        let cfg = self.cfg.get().await?;
45        let cfg = cfg.as_ref();
46        with_retry(&self.retry, false, || api::enable_live_logging(cfg))
47            .await
48            .map_err(map_manager_err)
49    }
50
51    /// Disable live logging.
52    pub async fn disable_live(&self) -> Result<models::DefaultV2MessageResponse, ManagerError> {
53        let cfg = self.cfg.get().await?;
54        let cfg = cfg.as_ref();
55        with_retry(&self.retry, false, || api::disable_live_logging(cfg))
56            .await
57            .map_err(map_manager_err)
58    }
59
60    /// Write a single entry to the live log.
61    pub async fn write(
62        &self,
63        body: models::WriteLogRequest,
64    ) -> Result<models::GenericItemResponse, ManagerError> {
65        let cfg = self.cfg.get().await?;
66        let cfg = cfg.as_ref();
67        with_retry(&self.retry, false, || api::write_log(cfg, body.clone()))
68            .await
69            .map_err(map_manager_err)
70    }
71}