doxing-emulator 0.2.0

//! Module for handling inline queries.

use super::dox_impl::{dox, get_full_info, get_user_full};
use frakti::{
    ParseMode,
    client_cyper::Bot,
    inline_mode::{
        InlineQuery, InlineQueryResultArticle, InlineQueryResultsButton, InputMessageContent,
        InputTextMessageContent,
    },
};
use log::info;

/// Handle inline queries.
pub async fn handle_inline_query(bot: &Bot, inline: &InlineQuery) -> InlineQueryResultArticle {
    info!("Handling inline query: {}", inline.query);
    // Reject users that the bot doesn't know
    let doxer = &inline.from;
    let Some(doxer_info) = get_full_info(bot, doxer.id).await else {
        return create_article(
            include_str!("./messages/doxer-identification-failed.html"),
            "ERR_DOXER_IDENTIFICATION_FAILED",
            "谁在说话？滚木吗？",
        );
    };
    // Actual doxing
    let query = inline.query.trim();
    if query.is_empty() {
        let report = dox(&doxer, Some(&doxer_info));
        create_article(report, format!("开盒 {}", doxer.first_name), "盒盒盒")
    } else if let Ok(user_id) = query.parse() {
        // Can be parsed as user_id
        match get_user_full(bot, user_id).await {
            Some((doxee, doxee_info)) => {
                let report = dox(&doxee, doxee_info.as_ref());
                create_article(report, format!("开盒 {}", doxee.first_name), "盒盒盒")
            }
            None => create_article(
                include_str!("./messages/doxee-identification-failed.html"),
                "ERR_DOXEE_IDENTIFICATION_FAILED",
                "马冬什么？马冬梅。什么冬梅啊？马冬梅啊。马什么梅啊？行，大爷，您先凉快吧。",
            ),
        }
    } else {
        // Not user id
        create_article(
            include_str!("./messages/not-user-id.html"),
            "ERR_NOT_USER_ID",
            "发的啥呀这是？",
        )
    }
}

/// Create an article with given message, title and description.
fn create_article(
    message: impl Into<String>,
    title: impl Into<String>,
    description: impl Into<String>,
) -> InlineQueryResultArticle {
    let content = InputTextMessageContent::builder()
        .message_text(message)
        .parse_mode(ParseMode::Html)
        .build();
    InlineQueryResultArticle::builder()
        .id("1")
        .title(title)
        .description(description)
        .input_message_content(InputMessageContent::Text(content))
        .build()
}

/// Create a button that sends `/start help` to the bot.
pub fn help_button() -> InlineQueryResultsButton {
    InlineQueryResultsButton::builder()
        .text("遇到 ERR? 点我查看帮助!")
        .start_parameter("help")
        .build()
}