tobi-wan-kenobi/bumblebee-status

# pylint: disable=C0111,R0903

"""
Displays the GitLab todo count:

    * https://docs.gitlab.com/ee/user/todos.html
    * https://docs.gitlab.com/ee/api/todos.html

Uses `xdg-open` or `x-www-browser` to open web-pages.

Requires the following library:
    * requests

Errors:
    if the GitLab todo query failed, the shown value is `n/a`

Parameters:
    * gitlab.token: GitLab personal access token, the token needs to have the "read_api" scope.
    * gitlab.host: Host of the GitLab instance, default is "gitlab.com".
    * gitlab.actions: Comma separated actions to be parsed (e.g.: gitlab.actions=assigned,approval_required)
"""

import shutil

import requests

import core.decorators
import core.input
import core.module
import core.widget
import util


class Module(core.module.Module):
    @core.decorators.every(minutes=5)
    def __init__(self, config, theme):
        super().__init__(config, theme, core.widget.Widget(self.gitlab))

        self.background = True
        self.__label = ""
        self.__host = self.parameter("host", "gitlab.com")

        self.__actions = []
        actions = self.parameter("actions", "")
        if actions:
            self.__actions = util.format.aslist(actions)

        self.__requests = requests.Session()
        self.__requests.headers.update({"PRIVATE-TOKEN": self.parameter("token", "")})

        cmd = "xdg-open"
        if not shutil.which(cmd):
            cmd = "x-www-browser"

        core.input.register(
            self,
            button=core.input.LEFT_MOUSE,
            cmd="{cmd} https:/{host}//dashboard/todos".format(
                cmd=cmd, host=self.__host
            ),
        )

    def gitlab(self, _):
        return self.__label

    def update(self):
        try:
            url = "https://{host}/api/v4/todos".format(host=self.__host)
            response = self.__requests.get(url)
            todos = response.json()
            if self.__actions:
                todos = [t for t in todos if t["action_name"] in self.__actions]
            self.__label = str(len(todos))
        except Exception as e:
            self.__label = "n/a"

    def state(self, widget):
        state = []

        try:
            if int(self.__label) > 0:
                state.append("warning")
        except ValueError:
            pass
        return state

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4