rippling_api/
leave_balances.rs

1use anyhow::Result;
2
3use crate::Client;
4#[derive(Clone, Debug)]
5pub struct LeaveBalances {
6    pub client: Client,
7}
8
9impl LeaveBalances {
10    #[doc(hidden)]
11    pub fn new(client: Client) -> Self {
12        Self { client }
13    }
14
15    #[doc = "List leave balances\n\nA List of leave balances\n- Requires: `API Tier 2`\n- Filterable fields: `worker_id`, `leave_type_id`\n- Expandable fields: `worker`, `leave_type`\n- Sortable fields: `id`, `created_at`, `updated_at`\n\n**Parameters:**\n\n- `cursor: Option<String>`\n- `expand: Option<String>`\n- `filter: Option<String>`\n- `order_by: Option<String>`\n\n```rust,no_run\nuse futures_util::TryStreamExt;\nasync fn example_leave_balances_list_stream() -> anyhow::Result<()> {\n    let client = rippling_api::Client::new_from_env();\n    let mut leave_balances = client.leave_balances();\n    let mut stream = leave_balances.list_stream(\n        Some(\"some-string\".to_string()),\n        Some(\"some-string\".to_string()),\n        Some(\"some-string\".to_string()),\n    );\n    loop {\n        match stream.try_next().await {\n            Ok(Some(item)) => {\n                println!(\"{:?}\", item);\n            }\n            Ok(None) => {\n                break;\n            }\n            Err(err) => {\n                return Err(err.into());\n            }\n        }\n    }\n\n    Ok(())\n}\n```"]
16    #[tracing::instrument]
17    pub async fn list<'a>(
18        &'a self,
19        cursor: Option<String>,
20        expand: Option<String>,
21        filter: Option<String>,
22        order_by: Option<String>,
23    ) -> Result<crate::types::ListLeaveBalancesResponse, crate::types::error::Error> {
24        let mut req = self.client.client.request(
25            http::Method::GET,
26            format!("{}/{}", self.client.base_url, "leave-balances"),
27        );
28        req = req.bearer_auth(&self.client.token);
29        let mut query_params = vec![];
30        if let Some(p) = cursor {
31            query_params.push(("cursor", p));
32        }
33
34        if let Some(p) = expand {
35            query_params.push(("expand", p));
36        }
37
38        if let Some(p) = filter {
39            query_params.push(("filter", p));
40        }
41
42        if let Some(p) = order_by {
43            query_params.push(("order_by", p));
44        }
45
46        req = req.query(&query_params);
47        let resp = req.send().await?;
48        let status = resp.status();
49        if status.is_success() {
50            let text = resp.text().await.unwrap_or_default();
51            serde_json::from_str(&text).map_err(|err| {
52                crate::types::error::Error::from_serde_error(
53                    format_serde_error::SerdeError::new(text.to_string(), err),
54                    status,
55                )
56            })
57        } else {
58            let text = resp.text().await.unwrap_or_default();
59            Err(crate::types::error::Error::Server {
60                body: text.to_string(),
61                status,
62            })
63        }
64    }
65
66    #[doc = "List leave balances\n\nA List of leave balances\n- Requires: `API Tier 2`\n- Filterable fields: `worker_id`, `leave_type_id`\n- Expandable fields: `worker`, `leave_type`\n- Sortable fields: `id`, `created_at`, `updated_at`\n\n**Parameters:**\n\n- `cursor: Option<String>`\n- `expand: Option<String>`\n- `filter: Option<String>`\n- `order_by: Option<String>`\n\n```rust,no_run\nuse futures_util::TryStreamExt;\nasync fn example_leave_balances_list_stream() -> anyhow::Result<()> {\n    let client = rippling_api::Client::new_from_env();\n    let mut leave_balances = client.leave_balances();\n    let mut stream = leave_balances.list_stream(\n        Some(\"some-string\".to_string()),\n        Some(\"some-string\".to_string()),\n        Some(\"some-string\".to_string()),\n    );\n    loop {\n        match stream.try_next().await {\n            Ok(Some(item)) => {\n                println!(\"{:?}\", item);\n            }\n            Ok(None) => {\n                break;\n            }\n            Err(err) => {\n                return Err(err.into());\n            }\n        }\n    }\n\n    Ok(())\n}\n```"]
67    #[tracing::instrument]
68    #[cfg(not(feature = "js"))]
69    pub fn list_stream<'a>(
70        &'a self,
71        expand: Option<String>,
72        filter: Option<String>,
73        order_by: Option<String>,
74    ) -> impl futures::Stream<Item = Result<crate::types::LeaveBalance, crate::types::error::Error>>
75           + Unpin
76           + '_ {
77        use futures::{StreamExt, TryFutureExt, TryStreamExt};
78
79        use crate::types::paginate::Pagination;
80        self.list(None, expand, filter, order_by)
81            .map_ok(move |result| {
82                let items = futures::stream::iter(result.items().into_iter().map(Ok));
83                let next_pages = futures::stream::try_unfold(
84                    (None, result),
85                    move |(prev_page_token, new_result)| async move {
86                        if new_result.has_more_pages()
87                            && !new_result.items().is_empty()
88                            && prev_page_token != new_result.next_page_token()
89                        {
90                            async {
91                                let mut req = self.client.client.request(
92                                    http::Method::GET,
93                                    format!("{}/{}", self.client.base_url, "leave-balances"),
94                                );
95                                req = req.bearer_auth(&self.client.token);
96                                let mut request = req.build()?;
97                                request = new_result.next_page(request)?;
98                                let resp = self.client.client.execute(request).await?;
99                                let status = resp.status();
100                                if status.is_success() {
101                                    let text = resp.text().await.unwrap_or_default();
102                                    serde_json::from_str(&text).map_err(|err| {
103                                        crate::types::error::Error::from_serde_error(
104                                            format_serde_error::SerdeError::new(
105                                                text.to_string(),
106                                                err,
107                                            ),
108                                            status,
109                                        )
110                                    })
111                                } else {
112                                    let text = resp.text().await.unwrap_or_default();
113                                    Err(crate::types::error::Error::Server {
114                                        body: text.to_string(),
115                                        status,
116                                    })
117                                }
118                            }
119                            .map_ok(|result: crate::types::ListLeaveBalancesResponse| {
120                                Some((
121                                    futures::stream::iter(result.items().into_iter().map(Ok)),
122                                    (new_result.next_page_token(), result),
123                                ))
124                            })
125                            .await
126                        } else {
127                            Ok(None)
128                        }
129                    },
130                )
131                .try_flatten();
132                items.chain(next_pages)
133            })
134            .try_flatten_stream()
135            .boxed()
136    }
137
138    #[doc = "Retrieve a specific leave balance\n\nRetrieve a specific leave balance\n\n**Parameters:**\n\n- `expand: Option<String>`\n- `id: &'astr`: ID of the resource to return (required)\n\n```rust,no_run\nasync fn example_leave_balances_get() -> anyhow::Result<()> {\n    let client = rippling_api::Client::new_from_env();\n    let result: rippling_api::types::GetLeaveBalancesResponse = client\n        .leave_balances()\n        .get(\n            Some(\"some-string\".to_string()),\n            \"d9797f8d-9ad6-4e08-90d7-2ec17e13471c\",\n        )\n        .await?;\n    println!(\"{:?}\", result);\n    Ok(())\n}\n```"]
139    #[tracing::instrument]
140    pub async fn get<'a>(
141        &'a self,
142        expand: Option<String>,
143        id: &'a str,
144    ) -> Result<crate::types::GetLeaveBalancesResponse, crate::types::error::Error> {
145        let mut req = self.client.client.request(
146            http::Method::GET,
147            format!(
148                "{}/{}",
149                self.client.base_url,
150                "leave-balances/{id}".replace("{id}", id)
151            ),
152        );
153        req = req.bearer_auth(&self.client.token);
154        let mut query_params = vec![];
155        if let Some(p) = expand {
156            query_params.push(("expand", p));
157        }
158
159        req = req.query(&query_params);
160        let resp = req.send().await?;
161        let status = resp.status();
162        if status.is_success() {
163            let text = resp.text().await.unwrap_or_default();
164            serde_json::from_str(&text).map_err(|err| {
165                crate::types::error::Error::from_serde_error(
166                    format_serde_error::SerdeError::new(text.to_string(), err),
167                    status,
168                )
169            })
170        } else {
171            let text = resp.text().await.unwrap_or_default();
172            Err(crate::types::error::Error::Server {
173                body: text.to_string(),
174                status,
175            })
176        }
177    }
178}