main.go

package clipboard_yt_dl

import (
	"github.com/beeker1121/goque"
	"log"
	"net/url"
	"time"
)

// create new ClipboardYtDl instance
func NewClipboardYtDl() *ClipboardYtDl {
	return &ClipboardYtDl{queue: openQueue()}
}

// open queue database
func openQueue() *goque.Queue {
	queue, err := goque.OpenQueue("data_dir")
	if err != nil {
		panic(err)
	}

	return queue
}

type ClipboardYtDl struct {
	queue  *goque.Queue
	stopCh chan struct{}
}

// iterate over each item in queue if download is enabled
func (c *ClipboardYtDl) StartQueue(stopCh <-chan bool, callback func(video *Video, length uint64)) {
	for {
		select {
		case <-stopCh:
			return
		default:
			time.Sleep(time.Second)

			if c.queue.Length() > 0 {
				item, err := c.queue.Dequeue()
				if err != nil {
					panic(err)
				}

				copiedUrl, err := url.Parse(item.ToString())
				if err != nil {
					panic(err)
				}

				video := c.downloadVideo(copiedUrl)

				if video != nil {
					callback(video, c.queue.Length())
				}
			}
		}
	}
}

// stop processing queue
func (c *ClipboardYtDl) StopQueue(stopCh chan bool) {
	stopCh <- true
}

// delete the queue database and open new queue
func (c *ClipboardYtDl) ClearQueue() {
	c.queue.Drop()
	c.queue = openQueue()
}

// add video object to queue
func (c *ClipboardYtDl) EnqueueVideo(url *url.URL) (*goque.Item, error) {
	return c.queue.EnqueueString(url.String())
}

// retrieve amount of queued videos
func (c *ClipboardYtDl) VideoLength() uint64 {
	return c.queue.Length()
}

// this method will download url
func (c *ClipboardYtDl) downloadVideo(url *url.URL) *Video {
	log.Printf("INFO: %s downloading ... \n", url.String())

	dl := YouTubeDl{}
	video, err := dl.Download(url)

	if err != nil {
		switch err {
		case UnsupportedError, UnknownServiceError, SSLCertificateVerifyFailedError:
			log.Printf("ERROR: %s %s \n", url, err.Error())
			return nil
		default:
			panic(err)
		}
	}

	log.Printf("INFO: %s finished download to \"%s\" \n", url.String(), video.Filename)

	return video
}

// close queue database
func (c *ClipboardYtDl) CloseQueue() {
	c.queue.Close()
}