use openlark_core::{
SDKResult,
api::{ApiRequest, ApiResponseTrait, ResponseFormat},
config::Config,
http::Transport,
};
use crate::common::{api_endpoints::DriveApi, api_utils::*};
use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Serialize)]
pub struct CreateFolderRequest {
#[serde(skip)]
config: Config,
pub name: String,
pub folder_token: String,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CreateFolderResponse {
pub token: String,
pub url: String,
}
impl ApiResponseTrait for CreateFolderResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl CreateFolderRequest {
pub fn new(config: Config, name: impl Into<String>, folder_token: impl Into<String>) -> Self {
Self {
config,
name: name.into(),
folder_token: folder_token.into(),
}
}
pub async fn execute(self) -> SDKResult<CreateFolderResponse> {
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<CreateFolderResponse> {
let name_len = self.name.len();
if name_len == 0 || name_len > 256 {
return Err(openlark_core::error::validation_error(
"name",
"name 长度必须在 1~256 字节之间",
));
}
let api_endpoint = DriveApi::CreateFolder;
let request = ApiRequest::<CreateFolderResponse>::post(&api_endpoint.to_url())
.body(serialize_params(&self, "新建文件夹")?);
let response = Transport::request(request, &self.config, Some(option)).await?;
extract_response_data(response, "新建文件夹")
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_create_folder_request() {
let config = Config::default();
let request = CreateFolderRequest::new(config, "test_folder", "parent_token");
assert_eq!(request.name, "test_folder");
assert_eq!(request.folder_token, "parent_token");
}
#[test]
fn test_create_folder_with_empty_name() {
let config = Config::default();
let request = CreateFolderRequest::new(config, "", "parent_token");
assert_eq!(request.name, "");
}
#[test]
fn test_create_folder_with_too_long_name() {
let config = Config::default();
let long_name = "a".repeat(257); let request = CreateFolderRequest::new(config, long_name, "parent_token");
assert!(request.name.len() > 256);
}
#[test]
fn test_create_folder_with_max_length_name() {
let config = Config::default();
let max_name = "a".repeat(256); let request = CreateFolderRequest::new(config, max_name.clone(), "parent_token");
assert_eq!(request.name, max_name);
assert_eq!(request.name.len(), 256);
}
#[test]
fn test_create_folder_in_root() {
let config = Config::default();
let request = CreateFolderRequest::new(config, "test_folder", "");
assert_eq!(request.folder_token, "");
}
}