darcs.py
208 lines
| 7.8 KiB
| text/x-python
|
PythonLexer
|
r8250 | # darcs.py - darcs support for the convert extension | ||
# | ||||
# Copyright 2007-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. | ||
|
r5359 | |||
|
r5512 | from common import NoRepo, checktool, commandline, commit, converter_source | ||
|
r5359 | from mercurial.i18n import _ | ||
|
r26587 | from mercurial import util, error | ||
|
r22296 | import os, shutil, tempfile, re, errno | ||
|
r5359 | |||
# The naming drift of ElementTree is fun! | ||||
|
r10282 | try: | ||
|
r12717 | from xml.etree.cElementTree import ElementTree, XMLParser | ||
|
r5359 | except ImportError: | ||
|
r10282 | try: | ||
|
r12717 | from xml.etree.ElementTree import ElementTree, XMLParser | ||
|
r5359 | except ImportError: | ||
|
r10282 | try: | ||
|
r12717 | from elementtree.cElementTree import ElementTree, XMLParser | ||
|
r5359 | except ImportError: | ||
|
r10282 | try: | ||
|
r12717 | from elementtree.ElementTree import ElementTree, XMLParser | ||
|
r10282 | except ImportError: | ||
|
r15457 | pass | ||
|
r5359 | |||
|
r5512 | class darcs_source(converter_source, commandline): | ||
|
r25748 | def __init__(self, ui, path, revs=None): | ||
converter_source.__init__(self, ui, path, revs=revs) | ||||
|
r5512 | commandline.__init__(self, ui, 'darcs') | ||
|
r5359 | |||
|
r12393 | # check for _darcs, ElementTree so that we can easily skip | ||
# test-convert-darcs if ElementTree is not around | ||||
|
r5520 | if not os.path.exists(os.path.join(path, '_darcs')): | ||
|
r10939 | raise NoRepo(_("%s does not look like a darcs repository") % path) | ||
|
r5359 | |||
|
r5497 | checktool('darcs') | ||
|
r9242 | version = self.run0('--version').splitlines()[0].strip() | ||
if version < '2.1': | ||||
|
r26587 | raise error.Abort(_('darcs version 2.1 or newer needed (found %r)') | ||
% version) | ||||
|
r5497 | |||
|
r15457 | if "ElementTree" not in globals(): | ||
|
r26587 | raise error.Abort(_("Python ElementTree module is not available")) | ||
|
r5359 | |||
|
r15381 | self.path = os.path.realpath(path) | ||
|
r5359 | |||
self.lastrev = None | ||||
self.changes = {} | ||||
self.parents = {} | ||||
self.tags = {} | ||||
|
r12393 | # Check darcs repository format | ||
format = self.format() | ||||
if format: | ||||
if format in ('darcs-1.0', 'hashed'): | ||||
raise NoRepo(_("%s repository format is unsupported, " | ||||
"please upgrade") % format) | ||||
else: | ||||
self.ui.warn(_('failed to detect repository format!')) | ||||
|
r5359 | def before(self): | ||
self.tmppath = tempfile.mkdtemp( | ||||
prefix='convert-' + os.path.basename(self.path) + '-') | ||||
output, status = self.run('init', repodir=self.tmppath) | ||||
self.checkexit(status) | ||||
|
r5512 | tree = self.xml('changes', xml_output=True, summary=True, | ||
repodir=self.path) | ||||
|
r5359 | tagname = None | ||
child = None | ||||
for elt in tree.findall('patch'): | ||||
node = elt.get('hash') | ||||
name = elt.findtext('name', '') | ||||
if name.startswith('TAG '): | ||||
tagname = name[4:].strip() | ||||
elif tagname is not None: | ||||
self.tags[tagname] = node | ||||
tagname = None | ||||
self.changes[node] = elt | ||||
self.parents[child] = [node] | ||||
child = node | ||||
self.parents[child] = [] | ||||
def after(self): | ||||
|
r9467 | self.ui.debug('cleaning up %s\n' % self.tmppath) | ||
|
r5362 | shutil.rmtree(self.tmppath, ignore_errors=True) | ||
|
r5359 | |||
|
r12717 | def recode(self, s, encoding=None): | ||
if isinstance(s, unicode): | ||||
# XMLParser returns unicode objects for anything it can't | ||||
# encode into ASCII. We convert them back to str to get | ||||
# recode's normal conversion behavior. | ||||
s = s.encode('latin-1') | ||||
return super(darcs_source, self).recode(s, encoding) | ||||
|
r5512 | def xml(self, cmd, **kwargs): | ||
|
r12252 | # NOTE: darcs is currently encoding agnostic and will print | ||
# patch metadata byte-for-byte, even in the XML changelog. | ||||
|
r5359 | etree = ElementTree() | ||
|
r12717 | # While we are decoding the XML as latin-1 to be as liberal as | ||
# possible, etree will still raise an exception if any | ||||
# non-printable characters are in the XML changelog. | ||||
parser = XMLParser(encoding='latin-1') | ||||
|
r17413 | p = self._run(cmd, **kwargs) | ||
etree.parse(p.stdout, parser=parser) | ||||
p.wait() | ||||
self.checkexit(p.returncode) | ||||
|
r5359 | return etree.getroot() | ||
|
r12393 | def format(self): | ||
output, status = self.run('show', 'repo', no_files=True, | ||||
repodir=self.path) | ||||
self.checkexit(status) | ||||
m = re.search(r'^\s*Format:\s*(.*)$', output, re.MULTILINE) | ||||
if not m: | ||||
return None | ||||
return ','.join(sorted(f.strip() for f in m.group(1).split(','))) | ||||
|
r9527 | def manifest(self): | ||
man = [] | ||||
output, status = self.run('show', 'files', no_directories=True, | ||||
repodir=self.tmppath) | ||||
self.checkexit(status) | ||||
for line in output.split('\n'): | ||||
path = line[2:] | ||||
if path: | ||||
man.append(path) | ||||
return man | ||||
|
r5359 | def getheads(self): | ||
return self.parents[None] | ||||
def getcommit(self, rev): | ||||
elt = self.changes[rev] | ||||
date = util.strdate(elt.get('local_date'), '%a %b %d %H:%M:%S %Z %Y') | ||||
desc = elt.findtext('name') + '\n' + elt.findtext('comment', '') | ||||
|
r12252 | # etree can return unicode objects for name, comment, and author, | ||
# so recode() is used to ensure str objects are emitted. | ||||
return commit(author=self.recode(elt.get('author')), | ||||
|
r16514 | date=util.datestr(date, '%Y-%m-%d %H:%M:%S %1%2'), | ||
|
r12252 | desc=self.recode(desc).strip(), | ||
parents=self.parents[rev]) | ||||
|
r5359 | |||
def pull(self, rev): | ||||
|
r5512 | output, status = self.run('pull', self.path, all=True, | ||
match='hash %s' % rev, | ||||
no_test=True, no_posthook=True, | ||||
external_merge='/bin/false', | ||||
|
r5359 | repodir=self.tmppath) | ||
if status: | ||||
if output.find('We have conflicts in') == -1: | ||||
self.checkexit(status, output) | ||||
|
r5512 | output, status = self.run('revert', all=True, repodir=self.tmppath) | ||
|
r5359 | self.checkexit(status, output) | ||
|
r22300 | def getchanges(self, rev, full): | ||
if full: | ||||
|
r26779 | raise error.Abort(_("convert from darcs does not support --full")) | ||
|
r5359 | copies = {} | ||
changes = [] | ||||
|
r9527 | man = None | ||
|
r5359 | for elt in self.changes[rev].find('summary').getchildren(): | ||
if elt.tag in ('add_directory', 'remove_directory'): | ||||
continue | ||||
if elt.tag == 'move': | ||||
|
r9527 | if man is None: | ||
man = self.manifest() | ||||
source, dest = elt.get('from'), elt.get('to') | ||||
if source in man: | ||||
# File move | ||||
changes.append((source, rev)) | ||||
changes.append((dest, rev)) | ||||
copies[dest] = source | ||||
else: | ||||
# Directory move, deduce file moves from manifest | ||||
source = source + '/' | ||||
for f in man: | ||||
if not f.startswith(source): | ||||
continue | ||||
fdest = dest + '/' + f[len(source):] | ||||
changes.append((f, rev)) | ||||
changes.append((fdest, rev)) | ||||
copies[fdest] = f | ||||
|
r5359 | else: | ||
changes.append((elt.text.strip(), rev)) | ||||
|
r9527 | self.pull(rev) | ||
|
r5359 | self.lastrev = rev | ||
|
r24395 | return sorted(changes), copies, set() | ||
|
r5359 | |||
def getfile(self, name, rev): | ||||
if rev != self.lastrev: | ||||
|
r26587 | raise error.Abort(_('internal calling inconsistency')) | ||
|
r11134 | path = os.path.join(self.tmppath, name) | ||
|
r22296 | try: | ||
data = util.readfile(path) | ||||
mode = os.lstat(path).st_mode | ||||
|
r25660 | except IOError as inst: | ||
|
r22296 | if inst.errno == errno.ENOENT: | ||
return None, None | ||||
raise | ||||
|
r25658 | mode = (mode & 0o111) and 'x' or '' | ||
|
r11134 | return data, mode | ||
|
r5359 | |||
def gettags(self): | ||||
return self.tags | ||||