1package music
  2
  3import (
  4	"os"
  5
  6	"github.com/birabittoh/disgord/src/mylog"
  7	"github.com/bwmarrin/discordgo"
  8	"github.com/kkdai/youtube/v2"
  9)
 10
 11var logger = mylog.NewLogger(os.Stdin, "music", mylog.DEBUG)
 12
 13type Queue struct {
 14	nowPlaying  *youtube.Video
 15	items       []*youtube.Video
 16	audioStream *Audio
 17	vc          *discordgo.VoiceConnection
 18}
 19
 20// queues stores all guild queues
 21var queues = map[string]*Queue{}
 22
 23// GetOrCreateQueue fetches or creates a new queue for the guild
 24func GetOrCreateQueue(vc *discordgo.VoiceConnection) (q *Queue) {
 25	q, ok := queues[vc.GuildID]
 26	if !ok {
 27		q = &Queue{vc: vc}
 28		queues[vc.GuildID] = q
 29		return
 30	}
 31
 32	if q.vc.Ready == false {
 33		q.vc = vc
 34	}
 35	return
 36}
 37
 38// GetQueue returns either nil or the queue for the requested guild
 39func GetQueue(guildID string) *Queue {
 40	q, ok := queues[guildID]
 41	if ok {
 42		return q
 43	}
 44	return nil
 45}
 46
 47// AddVideo adds a new video to the queue
 48func (q *Queue) AddVideo(video *youtube.Video) {
 49	q.items = append(q.items, video)
 50	if q.nowPlaying == nil {
 51		q.PlayNext()
 52	}
 53}
 54
 55// AddVideos adds a list of videos to the queue
 56func (q *Queue) AddVideos(videos []*youtube.Video) {
 57	q.items = append(q.items, videos...)
 58	if q.nowPlaying == nil {
 59		err := q.PlayNext()
 60		if err != nil {
 61			logger.Error(err)
 62		}
 63	}
 64}
 65
 66// PlayNext starts playing the next video in the queue
 67func (q *Queue) PlayNext() (err error) {
 68	if q.audioStream != nil {
 69		q.audioStream.Stop()
 70	}
 71
 72	if len(q.items) == 0 {
 73		q.nowPlaying = nil
 74		return q.vc.Disconnect()
 75	}
 76
 77	q.nowPlaying = q.items[0]
 78	q.items = q.items[1:]
 79
 80	formats := q.nowPlaying.Formats.Type("opus")
 81	if len(formats) == 0 {
 82		logger.Debug("no formats with audio channels available for video " + q.nowPlaying.ID)
 83		return q.PlayNext()
 84	}
 85
 86	q.audioStream, err = NewAudio(formats[0].URL, q.vc)
 87	if err != nil {
 88		return
 89	}
 90
 91	q.audioStream.Monitor(func() { q.PlayNext() })
 92	return
 93}
 94
 95// Stop stops the player and clears the queue
 96func (q *Queue) Stop() error {
 97	q.Clear()
 98	q.nowPlaying = nil
 99	q.audioStream.Stop()
100	return q.vc.Disconnect()
101}
102
103// Pause pauses the player
104func (q *Queue) Pause() {
105	q.audioStream.Pause()
106}
107
108// Resume resumes the player
109func (q *Queue) Resume() {
110	q.audioStream.Resume()
111}
112
113// Clear clears the video queue
114func (q *Queue) Clear() {
115	q.items = []*youtube.Video{}
116}
117
118// Videos returns all videos in the queue including the now playing one
119func (q *Queue) Videos() []*youtube.Video {
120	if q.nowPlaying != nil {
121		return append([]*youtube.Video{q.nowPlaying}, q.items...)
122	}
123	return q.items
124}
125
126func (q *Queue) VoiceChannelID() string {
127	return q.vc.ChannelID
128}