refactor: move sponsorblock in a different package

download.go

@@ -3,6 +3,7 @@ package main
 import (
 	"errors"
 	"github.com/TheTipo01/YADMB/queue"
+	"github.com/TheTipo01/YADMB/sponsorblock"
 	"github.com/bwmarrin/discordgo"
 	"github.com/bwmarrin/lit"
 	"github.com/goccy/go-json"
@@ -86,7 +87,7 @@ func downloadAndPlay(s *discordgo.Session, guildID, link, user string, i *discor
 		case "youtube":
 			el.ID = ytdl.ID + "-" + ytdl.Extractor
 			// SponsorBlock is supported only on YouTube
-			el.Segments = getSegments(ytdl.ID)
+			el.Segments = sponsorblock.GetSegments(ytdl.ID)
 
 			// If the song is on YouTube, we also add it with its compact url, for faster parsing
 			db.AddToDb(el, false)

sponsorblock.go → sponsorblock/sponsorblock.go

@@ -1,4 +1,4 @@
-package main
+package sponsorblock
 
 import (
 	"crypto/sha256"
@@ -9,8 +9,13 @@ import (
 	"time"
 )
 
-// Returns a map for skipping certain frames of a song
-func getSegments(videoID string) map[int]bool {
+const (
+	// How many DCA frames are needed for a second. It's not perfect, but good enough.
+	frameSeconds = 50.00067787
+)
+
+// GetSegments returns a map for skipping certain frames of a song
+func GetSegments(videoID string) map[int]bool {
 	// Gets segments
 	req, _ := http.NewRequest("GET", "https://sponsor.ajay.app/api/skipSegments/"+hash(videoID)+"?categories=[\"sponsor\",\"music_offtopic\"]", nil) // Sets timeout to one second, as sometime i
 	client := http.Client{Timeout: time.Second}

sponsorblock_test.go → sponsorblock/sponsorblock_test.go

@@ -1,4 +1,4 @@
-package main
+package sponsorblock
 
 import "testing"
 
@@ -10,7 +10,7 @@ func TestHash(t *testing.T) {
 }
 
 func TestGetSegments(t *testing.T) {
-	result := getSegments("kJQP7kiw5Fk")
+	result := GetSegments("kJQP7kiw5Fk")
 	if result == nil || len(result) == 0 {
 		t.Error("Getting segments failed. Expected map, got nil")
 	}

sponsorblock/structure.go

@@ -0,0 +1,10 @@
+package sponsorblock
+
+// SponsorBlock holds data for segments of sponsors in YouTube video
+type SponsorBlock []struct {
+	VideoID  string `json:"videoID"`
+	Segments []struct {
+		Category string    `json:"category"`
+		Segment  []float64 `json:"segment"`
+	} `json:"segments"`
+}

structures.go

@@ -56,15 +56,6 @@ type RequestedFormats []struct {
 	Resolution string `json:"resolution"`
 }
 
-// SponsorBlock holds data for segments of sponsors in YouTube video
-type SponsorBlock []struct {
-	VideoID  string `json:"videoID"`
-	Segments []struct {
-		Category string    `json:"category"`
-		Segment  []float64 `json:"segment"`
-	} `json:"segments"`
-}
-
 // Config holds data parsed from the config.yml
 type Config struct {
 	Token        string `fig:"token" validate:"required"`