feat(UI/Codebase): Made menu persistent. After starting the bot, the menu will remain on the screen and be interactable for the user while the bot is running

.gitignore

@@ -1 +1,2 @@
 /target
+cliff.toml

Cargo.toml

@@ -1,9 +1,11 @@
 [package]
 name = "RhythmWorks"
-version = "0.5.0"
+version = "0.6.0"
 edition = "2024"
 
 [dependencies]
+color-eyre = "0.6.5"
+crossterm = "0.29.0"
 dialoguer = "0.12.0"
 keyring = { version = "3.6.3", features = ["linux-native", "windows-native"] }
 opentelemetry = { version = "0.27.0", features = ["logs", "metrics", "trace"] }
@@ -11,6 +13,7 @@ opentelemetry-otlp = { version = "0.27.0", features = ["trace", "metrics", "grpc
 opentelemetry-proto = "0.27.0"
 opentelemetry-semantic-conventions = "0.27.0"
 opentelemetry_sdk = { version = "0.27.0", features = ["rt-tokio", "trace"] }
+ratatui = { version = "0.30.0", features = ["all-widgets"] }
 reqwest = { version = "0.12.28", features = ["blocking", "json"] }
 semver = "1.0.27"
 serde = "1.0.228"
@@ -19,6 +22,7 @@ serenity = { version = "0.12.5", features = ["client", "gateway", "voice"] }
 songbird = { version = "0.5.0", features = ["builtin-queue", "driver", "serenity"] }
 symphonia = { version = "0.5.5", features = ["aac", "alac", "isomp4", "mp3"] }
 tokio = "1.48.0"
+tokio-util = "0.7.18"
 tonic = { version = "0.12.3", features = ["tls-roots"] }
 tracing = "0.1.41"
 tracing-opentelemetry = "0.32.0"

changelog.md

@@ -0,0 +1 @@
+## [0.5.2] - 2026-01-21

src/main.rs

@@ -1,4 +1,3 @@
-use core::error;
 use dialoguer::Select;
 use dialoguer::theme::ColorfulTheme;
 use modules::*;
@@ -22,12 +21,19 @@ use songbird::input::Input;
 use std::env;
 use std::process::Command;
 use std::process::Stdio;
-pub mod modules;
+use tokio::task::JoinHandle;
+use tokio_util::sync::CancellationToken;
 
+pub mod modules;
+enum BotState {
+    Stopped,
+    Running,
+}
 struct TrackTraceHandler {
     otel_ctx: opentelemetry::Context,
     guild_id: String,
     track_url: String,
+    hls_url: String,
     event_name: &'static str,
 }
 
@@ -45,7 +51,15 @@ impl VoiceEventHandler for TrackTraceHandler {
                 .set_attribute(KeyValue::new("Guild_ID", self.guild_id.clone()));
             cx.span()
                 .set_attribute(KeyValue::new("Youtube_URL", self.track_url.clone()));
-            cx.span().add_event(self.event_name, vec![KeyValue::new("track_status", track_status.unwrap().to_string())]);
+            cx.span()
+                .set_attribute(KeyValue::new("Stream_URL", self.hls_url.clone()));
+            cx.span().add_event(
+                self.event_name,
+                vec![KeyValue::new(
+                    "track_status",
+                    track_status.unwrap().to_string(),
+                )],
+            );
         });
 
         None
@@ -215,7 +229,7 @@ impl EventHandler for Handler {
                 manager.join(guild_id, channel_id).await.unwrap()
             };
             let output = match Command::new("./yt-dlp")
-                .args(["-f", "bestaudio", "-g", &yt_url])
+                .args(["-f", "bestaudio", "-g", "--no-playlist", &yt_url])
                 .stdout(Stdio::piped())
                 .output()
             {
@@ -240,6 +254,7 @@ impl EventHandler for Handler {
                 .expect("Invalid UTF-8")
                 .trim()
                 .to_string();
+            let telemetry_stream_url = stream_url.clone();
             let mut call = handler_lock.lock().await;
             let input = Input::from(HlsRequest::new(reqwest::Client::new(), stream_url));
             let track_handle = call.enqueue_input(input).await;
@@ -265,6 +280,7 @@ impl EventHandler for Handler {
                     otel_ctx: otel_ctx.clone(),
                     guild_id: guild_id.to_string().clone(),
                     track_url: yt_url.clone(),
+                    hls_url: telemetry_stream_url.clone(),
                     event_name: "Track Started Playing",
                 },
             );
@@ -274,6 +290,7 @@ impl EventHandler for Handler {
                     otel_ctx: otel_ctx.clone(),
                     guild_id: guild_id.to_string().clone(),
                     track_url: yt_url.clone(),
+                    hls_url: telemetry_stream_url.clone(),
                     event_name: "Track Finished",
                 },
             );
@@ -283,6 +300,7 @@ impl EventHandler for Handler {
                     otel_ctx: otel_ctx.clone(),
                     guild_id: guild_id.to_string().clone(),
                     track_url: yt_url.clone(),
+                    hls_url: telemetry_stream_url.clone(),
                     event_name: "Playback had an error",
                 },
             );
