git.py
142 lines
| 4.5 KiB
| text/x-python
|
PythonLexer
|
r4536 | # git support for the convert extension | ||
import os | ||||
|
r5220 | from mercurial import util | ||
|
r4536 | |||
|
r5497 | from common import NoRepo, commit, converter_source, checktool | ||
|
r4536 | |||
class convert_git(converter_source): | ||||
|
r5217 | # Windows does not support GIT_DIR= construct while other systems | ||
# cannot remove environment variable. Just assume none have | ||||
# both issues. | ||||
if hasattr(os, 'unsetenv'): | ||||
def gitcmd(self, s): | ||||
prevgitdir = os.environ.get('GIT_DIR') | ||||
os.environ['GIT_DIR'] = self.path | ||||
try: | ||||
|
r7222 | return util.popen(s, 'rb') | ||
|
r5217 | finally: | ||
if prevgitdir is None: | ||||
del os.environ['GIT_DIR'] | ||||
else: | ||||
os.environ['GIT_DIR'] = prevgitdir | ||||
else: | ||||
def gitcmd(self, s): | ||||
|
r7222 | return util.popen('GIT_DIR=%s %s' % (self.path, s), 'rb') | ||
|
r4767 | |||
|
r4760 | def __init__(self, ui, path, rev=None): | ||
|
r4807 | super(convert_git, self).__init__(ui, path, rev=rev) | ||
|
r4536 | if os.path.isdir(path + "/.git"): | ||
path += "/.git" | ||||
if not os.path.exists(path + "/objects"): | ||||
|
r5521 | raise NoRepo("%s does not look like a Git repo" % path) | ||
|
r5497 | |||
|
r6837 | checktool('git', 'git') | ||
|
r5497 | |||
|
r4536 | self.path = path | ||
def getheads(self): | ||||
|
r4768 | if not self.rev: | ||
|
r6837 | return self.gitcmd('git rev-parse --branches').read().splitlines() | ||
|
r4768 | else: | ||
|
r6837 | fh = self.gitcmd("git rev-parse --verify %s" % self.rev) | ||
|
r4768 | return [fh.read()[:-1]] | ||
|
r4536 | |||
def catfile(self, rev, type): | ||||
if rev == "0" * 40: raise IOError() | ||||
|
r6837 | fh = self.gitcmd("git cat-file %s %s" % (type, rev)) | ||
|
r4536 | return fh.read() | ||
def getfile(self, name, rev): | ||||
return self.catfile(rev, "blob") | ||||
def getmode(self, name, rev): | ||||
return self.modecache[(name, rev)] | ||||
def getchanges(self, version): | ||||
self.modecache = {} | ||||
|
r6837 | fh = self.gitcmd("git diff-tree --root -m -r %s" % version) | ||
|
r4536 | changes = [] | ||
|
r5335 | seen = {} | ||
|
r4536 | for l in fh: | ||
|
r5335 | if "\t" not in l: | ||
continue | ||||
|
r4536 | m, f = l[:-1].split("\t") | ||
|
r5335 | if f in seen: | ||
continue | ||||
seen[f] = 1 | ||||
|
r4536 | m = m.split() | ||
h = m[3] | ||||
p = (m[1] == "100755") | ||||
s = (m[1] == "120000") | ||||
self.modecache[(f, h)] = (p and "x") or (s and "l") or "" | ||||
changes.append((f, h)) | ||||
|
r5121 | return (changes, {}) | ||
|
r4536 | |||
def getcommit(self, version): | ||||
c = self.catfile(version, "commit") # read the commit hash | ||||
end = c.find("\n\n") | ||||
message = c[end+2:] | ||||
|
r4759 | message = self.recode(message) | ||
|
r4536 | l = c[:end].splitlines() | ||
manifest = l[0].split()[1] | ||||
parents = [] | ||||
for e in l[1:]: | ||||
n, v = e.split(" ", 1) | ||||
if n == "author": | ||||
p = v.split() | ||||
tm, tz = p[-2:] | ||||
author = " ".join(p[:-2]) | ||||
if author[0] == "<": author = author[1:-1] | ||||
|
r4759 | author = self.recode(author) | ||
|
r4536 | if n == "committer": | ||
p = v.split() | ||||
tm, tz = p[-2:] | ||||
committer = " ".join(p[:-2]) | ||||
if committer[0] == "<": committer = committer[1:-1] | ||||
|
r4759 | committer = self.recode(committer) | ||
|
r4536 | message += "\ncommitter: %s\n" % committer | ||
if n == "parent": parents.append(v) | ||||
tzs, tzh, tzm = tz[-5:-4] + "1", tz[-4:-2], tz[-2:] | ||||
tz = -int(tzs) * (int(tzh) * 3600 + int(tzm)) | ||||
date = tm + " " + str(tz) | ||||
|
r4873 | c = commit(parents=parents, date=date, author=author, desc=message, | ||
rev=version) | ||||
|
r4536 | return c | ||
def gettags(self): | ||||
tags = {} | ||||
|
r6837 | fh = self.gitcmd('git ls-remote --tags "%s"' % self.path) | ||
|
r4536 | prefix = 'refs/tags/' | ||
for line in fh: | ||||
line = line.strip() | ||||
if not line.endswith("^{}"): | ||||
continue | ||||
node, tag = line.split(None, 1) | ||||
if not tag.startswith(prefix): | ||||
continue | ||||
tag = tag[len(prefix):-3] | ||||
tags[tag] = node | ||||
return tags | ||||
|
r5380 | |||
def getchangedfiles(self, version, i): | ||||
changes = [] | ||||
if i is None: | ||||
|
r6837 | fh = self.gitcmd("git diff-tree --root -m -r %s" % version) | ||
|
r5380 | for l in fh: | ||
if "\t" not in l: | ||||
continue | ||||
m, f = l[:-1].split("\t") | ||||
changes.append(f) | ||||
fh.close() | ||||
else: | ||||
|
r6837 | fh = self.gitcmd('git diff-tree --name-only --root -r %s "%s^%s" --' | ||
|
r5380 | % (version, version, i+1)) | ||
changes = [f.rstrip('\n') for f in fh] | ||||
fh.close() | ||||
return changes | ||||