git.py
205 lines
| 6.8 KiB
| text/x-python
|
PythonLexer
|
r8250 | # git.py - git support for the convert extension | ||
# | ||||
# Copyright 2005-2009 Matt Mackall <mpm@selenic.com> and others | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
|
r10263 | # GNU General Public License version 2 or any later version. | ||
|
r4536 | |||
import os | ||||
|
r5220 | from mercurial import util | ||
|
r12144 | from mercurial.node import hex, nullid | ||
|
r10939 | from mercurial.i18n import _ | ||
|
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'): | ||||
|
r13756 | def gitopen(self, s, noerr=False): | ||
|
r5217 | prevgitdir = os.environ.get('GIT_DIR') | ||
os.environ['GIT_DIR'] = self.path | ||||
try: | ||||
|
r13756 | if noerr: | ||
(stdin, stdout, stderr) = util.popen3(s) | ||||
return stdout | ||||
else: | ||||
return util.popen(s, 'rb') | ||||
|
r5217 | finally: | ||
if prevgitdir is None: | ||||
del os.environ['GIT_DIR'] | ||||
else: | ||||
os.environ['GIT_DIR'] = prevgitdir | ||||
else: | ||||
|
r13756 | def gitopen(self, s, noerr=False): | ||
if noerr: | ||||
(sin, so, se) = util.popen3('GIT_DIR=%s %s' % (self.path, s)) | ||||
|
r14177 | return so | ||
|
r13756 | else: | ||
|
r14735 | return util.popen('GIT_DIR=%s %s' % (self.path, s), 'rb') | ||
|
r4767 | |||
|
r10986 | def gitread(self, s): | ||
fh = self.gitopen(s) | ||||
data = fh.read() | ||||
return data, fh.close() | ||||
|
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"): | ||||
|
r10939 | raise NoRepo(_("%s does not look like a Git repository") % path) | ||
|
r5497 | |||
|
r6837 | checktool('git', 'git') | ||
|
r5497 | |||
|
r4536 | self.path = path | ||
def getheads(self): | ||||
|
r4768 | if not self.rev: | ||
|
r10986 | heads, ret = self.gitread('git rev-parse --branches --remotes') | ||
heads = heads.splitlines() | ||||
|
r4768 | else: | ||
|
r10986 | heads, ret = self.gitread("git rev-parse --verify %s" % self.rev) | ||
heads = [heads[:-1]] | ||||
if ret: | ||||
raise util.Abort(_('cannot retrieve git heads')) | ||||
return heads | ||||
|
r4536 | |||
def catfile(self, rev, type): | ||||
|
r12144 | if rev == hex(nullid): | ||
|
r10282 | raise IOError() | ||
|
r10986 | data, ret = self.gitread("git cat-file %s %s" % (type, rev)) | ||
if ret: | ||||
raise util.Abort(_('cannot read %r object at %s') % (type, rev)) | ||||
return data | ||||
|
r4536 | |||
def getfile(self, name, rev): | ||||
|
r11134 | data = self.catfile(rev, "blob") | ||
mode = self.modecache[(name, rev)] | ||||
return data, mode | ||||
|
r4536 | |||
def getchanges(self, version): | ||||
self.modecache = {} | ||||
|
r10985 | fh = self.gitopen("git diff-tree -z --root -m -r %s" % version) | ||
|
r4536 | changes = [] | ||
|
r8456 | seen = set() | ||
|
r7242 | entry = None | ||
for l in fh.read().split('\x00'): | ||||
if not entry: | ||||
if not l.startswith(':'): | ||||
continue | ||||
entry = l | ||||
|
r5335 | continue | ||
|
r7242 | f = l | ||
if f not in seen: | ||||
|
r8456 | seen.add(f) | ||
|
r7242 | entry = entry.split() | ||
h = entry[3] | ||||
p = (entry[1] == "100755") | ||||
s = (entry[1] == "120000") | ||||
self.modecache[(f, h)] = (p and "x") or (s and "l") or "" | ||||
changes.append((f, h)) | ||||
entry = None | ||||
|
r10987 | if fh.close(): | ||
raise util.Abort(_('cannot read changes in %s') % version) | ||||
|
r5121 | return (changes, {}) | ||
|
r4536 | |||
def getcommit(self, version): | ||||
c = self.catfile(version, "commit") # read the commit hash | ||||
end = c.find("\n\n") | ||||
|
r10282 | message = c[end + 2:] | ||
|
r4759 | message = self.recode(message) | ||
|
r4536 | l = c[:end].splitlines() | ||
parents = [] | ||||
|
r8271 | author = committer = None | ||
|
r4536 | 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) | ||
|
r10282 | if n == "parent": | ||
parents.append(v) | ||||
|
r4536 | |||
|
r8271 | if committer and committer != author: | ||
message += "\ncommitter: %s\n" % committer | ||||
|
r4536 | 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 = {} | ||||
|
r10985 | fh = self.gitopen('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 | ||||
|
r10987 | if fh.close(): | ||
raise util.Abort(_('cannot read tags from %s') % self.path) | ||||
|
r4536 | |||
return tags | ||||
|
r5380 | |||
def getchangedfiles(self, version, i): | ||||
changes = [] | ||||
if i is None: | ||||
|
r10985 | fh = self.gitopen("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) | ||||
else: | ||||
|
r10985 | fh = self.gitopen('git diff-tree --name-only --root -r %s "%s^%s" --' | ||
|
r10282 | % (version, version, i + 1)) | ||
|
r5380 | changes = [f.rstrip('\n') for f in fh] | ||
|
r10987 | if fh.close(): | ||
raise util.Abort(_('cannot read changes in %s') % version) | ||||
|
r5380 | |||
return changes | ||||
|
r13756 | |||
def getbookmarks(self): | ||||
bookmarks = {} | ||||
# Interesting references in git are prefixed | ||||
prefix = 'refs/heads/' | ||||
prefixlen = len(prefix) | ||||
# factor two commands | ||||
gitcmd = { 'remote/': 'git ls-remote --heads origin', | ||||
'': 'git show-ref'} | ||||
# Origin heads | ||||
for reftype in gitcmd: | ||||
try: | ||||
fh = self.gitopen(gitcmd[reftype], noerr=True) | ||||
for line in fh: | ||||
line = line.strip() | ||||
rev, name = line.split(None, 1) | ||||
if not name.startswith(prefix): | ||||
continue | ||||
name = '%s%s' % (reftype, name[prefixlen:]) | ||||
bookmarks[name] = rev | ||||
except: | ||||
pass | ||||
return bookmarks | ||||