use openlark_core::{
SDKResult,
api::{ApiRequest, ApiResponseTrait, ResponseFormat},
config::Config,
http::Transport,
validate_required,
};
use serde::{Deserialize, Serialize};
use crate::common::{api_endpoints::DriveApi, api_utils::*};
use super::models::PermissionMemberItem;
#[derive(Debug, Clone)]
pub struct ListPermissionMembersRequest {
config: Config,
pub token: String,
pub file_type: String,
pub fields: Option<String>,
pub perm_type: Option<String>,
}
impl ListPermissionMembersRequest {
pub fn new(config: Config, token: impl Into<String>, file_type: impl Into<String>) -> Self {
Self {
config,
token: token.into(),
file_type: file_type.into(),
fields: None,
perm_type: None,
}
}
pub fn fields(mut self, fields: impl Into<String>) -> Self {
self.fields = Some(fields.into());
self
}
pub fn perm_type(mut self, perm_type: impl Into<String>) -> Self {
self.perm_type = Some(perm_type.into());
self
}
pub async fn execute(self) -> SDKResult<ListPermissionMembersResponse> {
self.execute_with_options(openlark_core::req_option::RequestOption::default())
.await
}
pub async fn execute_with_options(
self,
option: openlark_core::req_option::RequestOption,
) -> SDKResult<ListPermissionMembersResponse> {
validate_required!(self.token, "token 不能为空");
validate_required!(self.file_type, "file_type 不能为空");
match self.file_type.as_str() {
"doc" | "sheet" | "file" | "wiki" | "bitable" | "docx" | "folder" | "mindnote"
| "minutes" | "slides" => {}
_ => {
return Err(openlark_core::error::validation_error(
"file_type",
"file_type 必须为 doc/sheet/file/wiki/bitable/docx/folder/mindnote/minutes/slides",
));
}
}
if let Some(perm_type) = &self.perm_type {
match perm_type.as_str() {
"container" | "single_page" => {}
_ => {
return Err(openlark_core::error::validation_error(
"perm_type",
"perm_type 必须为 container/single_page",
));
}
}
}
let api_endpoint = DriveApi::ListPermissionMembers(self.token.clone());
let mut api_request =
ApiRequest::<ListPermissionMembersResponse>::get(&api_endpoint.to_url());
api_request = api_request.query("type", &self.file_type);
if let Some(fields) = &self.fields {
api_request = api_request.query("fields", fields);
}
if let Some(perm_type) = &self.perm_type {
api_request = api_request.query("perm_type", perm_type);
}
let response = Transport::request(api_request, &self.config, Some(option)).await?;
extract_response_data(response, "列表")
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ListPermissionMembersResponse {
#[serde(default)]
pub items: Vec<PermissionMemberItem>,
}
impl ApiResponseTrait for ListPermissionMembersResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
#[cfg(test)]
mod tests {
use super::*;
use openlark_core::testing::prelude::test_runtime;
#[test]
fn test_list_permission_members_request_builder() {
let config = Config::default();
let request = ListPermissionMembersRequest::new(config, "file_token", "docx");
assert_eq!(request.token, "file_token");
assert_eq!(request.file_type, "docx");
}
#[test]
fn test_list_permission_members_request_with_params() {
let config = Config::default();
let request = ListPermissionMembersRequest::new(config, "file_token", "docx")
.fields("*")
.perm_type("container");
assert_eq!(request.token, "file_token");
assert_eq!(request.file_type, "docx");
assert_eq!(
request
.fields
.expect("fields should be set when .fields() is called"),
"*"
);
assert_eq!(
request
.perm_type
.expect("perm_type should be set when .perm_type() is called"),
"container"
);
}
#[test]
fn test_response_trait() {
assert_eq!(
ListPermissionMembersResponse::data_format(),
ResponseFormat::Data
);
}
#[test]
fn test_empty_token_validation() {
let config = Config::default();
let request = ListPermissionMembersRequest::new(config, "", "docx");
let result = std::thread::spawn(move || {
let rt = test_runtime();
rt.block_on(async move {
let _ = request.execute().await;
})
})
.join();
assert!(result.is_ok());
}
#[test]
fn test_file_type_validation() {
let config = Config::default();
let request = ListPermissionMembersRequest::new(config, "token", "invalid");
let result = std::thread::spawn(move || {
let rt = test_runtime();
rt.block_on(async move {
let _ = request.execute().await;
})
})
.join();
assert!(result.is_ok());
}
#[test]
fn test_perm_type_validation() {
let config = Config::default();
let request =
ListPermissionMembersRequest::new(config, "token", "docx").perm_type("invalid");
let result = std::thread::spawn(move || {
let rt = test_runtime();
rt.block_on(async move {
let _ = request.execute().await;
})
})
.join();
assert!(result.is_ok());
}
#[test]
fn test_supported_file_types() {
let config = Config::default();
for file_type in [
"doc", "sheet", "file", "wiki", "bitable", "docx", "folder", "mindnote", "minutes",
"slides",
] {
let request =
ListPermissionMembersRequest::new(config.clone(), "token", file_type.to_string());
assert_eq!(request.file_type, file_type);
}
}
#[test]
fn test_optional_parameters() {
let config = Config::default();
let request = ListPermissionMembersRequest::new(config.clone(), "token", "docx");
assert!(request.fields.is_none());
assert!(request.perm_type.is_none());
let request2 = ListPermissionMembersRequest::new(config.clone(), "token", "docx")
.fields("*")
.perm_type("container");
assert_eq!(request2.fields, Some("*".to_string()));
assert_eq!(request2.perm_type, Some("container".to_string()));
}
}