Add CLI flags for Steam avoidance and instant transitions

If called manually, ignore Steam by default. The systemd service checks for Steam as that is more likely to run at an annoying time.

btop/btop.conf

@@ -1,4 +1,4 @@
-#? Config file for btop v. 1.4.1
+#? Config file for btop v. 1.4.3
 
 #* Name of a btop++/bpytop/bashtop formatted ".theme" file, "Default" and "TTY" for builtin themes.
 #* Themes should be placed in "../share/btop/themes" relative to binary or "$HOME/.config/btop/themes"

eww/eww.yuck

@@ -122,7 +122,7 @@
 (defwidget wallpaper []
   (clicker :text "󰸉" :command "~/scripts/swww_change.py"))
 (defwidget sunset []
-  (clicker :text "" :command "systemctl --user start sunset.service"))
+  (clicker :text "" :command "~/scripts/sunset.py"))
 
 
 (defpoll brightness :interval 60 "~/.config/eww/scripts/backlight get")
@@ -169,7 +169,7 @@
       (gametime)(space)
       ;(colour_selector)(space)
       (sunset)(space)
-      (wallpaper)(space)
+      (wallpaper)(sep)
       (mountpoint :label "/" :mount "/")(space)
       (mountpoint :label "󰋊" :mount "/home/mbradley/hdd")(sep)
       (ram)(space)
