kovi-plugin-60s 0.1.1

kovi plugin
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

mod model;

use std::sync::Arc;
use kovi::log::{info};
use kovi::{serde_json, toml, Message, PluginBuilder as plugin, PluginBuilder, RuntimeBot};
use kovi::toml::{toml};
use kovi::utils::load_toml_data;
use crate::model::{Config, News, Response};

#[kovi::plugin]
async fn main() {
    let bot = PluginBuilder::get_runtime_bot();
    read_config(bot.clone());
    plugin::on_msg(move |event| {
        let bot = bot.clone();
        async move {
            let text = event.borrow_text().unwrap_or("");
            if text.starts_with("/60s") {
                let news = get_news(bot).await;
                let msg = Message::new()
                    .add_image(news.image.as_str());
                event.reply(msg);
            } else if text.starts_with("/help") {
                // let help = get_help(bot).await;
                // let help: String = {
                //     let mut result = String::new();
                //     for s in help {
                //         result.push_str(s.as_str());
                //     }
                //     result
                // };
                let msg = Message::new()
                    .add_text("- /60s");
                event.reply(msg);
            }
        }
    });
}

fn read_config(bot: Arc<RuntimeBot>) -> Config {
    let data_path = bot.get_data_path();
    let config_toml_path = data_path.join("config.toml");
    let default_config = toml! {
        url = "http://127.0.0.1:4399"
    };
    let config = load_toml_data(default_config, config_toml_path).unwrap();
    println!("{}", config.to_string());
    let config: Config = toml::from_str(&config.to_string()).unwrap();
    config
}

async fn get_news(bot: Arc<RuntimeBot>) -> News {
    let client = reqwest::Client::new();
    let config = read_config(bot);
    let res = client.get(format!("{}/v2/60s", config.url)).send().await.unwrap();
    let json = res.text().await.unwrap();
    info!("json = {:?}", json);
    let response: Response<News> = serde_json::from_str(json.as_str()).unwrap();
    response.data
}

// async fn get_help(bot: Arc<RuntimeBot>) -> Vec<String> {
//     let client = reqwest::Client::new();
//     let config = read_config(bot);
//     let res = client.get(format!("{}", config.url)).send().await.unwrap();
//     let json = res.text().await.unwrap();
//     info!("json = {:?}", json);
//     let response: Response<Vec<String>> = serde_json::from_str(json.as_str()).unwrap();
//     response.data
// }