bumblebee-status/modules/core/git.py

# pylint: disable=C0111,R0903

"""Print the branch and git status for the
currently focused window.

Requires:
    * xcwd
    * Python module 'pygit2'
"""

import os
import pygit2

import core.module
import core.widget

import util.cli


class Module(core.module.Module):
    def __init__(self, config, theme):
        super().__init__(config, theme, [])

        self.__error = False

    def hidden(self):
        return self.__error

    def update(self):
        state = {}
        new_widgets = []
        try:
            directory = util.cli.execute("xcwd").strip()
            directory = self.__get_git_root(directory)
            repo = pygit2.Repository(directory)

            new_widgets.append(
                core.widget.Widget(name="git.main", full_text=repo.head.shorthand)
            )

            for filepath, flags in repo.status().items():
                if (
                    flags == pygit2.GIT_STATUS_WT_NEW
                    or flags == pygit2.GIT_STATUS_INDEX_NEW
                ):
                    state["new"] = True
                if (
                    flags == pygit2.GIT_STATUS_WT_DELETED
                    or flags == pygit2.GIT_STATUS_INDEX_DELETED
                ):
                    state["deleted"] = True
                if (
                    flags == pygit2.GIT_STATUS_WT_MODIFIED
                    or flags == pygit2.GIT_STATUS_INDEX_MODIFIED
                ):
                    state["modified"] = True
            self.__error = False
            if "new" in state:
                new_widgets.append(core.widget.Widget(name="git.new"))
            if "modified" in state:
                new_widgets.append(core.widget.Widget(name="git.modified"))
            if "deleted" in state:
                new_widgets.append(core.widget.Widget(name="git.deleted"))

            self.widgets().clear()
            self.widget(new_widgets)

        except Exception as e:
            self.__error = True

    def state(self, widget):
        return widget.name.split(".")[1]

    def __get_git_root(self, directory):
        while len(directory) > 1:
            if os.path.exists(os.path.join(directory, ".git")):
                return directory
            directory = "/".join(directory.split("/")[0:-1])
        return "/"


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