@@ -201,7 +201,8 @@
 
 (defwindow left
   :monitor 1
-  :geometry (geometry :x "0px"
+  :geometry (geometry
+    :x "0px"
     :y "4px"
     :width "1912px"
     :height "24px"

hypr/hypridle.conf

@@ -5,6 +5,7 @@ general {
 	inhibit_sleep = 0
 }
 
+{% if host.name != "chonk" %}
 listener {
 	timeout = 120
 	on-timeout = brightnessctl -sd amdgpu_bl1 set 66-
@@ -32,3 +33,4 @@ listener {
 	timeout = 330
 	on-timeout = systemctl suspend
 }
+{% endif %}

scripts/sunset.py

@@ -1,8 +1,11 @@
 #!/bin/env python
 """Sets the monitor temperature based on the current time using Hyprsunset"""
+from argparse import ArgumentParser, Namespace
 from datetime import datetime
 from os import environ
+from pathlib import Path
 from socket import AF_UNIX, SOCK_STREAM, socket
+from typing import cast
 
 
 type Range = tuple[float, float]
@@ -24,32 +27,83 @@ def day_elapsed(hours: int = 0, minutes: int = 0, seconds: int = 0) -> float:
 	return ((((hours * 60) + minutes) * 60) + seconds) / 86400
 
 
-def main(sunrise: Range, sunset: Range, temp_range: Range) -> None:
+def now_elapsed() -> float:
-	"""Adjusts the monitor temperature based on the current time"""
+	"""How far through the day it is"""
 	now = datetime.now()
-	elapsed = day_elapsed(now.hour, now.minute, now.second)
+	return day_elapsed(now.hour, now.minute, now.second)
+
+def calculate_temperature(sunrise: Range, sunset: Range, temperature_range: Range) -> float:
+	"""What temperature the monitor should be"""
+	elapsed = now_elapsed()
 
 	if elapsed <= sunrise[0]:
-		temperature = temp_range[0]
+		return temperature_range[0]
 	elif elapsed < sunrise[1]:
-		temperature = smoothstep(lerped_amount(elapsed, sunrise), temp_range)
+		return smoothstep(lerped_amount(elapsed, sunrise), temperature_range)
 	elif elapsed <= sunset[0]:
-		temperature = temp_range[1]
+		return temperature_range[1]
 	elif elapsed < sunset[1]:
-		temperature = smoothstep(lerped_amount(elapsed, sunset), (temp_range[1], temp_range[0]))
+		return smoothstep(lerped_amount(elapsed, sunset), (temperature_range[1], temperature_range[0]))
 	else:
-		temperature = temp_range[0]
+		return temperature_range[0]
 
+
+def setup_socket() -> socket:
+	"""Connects to the Hyprsunset socket"""
 	hyprsunset = socket(AF_UNIX, SOCK_STREAM)
 	# In theory I should use $XDG_RUNTIME_DIR, but for me it's always `/run/user/1000/`
-	hyprsunset.connect(f"/run/user/1000/hypr/{environ["HYPRLAND_INSTANCE_SIGNATURE"]}/.hyprsunset.sock")
+	hyprsunset.connect(f"{environ["XDG_RUNTIME_DIR"]}/hypr/{environ["HYPRLAND_INSTANCE_SIGNATURE"]}/.hyprsunset.sock")
-	# In theory the message might not send in one go, but in practice it does do I won't bother handling it
+	return hyprsunset
-	_ = hyprsunset.send(f"temperature {temperature:.1f}".encode())
 
 
+def get_temperature(hyprsunset: socket) -> int:
+	"""Retrieves the current screen temperature"""
+	# In theory the message might not send in one go, but in practice it does so I won't bother handling the error
+	_ = hyprsunset.send(b"temperature")
+	# 4 bytes should be enough but why not 8 for comfort
+	# Just raise an error if it's not a number, nothing special to do here
+	return int(hyprsunset.recv(8))
+
+
+def set_temperature(hyprsunset: socket, temperature: float, /, instant: bool = False) -> None:
+	# In theory the message might not send in one go, but in practice it does so I won't bother handling the error
+	_ = hyprsunset.send(f"temperature {temperature:.1f}".encode())
+	if instant:
+		# Setting the temperature twice in quick succession sometimes skips the transition period
+		set_temperature(hyprsunset, temperature, instant=False)
+
+
+def on_steam() -> bool:
+	"""Whether the Steam PID is in use"""
+	pid_file = Path("~/.steampid").expanduser().resolve(strict=True)
+	pid = pid_file.read_text().strip()
+	pid_dir = Path("/proc") / pid
+	return pid_dir.is_dir()
+
+
+def main(sunrise: Range, sunset: Range, temperature_range: Range, /, check_steam: bool = False, instant: bool = False) -> None:
+	"""Adjusts the monitor temperature based on the current time"""
+	hyprsunset = setup_socket()
+	temperature = int(calculate_temperature(sunrise, sunset, temperature_range))
+	if temperature != get_temperature(hyprsunset):
+		if check_steam and on_steam():
+			return
+		set_temperature(hyprsunset, temperature, instant=instant)
+
+
+class Arguments(Namespace):
+	check_steam: bool = False
+	instant: bool = False
+
 if __name__ == "__main__":
+	parser = ArgumentParser(description="Adjusts the temperature of your screen based on the time of day")
+	_ = parser.add_argument("-s", "--check-steam", action="store_true", help="Don't adjust temperature if Steam is active")
+	_ = parser.add_argument("-i", "--instant", action="store_true", help="Try to instantly change temperature")
+	args = cast(Arguments, parser.parse_args())
 	main(
 		(day_elapsed(5), day_elapsed(7)),
 		(day_elapsed(21), day_elapsed(23)),
-		(2500.0, 6000.0)
+		(2500.0, 6000.0),
+		check_steam=args.check_steam,
+		instant=args.instant
 	)

systemd/sunset.service

@@ -3,4 +3,4 @@ Description=Sets the monitor temperature based on the time of day
 
 [Service]
 Type=oneshot
-ExecStart=/bin/env python -O /home/mbradley/scripts/sunset.py
+ExecStart=/bin/env python -O /home/mbradley/scripts/sunset.py --check-steam

yazi/yazi.toml

@@ -2,7 +2,7 @@
 # If you encounter any issues, please make an issue at https://github.com/yazi-rs/schemas.
 "$schema" = "https://yazi-rs.github.io/schemas/yazi.json"
 
-[manager]
+[mgr]
 ratio		= [ 1, 3, 4 ]
 show_hidden	= true