RMusicBot/src/player/commands/list.rs

use super::super::prelude::*;

use serenity::prelude::*;
use serenity::{
    framework::standard::{macros::command, Args, CommandResult},
    model::channel::Message,
};

#[command]
fn list(ctx: &mut Context, msg: &Message, _: Args) -> CommandResult {
    if !channel_contains_author(ctx, msg) {
        println!(
            "user {} is not in the same voice channel as the bot",
            msg.author.name
        );
        return Ok(());
    }

    let mut output = String::new();

    let data = ctx.data.read();
    let media = match data.get::<MediaData>() {
        Ok(media) => media,
        Err(_) => {
            msg.channel_id.say("could not find media data");
            return Ok(());
        }
    };

    let playlist = media.playlist();

    output += &format!(
        "{} {} queued\n",
        playlist.len(),
        if playlist.len() == 1 { "song" } else { "songs" }
    );

    let max_output = 5;

    for (i, song) in playlist.iter().enumerate() {
        if i < max_output {
            output += &format!("\t{}.\t{}\n", i + 1, song.name.clone());
        } else {
            output += &format!("\t... and {} more", playlist.len() - max_output);
            break;
        }
    }

    print_error!(msg.channel_id.say(output));

    Ok(())
}