githubinator.py

import os
import re
import sublime
import sublime_plugin


class GithubinatorCommand(sublime_plugin.TextCommand):
    '''This will allow you to highlight your code, activate the plugin, then see the
    highlighted results on GitHub.
    '''

    def load_config(self):
        s = sublime.load_settings("Githubinator.sublime-settings")
        global DEFAULT_GIT_REMOTE, DEFAULT_GITHUB_HOST
        DEFAULT_GIT_REMOTE = s.get("default_remote")
        if not isinstance(DEFAULT_GIT_REMOTE, list):
            DEFAULT_GIT_REMOTE = [DEFAULT_GIT_REMOTE]
        DEFAULT_GITHUB_HOST = s.get("default_host")
        if DEFAULT_GITHUB_HOST is None:
            DEFAULT_GITHUB_HOST = "github.com"


    def run(self, edit, copyonly = False, permalink = False, mode = 'blob', branch = None):
        self.load_config()
        if not self.view.file_name():
            return

        full_name = os.path.realpath(self.view.file_name())
        folder_name, file_name = os.path.split(full_name)

        git_path = self.recurse_dir(folder_name, '.git')
        if not git_path:
            sublime.status_message('Could not find .git directory.')
            print('Could not find .git directory.')
            return

        git_config_path = os.path.join(git_path, '.git', 'config')
        new_git_path = folder_name[len(git_path):]

        with open(git_config_path, "r") as git_config_file:
            config = git_config_file.read()

        sel = self.view.sel()[0]
        begin_line = self.view.rowcol(sel.begin())[0] + 1
        end_line = self.view.rowcol(sel.end())[0] + 1
        if begin_line == end_line:
            lines = begin_line
        else:
            lines = '%s-%s' % (begin_line, end_line)

        HTTP = 'https'
        result = re.search(r'url.*?=.*?((https?)://([^/]*)/)|(git@([^:]*):)', config)
        if result:
            matches = result.groups()
            if matches[0]:
                HTTP = matches[1]
                DEFAULT_GITHUB_HOST = matches[2]
            else:
                DEFAULT_GITHUB_HOST = matches[4]
        
        re_host = re.escape(DEFAULT_GITHUB_HOST)
        for remote in DEFAULT_GIT_REMOTE:
            regex = r'.*\s.*(?:https?://%s/|%s:|git://%s/)(.*)/(.*?)(?:\.git)?\r?\n' % (re_host, re_host, re_host)
            result = re.search(remote + regex, config)
            if not result:
                continue
            matches = result.groups()

            ref_path = open(os.path.join(git_path, '.git', 'HEAD'), "r").read().replace('ref: ', '')[:-1]
            if branch is None: branch = ref_path.replace('refs/heads/','')
            sha = open(os.path.join(git_path, '.git', ref_path), "r").read()[:-1]
            target = sha if permalink else branch

            full_link = HTTP + '://%s/%s/%s/%s/%s%s/%s#L%s' % \
                (DEFAULT_GITHUB_HOST, matches[0], matches[1], mode, target, new_git_path, file_name, lines)
            sublime.set_clipboard(full_link)
            sublime.status_message('Copied %s to clipboard.' % full_link)
            print('Copied %s to clipboard.' % full_link)
            if not copyonly:
                self.view.window().run_command('open_url', {"url": full_link})
            break


    def recurse_dir(self, path, folder):
        items = os.listdir(path)
        if folder in items and os.path.isdir(os.path.join(path, folder)):
            return path
        dirname = os.path.dirname(path)
        if dirname == path:
            return None
        return self.recurse_dir(dirname, folder)


    def is_enabled(self):
        if self.view.file_name() and len(self.view.file_name()) > 0:
            return True
        else:
            return False