77 lines
2.8 KiB
Python
77 lines
2.8 KiB
Python
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_GIT_REMOTE = s.get("default_remote")
|
|
|
|
def run(self, edit, permalink = False, mode = 'blob'):
|
|
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)
|
|
|
|
for remote in [DEFAULT_GIT_REMOTE]:
|
|
regex = r'.*\s.*(?:https://github\.com/|github\.com:|git://github\.com/)(.*)/(.*?)(?:\.git)?\r?\n'
|
|
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]
|
|
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 = 'https://github.com/%s/%s/%s/%s%s/%s#L%s' % \
|
|
(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)
|
|
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):
|
|
return self.view.file_name() and len(self.view.file_name()) > 0
|