# 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