use openlark_core::{
api::{ApiRequest, ApiResponseTrait, ResponseFormat},
config::Config,
http::Transport,
validate_required, SDKResult,
};
use serde::{Deserialize, Serialize};
use crate::ccm::wiki::v2::models::WikiSpaceNode;
use crate::common::{api_endpoints::WikiApiV2, api_utils::*};
pub struct CreateWikiSpaceNodeRequest {
space_id: String,
config: Config,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CreateWikiSpaceNodeParams {
pub obj_type: String,
pub obj_token: Option<String>,
pub parent_node_token: Option<String>,
pub node_type: Option<String>,
pub title: Option<String>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CreateWikiSpaceNodeResponse {
pub node: Option<WikiSpaceNode>,
}
impl ApiResponseTrait for CreateWikiSpaceNodeResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl CreateWikiSpaceNodeRequest {
pub fn new(config: Config) -> Self {
Self {
space_id: String::new(),
config,
}
}
pub fn space_id(mut self, space_id: impl Into<String>) -> Self {
self.space_id = space_id.into();
self
}
pub async fn execute(
self,
params: CreateWikiSpaceNodeParams,
) -> SDKResult<CreateWikiSpaceNodeResponse> {
self.execute_with_options(params, openlark_core::req_option::RequestOption::default())
.await
}
pub async fn execute_with_options(
self,
params: CreateWikiSpaceNodeParams,
option: openlark_core::req_option::RequestOption,
) -> SDKResult<CreateWikiSpaceNodeResponse> {
validate_required!(self.space_id, "知识空间ID不能为空");
let valid_types = [
"doc", "docx", "sheet", "slide", "mindnote", "bitable", "file",
];
if !valid_types.contains(¶ms.obj_type.as_str()) {
return Err(openlark_core::error::CoreError::validation_msg(
&format!(
"obj_type 参数无效: '{}'. 支持的类型为: doc, docx, sheet, slide, mindnote, bitable, file",
params.obj_type
)
));
}
validate_required!(params.obj_type, "obj_type不能为空");
let api_endpoint = WikiApiV2::SpaceNodeCreate(self.space_id.clone());
let api_request: ApiRequest<CreateWikiSpaceNodeResponse> =
ApiRequest::post(&api_endpoint.to_url())
.body(serialize_params(¶ms, "创建知识空间节点")?);
let response = Transport::request(api_request, &self.config, Some(option)).await?;
extract_response_data(response, "创建知识空间节点")
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_create_wiki_space_node_builder() {
let config = Config::default();
let request = CreateWikiSpaceNodeRequest::new(config).space_id("wiki_space_123");
assert_eq!(request.space_id, "wiki_space_123");
}
#[test]
fn test_create_doc_node_params() {
let params = CreateWikiSpaceNodeParams {
obj_type: "docx".to_string(),
parent_node_token: Some("parent_token".to_string()),
node_type: Some("origin".to_string()),
title: Some("测试文档".to_string()),
obj_token: None,
};
assert_eq!(params.obj_type, "docx");
assert_eq!(params.title, Some("测试文档".to_string()));
}
#[test]
fn test_create_sheet_node_params() {
let params = CreateWikiSpaceNodeParams {
obj_type: "sheet".to_string(),
parent_node_token: Some("parent_token".to_string()),
title: Some("项目进度表".to_string()),
obj_token: None,
node_type: None,
};
assert_eq!(params.obj_type, "sheet");
assert_eq!(params.title, Some("项目进度表".to_string()));
}
#[test]
fn test_create_wiki_space_node_response() {
let response = CreateWikiSpaceNodeResponse { node: None };
assert!(response.node.is_none());
}
#[test]
fn test_response_trait() {
assert_eq!(
CreateWikiSpaceNodeResponse::data_format(),
ResponseFormat::Data
);
}
#[test]
fn test_supported_object_types() {
let valid_types = [
"doc", "docx", "sheet", "slide", "mindnote", "bitable", "file",
];
for obj_type in valid_types {
let params = CreateWikiSpaceNodeParams {
obj_type: obj_type.to_string(),
parent_node_token: None,
node_type: None,
title: None,
obj_token: None,
};
assert_eq!(params.obj_type, obj_type);
}
}
#[test]
fn test_shortcut_node_type() {
let params = CreateWikiSpaceNodeParams {
obj_type: "docx".to_string(),
parent_node_token: Some("parent_token".to_string()),
node_type: Some("shortcut".to_string()),
title: Some("快捷方式".to_string()),
obj_token: Some("original_doc_token".to_string()),
};
assert_eq!(params.node_type, Some("shortcut".to_string()));
assert_eq!(params.obj_token, Some("original_doc_token".to_string()));
}
#[test]
fn test_root_node_no_parent() {
let params = CreateWikiSpaceNodeParams {
obj_type: "docx".to_string(),
parent_node_token: None,
node_type: Some("origin".to_string()),
title: Some("根目录文档".to_string()),
obj_token: None,
};
assert!(params.parent_node_token.is_none());
}
}