use openlark_core::{
SDKResult,
api::{ApiRequest, ApiResponseTrait, ResponseFormat},
config::Config,
http::Transport,
req_option::RequestOption,
validate_required,
};
use crate::common::{
api_endpoints::{CcmSheetApiOld, SheetsApiV3},
api_utils::*,
};
#[derive(Debug, Clone)]
pub struct SheetApi {
config: Config,
}
impl SheetApi {
pub fn new(config: Config) -> Self {
Self { config }
}
pub fn config(&self) -> &Config {
&self.config
}
}
pub mod models;
pub use models::{
AddSheetParams, AddSheetResponse, AddSheetResult, DeleteSheetParams, DeleteSheetResponse,
DeleteSheetResult, GetSheetParams, GetSheetResponse, SheetDetailInfo, UpdateSheetParams,
UpdateSheetResponse, UpdateSheetResult,
};
impl ApiResponseTrait for AddSheetResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl ApiResponseTrait for GetSheetResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl ApiResponseTrait for UpdateSheetResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl ApiResponseTrait for DeleteSheetResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
pub async fn add_sheet(
config: &Config,
spreadsheet_token: &str,
params: AddSheetParams,
) -> SDKResult<AddSheetResponse> {
add_sheet_with_options(config, spreadsheet_token, params, RequestOption::default()).await
}
pub async fn add_sheet_with_options(
config: &Config,
spreadsheet_token: &str,
params: AddSheetParams,
option: RequestOption,
) -> SDKResult<AddSheetResponse> {
validate_required!(spreadsheet_token.trim(), "表格Token不能为空");
validate_required!(params.title.trim(), "工作表标题不能为空");
let api_endpoint = CcmSheetApiOld::AddSheet(spreadsheet_token.to_string());
let api_request: ApiRequest<AddSheetResponse> = api_endpoint
.to_request()
.body(serialize_params(¶ms, "添加工作表")?);
Transport::request_typed(api_request, config, Some(option), "添加工作表").await
}
pub async fn get_sheet(
config: &Config,
spreadsheet_token: &str,
params: GetSheetParams,
) -> SDKResult<GetSheetResponse> {
get_sheet_with_options(config, spreadsheet_token, params, RequestOption::default()).await
}
pub async fn get_sheet_with_options(
config: &Config,
spreadsheet_token: &str,
params: GetSheetParams,
option: RequestOption,
) -> SDKResult<GetSheetResponse> {
validate_required!(spreadsheet_token.trim(), "表格Token不能为空");
validate_required!(params.sheet_id.trim(), "工作表ID不能为空");
let api_endpoint =
SheetsApiV3::GetSheet(spreadsheet_token.to_string(), params.sheet_id.to_string());
let api_request: ApiRequest<GetSheetResponse> = api_endpoint.to_request();
Transport::request_typed(api_request, config, Some(option), "获取工作表信息").await
}
pub async fn update_sheet(
config: &Config,
spreadsheet_token: &str,
params: UpdateSheetParams,
) -> SDKResult<UpdateSheetResponse> {
update_sheet_with_options(config, spreadsheet_token, params, RequestOption::default()).await
}
pub async fn update_sheet_with_options(
config: &Config,
spreadsheet_token: &str,
params: UpdateSheetParams,
option: RequestOption,
) -> SDKResult<UpdateSheetResponse> {
validate_required!(spreadsheet_token.trim(), "表格Token不能为空");
validate_required!(params.sheet_id.trim(), "工作表ID不能为空");
let api_endpoint = CcmSheetApiOld::UpdateSheet(spreadsheet_token.to_string());
let api_request: ApiRequest<UpdateSheetResponse> = api_endpoint
.to_request()
.body(serialize_params(¶ms, "更新工作表")?);
Transport::request_typed(api_request, config, Some(option), "更新工作表").await
}
pub async fn delete_sheet(
config: &Config,
spreadsheet_token: &str,
params: DeleteSheetParams,
) -> SDKResult<DeleteSheetResponse> {
delete_sheet_with_options(config, spreadsheet_token, params, RequestOption::default()).await
}
pub async fn delete_sheet_with_options(
config: &Config,
spreadsheet_token: &str,
params: DeleteSheetParams,
option: RequestOption,
) -> SDKResult<DeleteSheetResponse> {
validate_required!(spreadsheet_token.trim(), "表格Token不能为空");
validate_required!(params.sheet_id.trim(), "工作表ID不能为空");
let api_endpoint = CcmSheetApiOld::DeleteSheet(spreadsheet_token.to_string());
let api_request: ApiRequest<DeleteSheetResponse> = api_endpoint
.to_request()
.body(serialize_params(¶ms, "删除工作表")?);
Transport::request_typed(api_request, config, Some(option), "删除工作表").await
}
#[cfg(test)]
mod tests {
use super::*;
use serde_json::json;
use wiremock::MockServer;
use wiremock::matchers::{method, path};
use wiremock::{Mock, ResponseTemplate};
#[tokio::test]
async fn test_add_sheet_returns_data_on_success() {
let server = MockServer::start().await;
Mock::given(method("POST"))
.and(path(
"/open-apis/sheets/v2/spreadsheets/token001/sheets_batch_update",
))
.respond_with(ResponseTemplate::new(200).set_body_json(json!({
"code": 0, "msg": "success", "data": {}
})))
.mount(&server)
.await;
let config = Config::builder()
.app_id("ci_app_id")
.app_secret("ci_app_secret")
.base_url(server.uri())
.enable_token_cache(false)
.build();
let resp = add_sheet(
&config,
"token001",
AddSheetParams {
title: "新工作表".into(),
index: None,
},
)
.await
.expect("添加工作表应成功");
assert!(resp.data.is_none());
let received = server.received_requests().await.unwrap_or_default();
assert_eq!(received.len(), 1);
assert_eq!(
received[0].url.path(),
"/open-apis/sheets/v2/spreadsheets/token001/sheets_batch_update"
);
}
}