Show extra information for youtube livestreams (#678)

Author: M4tthewDE

CHANGELOG.md

@@ -4,6 +4,7 @@
 
 - Breaking: Go version 1.22.1 is now the minimum required version to build this project. (#667, #671)
 - Minor: Add playlist support to YouTube resolver. (#597, #601)
+- Minor: Add YouTube livestream support. (#678)
 - Fix: Do not resolve /results using YouTube channel resolver (#616)
 
 ## 2.0.3

internal/resolvers/youtube/initialize.go

@@ -41,13 +41,23 @@ const (
 <br><b>Videos:</b> {{.VideoCount}}
 <br><b>Published:</b> {{.PublishedAt}}
 </div>
+`
+
+	youtubeStreamTooltip = `<div style="text-align: left;">
+<b>{{.Title}}</b>
+<br><b>Channel:</b> {{.ChannelTitle}}
+<br><b>Uptime:</b> {{.Uptime}}
+<br><b>Viewers:</b> {{.Viewers}}
+<br><b><span style="color: #ff0000;">Live</span></b>&nbsp;•&nbsp;<span style="color: #2ecc71;">{{.LikeCount}} likes</span>
+</div>
 `
 )
 
 var (
 	youtubeVideoTooltipTemplate    = template.Must(template.New("youtubeVideoTooltip").Parse(youtubeVideoTooltip))
 	youtubeChannelTooltipTemplate  = template.Must(template.New("youtubeChannelTooltip").Parse(youtubeChannelTooltip))
 	youtubePlaylistTooltipTemplate = template.Must(template.New("youtubePlaylistTooltip").Parse(youtubePlaylistTooltip))
+	youtubeStreamTooltipTemplate   = template.Must(template.New("youtubeStreamTooltip").Parse(youtubeStreamTooltip))
 )
 
 func NewYouTubeVideoResolvers(ctx context.Context, cfg config.APIConfig, pool db.Pool, youtubeClient *youtubeAPI.Service) (resolver.Resolver, resolver.Resolver) {

internal/resolvers/youtube/video_loader.go

@@ -25,6 +25,14 @@ type youtubeVideoTooltipData struct {
 	AgeRestricted bool
 }
 
+type youtubeStreamTooltipData struct {
+	Title        string
+	ChannelTitle string
+	Uptime       string
+	Viewers      string
+	LikeCount    string
+}
+
 type VideoLoader struct {
 	youtubeClient *youtubeAPI.Service
 }
@@ -35,11 +43,13 @@ func (r *VideoLoader) Load(ctx context.Context, videoID string, req *http.Reques
 		"statistics",
 		"snippet",
 		"contentDetails",
+		"liveStreamingDetails",
 	}
 
 	log.Debugw("[YouTube] Get video",
 		"videoID", videoID,
 	)
+
 	youtubeResponse, err := r.youtubeClient.Videos.List(youtubeVideoParts).Id(videoID).Do()
 	if err != nil {
 		return resolver.InternalServerErrorf("YouTube API error: %s", err)
@@ -58,32 +68,60 @@ func (r *VideoLoader) Load(ctx context.Context, videoID string, req *http.Reques
 
 	video := youtubeResponse.Items[0]
 
-	if video.ContentDetails == nil {
+	if video.Snippet == nil {
 		return resolver.InternalServerErrorf("YouTube video unavailable")
 	}
 
-	// Check if a video is age resricted: https://stackoverflow.com/a/33750307
-	var ageRestricted = false
-	if video.ContentDetails.ContentRating != nil {
-		if video.ContentDetails.ContentRating.YtRating == "ytAgeRestricted" {
-			ageRestricted = true
+	var tooltip bytes.Buffer
+
+	if video.Snippet.LiveBroadcastContent == "live" {
+		if video.LiveStreamingDetails == nil {
+			return resolver.InternalServerErrorf("YouTube livestream unavailable")
 		}
-	}
 
-	data := youtubeVideoTooltipData{
-		Title:         video.Snippet.Title,
-		ChannelTitle:  video.Snippet.ChannelTitle,
-		Duration:      humanize.DurationPT(video.ContentDetails.Duration),
-		PublishDate:   humanize.CreationDateRFC3339(video.Snippet.PublishedAt),
-		Views:         humanize.Number(video.Statistics.ViewCount),
-		LikeCount:     humanize.Number(video.Statistics.LikeCount),
-		CommentCount:  humanize.Number(video.Statistics.CommentCount),
-		AgeRestricted: ageRestricted,
-	}
+		startTime, err := time.Parse(time.RFC3339, video.LiveStreamingDetails.ActualStartTime)
+		if err != nil {
+			return resolver.InternalServerErrorf("YouTube time parse error: %s", err)
+		}
 
-	var tooltip bytes.Buffer
-	if err := youtubeVideoTooltipTemplate.Execute(&tooltip, data); err != nil {
-		return resolver.InternalServerErrorf("YouTube template error: %s", err)
+		data := youtubeStreamTooltipData{
+			Title:        video.Snippet.Title,
+			ChannelTitle: video.Snippet.ChannelTitle,
+			Uptime:       humanize.Duration(time.Since(startTime)),
+			Viewers:      humanize.Number(video.LiveStreamingDetails.ConcurrentViewers),
+			LikeCount:    humanize.Number(video.Statistics.LikeCount),
+		}
+
+		if err := youtubeStreamTooltipTemplate.Execute(&tooltip, data); err != nil {
+			return resolver.InternalServerErrorf("YouTube template error: %s", err)
+		}
+	} else {
+		if video.ContentDetails == nil {
+			return resolver.InternalServerErrorf("YouTube video unavailable")
+		}
+
+		// Check if a video is age resricted: https://stackoverflow.com/a/33750307
+		var ageRestricted = false
+		if video.ContentDetails.ContentRating != nil {
+			if video.ContentDetails.ContentRating.YtRating == "ytAgeRestricted" {
+				ageRestricted = true
+			}
+		}
+
+		data := youtubeVideoTooltipData{
+			Title:         video.Snippet.Title,
+			ChannelTitle:  video.Snippet.ChannelTitle,
+			Duration:      humanize.DurationPT(video.ContentDetails.Duration),
+			PublishDate:   humanize.CreationDateRFC3339(video.Snippet.PublishedAt),
+			Views:         humanize.Number(video.Statistics.ViewCount),
+			LikeCount:     humanize.Number(video.Statistics.LikeCount),
+			CommentCount:  humanize.Number(video.Statistics.CommentCount),
+			AgeRestricted: ageRestricted,
+		}
+
+		if err := youtubeVideoTooltipTemplate.Execute(&tooltip, data); err != nil {
+			return resolver.InternalServerErrorf("YouTube template error: %s", err)
+		}
 	}
 
 	thumbnail := video.Snippet.Thumbnails.Default.Url