1use reqwest;
13use serde::{Deserialize, Serialize};
14use crate::{apis::ResponseContent, models};
15use super::{Error, configuration};
16
17
18#[derive(Debug, Clone, Serialize, Deserialize)]
20#[serde(untagged)]
21pub enum GetAssetsGraphError {
22 UnknownValue(serde_json::Value),
23}
24
25#[derive(Debug, Clone, Serialize, Deserialize)]
27#[serde(untagged)]
28pub enum ListAssetPartitionsError {
29 UnknownValue(serde_json::Value),
30}
31
32#[derive(Debug, Clone, Serialize, Deserialize)]
34#[serde(untagged)]
35pub enum ListAssetPartitionsInRangeError {
36 UnknownValue(serde_json::Value),
37}
38
39#[derive(Debug, Clone, Serialize, Deserialize)]
41#[serde(untagged)]
42pub enum ListAssetSchemasError {
43 UnknownValue(serde_json::Value),
44}
45
46#[derive(Debug, Clone, Serialize, Deserialize)]
48#[serde(untagged)]
49pub enum ListAssetsError {
50 UnknownValue(serde_json::Value),
51}
52
53#[derive(Debug, Clone, Serialize, Deserialize)]
55#[serde(untagged)]
56pub enum ListAssetsByUsageError {
57 UnknownValue(serde_json::Value),
58}
59
60#[derive(Debug, Clone, Serialize, Deserialize)]
62#[serde(untagged)]
63pub enum ListFavoriteAssetsError {
64 UnknownValue(serde_json::Value),
65}
66
67#[derive(Debug, Clone, Serialize, Deserialize)]
69#[serde(untagged)]
70pub enum ListPipelineFoldersError {
71 UnknownValue(serde_json::Value),
72}
73
74#[derive(Debug, Clone, Serialize, Deserialize)]
76#[serde(untagged)]
77pub enum ListWorkspaceMacrosError {
78 UnknownValue(serde_json::Value),
79}
80
81
82pub async fn get_assets_graph(configuration: &configuration::Configuration, workspace: &str, asset_kinds: Option<&str>, folder: Option<&str>, dbt_script_hash: Option<&str>) -> Result<models::AssetGraph, Error<GetAssetsGraphError>> {
83 let local_var_configuration = configuration;
84
85 let local_var_client = &local_var_configuration.client;
86
87 let local_var_uri_str = format!("{}/w/{workspace}/assets/graph", local_var_configuration.base_path, workspace=crate::apis::urlencode(workspace));
88 let mut local_var_req_builder = local_var_client.request(reqwest::Method::GET, local_var_uri_str.as_str());
89
90 if let Some(ref local_var_str) = asset_kinds {
91 local_var_req_builder = local_var_req_builder.query(&[("asset_kinds", &local_var_str.to_string())]);
92 }
93 if let Some(ref local_var_str) = folder {
94 local_var_req_builder = local_var_req_builder.query(&[("folder", &local_var_str.to_string())]);
95 }
96 if let Some(ref local_var_str) = dbt_script_hash {
97 local_var_req_builder = local_var_req_builder.query(&[("dbt_script_hash", &local_var_str.to_string())]);
98 }
99 if let Some(ref local_var_user_agent) = local_var_configuration.user_agent {
100 local_var_req_builder = local_var_req_builder.header(reqwest::header::USER_AGENT, local_var_user_agent.clone());
101 }
102 if let Some(ref local_var_token) = local_var_configuration.bearer_access_token {
103 local_var_req_builder = local_var_req_builder.bearer_auth(local_var_token.to_owned());
104 };
105
106 let local_var_req = local_var_req_builder.build()?;
107 let local_var_resp = local_var_client.execute(local_var_req).await?;
108
109 let local_var_status = local_var_resp.status();
110 let local_var_content = local_var_resp.text().await?;
111
112 if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
113 crate::from_str_patched(&local_var_content).map_err(Error::from)
114 } else {
115 let local_var_entity: Option<GetAssetsGraphError> = crate::from_str_patched(&local_var_content).ok();
116 let local_var_error = ResponseContent { status: local_var_status, content: local_var_content, entity: local_var_entity };
117 Err(Error::ResponseError(local_var_error))
118 }
119}
120
121pub async fn list_asset_partitions(configuration: &configuration::Configuration, workspace: &str, path: &str) -> Result<Vec<models::MaterializedPartition>, Error<ListAssetPartitionsError>> {
122 let local_var_configuration = configuration;
123
124 let local_var_client = &local_var_configuration.client;
125
126 let local_var_uri_str = format!("{}/w/{workspace}/assets/partitions", local_var_configuration.base_path, workspace=crate::apis::urlencode(workspace));
127 let mut local_var_req_builder = local_var_client.request(reqwest::Method::GET, local_var_uri_str.as_str());
128
129 local_var_req_builder = local_var_req_builder.query(&[("path", &path.to_string())]);
130 if let Some(ref local_var_user_agent) = local_var_configuration.user_agent {
131 local_var_req_builder = local_var_req_builder.header(reqwest::header::USER_AGENT, local_var_user_agent.clone());
132 }
133 if let Some(ref local_var_token) = local_var_configuration.bearer_access_token {
134 local_var_req_builder = local_var_req_builder.bearer_auth(local_var_token.to_owned());
135 };
136
137 let local_var_req = local_var_req_builder.build()?;
138 let local_var_resp = local_var_client.execute(local_var_req).await?;
139
140 let local_var_status = local_var_resp.status();
141 let local_var_content = local_var_resp.text().await?;
142
143 if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
144 crate::from_str_patched(&local_var_content).map_err(Error::from)
145 } else {
146 let local_var_entity: Option<ListAssetPartitionsError> = crate::from_str_patched(&local_var_content).ok();
147 let local_var_error = ResponseContent { status: local_var_status, content: local_var_content, entity: local_var_entity };
148 Err(Error::ResponseError(local_var_error))
149 }
150}
151
152pub async fn list_asset_partitions_in_range(configuration: &configuration::Configuration, workspace: &str, path: &str, from: String, to: String) -> Result<models::PartitionsInRange, Error<ListAssetPartitionsInRangeError>> {
153 let local_var_configuration = configuration;
154
155 let local_var_client = &local_var_configuration.client;
156
157 let local_var_uri_str = format!("{}/w/{workspace}/assets/partitions_in_range", local_var_configuration.base_path, workspace=crate::apis::urlencode(workspace));
158 let mut local_var_req_builder = local_var_client.request(reqwest::Method::GET, local_var_uri_str.as_str());
159
160 local_var_req_builder = local_var_req_builder.query(&[("path", &path.to_string())]);
161 local_var_req_builder = local_var_req_builder.query(&[("from", &from.to_string())]);
162 local_var_req_builder = local_var_req_builder.query(&[("to", &to.to_string())]);
163 if let Some(ref local_var_user_agent) = local_var_configuration.user_agent {
164 local_var_req_builder = local_var_req_builder.header(reqwest::header::USER_AGENT, local_var_user_agent.clone());
165 }
166 if let Some(ref local_var_token) = local_var_configuration.bearer_access_token {
167 local_var_req_builder = local_var_req_builder.bearer_auth(local_var_token.to_owned());
168 };
169
170 let local_var_req = local_var_req_builder.build()?;
171 let local_var_resp = local_var_client.execute(local_var_req).await?;
172
173 let local_var_status = local_var_resp.status();
174 let local_var_content = local_var_resp.text().await?;
175
176 if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
177 crate::from_str_patched(&local_var_content).map_err(Error::from)
178 } else {
179 let local_var_entity: Option<ListAssetPartitionsInRangeError> = crate::from_str_patched(&local_var_content).ok();
180 let local_var_error = ResponseContent { status: local_var_status, content: local_var_content, entity: local_var_entity };
181 Err(Error::ResponseError(local_var_error))
182 }
183}
184
185pub async fn list_asset_schemas(configuration: &configuration::Configuration, workspace: &str, path: &str) -> Result<Vec<models::AssetSchemaVersion>, Error<ListAssetSchemasError>> {
186 let local_var_configuration = configuration;
187
188 let local_var_client = &local_var_configuration.client;
189
190 let local_var_uri_str = format!("{}/w/{workspace}/assets/asset_schemas", local_var_configuration.base_path, workspace=crate::apis::urlencode(workspace));
191 let mut local_var_req_builder = local_var_client.request(reqwest::Method::GET, local_var_uri_str.as_str());
192
193 local_var_req_builder = local_var_req_builder.query(&[("path", &path.to_string())]);
194 if let Some(ref local_var_user_agent) = local_var_configuration.user_agent {
195 local_var_req_builder = local_var_req_builder.header(reqwest::header::USER_AGENT, local_var_user_agent.clone());
196 }
197 if let Some(ref local_var_token) = local_var_configuration.bearer_access_token {
198 local_var_req_builder = local_var_req_builder.bearer_auth(local_var_token.to_owned());
199 };
200
201 let local_var_req = local_var_req_builder.build()?;
202 let local_var_resp = local_var_client.execute(local_var_req).await?;
203
204 let local_var_status = local_var_resp.status();
205 let local_var_content = local_var_resp.text().await?;
206
207 if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
208 crate::from_str_patched(&local_var_content).map_err(Error::from)
209 } else {
210 let local_var_entity: Option<ListAssetSchemasError> = crate::from_str_patched(&local_var_content).ok();
211 let local_var_error = ResponseContent { status: local_var_status, content: local_var_content, entity: local_var_entity };
212 Err(Error::ResponseError(local_var_error))
213 }
214}
215
216pub async fn list_assets(configuration: &configuration::Configuration, workspace: &str, per_page: Option<i32>, cursor_created_at: Option<String>, cursor_id: Option<i64>, asset_path: Option<&str>, usage_path: Option<&str>, asset_kinds: Option<&str>, path: Option<&str>, columns: Option<&str>, broad_filter: Option<&str>) -> Result<models::ListAssets200Response, Error<ListAssetsError>> {
217 let local_var_configuration = configuration;
218
219 let local_var_client = &local_var_configuration.client;
220
221 let local_var_uri_str = format!("{}/w/{workspace}/assets/list", local_var_configuration.base_path, workspace=crate::apis::urlencode(workspace));
222 let mut local_var_req_builder = local_var_client.request(reqwest::Method::GET, local_var_uri_str.as_str());
223
224 if let Some(ref local_var_str) = per_page {
225 local_var_req_builder = local_var_req_builder.query(&[("per_page", &local_var_str.to_string())]);
226 }
227 if let Some(ref local_var_str) = cursor_created_at {
228 local_var_req_builder = local_var_req_builder.query(&[("cursor_created_at", &local_var_str.to_string())]);
229 }
230 if let Some(ref local_var_str) = cursor_id {
231 local_var_req_builder = local_var_req_builder.query(&[("cursor_id", &local_var_str.to_string())]);
232 }
233 if let Some(ref local_var_str) = asset_path {
234 local_var_req_builder = local_var_req_builder.query(&[("asset_path", &local_var_str.to_string())]);
235 }
236 if let Some(ref local_var_str) = usage_path {
237 local_var_req_builder = local_var_req_builder.query(&[("usage_path", &local_var_str.to_string())]);
238 }
239 if let Some(ref local_var_str) = asset_kinds {
240 local_var_req_builder = local_var_req_builder.query(&[("asset_kinds", &local_var_str.to_string())]);
241 }
242 if let Some(ref local_var_str) = path {
243 local_var_req_builder = local_var_req_builder.query(&[("path", &local_var_str.to_string())]);
244 }
245 if let Some(ref local_var_str) = columns {
246 local_var_req_builder = local_var_req_builder.query(&[("columns", &local_var_str.to_string())]);
247 }
248 if let Some(ref local_var_str) = broad_filter {
249 local_var_req_builder = local_var_req_builder.query(&[("broad_filter", &local_var_str.to_string())]);
250 }
251 if let Some(ref local_var_user_agent) = local_var_configuration.user_agent {
252 local_var_req_builder = local_var_req_builder.header(reqwest::header::USER_AGENT, local_var_user_agent.clone());
253 }
254 if let Some(ref local_var_token) = local_var_configuration.bearer_access_token {
255 local_var_req_builder = local_var_req_builder.bearer_auth(local_var_token.to_owned());
256 };
257
258 let local_var_req = local_var_req_builder.build()?;
259 let local_var_resp = local_var_client.execute(local_var_req).await?;
260
261 let local_var_status = local_var_resp.status();
262 let local_var_content = local_var_resp.text().await?;
263
264 if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
265 crate::from_str_patched(&local_var_content).map_err(Error::from)
266 } else {
267 let local_var_entity: Option<ListAssetsError> = crate::from_str_patched(&local_var_content).ok();
268 let local_var_error = ResponseContent { status: local_var_status, content: local_var_content, entity: local_var_entity };
269 Err(Error::ResponseError(local_var_error))
270 }
271}
272
273pub async fn list_assets_by_usage(configuration: &configuration::Configuration, workspace: &str, list_assets_by_usage_request: models::ListAssetsByUsageRequest) -> Result<Vec<Vec<models::ListAssetsByUsage200ResponseInnerInner>>, Error<ListAssetsByUsageError>> {
274 let local_var_configuration = configuration;
275
276 let local_var_client = &local_var_configuration.client;
277
278 let local_var_uri_str = format!("{}/w/{workspace}/assets/list_by_usages", local_var_configuration.base_path, workspace=crate::apis::urlencode(workspace));
279 let mut local_var_req_builder = local_var_client.request(reqwest::Method::POST, local_var_uri_str.as_str());
280
281 if let Some(ref local_var_user_agent) = local_var_configuration.user_agent {
282 local_var_req_builder = local_var_req_builder.header(reqwest::header::USER_AGENT, local_var_user_agent.clone());
283 }
284 if let Some(ref local_var_token) = local_var_configuration.bearer_access_token {
285 local_var_req_builder = local_var_req_builder.bearer_auth(local_var_token.to_owned());
286 };
287 local_var_req_builder = local_var_req_builder.json(&list_assets_by_usage_request);
288
289 let local_var_req = local_var_req_builder.build()?;
290 let local_var_resp = local_var_client.execute(local_var_req).await?;
291
292 let local_var_status = local_var_resp.status();
293 let local_var_content = local_var_resp.text().await?;
294
295 if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
296 crate::from_str_patched(&local_var_content).map_err(Error::from)
297 } else {
298 let local_var_entity: Option<ListAssetsByUsageError> = crate::from_str_patched(&local_var_content).ok();
299 let local_var_error = ResponseContent { status: local_var_status, content: local_var_content, entity: local_var_entity };
300 Err(Error::ResponseError(local_var_error))
301 }
302}
303
304pub async fn list_favorite_assets(configuration: &configuration::Configuration, workspace: &str) -> Result<Vec<models::ListFavoriteAssets200ResponseInner>, Error<ListFavoriteAssetsError>> {
305 let local_var_configuration = configuration;
306
307 let local_var_client = &local_var_configuration.client;
308
309 let local_var_uri_str = format!("{}/w/{workspace}/assets/list_favorites", local_var_configuration.base_path, workspace=crate::apis::urlencode(workspace));
310 let mut local_var_req_builder = local_var_client.request(reqwest::Method::GET, local_var_uri_str.as_str());
311
312 if let Some(ref local_var_user_agent) = local_var_configuration.user_agent {
313 local_var_req_builder = local_var_req_builder.header(reqwest::header::USER_AGENT, local_var_user_agent.clone());
314 }
315 if let Some(ref local_var_token) = local_var_configuration.bearer_access_token {
316 local_var_req_builder = local_var_req_builder.bearer_auth(local_var_token.to_owned());
317 };
318
319 let local_var_req = local_var_req_builder.build()?;
320 let local_var_resp = local_var_client.execute(local_var_req).await?;
321
322 let local_var_status = local_var_resp.status();
323 let local_var_content = local_var_resp.text().await?;
324
325 if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
326 crate::from_str_patched(&local_var_content).map_err(Error::from)
327 } else {
328 let local_var_entity: Option<ListFavoriteAssetsError> = crate::from_str_patched(&local_var_content).ok();
329 let local_var_error = ResponseContent { status: local_var_status, content: local_var_content, entity: local_var_entity };
330 Err(Error::ResponseError(local_var_error))
331 }
332}
333
334pub async fn list_pipeline_folders(configuration: &configuration::Configuration, workspace: &str) -> Result<Vec<models::ListPipelineFolders200ResponseInner>, Error<ListPipelineFoldersError>> {
335 let local_var_configuration = configuration;
336
337 let local_var_client = &local_var_configuration.client;
338
339 let local_var_uri_str = format!("{}/w/{workspace}/assets/pipelines", local_var_configuration.base_path, workspace=crate::apis::urlencode(workspace));
340 let mut local_var_req_builder = local_var_client.request(reqwest::Method::GET, local_var_uri_str.as_str());
341
342 if let Some(ref local_var_user_agent) = local_var_configuration.user_agent {
343 local_var_req_builder = local_var_req_builder.header(reqwest::header::USER_AGENT, local_var_user_agent.clone());
344 }
345 if let Some(ref local_var_token) = local_var_configuration.bearer_access_token {
346 local_var_req_builder = local_var_req_builder.bearer_auth(local_var_token.to_owned());
347 };
348
349 let local_var_req = local_var_req_builder.build()?;
350 let local_var_resp = local_var_client.execute(local_var_req).await?;
351
352 let local_var_status = local_var_resp.status();
353 let local_var_content = local_var_resp.text().await?;
354
355 if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
356 crate::from_str_patched(&local_var_content).map_err(Error::from)
357 } else {
358 let local_var_entity: Option<ListPipelineFoldersError> = crate::from_str_patched(&local_var_content).ok();
359 let local_var_error = ResponseContent { status: local_var_status, content: local_var_content, entity: local_var_entity };
360 Err(Error::ResponseError(local_var_error))
361 }
362}
363
364pub async fn list_workspace_macros(configuration: &configuration::Configuration, workspace: &str) -> Result<Vec<models::ListWorkspaceMacros200ResponseInner>, Error<ListWorkspaceMacrosError>> {
365 let local_var_configuration = configuration;
366
367 let local_var_client = &local_var_configuration.client;
368
369 let local_var_uri_str = format!("{}/w/{workspace}/assets/macros", local_var_configuration.base_path, workspace=crate::apis::urlencode(workspace));
370 let mut local_var_req_builder = local_var_client.request(reqwest::Method::GET, local_var_uri_str.as_str());
371
372 if let Some(ref local_var_user_agent) = local_var_configuration.user_agent {
373 local_var_req_builder = local_var_req_builder.header(reqwest::header::USER_AGENT, local_var_user_agent.clone());
374 }
375 if let Some(ref local_var_token) = local_var_configuration.bearer_access_token {
376 local_var_req_builder = local_var_req_builder.bearer_auth(local_var_token.to_owned());
377 };
378
379 let local_var_req = local_var_req_builder.build()?;
380 let local_var_resp = local_var_client.execute(local_var_req).await?;
381
382 let local_var_status = local_var_resp.status();
383 let local_var_content = local_var_resp.text().await?;
384
385 if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
386 crate::from_str_patched(&local_var_content).map_err(Error::from)
387 } else {
388 let local_var_entity: Option<ListWorkspaceMacrosError> = crate::from_str_patched(&local_var_content).ok();
389 let local_var_error = ResponseContent { status: local_var_status, content: local_var_content, entity: local_var_entity };
390 Err(Error::ResponseError(local_var_error))
391 }
392}
393