From ad14ff81a722632e52a841b1cd8df0d1272d36b5 Mon Sep 17 00:00:00 2001
From: tobi-wan-kenobi <tobi-wan-kenobi@users.noreply.github.com>
Date: Sun, 19 Apr 2020 10:30:22 +0200
Subject: [PATCH] [modules] re-enable rotation

---
 modules/contrib/rotation.py | 61 +++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 modules/contrib/rotation.py

diff --git a/modules/contrib/rotation.py b/modules/contrib/rotation.py
new file mode 100644
index 0000000..23552d0
--- /dev/null
+++ b/modules/contrib/rotation.py
@@ -0,0 +1,61 @@
+# pylint: disable=C0111,R0903
+
+"""Shows a widget for each connected screen and allows the user to loop through different orientations.
+
+Requires the following executable:
+    * xrandr
+"""
+
+import bumblebee.util
+import bumblebee.input
+import bumblebee.output
+import bumblebee.engine
+
+possible_orientations = ["normal", "left", "inverted", "right"]
+
+class Module(bumblebee.engine.Module):
+    def __init__(self, engine, config):
+        widgets = []
+        self._engine = engine
+        super(Module, self).__init__(engine, config, widgets)
+        self.update_widgets(widgets)
+
+    def update_widgets(self, widgets):
+        for line in bumblebee.util.execute("xrandr -q").split("\n"):
+            if not " connected" in line:
+                continue
+            display = line.split(" ", 2)[0]
+
+            orientation = "normal"
+            for curr_orient in possible_orientations:
+                if((line.split(" ")).count(curr_orient) > 1):
+                    orientation = curr_orient
+                    break
+
+            widget = self.widget(display)
+            if not widget:
+                widget = bumblebee.output.Widget(full_text=display, name=display)
+                self._engine.input.register_callback(widget, button=bumblebee.input.LEFT_MOUSE, cmd=self._toggle)
+            widget.set("orientation", orientation)
+            widgets.append(widget)
+
+    def update(self, widgets):
+        if len(widgets) <= 0:
+            self.update_widgets(widgets)
+
+    def state(self, widget):
+        return widget.get("orientation", "normal")
+
+    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("xrandr --output {} --rotation {}".format(widget.name, new_orientation))
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4