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::*};
#[derive(Debug, Clone)]
pub struct CreatePermissionPublicPasswordRequest {
config: Config,
pub token: String,
pub r#type: String,
}
impl CreatePermissionPublicPasswordRequest {
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<CreatePermissionPublicPasswordResponse> {
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<CreatePermissionPublicPasswordResponse> {
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",
));
}
}
if self.r#type == "minutes" {
return Err(openlark_core::error::validation_error(
"type",
"type=minutes 暂不支持启用云文档密码",
));
}
let api_endpoint = DriveApi::CreatePublicPassword(self.token);
let request =
ApiRequest::<CreatePermissionPublicPasswordResponse>::post(&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 CreatePermissionPublicPasswordResponse {
pub password: String,
}
impl ApiResponseTrait for CreatePermissionPublicPasswordResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
#[cfg(test)]
mod tests {
use super::*;
use openlark_core::testing::prelude::test_runtime;
#[test]
fn test_create_permission_public_password_request_builder() {
let config = Config::default();
let request = CreatePermissionPublicPasswordRequest::new(config, "token", "docx");
assert_eq!(request.token, "token");
assert_eq!(request.r#type, "docx");
}
#[test]
fn test_response_trait() {
assert_eq!(
CreatePermissionPublicPasswordResponse::data_format(),
ResponseFormat::Data
);
}
#[test]
fn test_empty_token() {
let config = Config::default();
let request = CreatePermissionPublicPasswordRequest::new(config, "", "docx");
let rt = test_runtime();
let result = rt.block_on(request.execute());
assert!(result.is_err());
let err = result.unwrap_err();
assert!(err.to_string().contains("token"));
}
#[test]
fn test_empty_type() {
let config = Config::default();
let request = CreatePermissionPublicPasswordRequest::new(config, "token", "");
let rt = test_runtime();
let result = rt.block_on(request.execute());
assert!(result.is_err());
let err = result.unwrap_err();
assert!(err.to_string().contains("type"));
}
#[test]
fn test_invalid_type() {
let config = Config::default();
let request = CreatePermissionPublicPasswordRequest::new(config, "token", "invalid_type");
let rt = test_runtime();
let result = rt.block_on(request.execute());
assert!(result.is_err());
let err = result.unwrap_err();
assert!(err.to_string().contains("type"));
}
#[test]
fn test_minutes_not_supported() {
let config = Config::default();
let request = CreatePermissionPublicPasswordRequest::new(config, "token", "minutes");
let rt = test_runtime();
let result = rt.block_on(request.execute());
assert!(result.is_err());
let err = result.unwrap_err();
assert!(err.to_string().contains("minutes"));
}
}