use crate::internal::path_escape;
use crate::runtime::{self, Error};
#[derive(Clone, Debug, Default)]
pub struct MembershipsListUserOptions {
pub limit: Option<i64>,
pub offset: Option<i64>,
}
#[derive(Clone, Debug, Default)]
pub struct MembershipsListGroupOptions {
pub limit: Option<i64>,
pub offset: Option<i64>,
}
#[derive(Clone, Debug, Default)]
pub struct MembershipsCreateGroupOptions {
pub fields: Option<Vec<String>>,
}
#[derive(Clone, Debug, Default)]
pub struct MembershipsGetGroupOptions {
pub fields: Option<Vec<String>>,
}
#[derive(Clone, Debug, Default)]
pub struct MembershipsUpdateGroupOptions {
pub fields: Option<Vec<String>>,
}
pub struct MembershipsListUserPaginator {
manager: MembershipsManager,
user_id: String,
options: MembershipsListUserOptions,
buffer: std::vec::IntoIter<crate::models::schemas::GroupMembership>,
done: bool,
}
impl MembershipsListUserPaginator {
pub async fn next(&mut self) -> Option<Result<crate::models::schemas::GroupMembership, Error>> {
loop {
if let Some(item) = self.buffer.next() {
return Some(Ok(item));
}
if self.done {
return None;
}
let page = match self
.manager
.list_user_page(self.user_id.clone(), Some(self.options.clone()))
.await
{
Ok(page) => page,
Err(err) => {
self.done = true;
return Some(Err(err));
}
};
let items = page.entries.unwrap_or_default();
if items.is_empty() {
self.done = true;
} else {
let next = self.options.offset.unwrap_or(0) + items.len() as i64;
self.options.offset = Some(next);
}
self.buffer = items.into_iter();
}
}
}
pub struct MembershipsListGroupPaginator {
manager: MembershipsManager,
group_id: String,
options: MembershipsListGroupOptions,
buffer: std::vec::IntoIter<crate::models::schemas::GroupMembership>,
done: bool,
}
impl MembershipsListGroupPaginator {
pub async fn next(&mut self) -> Option<Result<crate::models::schemas::GroupMembership, Error>> {
loop {
if let Some(item) = self.buffer.next() {
return Some(Ok(item));
}
if self.done {
return None;
}
let page = match self
.manager
.list_group_page(self.group_id.clone(), Some(self.options.clone()))
.await
{
Ok(page) => page,
Err(err) => {
self.done = true;
return Some(Err(err));
}
};
let items = page.entries.unwrap_or_default();
if items.is_empty() {
self.done = true;
} else {
let next = self.options.offset.unwrap_or(0) + items.len() as i64;
self.options.offset = Some(next);
}
self.buffer = items.into_iter();
}
}
}
pub struct MembershipsManager {
session: std::sync::Arc<runtime::Client>,
}
impl MembershipsManager {
pub(crate) fn new(session: std::sync::Arc<runtime::Client>) -> Self {
Self { session }
}
async fn list_user_page(
&self,
user_id: String,
opts: Option<MembershipsListUserOptions>,
) -> Result<crate::models::schemas::GroupMemberships, Error> {
let mut url = self.session.base_url("api");
url.push_str("/users");
url.push('/');
let seg = path_escape(&user_id);
url.push_str(&seg);
url.push_str("/memberships");
let mut req = self.session.new_request("GET", &url);
let opts = opts.unwrap_or_default();
if let Some(value) = opts.limit {
req = runtime::with_query(req, "limit", &value.to_string());
}
if let Some(value) = opts.offset {
req = runtime::with_query(req, "offset", &value.to_string());
}
let resp = self.session.fetch(req).await?;
let data = runtime::response_bytes(&resp)?;
Ok(serde_json::from_slice(&data)?)
}
pub fn list_user(
&self,
user_id: String,
opts: Option<MembershipsListUserOptions>,
) -> MembershipsListUserPaginator {
MembershipsListUserPaginator {
manager: MembershipsManager::new(self.session.clone()),
user_id,
options: opts.unwrap_or_default(),
buffer: Vec::new().into_iter(),
done: false,
}
}
async fn list_group_page(
&self,
group_id: String,
opts: Option<MembershipsListGroupOptions>,
) -> Result<crate::models::schemas::GroupMemberships, Error> {
let mut url = self.session.base_url("api");
url.push_str("/groups");
url.push('/');
let seg = path_escape(&group_id);
url.push_str(&seg);
url.push_str("/memberships");
let mut req = self.session.new_request("GET", &url);
let opts = opts.unwrap_or_default();
if let Some(value) = opts.limit {
req = runtime::with_query(req, "limit", &value.to_string());
}
if let Some(value) = opts.offset {
req = runtime::with_query(req, "offset", &value.to_string());
}
let resp = self.session.fetch(req).await?;
let data = runtime::response_bytes(&resp)?;
Ok(serde_json::from_slice(&data)?)
}
pub fn list_group(
&self,
group_id: String,
opts: Option<MembershipsListGroupOptions>,
) -> MembershipsListGroupPaginator {
MembershipsListGroupPaginator {
manager: MembershipsManager::new(self.session.clone()),
group_id,
options: opts.unwrap_or_default(),
buffer: Vec::new().into_iter(),
done: false,
}
}
pub async fn create_group(
&self,
body: crate::models::schemas::GroupCreateRequest2,
opts: Option<MembershipsCreateGroupOptions>,
) -> Result<crate::models::schemas::GroupMembership, Error> {
let mut url = self.session.base_url("api");
url.push_str("/group_memberships");
let mut req = self.session.new_request("POST", &url);
let opts = opts.unwrap_or_default();
if let Some(value) = opts.fields {
req = runtime::with_query(req, "fields", &value.join(","));
}
let payload = serde_json::to_vec(&body)?;
req = runtime::with_json_body(req, &payload);
let resp = self.session.fetch(req).await?;
let data = runtime::response_bytes(&resp)?;
Ok(serde_json::from_slice(&data)?)
}
pub async fn get_group(
&self,
group_membership_id: String,
opts: Option<MembershipsGetGroupOptions>,
) -> Result<crate::models::schemas::GroupMembership, Error> {
let mut url = self.session.base_url("api");
url.push_str("/group_memberships");
url.push('/');
let seg = path_escape(&group_membership_id);
url.push_str(&seg);
let mut req = self.session.new_request("GET", &url);
let opts = opts.unwrap_or_default();
if let Some(value) = opts.fields {
req = runtime::with_query(req, "fields", &value.join(","));
}
let resp = self.session.fetch(req).await?;
let data = runtime::response_bytes(&resp)?;
Ok(serde_json::from_slice(&data)?)
}
pub async fn update_group(
&self,
group_membership_id: String,
body: crate::models::schemas::GroupUpdateRequest2,
opts: Option<MembershipsUpdateGroupOptions>,
) -> Result<crate::models::schemas::GroupMembership, Error> {
let mut url = self.session.base_url("api");
url.push_str("/group_memberships");
url.push('/');
let seg = path_escape(&group_membership_id);
url.push_str(&seg);
let mut req = self.session.new_request("PUT", &url);
let opts = opts.unwrap_or_default();
if let Some(value) = opts.fields {
req = runtime::with_query(req, "fields", &value.join(","));
}
let payload = serde_json::to_vec(&body)?;
req = runtime::with_json_body(req, &payload);
let resp = self.session.fetch(req).await?;
let data = runtime::response_bytes(&resp)?;
Ok(serde_json::from_slice(&data)?)
}
pub async fn delete_group(&self, group_membership_id: String) -> Result<(), Error> {
let mut url = self.session.base_url("api");
url.push_str("/group_memberships");
url.push('/');
let seg = path_escape(&group_membership_id);
url.push_str(&seg);
let req = self.session.new_request("DELETE", &url);
let _ = self.session.fetch(req).await?;
Ok(())
}
}