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::PermissionPublic;
#[derive(Debug, Clone)]
pub struct GetPublicPermissionRequest {
config: Config,
pub token: String,
pub r#type: String,
}
impl GetPublicPermissionRequest {
pub fn new(config: Config, token: impl Into<String>, r#type: impl Into<String>) -> Self {
Self {
config,
token: token.into(),
r#type: r#type.into(),
}
}
pub async fn execute(self) -> SDKResult<GetPublicPermissionResponse> {
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<GetPublicPermissionResponse> {
validate_required!(self.token, "token 不能为空");
if self.r#type.is_empty() {
return Err(openlark_core::error::validation_error(
"type",
"type 不能为空",
));
}
match self.r#type.as_str() {
"doc" | "sheet" | "file" | "wiki" | "bitable" | "docx" | "mindnote" | "minutes"
| "slides" => {}
_ => {
return Err(openlark_core::error::validation_error(
"type",
"type 必须为 doc/sheet/file/wiki/bitable/docx/mindnote/minutes/slides",
));
}
}
let api_endpoint = DriveApi::GetPublicPermission(self.token);
let request = ApiRequest::<GetPublicPermissionResponse>::get(&api_endpoint.to_url())
.query("type", self.r#type);
let response = Transport::request(request, &self.config, Some(option)).await?;
extract_response_data(response, "获取")
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct GetPublicPermissionResponse {
pub permission_public: PermissionPublic,
}
impl ApiResponseTrait for GetPublicPermissionResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_get_public_permission_request_builder() {
let config = Config::default();
let request = GetPublicPermissionRequest::new(config, "doc_token", "docx");
assert_eq!(request.token, "doc_token");
assert_eq!(request.r#type, "docx");
}
#[test]
fn test_response_trait() {
assert_eq!(
GetPublicPermissionResponse::data_format(),
ResponseFormat::Data
);
}
#[test]
fn test_empty_token_validation() {
let config = Config::default();
let request = GetPublicPermissionRequest::new(config, "", "docx");
assert_eq!(request.token, "");
}
#[test]
fn test_empty_type_validation() {
let config = Config::default();
let request = GetPublicPermissionRequest::new(config, "doc_token", "");
assert_eq!(request.r#type, "");
}
#[test]
fn test_invalid_type_validation() {
let config = Config::default();
let request = GetPublicPermissionRequest::new(config, "doc_token", "invalid_type");
assert_eq!(request.r#type, "invalid_type");
}
#[test]
fn test_valid_file_types() {
let config = Config::default();
let valid_types = vec![
"doc", "sheet", "file", "wiki", "bitable", "docx", "mindnote", "minutes", "slides",
];
for file_type in valid_types {
let request = GetPublicPermissionRequest::new(config.clone(), "doc_token", file_type);
assert_eq!(request.r#type, file_type);
}
}
}