forked from ehamiter/GitHubinator
-
Notifications
You must be signed in to change notification settings - Fork 0
/
githubinator.py
76 lines (60 loc) · 2.74 KB
/
githubinator.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
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):
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)?\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/blob/%s%s/%s#L%s' % \
(matches[0], matches[1], 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