use crate::api_impl::{BotApi, resource};
use crate::error::Result;
use crate::models::{
api::MessageResponse,
message::{C2CMessageParams, GroupMessageParams, MessageToCreate},
};
use reqwest::Method;
use tracing::debug;
impl BotApi {
pub async fn send_group_message(
&self,
group_openid: &str,
params: GroupMessageParams,
) -> Result<MessageResponse> {
debug!("Sending group message to {}", group_openid);
let body = MessageToCreate::from(params);
let path = resource::group_messages(group_openid);
self.request_message_response_body(Method::POST, &path, &body)
.await
}
pub async fn send_c2c_message(
&self,
openid: &str,
params: C2CMessageParams,
) -> Result<MessageResponse> {
debug!("Sending C2C message to {}", openid);
let body = MessageToCreate::from(params);
let path = resource::c2c_messages(openid);
self.request_message_response_body(Method::POST, &path, &body)
.await
}
}
#[cfg(test)]
mod tests {
use super::*;
use tokio::io::{AsyncReadExt, AsyncWriteExt};
use tokio::net::TcpListener;
use tokio::sync::oneshot;
async fn test_api(base_url: String) -> BotApi {
let token = crate::Token::new("APPID_XXXXXX", "SECRET_XXXXXX");
token
.set_cached_access_token_for_test("ACCESS_TOKEN_XXXXXX")
.await;
let mut http = crate::http::HttpClient::new(30, false).unwrap();
http.base_url = base_url;
BotApi::new(http, token)
}
async fn spawn_capture_server() -> (
String,
oneshot::Receiver<String>,
tokio::task::JoinHandle<()>,
) {
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
let addr = listener.local_addr().unwrap();
let (tx, rx) = oneshot::channel();
let handle = tokio::spawn(async move {
let (mut stream, _) = listener.accept().await.unwrap();
let mut request_bytes = Vec::new();
let mut buffer = [0_u8; 4096];
loop {
let n = stream.read(&mut buffer).await.unwrap();
request_bytes.extend_from_slice(&buffer[..n]);
let request = String::from_utf8_lossy(&request_bytes);
let Some(header_end) = request.find("\r\n\r\n") else {
continue;
};
let content_length = request
.lines()
.find_map(|line| {
let (name, value) = line.split_once(':')?;
name.eq_ignore_ascii_case("content-length")
.then(|| value.trim().parse::<usize>().ok())
.flatten()
})
.unwrap_or(0);
let body_start = header_end + 4;
if request_bytes.len().saturating_sub(body_start) >= content_length {
break;
}
}
let request = String::from_utf8_lossy(&request_bytes).to_string();
let _ = tx.send(request);
let body = r#"{"id":"message-1","timestamp":"2026-01-01T00:00:00+08:00"}"#;
let response = format!(
"HTTP/1.1 200 OK\r\ncontent-type: application/json\r\ncontent-length: {}\r\nconnection: close\r\n\r\n{}",
body.len(),
body
);
stream.write_all(response.as_bytes()).await.unwrap();
});
(format!("http://{addr}"), rx, handle)
}
fn request_body(request: &str) -> serde_json::Value {
let body = request.split("\r\n\r\n").nth(1).unwrap_or_default();
serde_json::from_str(body).unwrap()
}
#[tokio::test]
async fn group_message_params_send_message_body() {
let (base_url, request, server) = spawn_capture_server().await;
let api = test_api(base_url).await;
let response = api
.send_group_message("group-openid-1", GroupMessageParams::new_text("hello"))
.await
.unwrap();
assert_eq!(response.id.as_deref(), Some("message-1"));
let request = request.await.unwrap();
assert!(request.starts_with("POST /v2/groups/group-openid-1/messages HTTP/1.1"));
assert_eq!(
request_body(&request),
serde_json::json!({
"content": "hello"
})
);
server.await.unwrap();
}
#[tokio::test]
async fn c2c_message_params_send_message_body() {
let (base_url, request, server) = spawn_capture_server().await;
let api = test_api(base_url).await;
let response = api
.send_c2c_message("openid-1", C2CMessageParams::new_text("hello"))
.await
.unwrap();
assert_eq!(response.id.as_deref(), Some("message-1"));
let request = request.await.unwrap();
assert!(request.starts_with("POST /v2/users/openid-1/messages HTTP/1.1"));
assert_eq!(
request_body(&request),
serde_json::json!({
"content": "hello"
})
);
server.await.unwrap();
}
}