use openlark_core::{
api::{ApiRequest, ApiResponseTrait, ResponseFormat},
config::Config,
error::SDKResult,
http::Transport,
validate_required,
};
use serde::{Deserialize, Serialize};
#[derive(Debug, Clone)]
pub struct ListTablesRequest {
app_token: String,
page_size: Option<i32>,
page_token: Option<String>,
config: Config,
}
impl ListTablesRequest {
pub fn new(config: Config) -> Self {
Self {
app_token: String::new(),
page_size: None,
page_token: None,
config,
}
}
pub fn app_token(mut self, app_token: String) -> Self {
self.app_token = app_token;
self
}
pub fn page_size(mut self, page_size: i32) -> Self {
self.page_size = Some(page_size.min(100)); self
}
pub fn page_token(mut self, page_token: String) -> Self {
self.page_token = Some(page_token);
self
}
pub async fn execute(self) -> SDKResult<ListTablesResponse> {
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<ListTablesResponse> {
validate_required!(self.app_token.trim(), "app_token");
if let Some(page_size) = self.page_size
&& page_size <= 0
{
return Err(openlark_core::error::validation_error(
"page_size",
"分页大小必须大于0",
));
}
use crate::common::api_endpoints::BitableApiV1;
let api_endpoint = BitableApiV1::TableList(self.app_token.clone());
let mut api_request: ApiRequest<ListTablesResponse> =
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(ref 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?;
response.data.ok_or_else(|| {
openlark_core::error::validation_error("响应数据为空", "服务器没有返回有效的数据")
})
}
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
pub struct TableInfo {
#[serde(skip_serializing_if = "Option::is_none")]
pub table_id: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub revision: Option<i32>,
#[serde(skip_serializing_if = "Option::is_none")]
pub name: Option<String>,
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
pub struct ListTablesResponse {
#[serde(default)]
pub has_more: bool,
#[serde(skip_serializing_if = "Option::is_none")]
pub page_token: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub total: Option<i32>,
#[serde(default)]
pub items: Vec<TableInfo>,
}
impl ApiResponseTrait for ListTablesResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
#[cfg(test)]
mod tests {
use super::*;
use openlark_core::testing::prelude::test_runtime;
#[test]
fn test_empty_app_token() {
let config = Config::default();
let request = ListTablesRequest::new(config).app_token("".to_string());
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("app_token"));
}
#[test]
fn test_invalid_page_size() {
let config = Config::default();
let request = ListTablesRequest::new(config)
.app_token("app_token".to_string())
.page_size(0);
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("page_size"));
}
#[test]
fn test_response_trait() {
assert_eq!(ListTablesResponse::data_format(), ResponseFormat::Data);
}
}