From be22c138146e1e0fca2ba0de6b22f35e8583664f Mon Sep 17 00:00:00 2001
From: ByteDice <bytedice.official@gmail.com>
Date: Thu, 30 Jan 2025 22:36:56 +0100
Subject: [PATCH] added way too many features for a single commit. Who cares
 though?

---
 data/status.txt |  1 +
 src/cmds.rs     | 90 +++++++++++++++++++++++++++++++++++++++++++++++--
 src/events.rs   | 14 ++++++--
 src/main.rs     | 19 +++++++++--
 4 files changed, 115 insertions(+), 9 deletions(-)
 create mode 100644 data/status.txt

diff --git a/data/status.txt b/data/status.txt
new file mode 100644
index 0000000..4251794
--- /dev/null
+++ b/data/status.txt
@@ -0,0 +1 @@
+🎲 https://bytedice.net
\ No newline at end of file
diff --git a/src/cmds.rs b/src/cmds.rs
index 7026560..ca8d5ed 100644
--- a/src/cmds.rs
+++ b/src/cmds.rs
@@ -1,11 +1,95 @@
 use crate::{Context, Error};
 
+use poise::{serenity_prelude::{Color, CreateEmbed, OnlineStatus, Timestamp}, CreateReply};
+
+
 #[poise::command(slash_command, prefix_command)]
 pub async fn ping(
   ctx: Context<'_>,
-  #[description = "The text to echo back"] text: String,
+  #[description = "The text to echo back"] text: Option<String>,
 ) -> Result<(), Error>
 {
-  ctx.reply(text).await?;
-  Ok(())
+  let t = text.unwrap_or_else(|| "Pong".to_string());
+
+  let r = CreateReply {
+    content: Some(t),
+    ephemeral: Some(true),
+    ..Default::default()
+  };
+
+  ctx.send(r).await?;
+
+  return Ok(());
+}
+
+
+#[poise::command(slash_command, prefix_command, default_member_permissions = "ADMINISTRATOR")]
+pub async fn stop(
+  ctx: Context<'_>,
+  #[description = "Type \"i want to stop the bot now\" to confirm"] confirmation: Option<String>,
+) -> Result<(), Error>
+{
+  let dev_enabled = ctx.data().dev;
+  let should_stop = dev_enabled
+    || confirmation.unwrap_or_else(|| "".to_string()).to_lowercase() == "i want to stop the bot now";
+
+  let r_success = CreateReply {
+    content: Some("Shutting down...".to_string()),
+    ephemeral: Some(true),
+    ..Default::default()
+  };
+
+  let r_fail = CreateReply {
+    content: Some("Failed to shut down.".to_string()),
+    ephemeral: Some(true),
+    ..Default::default()
+  };
+
+  if should_stop {
+    ctx.send(r_success).await?;
+    ctx.serenity_context().set_presence(None, OnlineStatus::Invisible);
+    ctx.framework().shard_manager.shutdown_all().await;
+  }
+  else {
+    ctx.send(r_fail).await?;
+  }
+
+  return Ok(());
+}
+
+
+fn none_to_empty(string: Option<String>) -> String {
+  return string.unwrap_or_else(|| "".to_string());
+}
+
+
+#[poise::command(
+  slash_command,
+  prefix_command,
+  default_member_permissions = "ADMINISTRATOR"
+)]
+pub async fn embed(
+  ctx: Context<'_>,
+  #[description = "Title of embed."] title: Option<String>,
+  #[description = "Body text of embed."] description: String,
+  #[description = "Color of side strip."] color: Option<u32>,
+  #[description = "A URL the title is bound to."] url: Option<String>,
+  #[description = "Timestamp at bottom (best to leave empty)."] timestamp: Option<Timestamp>,
+) -> Result<(), Error> 
+{
+  let embed = CreateEmbed::new()
+    .title      (none_to_empty(title))
+    .description(description)
+    .colour     (Color::new(color.unwrap_or_else(|| 5793266)))
+    .url        (none_to_empty(url))
+    .timestamp  (timestamp.unwrap_or_else(|| Timestamp::now()));
+
+  let r = CreateReply {
+    embeds: vec![embed],
+    ..Default::default()
+  };
+
+  ctx.send(r).await?;
+
+  return Ok(());
 }
\ No newline at end of file
diff --git a/src/events.rs b/src/events.rs
index 93d1ef4..5b443ec 100644
--- a/src/events.rs
+++ b/src/events.rs
@@ -1,12 +1,12 @@
 use crate::{Data, Error};
 
-use poise::serenity_prelude as serenity;
+use poise::serenity_prelude::{self as serenity, ActivityData};
 
 use std::future::Future;
 use std::pin::Pin;
 
 pub fn event_handler<'a>(
-    _ctx: &'a serenity::Context,
+    ctx: &'a serenity::Context,
     event: &'a serenity::FullEvent,
     _framework: poise::FrameworkContext<'a, Data, Error>,
     _data: &'a Data,
@@ -18,7 +18,15 @@ pub fn event_handler<'a>(
         data_about_bot.user.name,
         data_about_bot.user.id
       );
+
+      let file_text = std::fs::read_to_string("./data/status.txt").unwrap();
+      let custom_activity = ActivityData::custom(file_text);
+      // TODO: make custom rich presence
+      //let playing_activity
+
+      ctx.online();
+      ctx.set_activity(Some(custom_activity));
     }
-    Ok(())
+    return Ok(());
   })
 }
\ No newline at end of file
diff --git a/src/main.rs b/src/main.rs
index 5fa9df1..15a5d0c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,15 +1,24 @@
 mod cmds;
 mod events;
 
+use std::env;
+
 use poise::serenity_prelude as serenity;
 
-struct Data {}
+struct Data {
+  dev: bool
+}
 type Error = Box<dyn std::error::Error + Send + Sync>;
 type Context<'a> = poise::Context<'a, Data, Error>;
 
 
 #[tokio::main]
 async fn main() {
+  let args: Vec<String> = env::args().collect();
+  let data = Data {
+    dev: args.contains(&"--dev".to_string())
+  };
+
   let token = std::env::var("ASSISTANT_TOKEN").expect("missing ASSISTANT_TOKEN env var");
   let intents = serenity::GatewayIntents::all();
 
@@ -21,14 +30,18 @@ async fn main() {
 
   let framework = poise::Framework::builder()
     .options(poise::FrameworkOptions {
-      commands: vec![cmds::ping()],
+      commands: vec![
+        cmds::ping(),
+        cmds::embed(),
+        cmds::stop()
+      ],
       event_handler: events::event_handler,
       ..Default::default()
     })
     .setup(|ctx, _ready, framework| {
       Box::pin(async move {
         poise::builtins::register_globally(ctx, &framework.options().commands).await?;
-        Ok(Data {})
+        return Ok(data);
       })
     })
     .build();