feat: add time left and apply rich presence on startup

Author: sammy-ette

main.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"os"
 	"strings"
+	"time"
 
 	"github.com/godbus/dbus/v5"
 	"github.com/TorchedSammy/go-mpris"
@@ -20,6 +21,7 @@ func main() {
 	if err != nil {
 		panic(err)
 	}
+
 	if len(names) == 0 {
 		fmt.Println("No MPRIS player found.")
 		os.Exit(1)
@@ -32,38 +34,66 @@ func main() {
 	var rules = []string{
 		"type='signal',member='PropertiesChanged',path='/org/mpris/MediaPlayer2',interface='org.freedesktop.DBus.Properties'",
 	}
+
 	var flag uint = 0
 
+	data := dbus.Variant{}
+	elapsedFromDbus := dbus.Variant{}
+	conn.Object(name, "/org/mpris/MediaPlayer2").Call("org.freedesktop.DBus.Properties.Get", 0, "org.mpris.MediaPlayer2.Player", "Metadata").Store(&data)
+	conn.Object(name, "/org/mpris/MediaPlayer2").Call("org.freedesktop.DBus.Properties.Get", 0, "org.mpris.MediaPlayer2.Player", "Position").Store(&elapsedFromDbus)
+	initialMetadata := data.Value().(map[string]dbus.Variant)
+	elapsed := elapsedFromDbus.Value().(int64)
+
 	call := conn.BusObject().Call("org.freedesktop.DBus.Monitoring.BecomeMonitor", 0, rules, flag)
 	if call.Err != nil {
 		fmt.Fprintln(os.Stderr, "Failed to become monitor:", call.Err)
 		os.Exit(1)
 	}
 
 	client.Login("902662551119224852")
+	// the reason why we add a subtracted Duration is because
+	// time.Sub returns a Duration but i need a Time
+	// confusing huh? thanks go
+	setPresence(initialMetadata, time.Now().Add(-time.Duration(elapsed) * time.Microsecond))
 	c := make(chan *dbus.Message, 10)
 	conn.Eavesdrop(c)
 	for msg := range c {
 		if len(msg.Body) <= 1 {
 			continue
 		}
-		bodyMap := msg.Body[1].(map[string]dbus.Variant)
-		if bodyMap["Metadata"].Value() == nil {
+		metadata := getMetadata(msg.Body[1])
+		if metadata == nil {
 			continue
 		}
-		metadata := bodyMap["Metadata"].Value().(map[string]dbus.Variant)
-		artists := strings.Join(metadata["xesam:artist"].Value().([]string), ", ")
-		client.SetActivity(client.Activity{
-			Details:    metadata["xesam:title"].Value().(string),
-			State:      "on " + metadata["xesam:album"].Value().(string) + " by " + artists,
-			LargeImage: "music",
-			LargeText:  "cmus",
-			SmallImage: "playing",
-			SmallText:  "Playing",
-			/*Timestamps: &client.Timestamps{
-				Start: &start,
-			},*/
-		})
+		setPresence(*metadata, time.Now())
 	}
+}
+
+func getMetadata(msgbody interface{}) *map[string]dbus.Variant {
+	bodyMap := msgbody.(map[string]dbus.Variant)
+	if bodyMap["Metadata"].Value() == nil {
+		return nil
+	}
+	metadataMap := bodyMap["Metadata"].Value().(map[string]dbus.Variant)
+
+	return &metadataMap
+}
+
+func setPresence(metadata map[string]dbus.Variant, songstamp time.Time) {
+	songLength := metadata["mpris:length"].Value().(int64)
+	stampTime := songstamp.Add(time.Duration(songLength) * time.Microsecond)
 
+	artists := strings.Join(metadata["xesam:artist"].Value().([]string), ", ")
+	client.SetActivity(client.Activity{
+		Details:    metadata["xesam:title"].Value().(string),
+		State:      "on " + metadata["xesam:album"].Value().(string) + " by " + artists,
+		LargeImage: "music",
+		LargeText:  "cmus",
+		SmallImage: "playing",
+		SmallText:  "Playing",
+		Timestamps: &client.Timestamps{
+			Start: &songstamp,
+			End: &stampTime,
+		},
+	})
 }