use openlark_core::{
SDKResult,
api::{ApiRequest, ApiResponseTrait, ResponseFormat},
config::Config,
http::Transport,
req_option::RequestOption,
validate_required,
};
use crate::common::{api_endpoints::CcmSheetApiOld, api_utils::*};
#[derive(Debug, Clone)]
pub struct DataIOApi {
config: Config,
}
impl DataIOApi {
pub fn new(config: Config) -> Self {
Self { config }
}
pub fn config(&self) -> &Config {
&self.config
}
}
pub mod models;
pub use models::{
AppendResult, AppendValuesParams, AppendValuesResponse, BatchUpdateResult, BatchWriteData,
BatchWriteRangesParams, BatchWriteRangesResponse, ImageResult, ImageValue, InsertResult,
InsertValuesParams, InsertValuesResponse, MultipleRangeData, RangeData,
ReadMultipleRangesParams, ReadMultipleRangesResponse, ReadSingleRangeParams,
ReadSingleRangeResponse, UpdateRangeResult, ValuesImageParams, ValuesImageResponse,
WriteSingleRangeParams, WriteSingleRangeResponse,
};
impl ApiResponseTrait for ReadSingleRangeResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl ApiResponseTrait for ReadMultipleRangesResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl ApiResponseTrait for WriteSingleRangeResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl ApiResponseTrait for BatchWriteRangesResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl ApiResponseTrait for AppendValuesResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl ApiResponseTrait for InsertValuesResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl ApiResponseTrait for ValuesImageResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
pub async fn read_single_range(
config: &Config,
spreadsheet_token: &str,
params: ReadSingleRangeParams,
) -> SDKResult<ReadSingleRangeResponse> {
read_single_range_with_options(config, spreadsheet_token, params, RequestOption::default())
.await
}
pub async fn read_single_range_with_options(
config: &Config,
spreadsheet_token: &str,
params: ReadSingleRangeParams,
option: RequestOption,
) -> SDKResult<ReadSingleRangeResponse> {
validate_required!(spreadsheet_token.trim(), "表格Token不能为空");
validate_required!(params.value_range, "数据范围不能为空");
let api_endpoint =
CcmSheetApiOld::ReadSingleRange(spreadsheet_token.to_string(), params.value_range.clone());
let api_request: ApiRequest<ReadSingleRangeResponse> = ApiRequest::get(&api_endpoint.to_url())
.query_opt("valueRenderOption", params.value_render_option.as_ref())
.query_opt("dateTimeRenderOption", params.date_render_option.as_ref());
let response = Transport::request(api_request, config, Some(option)).await?;
extract_response_data(response, "读取单个范围")
}
pub async fn read_multiple_ranges(
config: &Config,
spreadsheet_token: &str,
params: ReadMultipleRangesParams,
) -> SDKResult<ReadMultipleRangesResponse> {
read_multiple_ranges_with_options(config, spreadsheet_token, params, RequestOption::default())
.await
}
pub async fn read_multiple_ranges_with_options(
config: &Config,
spreadsheet_token: &str,
params: ReadMultipleRangesParams,
option: RequestOption,
) -> SDKResult<ReadMultipleRangesResponse> {
validate_required!(spreadsheet_token.trim(), "表格Token不能为空");
if params.ranges.is_empty() {
return Err(openlark_core::error::CoreError::validation(
"ranges",
"范围列表不能为空",
));
}
let api_endpoint = CcmSheetApiOld::ReadMultipleRanges(spreadsheet_token.to_string());
let ranges = params.ranges.join(",");
let api_request: ApiRequest<ReadMultipleRangesResponse> =
ApiRequest::get(&api_endpoint.to_url())
.query("ranges", ranges)
.query_opt("valueRenderOption", params.value_render_option.as_ref())
.query_opt("dateTimeRenderOption", params.date_render_option.as_ref());
let response = Transport::request(api_request, config, Some(option)).await?;
extract_response_data(response, "读取多个范围")
}
pub async fn write_single_range(
config: &Config,
spreadsheet_token: &str,
params: WriteSingleRangeParams,
) -> SDKResult<WriteSingleRangeResponse> {
write_single_range_with_options(config, spreadsheet_token, params, RequestOption::default())
.await
}
pub async fn write_single_range_with_options(
config: &Config,
spreadsheet_token: &str,
params: WriteSingleRangeParams,
option: RequestOption,
) -> SDKResult<WriteSingleRangeResponse> {
validate_required!(spreadsheet_token.trim(), "表格Token不能为空");
validate_required!(params.range, "数据范围不能为空");
let api_endpoint = CcmSheetApiOld::WriteSingleRange(spreadsheet_token.to_string());
let api_request: ApiRequest<WriteSingleRangeResponse> =
ApiRequest::put(&api_endpoint.to_url()).body(serialize_params(¶ms, "写入单个范围")?);
let response = Transport::request(api_request, config, Some(option)).await?;
extract_response_data(response, "写入单个范围")
}
pub async fn batch_write_ranges(
config: &Config,
spreadsheet_token: &str,
params: BatchWriteRangesParams,
) -> SDKResult<BatchWriteRangesResponse> {
batch_write_ranges_with_options(config, spreadsheet_token, params, RequestOption::default())
.await
}
pub async fn batch_write_ranges_with_options(
config: &Config,
spreadsheet_token: &str,
params: BatchWriteRangesParams,
option: RequestOption,
) -> SDKResult<BatchWriteRangesResponse> {
validate_required!(spreadsheet_token.trim(), "表格Token不能为空");
if params.data.is_empty() {
return Err(openlark_core::error::CoreError::validation(
"data",
"写入数据列表不能为空",
));
}
let api_endpoint = CcmSheetApiOld::BatchWriteRanges(spreadsheet_token.to_string());
let api_request: ApiRequest<BatchWriteRangesResponse> =
ApiRequest::post(&api_endpoint.to_url()).body(serialize_params(¶ms, "批量写入范围")?);
let response = Transport::request(api_request, config, Some(option)).await?;
extract_response_data(response, "批量写入范围")
}
pub async fn append_values(
config: &Config,
spreadsheet_token: &str,
params: AppendValuesParams,
) -> SDKResult<AppendValuesResponse> {
append_values_with_options(config, spreadsheet_token, params, RequestOption::default()).await
}
pub async fn append_values_with_options(
config: &Config,
spreadsheet_token: &str,
params: AppendValuesParams,
option: RequestOption,
) -> SDKResult<AppendValuesResponse> {
validate_required!(spreadsheet_token.trim(), "表格Token不能为空");
validate_required!(params.range, "追加范围不能为空");
let api_endpoint = CcmSheetApiOld::AppendValues(spreadsheet_token.to_string());
let api_request: ApiRequest<AppendValuesResponse> =
ApiRequest::post(&api_endpoint.to_url()).body(serialize_params(¶ms, "追加数据")?);
let response = Transport::request(api_request, config, Some(option)).await?;
extract_response_data(response, "追加数据")
}
pub async fn insert_values(
config: &Config,
spreadsheet_token: &str,
params: InsertValuesParams,
) -> SDKResult<InsertValuesResponse> {
insert_values_with_options(config, spreadsheet_token, params, RequestOption::default()).await
}
pub async fn insert_values_with_options(
config: &Config,
spreadsheet_token: &str,
params: InsertValuesParams,
option: RequestOption,
) -> SDKResult<InsertValuesResponse> {
validate_required!(spreadsheet_token.trim(), "表格Token不能为空");
validate_required!(params.range, "插入范围不能为空");
let api_endpoint = CcmSheetApiOld::InsertValues(spreadsheet_token.to_string());
let api_request: ApiRequest<InsertValuesResponse> =
ApiRequest::post(&api_endpoint.to_url()).body(serialize_params(¶ms, "插入数据")?);
let response = Transport::request(api_request, config, Some(option)).await?;
extract_response_data(response, "插入数据")
}
pub async fn values_image(
config: &Config,
spreadsheet_token: &str,
params: ValuesImageParams,
) -> SDKResult<ValuesImageResponse> {
values_image_with_options(config, spreadsheet_token, params, RequestOption::default()).await
}
pub async fn values_image_with_options(
config: &Config,
spreadsheet_token: &str,
params: ValuesImageParams,
option: RequestOption,
) -> SDKResult<ValuesImageResponse> {
validate_required!(spreadsheet_token.trim(), "表格Token不能为空");
validate_required!(params.range, "图片范围不能为空");
let api_endpoint = CcmSheetApiOld::ValuesImage(spreadsheet_token.to_string());
let api_request: ApiRequest<ValuesImageResponse> =
ApiRequest::post(&api_endpoint.to_url()).body(serialize_params(¶ms, "写入图片")?);
let response = Transport::request(api_request, config, Some(option)).await?;
extract_response_data(response, "写入图片")
}
#[cfg(test)]
mod tests {
use serde_json;
#[test]
fn test_serialization_roundtrip() {
let json = r#"{"test": "value"}"#;
assert!(serde_json::from_str::<serde_json::Value>(json).is_ok());
}
#[test]
fn test_deserialization_from_json() {
let json = r#"{"field": "data"}"#;
let value: serde_json::Value = serde_json::from_str(json).expect("JSON 反序列化失败");
assert_eq!(value["field"], "data");
}
}