fix(device): Fix ad skip being blocked by API lookups

Ref #27
gabe565 · Oct 3, 2023 · 47cd3d4 · 47cd3d4
1 parent a01ac79
commit 47cd3d4
Showing 1 changed file with 22 additions and 12 deletions.
diff --git a/internal/device/watch.go b/internal/device/watch.go
@@ -181,10 +181,14 @@ func (d *Device) tick() error {
 		}
 
 		if castMedia.Media.ContentId != d.prevVideoId {
-			d.logger.Info("Detected video stream.", "video_id", castMedia.Media.ContentId)
+			if castMedia.Media.ContentId != "" {
+				d.logger.Info("Detected video stream.", "video_id", castMedia.Media.ContentId)
+			}
 			d.prevVideoId = castMedia.Media.ContentId
 			d.unmuteSegment()
-			d.querySegments(castMedia)
+			d.segments = nil
+			go d.querySegments(castMedia)
+			break
 		}
 
 		for i, segment := range d.segments {
@@ -303,17 +307,19 @@ func (d *Device) queryVideoId(castMedia *cast.Media) {
 			d.logger.Info("Video ID not found. Searching for video on YouTube...")
 			d.prevArtist = currArtist
 			d.prevTitle = currTitle
-			if err := util.Retry(d.ctx, 3, time.Second, func(try uint) (err error) {
-				castMedia.Media.ContentId, err = youtube.QueryVideoId(d.ctx, currArtist, currTitle)
-				if err != nil {
-					d.logger.Error("YouTube search failed.", "error", err.Error())
+			go func() {
+				if err := util.Retry(d.ctx, 3, time.Second, func(try uint) (err error) {
+					castMedia.Media.ContentId, err = youtube.QueryVideoId(d.ctx, currArtist, currTitle)
+					if err != nil {
+						d.logger.Error("YouTube search failed.", "error", err.Error())
+					}
+					return err
+				}); err != nil {
+					d.logger.Debug("Halting YouTube search retries.")
+					return
 				}
-				return err
-			}); err != nil {
-				d.logger.Debug("Halting YouTube search retries.")
-				return
-			}
-			d.logger.Debug("YouTube search found video ID", "video_id", castMedia.Media.ContentId)
+				d.logger.Debug("YouTube search found video ID", "video_id", castMedia.Media.ContentId)
+			}()
 		}
 	}
 }
@@ -378,6 +384,10 @@ func (d *Device) unmuteSegment() {
 }
 
 func (d *Device) querySegments(castMedia *cast.Media) {
+	if castMedia.Media.ContentId == "" {
+		return
+	}
+
 	if err := util.Retry(d.ctx, 10, 500*time.Millisecond, func(try uint) (err error) {
 		d.segments, err = sponsorblock.QuerySegments(d.ctx, castMedia.Media.ContentId)
 		return err