@@ -337,6 +355,30 @@ impl EventHandler for Handler {
         println!("{} is connected!", ready.user.name);
     }
 }
+async fn run_bot(token: String, shutdown: CancellationToken) {
+    let intents = GatewayIntents::GUILD_MESSAGES
+        | GatewayIntents::GUILDS
+        | GatewayIntents::DIRECT_MESSAGES
+        | GatewayIntents::MESSAGE_CONTENT
+        | GatewayIntents::GUILD_VOICE_STATES;
+
+    let mut client = Client::builder(&token, intents)
+        .event_handler(Handler)
+        .register_songbird()
+        .await
+        .expect("Err creating client");
+
+    tokio::select! {
+        result = client.start() => {
+            if let Err(why) = result {
+                println!("Client error: {why:?}");
+            }
+        }
+        _ = shutdown.cancelled() => {
+            println!("Shutting down bot...");
+        }
+    }
+}
 
 #[tokio::main]
 async fn main() {
@@ -353,9 +395,18 @@ This is free software, and you are welcome to redistribute it under certain cond
         .await;
     let service_name = "RhythmWorks";
     let username = env::var("USER").or_else(|_| env::var("USERNAME")).unwrap();
+    let mut bot_task: Option<JoinHandle<()>> = None;
+    let mut shutdown_token: Option<CancellationToken> = None;
+    let mut bot_state = BotState::Stopped;
     loop {
         let token = credentialmanager::get_token(service_name, &username);
-        let options: Vec<&'static str> = vec!["- Start Bot", "- Change Token", "- Delete Token"];
+        let options: Vec<&'static str> = vec![
+            "- Start Bot",
+            "- Stop Bot",
+            "- Change Token",
+            "- Delete Token",
+            "- Exit",
+        ];
         let selection = Select::with_theme(&ColorfulTheme::default())
             .with_prompt("Choose an Option")
             .default(0)
@@ -364,26 +415,44 @@ This is free software, and you are welcome to redistribute it under certain cond
             .unwrap();
         match selection {
             0 => {
-                let intents = GatewayIntents::GUILD_MESSAGES
-                    | GatewayIntents::GUILDS
-                    | GatewayIntents::DIRECT_MESSAGES
-                    | GatewayIntents::MESSAGE_CONTENT
-                    | GatewayIntents::GUILD_VOICE_STATES;
-                let mut client = Client::builder(&token, intents)
-                    .event_handler(Handler)
-                    .register_songbird()
-                    .await
-                    .expect("Err creating client");
-                if let Err(why) = client.start().await {
-                    println!("Client error: {why:?}");
+                if matches!(bot_state, BotState::Running) {
+                    println!("[x] Bot is already running.");
+                    continue;
                 }
+
+                let token = credentialmanager::get_token(service_name, &username);
+                let shutdown = CancellationToken::new();
+                let shutdown_clone = shutdown.clone();
+
+                let task = tokio::spawn(run_bot(token, shutdown_clone));
+
+                bot_task = Some(task);
+                shutdown_token = Some(shutdown);
+                bot_state = BotState::Running;
+
+                println!("✅ Bot started.");
             }
             1 => {
-                credentialmanager::change_token(service_name, &username);
+                if let Some(token) = shutdown_token.take() {
+                    token.cancel();
+                }
+                if let Some(task) = bot_task.take() {
+                    let _ = task.await;
+                }
+                println!("Bot stopped.");
             }
             2 => {
+                credentialmanager::change_token(service_name, &username);
+            }
+            3 => {
                 credentialmanager::remove_token(service_name, &username);
             }
+            4 => {
+                if let Some(token) = shutdown_token {
+                    token.cancel();
+                }
+                break;
+            }
             _ => println!("Invalid Option"),
         }
     }