main.go

package main

import (
	"fmt"
	"net/url"
	"strings"
	"time"

	_ "embed"

	"github.com/Code-Hex/go-generics-cache/policy/fifo"
	"github.com/altfoxie/drpc"
	"github.com/getlantern/systray"
)

const AppID = "991335093878673448"

//go:embed assets/tray.png
var icon []byte

var firstSecond = time.Unix(1, 0)

func main() {
	systray.Run(onReady, nil)
}

func onReady() {
	client, err := drpc.New(AppID)
	if err != nil {
		panic(err)
	}

	systray.SetIcon(icon)

	state := systray.AddMenuItem("No state", "")
	state.Disable()

	systray.AddSeparator()
	restart := systray.AddMenuItem("Restart Discord RPC", "")
	quit := systray.AddMenuItem("Quit", "")

	go func() {
		cache := fifo.NewCache[string, string](fifo.WithCapacity(100))
		ticker := time.NewTicker(5 * time.Second)
		// hacky trick to force first tick
		for ; true; <-ticker.C {
			result, err := executeScript()
			if err != nil {
				state.SetTitle("Script error")
				client.Close()
				continue
			}

			if result.state == "" {
				state.SetTitle("Apple Music is not running")
				client.Close()
				continue
			}

			if result.name == "" && result.artist == "" {
				state.SetTitle("No song")
				client.Close()
				continue
			}

			if err = client.Connect(); err != nil {
				state.SetTitle("Discord RPC error")
				continue
			}

			song := result.artist + " – " + result.name
			switch {
			case result.artist == "":
				song = result.name
			case result.name == "":
				song = result.artist
			}

			state.SetTitle(fmt.Sprintf("%s (%d:%02d / %d:%02d)",
				song, int(result.position/60), int(result.position)%60,
				int(result.duration/60), int(result.duration)%60))

			activity := drpc.Activity{
				Details: result.name,
				State:   result.artist,
				Timestamps: &drpc.Timestamps{
					Start: time.Now().Add(-time.Duration(result.position) * time.Second),
					End:   firstSecond,
				},
				Assets: &drpc.Assets{
					LargeImage: "music",
					SmallImage: "pause",
				},
				Buttons: []drpc.Button{
					{
						Label: "Search on YouTube",
						URL: "https://www.youtube.com/results?" + url.Values{
							"search_query": {song},
						}.Encode(),
					},
				},
			}
			if result.state == StatePlaying {
				activity.Assets.SmallImage = "play"
				activity.Timestamps = &drpc.Timestamps{
					Start: time.Now().Add(-time.Duration(result.position) * time.Second),
					End: time.Now().
						Add(time.Duration((result.duration - result.position) * float64(time.Second))),
				}
			}

			artwork, _ := cache.Get(song)
			query := strings.TrimSpace(result.artist + " " + result.name)
			if artwork == "" {
				artwork = artworkSearchITunes(query)
			}
			if artwork == "" {
				artwork = artworkSearchMusixmatch(query)
			}
			if artwork != "" {
				cache.Set(song, artwork)
				activity.Assets.LargeImage = artwork
			}

			client.SetActivity(activity)
		}
	}()

	for {
		select {
		case <-restart.ClickedCh:
			client.Close()
		case <-quit.ClickedCh:
			systray.Quit()
		}
	}
}