use openlark_core::{
api::{ApiRequest, ApiResponseTrait, ResponseFormat},
config::Config,
http::Transport,
SDKResult,
};
use serde::{Deserialize, Serialize};
use crate::common::{api_endpoints::DriveApi, api_utils::*};
use super::models::ReplyContent;
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct UpdateReplyRequest {
pub file_token: String,
pub comment_id: String,
pub reply_id: String,
pub file_type: String,
pub user_id_type: Option<String>,
pub content: ReplyContent,
}
impl UpdateReplyRequest {
pub fn new(
file_token: impl Into<String>,
comment_id: impl Into<String>,
reply_id: impl Into<String>,
file_type: impl Into<String>,
content: ReplyContent,
) -> Self {
Self {
file_token: file_token.into(),
comment_id: comment_id.into(),
reply_id: reply_id.into(),
file_type: file_type.into(),
user_id_type: None,
content,
}
}
pub fn user_id_type(mut self, user_id_type: impl Into<String>) -> Self {
self.user_id_type = Some(user_id_type.into());
self
}
}
#[derive(Debug, Serialize)]
struct UpdateReplyRequestBody {
content: ReplyContent,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct UpdateReplyResponse {}
impl ApiResponseTrait for UpdateReplyResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
pub async fn update_reply(
request: UpdateReplyRequest,
config: &Config,
option: Option<openlark_core::req_option::RequestOption>,
) -> SDKResult<UpdateReplyResponse> {
if request.file_token.trim().is_empty() {
return Err(openlark_core::error::validation_error(
"file_token",
"file_token 不能为空",
));
}
if request.comment_id.trim().is_empty() {
return Err(openlark_core::error::validation_error(
"comment_id",
"comment_id 不能为空",
));
}
if request.reply_id.trim().is_empty() {
return Err(openlark_core::error::validation_error(
"reply_id",
"reply_id 不能为空",
));
}
if request.file_type.trim().is_empty() {
return Err(openlark_core::error::validation_error(
"file_type",
"file_type 不能为空",
));
}
super::super::validate_comment_file_type_for_list_like(&request.file_type)?;
let api_endpoint = DriveApi::UpdateCommentReply(
request.file_token.clone(),
request.comment_id.clone(),
request.reply_id.clone(),
);
let mut api_request: ApiRequest<UpdateReplyResponse> = ApiRequest::put(&api_endpoint.to_url())
.body(serialize_params(
&UpdateReplyRequestBody {
content: request.content,
},
"更新回复的内容",
)?);
api_request = api_request.query("file_type", &request.file_type);
if let Some(user_id_type) = &request.user_id_type {
api_request = api_request.query("user_id_type", user_id_type);
}
let response = Transport::request(api_request, config, option).await?;
extract_response_data(response, "更新回复的内容")
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_update_reply_request_builder() {
let content = ReplyContent { elements: vec![] };
let request =
UpdateReplyRequest::new("file_token", "comment_123", "reply_456", "docx", content);
assert_eq!(request.file_token, "file_token");
assert_eq!(request.comment_id, "comment_123");
assert_eq!(request.reply_id, "reply_456");
assert_eq!(request.file_type, "docx");
}
#[test]
fn test_update_reply_request_with_user_id_type() {
let content = ReplyContent { elements: vec![] };
let request =
UpdateReplyRequest::new("file_token", "comment_123", "reply_456", "docx", content)
.user_id_type("union_id");
assert_eq!(request.user_id_type, Some("union_id".to_string()));
}
#[test]
fn test_update_reply_request_empty_fields() {
let content = ReplyContent { elements: vec![] };
let request = UpdateReplyRequest::new("", "comment_123", "reply_456", "docx", content);
assert!(request.file_token.is_empty());
let content2 = ReplyContent { elements: vec![] };
let request2 = UpdateReplyRequest::new("token", "", "reply_456", "docx", content2);
assert!(request2.comment_id.is_empty());
let content3 = ReplyContent { elements: vec![] };
let request3 = UpdateReplyRequest::new("token", "comment_123", "", "docx", content3);
assert!(request3.reply_id.is_empty());
let content4 = ReplyContent { elements: vec![] };
let request4 = UpdateReplyRequest::new("token", "comment_123", "reply_456", "", content4);
assert!(request4.file_type.is_empty());
}
#[test]
fn test_response_trait() {
assert_eq!(UpdateReplyResponse::data_format(), ResponseFormat::Data);
}
}