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

use colored::Colorize;
use rust_socketio::asynchronous::Client;
use rust_socketio::{Event, Payload};
use tracing::{info, warn};

use crate::client::{send_message, IcalinguaStatus};
use crate::data_struct::all_rooms::Room;
use crate::data_struct::messages::NewMessage;
use crate::data_struct::online_data::OnlineData;
use crate::{py, VERSION};

/// 获取在线数据
pub async fn get_online_data(payload: Payload, _client: Client) {
    if let Payload::Text(values) = payload {
        if let Some(value) = values.first() {
            let online_data = OnlineData::new_from_json(value);
            info!("update_online_data {}", format!("{:?}", online_data).cyan());
            unsafe {
                crate::ClientStatus.update_online_data(online_data);
            }
        }
    }
}

/// 接收消息
pub async fn add_message(payload: Payload, client: Client) {
    if let Payload::Text(values) = payload {
        if let Some(value) = values.first() {
            let message = NewMessage::new_from_json(value);
            // 检测是否在过滤列表内
            if IcalinguaStatus::get_config()
                .filter_list
                .contains(&message.sender_id)
            {
                return;
            }
            info!("add_message {}", message.output().cyan());
            // info!("add_message {}", format!("{:#?}", message).cyan());
            // 就在这里处理掉最基本的消息
            // 之后的处理交给插件
            if message.content.eq("/bot-rs") && !message.is_from_self() && !message.is_reply() {
                let reply = message.reply_with(&format!("ica-async-rs pong v{}", VERSION));
                send_message(&client, &reply).await;
            }
            // python 插件
            py::new_message_py(&message, &client).await;
        }
    }
}

/// 撤回消息
pub async fn delete_message(payload: Payload, _client: Client) {
    if let Payload::Text(values) = payload {
        // 消息 id
        if let Some(value) = values.first() {
            if let Some(msg_id) = value.as_str() {
                info!("delete_message {}", format!("{}", msg_id).yellow());
            }
        }
    }
}

pub async fn update_all_room(payload: Payload, _client: Client) {
    if let Payload::Text(values) = payload {
        if let Some(value) = values.first() {
            if let Some(raw_rooms) = value.as_array() {
                let rooms: Vec<Room> = raw_rooms
                    .iter()
                    .map(|room| Room::new_from_json(room))
                    .collect();
                unsafe {
                    crate::ClientStatus.update_rooms(rooms.clone());
                }
                info!("update_all_room {}", rooms.len());
            }
        }
    }
}

/// 所有
pub async fn any_event(event: Event, payload: Payload, _client: Client) {
    let handled = vec![
        // 真正处理过的
        "authSucceed",
        "authFailed",
        "authRequired",
        "requireAuth",
        "onlineData",
        "addMessage",
        "deleteMessage",
        "setAllRooms",
        // 忽略的
        "notify",
        "closeLoading", // 发送消息/加载新聊天 有一个 loading
        "updateRoom",
        "syncRead",
    ];
    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());
                    }
                }
                _ => (),
            }
            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.to_string());
            }
        }
        _ => (),
    }
}

pub async fn connect_callback(payload: Payload, _client: Client) {
    match payload {
        Payload::Text(values) => {
            if let Some(value) = values.first() {
                match value.as_str() {
                    Some("authSucceed") => {
                        info!("{}", "已经登录到 icalingua!".green())
                    }
                    Some("authFailed") => {
                        info!("{}", "登录到 icalingua 失败!".red());
                        panic!("登录失败")
                    }
                    Some("authRequired") => {
                        warn!("{}", "需要登录到 icalingua!".yellow())
                    }
                    Some(msg) => {
                        warn!("未知消息 {}", msg.yellow())
                    }
                    None => (),
                }
            }
        }
        _ => (),
    }
}