use openlark_core::{
api::{ApiRequest, ApiResponseTrait, ResponseFormat},
config::Config,
http::Transport,
SDKResult,
};
use serde::{Deserialize, Serialize};
use super::super::models::WikiSpace;
use crate::common::{api_endpoints::WikiApiV2, api_utils::*};
pub struct ListWikiSpacesRequest {
config: Config,
page_size: Option<i32>,
page_token: Option<String>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ListWikiSpacesResponse {
#[serde(default)]
pub items: Vec<WikiSpace>,
pub has_more: Option<bool>,
pub page_token: Option<String>,
}
impl ApiResponseTrait for ListWikiSpacesResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl ListWikiSpacesRequest {
pub fn new(config: Config) -> Self {
Self {
config,
page_size: None,
page_token: None,
}
}
pub fn page_size(mut self, page_size: i32) -> Self {
self.page_size = Some(page_size);
self
}
pub fn page_token(mut self, page_token: impl Into<String>) -> Self {
self.page_token = Some(page_token.into());
self
}
pub async fn execute(self) -> SDKResult<ListWikiSpacesResponse> {
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<ListWikiSpacesResponse> {
if let Some(page_size) = self.page_size {
if page_size < 1 || page_size > 50 {
return Err(openlark_core::error::validation_error(
"page_size",
"page_size 必须在 1~50 之间",
));
}
}
let api_endpoint = WikiApiV2::SpaceList;
let mut api_request: ApiRequest<ListWikiSpacesResponse> =
ApiRequest::get(&api_endpoint.to_url());
if let Some(page_size) = self.page_size {
api_request = api_request.query("page_size", &page_size.to_string());
}
if let Some(page_token) = self.page_token {
api_request = api_request.query("page_token", &page_token);
}
let response = Transport::request(api_request, &self.config, Some(option)).await?;
extract_response_data(response, "获取知识空间列表")
}
}
#[deprecated(
since = "0.16.0",
note = "请使用 ListWikiSpacesRequest 的流式 Builder 模式"
)]
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ListWikiSpacesParams {
pub page_size: Option<i32>,
pub page_token: Option<String>,
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_list_wiki_spaces_builder() {
let config = Config::default();
let request = ListWikiSpacesRequest::new(config)
.page_size(20)
.page_token("token123");
assert_eq!(request.page_size, Some(20));
assert_eq!(request.page_token, Some("token123".to_string()));
}
#[test]
fn test_list_wiki_spaces_no_params() {
let config = Config::default();
let request = ListWikiSpacesRequest::new(config);
assert!(request.page_size.is_none());
assert!(request.page_token.is_none());
}
#[test]
fn test_list_wiki_spaces_response() {
let response = ListWikiSpacesResponse {
items: vec![],
has_more: Some(true),
page_token: Some("next_token".to_string()),
};
assert!(response.items.is_empty());
assert_eq!(response.has_more, Some(true));
}
#[test]
fn test_response_trait() {
assert_eq!(ListWikiSpacesResponse::data_format(), ResponseFormat::Data);
}
#[test]
fn test_page_size_boundaries() {
let config = Config::default();
let min_request = ListWikiSpacesRequest::new(config.clone()).page_size(1);
assert_eq!(min_request.page_size, Some(1));
let max_request = ListWikiSpacesRequest::new(config).page_size(50);
assert_eq!(max_request.page_size, Some(50));
}
#[test]
fn test_empty_response() {
let response = ListWikiSpacesResponse {
items: vec![],
has_more: Some(false),
page_token: None,
};
assert!(response.items.is_empty());
assert_eq!(response.has_more, Some(false));
}
#[test]
fn test_page_token_setting() {
let config = Config::default();
let request = ListWikiSpacesRequest::new(config).page_token("custom_token_456");
assert_eq!(request.page_token, Some("custom_token_456".to_string()));
}
#[test]
#[allow(deprecated)]
fn test_deprecated_params() {
let params = ListWikiSpacesParams {
page_size: Some(30),
page_token: Some("old_token".to_string()),
};
assert_eq!(params.page_size, Some(30));
assert_eq!(params.page_token, Some("old_token".to_string()));
}
}