# darcs support for the convert extension

from mercurial.i18n import _

from mercurial import util

import os, shutil, tempfile

# The naming drift of ElementTree is fun!

try: from xml.etree.cElementTree import ElementTree

except ImportError:

    try: from xml.etree.ElementTree import ElementTree

    except ImportError:

        try: from elementtree.cElementTree import ElementTree

        except ImportError:

            try: from elementtree.ElementTree import ElementTree

            except ImportError: ElementTree = None

    def __init__(self, ui, path, rev=None):

        if not os.path.exists(os.path.join(path, '_darcs', 'inventory')):

            raise NoRepo("couldn't open darcs repo %s" % path)

        checktool('darcs')

        if ElementTree is None:

            raise util.Abort(_("Python ElementTree module is not available"))

        self.path = os.path.realpath(path)

        self.lastrev = None

        self.changes = {}

        self.parents = {}

        self.tags = {}

    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)

        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):

        self.ui.debug('cleaning up %s\n' % self.tmppath)

        shutil.rmtree(self.tmppath, ignore_errors=True)

        etree = ElementTree()

        etree.parse(fp)

        self.checkexit(fp.close())

        return etree.getroot()

    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', '')

        return commit(author=elt.get('author'), date=util.datestr(date),

                      desc=desc.strip(), parents=self.parents[rev])

    def pull(self, rev):

                                  repodir=self.tmppath)

        if status:

            if output.find('We have conflicts in') == -1:

                self.checkexit(status, output)

            self.checkexit(status, output)

    def getchanges(self, rev):

        self.pull(rev)

        copies = {}

        changes = []

        for elt in self.changes[rev].find('summary').getchildren():

            if elt.tag in ('add_directory', 'remove_directory'):

                continue

            if elt.tag == 'move':

                changes.append((elt.get('from'), rev))

                copies[elt.get('from')] = elt.get('to')

            else:

                changes.append((elt.text.strip(), rev))

        changes.sort()

        self.lastrev = rev

        return changes, copies

    def getfile(self, name, rev):

        if rev != self.lastrev:

            raise util.Abort(_('internal calling inconsistency'))

        return open(os.path.join(self.tmppath, name), 'rb').read()

    def getmode(self, name, rev):

        mode = os.lstat(os.path.join(self.tmppath, name)).st_mode

        return (mode & 0111) and 'x' or ''

    def gettags(self):

        return self.tags