bendns' repos / plent
smol bot
Diffstat
-rw-r--r--Cargo.toml1
-rw-r--r--src/bot/mod.rs65
2 files changed, 53 insertions, 13 deletions
diff --git a/Cargo.toml b/Cargo.toml
index 4192b6e..5b43311 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -63,6 +63,7 @@ sled = { version = "0.34.7", features = ["compression"] }
remapper = { version = "0.1.0", path = "../remapper" }
implicit-fn = "0.1.0"
sql = "0.4.3"
+charts-rs = { version = "0.3.24", features = ["image", "image-encoder", "resvg"] }
[features]
server = ["axum"]
diff --git a/src/bot/mod.rs b/src/bot/mod.rs
index 08dcbb2..24f0cc0 100644
--- a/src/bot/mod.rs
+++ b/src/bot/mod.rs
@@ -7,6 +7,7 @@ pub mod search;
mod sorter;
mod db;
mod data;
+use charts_rs::{Series, THEME_GRAFANA};
pub use data::log;
@@ -227,18 +228,6 @@ async fn handle_message(
.unwrap_or(new_message.author.name.clone());
let post = EXTRA.get(&new_message.channel_id.get()).map(|x| x.clone());
let (dir, l, repo) = sep(SPECIAL.get(&new_message.channel_id.get()).or(post.as_ref()));
- if new_message.author.id.get()==OWNER&&new_message.content.contains("meow"){
- new_message.author
- .direct_message(
- c,
- poise::serenity_prelude::CreateMessage::new().content(format!(
- "your match `meow` was satisfied on message ```\n{}\n``` {}",
- new_message.content.replace('`', "​`"),
- new_message.link()
- )),
- )
- .await?;
- }
let m = Msg {
author: who.clone(),
locale: new_message.author.locale.clone().unwrap_or("unknown locale".to_string()),
@@ -321,7 +310,7 @@ impl Bot {
std::env::var("TOKEN").unwrap_or_else(|_| read_to_string("token").expect("wher token"));
let f = poise::Framework::builder()
.options(poise::FrameworkOptions {
- commands: vec![logic::run(), lb(), logic::run_file(), sorter::sorter(), sorter::mapper(), schembrowser_instructions(), lb_no_vds(), ping(), help(), scour(), search::search(), search::file(), rename(), rename_file(), render(), render_file(), render_message(), map::render_message()],
+ commands: vec![logic::run(), lb(), logic::run_file(), sorter::sorter(), sorter::mapper(), schembrowser_instructions(), lb_no_vds(), ping(), help(), scour(), search::search(), search::file(), rename(), rename_file(), render(), render_file(), render_message(), map::render_message(), stats()],
event_handler: |c, e, _, d| {
Box::pin(async move {
match e {
@@ -507,6 +496,7 @@ impl Bot {
render_message(),
rename(),
rename_file(),
+ stats(),
map::render_message(),
logic::run_file(),
sorter::sorter(),
@@ -1058,4 +1048,53 @@ pub async fn schembrowser_instructions(c: Context<'_>) -> Result<()> {
)
.await?;
Ok(())
+}
+
+
+#[poise::command(
+ slash_command,
+ install_context = "Guild",
+ interaction_context = "Guild|PrivateChannel"
+)]
+/// Statistics
+#[implicit_fn::implicit_fn]
+pub async fn stats(c: Context<'_>) -> Result<()> {
+ let mut guilds = HashMap::<_, u64>::default();
+ let mut schem_calls = 0;
+ let mut map_calls = 0;
+ let mut eval_calls = 0;
+ for x in std::fs::read_to_string("data").unwrap().lines().map(serde_json::from_str::<serde_json::Value>).filter_map(Result::ok) {
+ *guilds.entry(x.get("guild").unwrap().as_u64().unwrap()).or_default() += 1;
+ let x = x.get("cname").unwrap().as_str().unwrap();
+ if x.contains("schematic"){
+ schem_calls+=1;
+ }
+ if x.contains("map") {
+ map_calls +=1;
+ }
+ if x.contains("eval") {
+ eval_calls+=1;
+ }
+ }
+ use futures::stream;
+
+ let mut x = stream::iter(guilds.into_iter().filter(_.0 != 0).filter(_.1 > 25))
+ .map(async |(k,v)| {GuildId::new(k).to_partial_guild(c.http()).await.map(|x| (x.name, v)).unwrap_or(("DM".to_string(), v))})
+ .buffer_unordered(16).collect::<Vec<_>>().await.into_iter()
+ .map(|(a,b)| Series::new(a, vec![b as f32]))
+ .collect::<Vec<_>>();
+ x.sort_by_key(|x| x.data[0] as u64);
+
+ let mut ch = charts_rs::PieChart::new_with_theme(x, THEME_GRAFANA);
+ ch.title_text = "usage".into();
+ // ch.font_family = "Verdana".into();
+ ch.width = 800.0;
+ ch.rose_type = Some(false);
+ ch.inner_radius = 20.0;
+ ch.height = 300.0;
+
+ use emoji::named::*;
+ let x = charts_rs::svg_to_webp(&ch.svg().unwrap()).unwrap();
+ poise::send_reply(c, poise::CreateReply::default().attachment(CreateAttachment::bytes(x, "chart.webp")).content(format!("{EDIT} total schematics rendered: {schem_calls}\n{MAP} total maps rendered: {map_calls}\n{WORLD_PROCESSOR} eval calls: {eval_calls}"))).await?;
+ Ok(())
} \ No newline at end of file