feat: skip multiple tracks at once

Author: BomBardyGamer

commands/music.go

@@ -240,6 +240,13 @@ var music = discord.SlashCommandCreate{
 		discord.ApplicationCommandOptionSubCommand{
 			Name:        "skip",
 			Description: "Skips the current track",
+			Options: []discord.ApplicationCommandOption{
+				discord.ApplicationCommandOptionInt{
+					Name:        "count",
+					Description: "The number of tracks to skip",
+					Required:    false,
+				},
+			},
 		},
 		discord.ApplicationCommandOptionSubCommand{
 			Name:        "pause",

commands/skip.go

@@ -2,31 +2,38 @@ package commands
 
 import (
 	"context"
+	"fmt"
 	"time"
 
 	"github.com/disgoorg/disgo/discord"
 	"github.com/disgoorg/disgo/handler"
 	"github.com/disgoorg/disgolink/v3/lavalink"
 )
 
-func (c *Commands) Skip(_ discord.SlashCommandInteractionData, e *handler.CommandEvent) error {
+func (c *Commands) Skip(data discord.SlashCommandInteractionData, e *handler.CommandEvent) error {
 	ctx, cancel := context.WithTimeout(e.Ctx, 10*time.Second)
 	defer cancel()
+
+	var trackCount = data.Int("count")
+	if trackCount == 0 {
+		trackCount = 1
+	}
+
 	player := c.Lavalink.ExistingPlayer(*e.GuildID())
-	track, ok := c.MusicQueue.Next(*e.GuildID())
+	track, ok := c.MusicQueue.NextCount(*e.GuildID(), trackCount)
 	if !ok {
 		return e.CreateMessage(discord.MessageCreate{
-			Content: "No more tracks in queue",
+			Content: "Not enough tracks to skip",
 			Flags:   discord.MessageFlagEphemeral,
 		})
 	}
 	if err := player.Update(ctx, lavalink.WithTrack(track)); err != nil {
 		return e.CreateMessage(discord.MessageCreate{
-			Content: "Failed to play skip track",
+			Content: "Failed to skip track(s)",
 		})
 	}
 
 	return e.CreateMessage(discord.MessageCreate{
-		Content: "⏭ Skipped track",
+		Content: fmt.Sprintf("⏭ Skipped %d track(s)", trackCount),
 	})
 }

lavalinkbot/queue.go

@@ -132,22 +132,27 @@ func (q *PlayerManager) SetRepeatMode(guildID snowflake.ID, mode RepeatMode) {
 }
 
 func (q *PlayerManager) Next(guildID snowflake.ID) (lavalink.Track, bool) {
+	return q.NextCount(guildID, 1)
+}
+
+func (q *PlayerManager) NextCount(guildID snowflake.ID, count int) (lavalink.Track, bool) {
 	q.mu.Lock()
 	defer q.mu.Unlock()
 
 	qq, ok := q.queues[guildID]
 	if !ok {
 		return lavalink.Track{}, false
 	}
-	if len(qq.tracks) == 0 {
+	if len(qq.tracks) < count {
 		return lavalink.Track{}, false
 	}
-	track := qq.tracks[0]
+
+	track := qq.tracks[count-1]
 	if qq.mode != RepeatModeTrack {
 		if qq.mode == RepeatModeQueue {
 			qq.tracks = append(qq.tracks, track)
 		}
-		qq.tracks = qq.tracks[1:]
+		qq.tracks = qq.tracks[count:]
 	}
 	return track, true
 }