use crate::common::{api_endpoints::DriveApi, api_utils::*};
use openlark_core::{
SDKResult, api::ApiRequest, config::Config, http::Transport, validate_required,
};
use super::models::FileVersionInfo;
#[derive(Debug, Clone)]
pub struct GetFileVersionRequest {
config: Config,
pub file_token: String,
pub version_id: String,
pub obj_type: String,
pub user_id_type: Option<String>,
}
impl GetFileVersionRequest {
pub fn new(
config: Config,
file_token: impl Into<String>,
version_id: impl Into<String>,
obj_type: impl Into<String>,
) -> Self {
Self {
config,
file_token: file_token.into(),
version_id: version_id.into(),
obj_type: obj_type.into(),
user_id_type: None,
}
}
pub fn user_id_type(mut self, user_id_type: impl Into<String>) -> Self {
self.user_id_type = Some(user_id_type.into());
self
}
pub async fn execute(self) -> SDKResult<GetFileVersionResponse> {
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<GetFileVersionResponse> {
validate_required!(self.file_token, "file_token 不能为空");
validate_required!(self.version_id, "version_id 不能为空");
match self.obj_type.as_str() {
"docx" | "sheet" => {}
_ => {
return Err(openlark_core::error::validation_error(
"obj_type",
"obj_type 仅支持 docx/sheet",
));
}
}
let api_endpoint = DriveApi::GetFileVersion(self.file_token, self.version_id);
let request = ApiRequest::<GetFileVersionResponse>::get(&api_endpoint.to_url())
.query("obj_type", self.obj_type)
.query_opt("user_id_type", self.user_id_type);
let response = Transport::request(request, &self.config, Some(option)).await?;
extract_response_data(response, "获取")
}
}
pub type GetFileVersionResponse = FileVersionInfo;
#[cfg(test)]
mod tests {
use super::*;
use openlark_core::api::ApiResponseTrait;
use openlark_core::testing::prelude::test_runtime;
#[test]
fn test_get_file_version_request_builder() {
let config = Config::default();
let request = GetFileVersionRequest::new(config, "file_token", "fnJfyX", "docx")
.user_id_type("open_id");
assert_eq!(request.file_token, "file_token");
assert_eq!(request.version_id, "fnJfyX");
assert_eq!(request.obj_type, "docx");
assert_eq!(request.user_id_type, Some("open_id".to_string()));
}
#[test]
fn test_response_trait() {
assert_eq!(
<FileVersionInfo as ApiResponseTrait>::data_format(),
openlark_core::api::ResponseFormat::Data
);
}
#[test]
fn test_empty_file_token_validation() {
let config = Config::default();
let request = GetFileVersionRequest::new(config, "", "version_id", "docx");
let result = std::thread::spawn(move || {
let rt = test_runtime();
rt.block_on(async move {
let _ = request.execute().await;
})
})
.join();
assert!(result.is_ok());
}
#[test]
fn test_empty_version_id_validation() {
let config = Config::default();
let request = GetFileVersionRequest::new(config, "token", "", "docx");
let result = std::thread::spawn(move || {
let rt = test_runtime();
rt.block_on(async move {
let _ = request.execute().await;
})
})
.join();
assert!(result.is_ok());
}
#[test]
fn test_obj_type_validation() {
let config = Config::default();
let request = GetFileVersionRequest::new(config, "token", "version", "invalid");
let result = std::thread::spawn(move || {
let rt = test_runtime();
rt.block_on(async move {
let _ = request.execute().await;
})
})
.join();
assert!(result.is_ok());
}
#[test]
fn test_supported_obj_types() {
let config = Config::default();
for obj_type in ["docx", "sheet"] {
let request = GetFileVersionRequest::new(
config.clone(),
"token",
"version",
obj_type.to_string(),
);
assert_eq!(request.obj_type, obj_type);
}
}
#[test]
fn test_user_id_type_optional() {
let config = Config::default();
let request1 = GetFileVersionRequest::new(config.clone(), "token", "version", "docx");
assert!(request1.user_id_type.is_none());
let request2 =
GetFileVersionRequest::new(config, "token", "version", "docx").user_id_type("user_id");
assert_eq!(request2.user_id_type, Some("user_id".to_string()));
}
}