bendns' repos / plent
smol bot
Diffstat (limited to 'src/bot/schematic.rs')
-rw-r--r--src/bot/schematic.rs95
1 files changed, 95 insertions, 0 deletions
diff --git a/src/bot/schematic.rs b/src/bot/schematic.rs
new file mode 100644
index 0000000..f409952
--- /dev/null
+++ b/src/bot/schematic.rs
@@ -0,0 +1,95 @@
+use anyhow::{anyhow, Result};
+use mindus::data::DataRead;
+use mindus::*;
+use poise::serenity_prelude::*;
+use regex::Regex;
+use std::fmt::Write;
+use std::sync::LazyLock;
+use std::{borrow::Cow, ops::ControlFlow};
+
+use super::{strip_colors, Msg, SUCCESS};
+
+static RE: LazyLock<Regex> =
+ LazyLock::new(|| Regex::new(r"(```)?(\n)?([^`]+)(\n)?(```)?").unwrap());
+
+async fn from_attachments(attchments: &[Attachment]) -> Result<Option<Schematic>> {
+ for a in attchments {
+ if a.filename.ends_with("msch") {
+ let s = a.download().await?;
+ let mut s = DataRead::new(&s);
+ let Ok(s) = Schematic::deserialize(&mut s) else {
+ continue;
+ };
+ return Ok(Some(s));
+ // discord uploads base64 as a file when its too long
+ } else if a.filename == "message.txt" {
+ let Ok(s) = String::from_utf8(a.download().await?) else {
+ continue;
+ };
+ let Ok(s) = Schematic::deserialize_base64(&s) else {
+ continue;
+ };
+ return Ok(Some(s));
+ }
+ }
+ Ok(None)
+}
+
+pub async fn with(m: Msg, c: &serenity::client::Context) -> Result<ControlFlow<Message, ()>> {
+ let author = m.author;
+ let send = |v: Schematic| async move {
+ let d = v.tags.get("description").map(|t| crate::conv::replace(t));
+ let name = crate::conv::replace(&strip_colors(v.tags.get("name").unwrap()));
+ let cost = v.compute_total_cost().0;
+ let p = tokio::task::spawn_blocking(move || to_png(&v)).await?;
+ anyhow::Ok(
+ m.channel
+ .send_message(c, |m| {
+ m.add_file(AttachmentType::Bytes {
+ data: Cow::Owned(p),
+ filename: "image.png".to_string(),
+ })
+ .embed(|e| {
+ e.attachment("image.png");
+ d.map(|v| e.description(v));
+ let mut s = String::new();
+ for (i, n) in cost.iter() {
+ if n == 0 {
+ continue;
+ }
+ write!(s, "{} {n} ", crate::conv::item(i)).unwrap();
+ }
+ e.field("req", s, true);
+ e.title(name)
+ .footer(|f| f.text(format!("requested by {author}")))
+ .color(SUCCESS)
+ })
+ })
+ .await?,
+ )
+ };
+
+ if let Ok(Some(v)) = from_attachments(&m.attachments).await {
+ println!("sent {}", v.tags.get("name").unwrap());
+ return Ok(ControlFlow::Break(send(v).await?));
+ }
+ if let Ok(v) = from_msg(&m.content) {
+ println!("sent {}", v.tags.get("name").unwrap());
+ return Ok(ControlFlow::Break(send(v).await?));
+ }
+ Ok(ControlFlow::Continue(()))
+}
+
+pub fn to_png(s: &Schematic) -> Vec<u8> {
+ super::png(s.render())
+}
+
+fn from_msg(msg: &str) -> Result<Schematic> {
+ let schem_text = RE
+ .captures(msg)
+ .ok_or(anyhow!("couldnt find schematic"))?
+ .get(3)
+ .unwrap()
+ .as_str();
+ Ok(Schematic::deserialize_base64(schem_text)?)
+}