Skip to main content

babelforce_manager_sdk/resources/
logs.rs

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