From b45ff330c9d2044dda0a43503fab5d45973bd983 Mon Sep 17 00:00:00 2001
From: Ludwig Behm <ludwig@behm.me>
Date: Wed, 8 Nov 2023 00:28:29 +0100
Subject: [PATCH] wlrotation: init

---
 .../modules/contrib/wlrotation.py             | 69 +++++++++++++++++++
 1 file changed, 69 insertions(+)
 create mode 100644 bumblebee_status/modules/contrib/wlrotation.py

diff --git a/bumblebee_status/modules/contrib/wlrotation.py b/bumblebee_status/modules/contrib/wlrotation.py
new file mode 100644
index 0000000..54b43a9
--- /dev/null
+++ b/bumblebee_status/modules/contrib/wlrotation.py
@@ -0,0 +1,69 @@
+# pylint: disable=C0111,R0903
+# -*- coding: utf-8 -*-
+
+"""Shows a widget for each connected screen and allows the user to loop through different orientations.
+
+Requires the following executable:
+    * swaymsg
+"""
+
+import bumblebee.util
+import bumblebee.input
+import bumblebee.output
+import bumblebee.engine
+import json
+from os import environ, path
+
+possible_orientations = ["normal", "90", "180", "270"]
+
+class Module(bumblebee.engine.Module):
+    def __init__(self, engine, config):
+        self.service = "sway-auto-rotate@%s" % path.basename(environ['SWAYSOCK']).replace('-', '\\\\x2d')
+        widgets = []
+        self._engine = engine
+        super(Module, self).__init__(engine, config, widgets)
+        self.update_widgets(widgets)
+
+    def update_widgets(self, widgets):
+        for display in json.loads(bumblebee.util.execute("/usr/bin/swaymsg -t get_outputs -r")):
+            name = display['name']
+            widget = self.widget(name)
+            if not widget:
+                widget = bumblebee.output.Widget(name=name, full_text="若 "+name)
+                widget.set("auto", bumblebee.util.execute("systemctl --user show %s -p ActiveState" % self.service).split("\n")[0].split("=")[1])
+                self._engine.input.register_callback(widget, button=bumblebee.input.LEFT_MOUSE, cmd=self._toggle)
+                self._engine.input.register_callback(widget, button=bumblebee.input.RIGHT_MOUSE, cmd=self._toggle_auto)
+            widget.set("orientation", display['transform'])
+            widgets.append(widget)
+
+    def update(self, widgets):
+        if len(widgets) <= 0:
+            self.update_widgets(widgets)
+
+    def state(self, widget):
+        curr = widget.get("auto", "inactive")
+        if curr == "inactive":
+            return ["warning"]
+        return [curr]
+
+    def _toggle_auto(self, event):
+        widget = self.widget_by_id(event["instance"])
+        if widget.get("auto") == "inactive":
+            bumblebee.util.execute("systemctl --user start %s" % self.service)
+        else:
+            bumblebee.util.execute("systemctl --user stop %s" % self.service)
+        widget.set("auto", bumblebee.util.execute("systemctl --user show %s -p ActiveState" % self.service).split("\n")[0].split("=")[1])
+
+    def _toggle(self, event):
+        widget = self.widget_by_id(event["instance"])
+
+        # compute new orientation based on current orientation
+        idx = possible_orientations.index(widget.get("orientation"))
+        idx = (idx + 1) % len(possible_orientations)
+        new_orientation = possible_orientations[idx]
+
+        widget.set("orientation", new_orientation)
+
+        bumblebee.util.execute("swaymsg 'output {} transform {}'".format(widget.name, new_orientation))
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4