rana-cli/wakatime/projects/git.py

# -*- coding: utf-8 -*-
"""
    wakatime.projects.git
    ~~~~~~~~~~~~~~~~~~~~~

    Information about the git project for a given file.

    :copyright: (c) 2013 Alan Hamlett.
    :license: BSD, see LICENSE for more details.
"""

import logging
import os
from subprocess import Popen, PIPE

from .base import BaseProject
try:
    from collections import OrderedDict
except ImportError:
    from ..packages.ordereddict import OrderedDict


log = logging.getLogger(__name__)


class Git(BaseProject):

    def process(self):
        self.config = self._find_config(self.path)
        if self.config:
            return True
        return False

    def name(self):
        base = self._project_base()
        if base:
            return os.path.basename(base)
        return None

    def tags(self):
        tags = []
        if self.config:
            proj_name = self.name()
            if proj_name:
                tags.append(proj_name)
            sections = self._parse_config()
            for section in sections:
                if section.split(' ', 1)[0] == 'remote' and 'url' in sections[section]:
                    remote = sections[section]['url'].rsplit(':', 1)[-1].rsplit('/', 1)[-1].split('.git', 1)[0]
                    if remote:
                        tags.append(remote)
            branch = self._current_branch()
            if branch is not None:
                tags.append(branch)
        return tags

    def _project_base(self):
        if self.config:
            return os.path.dirname(os.path.dirname(self.config))
        return None

    def _current_branch(self):
        stdout = None
        try:
            stdout, stderr = Popen([
                    'git', 'branch', '--no-color'
                ], stdout=PIPE, stderr=PIPE, cwd=self._project_base()
            ).communicate()
        except OSError:
            pass
        if stdout:
            for line in stdout.splitlines():
                if isinstance(line, bytes):
                    line = bytes.decode(line)
                line = line.split(' ', 1)
                if line[0] == '*':
                    return line[1]
        return None

    def _find_config(self, path):
        path = os.path.realpath(path)
        if os.path.isfile(path):
            path = os.path.split(path)[0]
        if os.path.isfile(os.path.join(path, '.git', 'config')):
            return os.path.join(path, '.git', 'config')
        split_path = os.path.split(path)
        if split_path[1] == '':
            return None
        return self._find_config(split_path[0])

    def _parse_config(self):
        sections = {}
        try:
            f = open(self.config, 'r')
        except IOError as e:
            log.exception("Exception:")
        else:
            with f:
                section = None
                for line in f.readlines():
                    line = line.lstrip()
                    if len(line) > 0 and line[0] == '[':
                        section = line[1:].split(']', 1)[0]
                        temp = section.split(' ', 1)
                        section = temp[0].lower()
                        if len(temp) > 1:
                            section = ' '.join([section, temp[1]])
                        sections[section] = {}
                    else:
                        try:
                            (setting, value) = line.split('=', 1)
                        except ValueError:
                            setting = line.split('#', 1)[0].split(';', 1)[0]
                            value = 'true'
                        setting = setting.strip().lower()
                        value = value.split('#', 1)[0].split(';', 1)[0].strip()
                        sections[section][setting] = value
                f.close()
        return sections