handy scripts

Cargo.lock

@@ -892,7 +892,7 @@ dependencies = [
 
 [[package]]
 name = "lyra"
-version = "0.1.0"
+version = "0.3.0"
 dependencies = [
  "dotenv",
  "openssl",

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "lyra"
-version = "0.1.0"
+version = "0.3.0"
 edition = "2021"
 
 [dependencies]

scripts/cross-build.sh

@@ -0,0 +1,3 @@
+#!/bin/bash
+
+cross build -r --target aarch64-unknown-linux-gnu

scripts/launch.sh

@@ -0,0 +1,3 @@
+#!/bin/bash
+
+nohup ./lyra > lyra.log 2> lyra.err < /dev/null &

src/commands/music/deafen.rs

@@ -27,7 +27,15 @@ async fn deafen(ctx: &Context, msg: &Message) -> CommandResult {
     let mut handler = handler_lock.lock().await;
 
     if handler.is_deaf() {
-        check_msg(msg.channel_id.say(&ctx.http, "Already deafened").await);
+        if let Err(err) = handler.deafen(false).await {
+            check_msg(
+                msg.channel_id
+                    .say(&ctx.http, format!("Failed: {:?}", err))
+                    .await,
+            );
+        }
+
+        check_msg(msg.channel_id.say(&ctx.http, "Undeafened").await);
     } else {
         if let Err(err) = handler.deafen(true).await {
             check_msg(

src/commands/music/join.rs

@@ -9,15 +9,8 @@ use crate::commands::{misc::check_msg, music::misc::TrackErrorNotifier};
 #[command]
 #[only_in(guilds)]
 async fn join(ctx: &Context, msg: &Message) -> CommandResult {
-    let (guild_id, channel_id) = {
-        let guild = msg.guild(&ctx.cache).unwrap();
-        let channel_id = guild
-            .voice_states
-            .get(&msg.author.id)
-            .and_then(|voice_state| voice_state.channel_id);
-
-        (guild.id, channel_id)
-    };
+    let guild_id = msg.guild_id.unwrap();
+    let channel_id = msg.guild(&ctx.cache).unwrap().voice_states.get(&msg.author.id).and_then(|voice_state| voice_state.channel_id);
 
     let connect_to = match channel_id {
         Some(channel) => channel,

src/commands/music/loopcurrent.rs

@@ -0,0 +1,83 @@
+use serenity::framework::standard::macros::command;
+use serenity::framework::standard::{Args, CommandResult};
+use serenity::model::prelude::*;
+use serenity::prelude::*;
+use songbird::tracks::LoopState;
+
+use crate::commands::misc::check_msg;
+
+#[command]
+#[aliases(loop)]
+#[only_in(guilds)]
+async fn loopcurrent(ctx: &Context, msg: &Message, mut args: Args) -> CommandResult {
+    let guild_id = msg.guild_id.unwrap();
+
+    let manager = songbird::get(ctx)
+        .await
+        .expect("Client placed at init")
+        .clone();
+
+
+    if let Some(handler_lock) = manager.get(guild_id) {
+        let handler = handler_lock.lock().await;
+        let queue = handler.queue();
+
+        let track = queue.current().unwrap().get_info().await;
+        let is_looped = track.unwrap().loops;
+
+
+        let count = match args.single::<usize>() {
+            Ok(count) => count,
+            Err(_) => 100,
+        };
+
+        match is_looped {
+            LoopState::Infinite => {
+                let _ = queue.current().unwrap().disable_loop();
+
+                check_msg(
+                    msg.channel_id
+                        .say(
+                            &ctx.http,
+                            format!("Song unlooped."),
+                        )
+                    .await,
+                );
+            }
+            LoopState::Finite(_) => { 
+                if count < 100 {
+                    let _ = queue.current().unwrap().loop_for(count);
+                    check_msg(
+                        msg.channel_id
+                            .say(
+                                &ctx.http,
+                                format!("Song looped forever (a very long time)."),
+                            )
+                        .await,
+                    )
+                }
+                else {
+                    let _ = queue.current().unwrap().enable_loop();
+
+                    check_msg(
+                        msg.channel_id
+                            .say(
+                                &ctx.http,
+                                format!("Song looped {} times.", count),
+                            )
+                        .await,
+                    )
+                }
+
+            }
+        }
+    } else {
+        check_msg(
+            msg.channel_id
+                .say(&ctx.http, "Not in a voice channel to play in")
+            .await,
+        );
+    }
+
+    Ok(())
+}

src/commands/music/mod.rs

@@ -7,5 +7,6 @@ pub mod play;
 pub mod queue;
 pub mod skip;
 pub mod stop;
-pub mod undeafen;
-pub mod unmute;
+pub mod loopcurrent;
+pub mod pause;
+pub mod resume;

src/commands/music/mute.rs

@@ -27,7 +27,15 @@ async fn mute(ctx: &Context, msg: &Message) -> CommandResult {
     let mut handler = handler_lock.lock().await;
 
     if handler.is_mute() { 
-        check_msg(msg.channel_id.say(&ctx.http, "Already muted").await);
+        if let Err(e) = handler.mute(false).await {
+            check_msg(
+                msg.channel_id
+                    .say(&ctx.http, format!("failed: {:?}", e))
+                    .await,
+            );
+        }
+
+        check_msg(msg.channel_id.say(&ctx.http, "Unmuted").await);
     } else {
         if let Err(err) = handler.mute(true).await {
             check_msg(

src/commands/music/pause.rs

@@ -7,7 +7,7 @@ use crate::commands::misc::check_msg;
 
 #[command]
 #[only_in(guilds)]
-async fn undeafen(ctx: &Context, msg: &Message) -> CommandResult {
+async fn pause(ctx: &Context, msg: &Message) -> CommandResult {
     let guild_id = msg.guild_id.unwrap();
 
     let manager = songbird::get(ctx)
@@ -16,20 +16,22 @@ async fn undeafen(ctx: &Context, msg: &Message) -> CommandResult {
         .clone();
 
     if let Some(handler_lock) = manager.get(guild_id) {
-        let mut handler = handler_lock.lock().await;
-        if let Err(e) = handler.deafen(false).await {
+        let handler = handler_lock.lock().await;
+        let queue = handler.queue();
+        let _ = queue.pause();
+
         check_msg(
             msg.channel_id
-                    .say(&ctx.http, format!("Failed: {:?}", e))
+                .say(
+                    &ctx.http,
+                    format!("Song paused."),
+                )
                 .await,
         );
-        }
-
-        check_msg(msg.channel_id.say(&ctx.http, "Undeafened").await);
     } else {
         check_msg(
             msg.channel_id
-                .say(&ctx.http, "Not in a voice channel to undeafen in")
+                .say(&ctx.http, "Not in a voice channel to play in")
                 .await,
         );
     }

src/commands/music/play.rs

@@ -3,7 +3,7 @@ use serenity::framework::standard::{Args, CommandResult};
 use serenity::model::prelude::*;
 use serenity::prelude::*;
 use reqwest::Client as HttpClient;
-use songbird::input::YoutubeDl;
+use songbird::input::{Compose, YoutubeDl};
 use songbird::events::TrackEvent;
 
 use crate::commands::{misc::check_msg, music::misc::TrackErrorNotifier};
@@ -31,17 +31,10 @@ async fn play(ctx: &Context, msg: &Message, mut args: Args) -> CommandResult {
         }
     };
 
-    let do_search = !url.starts_with("http");
+    let is_search = !url.starts_with("http");
 
-    let (guild_id, channel_id) = {
-        let guild = msg.guild(&ctx.cache).unwrap();
-        let channel_id = guild
-            .voice_states
-            .get(&msg.author.id)
-            .and_then(|voice_state| voice_state.channel_id);
-
-        (guild.id, channel_id)
-    };
+    let guild_id = msg.guild_id.unwrap();
+    let channel_id = msg.guild(&ctx.cache).unwrap().voice_states.get(&msg.author.id).and_then(|voice_state| voice_state.channel_id);
 
     let connect_to = match channel_id {
         Some(channel) => channel,
@@ -67,19 +60,21 @@ async fn play(ctx: &Context, msg: &Message, mut args: Args) -> CommandResult {
     if let Ok(handler_lock) = manager.join(guild_id, connect_to).await {
         let mut handler = handler_lock.lock().await;
        
+        // if let Err(err) = handler.deafen(true).await {println!("Failed to deafen: {:?}", err)};
         handler.add_global_event(TrackEvent::Error.into(), TrackErrorNotifier);
 
-        let src = if do_search {
-            YoutubeDl::new_ytdl_like("yt-dlp", http_client, url)
+        let mut src = if is_search {
+            println!("ytsearch:{}", url);
+            YoutubeDl::new_ytdl_like("yt-dlp", http_client, format!("ytsearch:{}", args.clone().message()))
         } else {
-            YoutubeDl::new(http_client, url)
+            YoutubeDl::new_ytdl_like("yt-dlp", http_client, url)
         };
         
         let _ = handler.enqueue_input(src.clone().into()).await;
         
-        // let _ = handler.play_input(src.clone().into());
+        let metadata = src.aux_metadata().await.unwrap();
 
-        check_msg(msg.channel_id.say(&ctx.http, "Playing song").await);
+        check_msg(msg.channel_id.say(&ctx.http, format!("Playing song: {}", metadata.title.unwrap())).await);
     } else {
         check_msg(
             msg.channel_id

src/commands/music/queue.rs

@@ -3,10 +3,46 @@ use serenity::framework::standard::CommandResult;
 use serenity::model::prelude::*;
 use serenity::prelude::*;
 
-// use crate::commands::misc::check_msg;
+use crate::commands::misc::check_msg;
 
 #[command]
 #[only_in(guilds)]
-async fn queue(_ctx: &Context, _msg: &Message) -> CommandResult {
+async fn queue(ctx: &Context, msg: &Message) -> CommandResult {
+    let guild_id = msg.guild_id.unwrap();
+    
+    let manager = songbird::get(ctx)
+        .await
+        .expect("Client placed at init")
+        .clone();
+
+    if let Some(handler_lock) = manager.get(guild_id) {
+        let handler = handler_lock.lock().await;
+        let queue = handler.queue();
+        let mut queue_res = String::from("Queue: \n");
+
+        for (i, song) in queue.current_queue().iter().enumerate() {
+            queue_res.push_str(&format!(
+                "{}. {} - {}\n",
+                i + 1,
+                song.uuid(),
+                "Artist"
+                // song.metadata().artist.clone().unwrap_or_else(|| String::from("Unknown"))
+        ));
+        }
+        
+        check_msg(
+            msg.channel_id
+                .say(&ctx.http, queue_res)
+                .await,
+        );
+
+    } else {
+        check_msg(
+            msg.channel_id
+                .say(&ctx.http, "Not in a voice channel!")
+                .await,
+        );
+    }
+
     Ok(())
 }

src/commands/music/resume.rs

@@ -7,7 +7,7 @@ use crate::commands::misc::check_msg;
 
 #[command]
 #[only_in(guilds)]
-async fn unmute(ctx: &Context, msg: &Message) -> CommandResult {
+async fn resume(ctx: &Context, msg: &Message) -> CommandResult {
     let guild_id = msg.guild_id.unwrap();
 
     let manager = songbird::get(ctx)
@@ -16,20 +16,22 @@ async fn unmute(ctx: &Context, msg: &Message) -> CommandResult {
         .clone();
 
     if let Some(handler_lock) = manager.get(guild_id) {
-        let mut handler = handler_lock.lock().await;
-        if let Err(e) = handler.mute(false).await {
+        let handler = handler_lock.lock().await;
+        let queue = handler.queue();
+        let _ = queue.resume();
+
         check_msg(
             msg.channel_id
-                    .say(&ctx.http, format!("Failed: {:?}", e))
+                .say(
+                    &ctx.http,
+                    format!("Song resumed."),
+                )
                 .await,
         );
-        }
-
-        check_msg(msg.channel_id.say(&ctx.http, "Unmuted").await);
     } else {
         check_msg(
             msg.channel_id
-                .say(&ctx.http, "Not in a voice channel to unmute in")
+                .say(&ctx.http, "Not in a voice channel to play in")
                 .await,
         );
     }

src/commands/music/skip.rs

@@ -3,10 +3,38 @@ use serenity::framework::standard::CommandResult;
 use serenity::model::prelude::*;
 use serenity::prelude::*;
 
-// use crate::commands::misc::check_msg;
+use crate::commands::misc::check_msg;
 
 #[command]
 #[only_in(guilds)]
-async fn skip(_ctx: &Context, _msg: &Message) -> CommandResult {
+async fn skip(ctx: &Context, msg: &Message) -> CommandResult {
+    let guild_id = msg.guild_id.unwrap();
+
+    let manager = songbird::get(ctx)
+        .await
+        .expect("Client placed at init")
+        .clone();
+
+    if let Some(handler_lock) = manager.get(guild_id) {
+        let handler = handler_lock.lock().await;
+        let queue = handler.queue();
+        let _ = queue.skip();
+
+        check_msg(
+            msg.channel_id
+                .say(
+                    &ctx.http,
+                    format!("Song skipped: {} in queue.", queue.len()),
+                )
+                .await,
+        );
+    } else {
+        check_msg(
+            msg.channel_id
+                .say(&ctx.http, "Not in a voice channel to play in")
+                .await,
+        );
+    }
+
     Ok(())
 }

src/commands/music/stop.rs

@@ -16,22 +16,15 @@ async fn stop(ctx: &Context, msg: &Message) -> CommandResult {
         .clone();
 
     if let Some(handler_lock) = manager.get(guild_id) {
-        let mut handler = handler_lock.lock().await;
-        let _queue = handler.queue();
-
-        if let Err(e) = handler.deafen(false).await {
-            check_msg(
-                msg.channel_id
-                    .say(&ctx.http, format!("Failed: {:?}", e))
-                    .await,
-            );
-        }
+        let handler = handler_lock.lock().await;
+        let queue = handler.queue();
+        queue.stop();
 
         check_msg(msg.channel_id.say(&ctx.http, "Playback stopped!").await);
     } else {
         check_msg(
             msg.channel_id
-                .say(&ctx.http, "Not in a voice channel to undeafen in")
+                .say(&ctx.http, "Not in a voice channel!")
                 .await,
         );
     }

src/main.rs

@@ -30,8 +30,9 @@ use crate::commands::music::play::*;
 use crate::commands::music::queue::*;
 use crate::commands::music::skip::*;
 use crate::commands::music::stop::*;
-use crate::commands::music::undeafen::*;
-use crate::commands::music::unmute::*;
+use crate::commands::music::loopcurrent::*;
+use crate::commands::music::pause::*;
+use crate::commands::music::resume::*;
 
 // tools
 use crate::commands::tools::ping::*;
@@ -61,7 +62,7 @@ async fn before(_: &Context, msg: &Message, command_name: &str) -> bool {
 
 #[group]
 #[commands(
-    join, deafen, leave, mute, play, unmute, undeafen, ping, kashi, queue, stop, skip
+    join, deafen, leave, mute, play, ping, kashi, queue, stop, skip, loopcurrent, pause, resume
 )]
 struct General;