pollo/lib/spotify.go

package lib

import (
	"context"
	"os"

	"atri.dad/lib/pubsub"
	"github.com/zmb3/spotify"
	"golang.org/x/oauth2"
)

var spotifyOAuth2Endpoint = oauth2.Endpoint{
	TokenURL: "https://accounts.spotify.com/api/token",
	AuthURL:  "https://accounts.spotify.com/authorize",
}

func GetCurrentlyPlayingTrack(clientID string, clientSecret string, refreshToken string) (*spotify.CurrentlyPlaying, error) {
	// OAuth2 config
	config := &oauth2.Config{
		ClientID:     clientID,
		ClientSecret: clientSecret,
		Scopes:       []string{spotify.ScopeUserReadCurrentlyPlaying},
		Endpoint:     spotifyOAuth2Endpoint,
	}

	// Token source
	tokenSource := config.TokenSource(context.Background(), &oauth2.Token{RefreshToken: refreshToken})

	// Get new token
	newToken, err := tokenSource.Token()
	if err != nil {
		return nil, err
	}

	// Create new client
	client := spotify.Authenticator{}.NewClient(newToken)

	// Get currently playing track
	playing, err := client.PlayerCurrentlyPlaying()
	if err != nil {
		return nil, err
	}

	return playing, nil
}

func CurrentlyPlayingTrackSSE(ctx context.Context, pubSub pubsub.PubSub) error {
	clientID := os.Getenv("SPOTIFY_CLIENT_ID")
	clientSecret := os.Getenv("SPOTIFY_CLIENT_SECRET")
	refreshToken := os.Getenv("SPOTIFY_REFRESH_TOKEN")

	playing, err := GetCurrentlyPlayingTrack(clientID, clientSecret, refreshToken)
	if err != nil {
		return err
	}

	if playing.Item != nil && playing.Playing {
		SendSSE(ctx, pubSub, "spotify", `<div class="indicator-item badge badge-success"><a _="on mouseover put '🔥 Listening to `+playing.Item.Name+" by "+playing.Item.Artists[0].Name+` 🔥' into my.textContent on mouseout put '🔥' into my.textContent" href="`+playing.Item.ExternalURLs["spotify"]+`" rel="noreferrer" target="_blank">🔥</a></div>`)
	} else {
		SendSSE(ctx, pubSub, "spotify", "")
	}

	return nil
}