socket-bot/ica-rs/src/tailchat/events.rs

75 lines
2.3 KiB
Rust
Raw Normal View History

2024-03-30 18:30:43 +08:00
use colored::Colorize;
use rust_socketio::asynchronous::Client;
use rust_socketio::{Event, Payload};
use tracing::info;
2024-03-30 14:24:19 +08:00
2024-06-04 00:22:36 +08:00
use crate::data_struct::tailchat::messages::ReciveMessage;
2024-03-30 18:30:43 +08:00
/// 所有
pub async fn any_event(event: Event, payload: Payload, _client: Client) {
2024-06-02 23:52:12 +08:00
let handled = [
2024-03-30 18:30:43 +08:00
// 真正处理过的
2024-06-04 00:22:36 +08:00
"notify:chat.message.add",
"notify:chat.message.delete",
2024-06-02 23:52:12 +08:00
// 也许以后会用到
2024-06-04 00:22:36 +08:00
"notify:chat.message.update",
"notify:chat.message.addReaction",
"notify:chat.message.removeReaction",
2024-06-02 23:52:12 +08:00
// 忽略的
2024-06-04 00:22:36 +08:00
"notify:chat.inbox.append", // 被 @ 之类的事件
2024-03-30 18:30:43 +08:00
];
match &event {
Event::Custom(event_name) => {
if handled.contains(&event_name.as_str()) {
return;
}
}
Event::Message => {
match payload {
Payload::Text(values) => {
if let Some(value) = values.first() {
if handled.contains(&value.as_str().unwrap()) {
return;
}
info!("收到消息 {}", value.to_string().yellow());
}
}
2024-06-02 23:52:12 +08:00
_ => {
return;
}
2024-03-30 18:30:43 +08:00
}
return;
}
_ => (),
}
match payload {
Payload::Binary(ref data) => {
println!("event: {} |{:?}", event, data)
}
Payload::Text(ref data) => {
print!("event: {}", event.as_str().purple());
for value in data {
println!("|{}", value);
}
}
_ => (),
}
}
2024-06-04 23:48:01 +08:00
pub async fn on_message(payload: Payload, client: Client) {
if let Payload::Text(values) = payload {
if let Some(value) = values.first() {
let message: ReciveMessage = serde_json::from_value(value.clone()).unwrap();
info!("收到消息 {:?}", message);
crate::py::call::tailchat_new_message_py(&message, &client).await;
2024-03-30 18:30:43 +08:00
}
}
}
2024-06-04 00:22:36 +08:00
pub async fn on_msg_delete(payload: Payload, _client: Client) {
2024-06-04 23:48:01 +08:00
if let Payload::Text(values) = payload {
if let Some(value) = values.first() {
info!("删除消息 {}", value.to_string().red());
2024-06-04 00:22:36 +08:00
}
}
2024-06-02 23:52:12 +08:00
}