From 4d50971704e6dd9fff6a74899fc2b6e3a81228e2 Mon Sep 17 00:00:00 2001
From: ByteDice <bytedice.official@gmail.com>
Date: Sun, 16 Feb 2025 18:35:41 +0100
Subject: [PATCH] added token for a non-production bot

---
 src/main.rs | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 00ff5b6..4ef7fb6 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -97,8 +97,8 @@ async fn main() {
 
 
 async fn start(args: Args) {
-  let data = gen_data(args).await;
-  let mut bot = gen_bot(data).await;
+  let data = gen_data(args.clone()).await;
+  let mut bot = gen_bot(data, args).await;
 
   rs_println!("Starting bot...");
   bot.start().await.unwrap();
@@ -127,8 +127,15 @@ async fn gen_data(args: Args) -> Data {
 }
 
 
-async fn gen_bot(data: Data) -> Client {
-  let token = std::env::var("ASSISTANT_TOKEN").expect("missing ASSISTANT_TOKEN env var");
+async fn gen_bot(data: Data, args: Args) -> Client {
+  let token;
+  if !args.dev {
+    token = std::env::var("ASSISTANT_TOKEN").expect("Missing ASSISTANT_TOKEN env var!");
+  }
+  else {
+    token = std::env::var("ASSISTANT_TOKEN_TEST").expect("Missing ASSISTANT_TOKEN_TEST env var!");
+  }
+
   let intents = serenity::GatewayIntents::all();
 
   let peek_len = 27;