From 454031573563a0f691536d2b1893a4d174705e57 Mon Sep 17 00:00:00 2001
From: Midgard <midgard@zeus.ugent.be>
Date: Sun, 8 Aug 2021 16:10:03 +0200
Subject: [PATCH] Use mmcli for statuses, remove unnecessary code, expire at
 song end

---
 mpd_mm.py | 117 ++++++++++++++++++------------------------------------
 1 file changed, 39 insertions(+), 78 deletions(-)

diff --git a/mpd_mm.py b/mpd_mm.py
index 3c6aab6..9ef98c3 100755
--- a/mpd_mm.py
+++ b/mpd_mm.py
@@ -1,23 +1,17 @@
 #!/usr/bin/env python3
 
 import sys
-import os
-import requests
+import subprocess
 import datetime
-import mpd
 import time
 import logging
+import mpd
 
 NAME = "MPD MM now playing status"
 VERSION = "0.0.1"
 
 ### Config ########
 
-# E.g. https://mattermost.zeus.gent/api
-MM_API = os.getenv("MM_API")
-# Get one with mmcli, or by looking in your browser cookie. E.g. gh3cd6ef10abij2kl2nm9op8qr
-MM_API_ACCESS_TOKEN = os.getenv("MM_API_ACCESS_TOKEN")
-
 MPD_HOST = "/var/lib/mpd/socket"
 MPD_PORT = None
 # or to connect with TCP
@@ -27,81 +21,32 @@ MPD_PORT = None
 EMOJI_PAUSED = "pause_button"
 EMOJI_PLAYING = "musical_note"
 
-# Make the status automatically expire after this many minutes in case this script fails to clear
-# its status when it's done, e.g. due to a loss of internet connection.
-# This expiration is handled by the Mattermost server.
-# Set to 0 or None to disable.
-AUTO_EXPIRE_MINUTES = 60
-
 ###################
 
 
-HEADERS = {
-	"Authorization": f"Bearer {MM_API_ACCESS_TOKEN}",
-	"User-Agent": f"{NAME}/{VERSION}"
-}
-
-LOGGER = logging.getLogger("MPD MM now playing")
+LOGGER = logging.getLogger("mpd_mm")
 
 MPD_STATE_PLAY = "play"
 MPD_STATE_PAUSE = "pause"
 
-# Util {{{1
-# ----
-
-class FastLoopAvoider:
-	def __init__(self, logger, threshold_seconds=1, warn_count=10, stop_count=20, sleep=0):
-		self.threshold_seconds = threshold_seconds
-		self.warn_count = warn_count
-		self.stop_count = stop_count
-		self.sleep = sleep
-		self._last_loop_start = 0
-		self._fast_loops = 0
-		self.logger = logging.getLogger(logger)
-
-	def loop(self):
-		now = time.time()
-		if now - self._last_loop_start < self.threshold_seconds:
-			self._fast_loops += 1
-		self._last_loop_start = now
-
-		if self.stop_count and self._fast_loops == self.stop_count:
-			self.logger.error(f"Stopping because loop reached {self._fast_loops} fast runs")
-			return False
-		if self.warn_count and self._fast_loops == self.warn_count:
-			if self.stop_count:
-				self.logger.warn(f"Loop reached {self._fast_loops} fast runs, stopping at {self.stop_count}")
-			else:
-				self.logger.warn(f"Loop reached {self._fast_loops} fast runs")
-
-		if self.sleep:
-			time.sleep(self.sleep)
-
-		return True
 
 # Mattermost stuff {{{1
 # ----------------
 
 def set_status(emoji, text, expires_datetime):
-	expires = expires_datetime.isoformat() if expires_datetime else None
-	LOGGER.info(f"Setting custom status to :{emoji}: {text} expiring {expires or 'never'}")
-
-	r = requests.put(
-		f"{MM_API}/v4/users/me/status/custom",
-		json={
-			"emoji": emoji,
-			"text": text,
-			"expires_at": expires
-		},
-		headers=HEADERS
-	)
-	r.raise_for_status()
+	LOGGER.info(f"Custom status expiring {expires_datetime or 'never'}: :{emoji}: {text}")
+	subprocess.run([
+		"mmcli", "customstatus",
+		"--until", expires_datetime.isoformat() if expires_datetime else "",
+		"--emoji", emoji,
+		"--", text
+	], check=True)
 
 
 def clear_status():
-	LOGGER.info("Clearing custom status")
-	r = requests.delete(f"{MM_API}/v4/users/me/status/custom", headers=HEADERS)
-	r.raise_for_status()
+	LOGGER.info("Clearing status")
+	subprocess.run(["mmcli", "customstatus"], check=True)
+
 
 # MPD stuff {{{1
 # ---------
@@ -112,33 +57,49 @@ def song_string(song_info):
 	return f"{artist} – {title}"
 
 
-def set_status_from_mpd(mpd_client):
+def formatted_status(mpd_client):
 	status = mpd_client.status()
 	state = status.get("state")
 	LOGGER.debug(f"Player state: {state!r}")
 	if state not in (MPD_STATE_PLAY, MPD_STATE_PAUSE):
-		clear_status()
-		return
+		return None
 
 	song = mpd_client.currentsong()
 	song_str = song_string(song)
 	emoji = EMOJI_PAUSED if state == MPD_STATE_PAUSE else EMOJI_PLAYING
 
-	if AUTO_EXPIRE_MINUTES:
+	expire = None
+	elapsed = status.get("elapsed")
+	duration = status.get("duration")
+	if state == MPD_STATE_PLAY and elapsed is not None and duration:
 		now = datetime.datetime.now().astimezone()
-		expire = now + datetime.timedelta(seconds=AUTO_EXPIRE_MINUTES * 60)
-	else:
-		expire = None
+		try:
+			# 1 second extra to allow some time to set the new song without flickering status
+			expire = now + datetime.timedelta(seconds=1 + float(duration) - float(elapsed))
+		except ValueError as e:
+			LOGGER.error("Could not calculate expiry time", exc_info=e)
 
-	set_status(emoji, song_str, expire)
+	return (emoji, song_str, expire)
+
+
+# Driving stuff {{{1
+# -------------
+
+def set_status_from_mpd(mpd_client):
+	status = formatted_status(mpd_client)
+	if status is None:
+		clear_status()
+	else:
+		emoji, song_str, expire = status
+		set_status(emoji, song_str, expire)
 
 
 def loop(mpd_client, on_status_change):
-	fla = FastLoopAvoider("MPD-MM main loop", sleep=1)
-	while fla.loop():
+	while True:
 		mpd_client.idle("player")
 		LOGGER.debug("Got event from MPD")
 		on_status_change(mpd_client)
+		time.sleep(1)
 
 
 def main(mpd_host, mpd_port):