# subrepo.py - sub-repository handling for Mercurial

#

# Copyright 2009-2010 Matt Mackall <mpm@selenic.com>

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

import errno, os, re, xml.dom.minidom, shutil, posixpath

import stat, subprocess, tarfile

from i18n import _

import config, scmutil, util, node, error, cmdutil, bookmarks

hg = None

propertycache = util.propertycache

nullstate = ('', '', 'empty')

def state(ctx, ui):

    """return a state dict, mapping subrepo paths configured in .hgsub

    to tuple: (source from .hgsub, revision from .hgsubstate, kind

    (key in types dict))

    """

    p = config.config()

    def read(f, sections=None, remap=None):

        if f in ctx:

            try:

                data = ctx[f].data()

            except IOError, err:

                if err.errno != errno.ENOENT:

                    raise

                # handle missing subrepo spec files as removed

                ui.warn(_("warning: subrepo spec file %s not found\n") % f)

                return

            p.parse(f, data, sections, remap, read)

        else:

            raise util.Abort(_("subrepo spec file %s not found") % f)

    if '.hgsub' in ctx:

        read('.hgsub')

    for path, src in ui.configitems('subpaths'):

        p.set('subpaths', path, src, ui.configsource('subpaths', path))

    rev = {}

    if '.hgsubstate' in ctx:

        try:

            for l in ctx['.hgsubstate'].data().splitlines():

                revision, path = l.split(" ", 1)

                rev[path] = revision

        except IOError, err:

            if err.errno != errno.ENOENT:

                raise

    def remap(src):

        for pattern, repl in p.items('subpaths'):

            # Turn r'C:\foo\bar' into r'C:\\foo\\bar' since re.sub

            # does a string decode.

            repl = repl.encode('string-escape')

            # However, we still want to allow back references to go

            # through unharmed, so we turn r'\\1' into r'\1'. Again,

            # extra escapes are needed because re.sub string decodes.

            repl = re.sub(r'\\\\([0-9]+)', r'\\\1', repl)

            try:

                src = re.sub(pattern, repl, src, 1)

            except re.error, e:

                raise util.Abort(_("bad subrepository pattern in %s: %s")

                                 % (p.source('subpaths', pattern), e))

        return src

    state = {}

    for path, src in p[''].items():

        kind = 'hg'

        if src.startswith('['):

            if ']' not in src:

                raise util.Abort(_('missing ] in subrepo source'))

            kind, src = src.split(']', 1)

            kind = kind[1:]

            src = src.lstrip() # strip any extra whitespace after ']'

        if not util.url(src).isabs():

            parent = _abssource(ctx._repo, abort=False)

            if parent:

                parent = util.url(parent)

                parent.path = posixpath.join(parent.path or '', src)

                parent.path = posixpath.normpath(parent.path)

                joined = str(parent)

                # Remap the full joined path and use it if it changes,

                # else remap the original source.

                remapped = remap(joined)

                if remapped == joined:

                    src = remap(src)

                else:

                    src = remapped

        src = remap(src)

        state[path] = (src.strip(), rev.get(path, ''), kind)

    return state

def writestate(repo, state):

    """rewrite .hgsubstate in (outer) repo with these subrepo states"""

    lines = ['%s %s\n' % (state[s][1], s) for s in sorted(state)]

    repo.wwrite('.hgsubstate', ''.join(lines), '')

def submerge(repo, wctx, mctx, actx, overwrite):

    """delegated from merge.applyupdates: merging of .hgsubstate file

    in working context, merging context and ancestor context"""

    if mctx == actx: # backwards?

        actx = wctx.p1()

    s1 = wctx.substate

    s2 = mctx.substate

    sa = actx.substate

    sm = {}

    repo.ui.debug("subrepo merge %s %s %s\n" % (wctx, mctx, actx))

    def debug(s, msg, r=""):

        if r:

            r = "%s:%s:%s" % r

        repo.ui.debug("  subrepo %s: %s %s\n" % (s, msg, r))

    for s, l in s1.items():

        a = sa.get(s, nullstate)

        ld = l # local state with possible dirty flag for compares

        if wctx.sub(s).dirty():

            ld = (l[0], l[1] + "+")

        if wctx == actx: # overwrite

            a = ld

        if s in s2:

            r = s2[s]

            if ld == r or r == a: # no change or local is newer

                sm[s] = l

                continue

            elif ld == a: # other side changed

                debug(s, "other changed, get", r)

                wctx.sub(s).get(r, overwrite)

                sm[s] = r

            elif ld[0] != r[0]: # sources differ

                if repo.ui.promptchoice(

                    _(' subrepository sources for %s differ\n'

                      'use (l)ocal source (%s) or (r)emote source (%s)?')

                      % (s, l[0], r[0]),

                      (_('&Local'), _('&Remote')), 0):

                    debug(s, "prompt changed, get", r)

                    wctx.sub(s).get(r, overwrite)

                    sm[s] = r

            elif ld[1] == a[1]: # local side is unchanged

                debug(s, "other side changed, get", r)

                wctx.sub(s).get(r, overwrite)

                sm[s] = r

            else:

                debug(s, "both sides changed, merge with", r)

                wctx.sub(s).merge(r)

                sm[s] = l

        elif ld == a: # remote removed, local unchanged

            debug(s, "remote removed, remove")

            wctx.sub(s).remove()

        elif a == nullstate: # not present in remote or ancestor

            debug(s, "local added, keep")

            sm[s] = l

            continue

        else:

            if repo.ui.promptchoice(

                _(' local changed subrepository %s which remote removed\n'

                  'use (c)hanged version or (d)elete?') % s,

                (_('&Changed'), _('&Delete')), 0):

                debug(s, "prompt remove")

                wctx.sub(s).remove()

    for s, r in sorted(s2.items()):

        if s in s1:

            continue

        elif s not in sa:

            debug(s, "remote added, get", r)

            mctx.sub(s).get(r)

            sm[s] = r

        elif r != sa[s]:

            if repo.ui.promptchoice(

                _(' remote changed subrepository %s which local removed\n'

                  'use (c)hanged version or (d)elete?') % s,

                (_('&Changed'), _('&Delete')), 0) == 0:

                debug(s, "prompt recreate", r)

                wctx.sub(s).get(r)

                sm[s] = r

    # record merged .hgsubstate

    writestate(repo, sm)

def _updateprompt(ui, sub, dirty, local, remote):

    if dirty:

        msg = (_(' subrepository sources for %s differ\n'

                 'use (l)ocal source (%s) or (r)emote source (%s)?\n')

               % (subrelpath(sub), local, remote))

    else:

        msg = (_(' subrepository sources for %s differ (in checked out version)\n'

                 'use (l)ocal source (%s) or (r)emote source (%s)?\n')

               % (subrelpath(sub), local, remote))

    return ui.promptchoice(msg, (_('&Local'), _('&Remote')), 0)

def reporelpath(repo):

    """return path to this (sub)repo as seen from outermost repo"""

    parent = repo

    while util.safehasattr(parent, '_subparent'):

        parent = parent._subparent

    p = parent.root.rstrip(os.sep)

    return repo.root[len(p) + 1:]

def subrelpath(sub):

    """return path to this subrepo as seen from outermost repo"""

    if util.safehasattr(sub, '_relpath'):

        return sub._relpath

    if not util.safehasattr(sub, '_repo'):

        return sub._path

    return reporelpath(sub._repo)

def _abssource(repo, push=False, abort=True):

    """return pull/push path of repo - either based on parent repo .hgsub info

    or on the top repo config. Abort or return None if no source found."""

    if util.safehasattr(repo, '_subparent'):

        source = util.url(repo._subsource)

        if source.isabs():

            return str(source)

        source.path = posixpath.normpath(source.path)

        parent = _abssource(repo._subparent, push, abort=False)

        if parent:

            parent = util.url(parent)

            parent.path = posixpath.join(parent.path or '', source.path)

            parent.path = posixpath.normpath(parent.path)

            return str(parent)

    else: # recursion reached top repo

        if util.safehasattr(repo, '_subtoppath'):

            return repo._subtoppath

        if push and repo.ui.config('paths', 'default-push'):

            return repo.ui.config('paths', 'default-push')

        if repo.ui.config('paths', 'default'):

            return repo.ui.config('paths', 'default')

    if abort:

        raise util.Abort(_("default path for subrepository %s not found") %

            reporelpath(repo))

def itersubrepos(ctx1, ctx2):

    """find subrepos in ctx1 or ctx2"""

    # Create a (subpath, ctx) mapping where we prefer subpaths from

    # ctx1. The subpaths from ctx2 are important when the .hgsub file

    # has been modified (in ctx2) but not yet committed (in ctx1).

    subpaths = dict.fromkeys(ctx2.substate, ctx2)

    subpaths.update(dict.fromkeys(ctx1.substate, ctx1))

    for subpath, ctx in sorted(subpaths.iteritems()):

        yield subpath, ctx.sub(subpath)

def subrepo(ctx, path):

    """return instance of the right subrepo class for subrepo in path"""

    # subrepo inherently violates our import layering rules

    # because it wants to make repo objects from deep inside the stack

    # so we manually delay the circular imports to not break

    # scripts that don't use our demand-loading

    global hg

    import hg as h

    hg = h

    scmutil.pathauditor(ctx._repo.root)(path)

    state = ctx.substate.get(path, nullstate)

    if state[2] not in types:

        raise util.Abort(_('unknown subrepo type %s') % state[2])

    return types[state[2]](ctx, path, state[:2])

# subrepo classes need to implement the following abstract class:

class abstractsubrepo(object):

    def dirty(self, ignoreupdate=False):

        """returns true if the dirstate of the subrepo is dirty or does not

        match current stored state. If ignoreupdate is true, only check

        whether the subrepo has uncommitted changes in its dirstate.

        """

        raise NotImplementedError

    def checknested(self, path):

        """check if path is a subrepository within this repository"""

        return False

    def commit(self, text, user, date):

        """commit the current changes to the subrepo with the given

        log message. Use given user and date if possible. Return the

        new state of the subrepo.

        """

        raise NotImplementedError

    def remove(self):

        """remove the subrepo

        (should verify the dirstate is not dirty first)

        """

        raise NotImplementedError

    def get(self, state, overwrite=False):

        """run whatever commands are needed to put the subrepo into

        this state

        """

        raise NotImplementedError

    def merge(self, state):

        """merge currently-saved state with the new state."""

        raise NotImplementedError

    def push(self, force):

        """perform whatever action is analogous to 'hg push'

        This may be a no-op on some systems.

        """

        raise NotImplementedError

    def add(self, ui, match, dryrun, prefix):

        return []

    def status(self, rev2, **opts):

        return [], [], [], [], [], [], []

    def diff(self, diffopts, node2, match, prefix, **opts):

        pass

    def outgoing(self, ui, dest, opts):

        return 1

    def incoming(self, ui, source, opts):

        return 1

    def files(self):

        """return filename iterator"""

        raise NotImplementedError

    def filedata(self, name):

        """return file data"""

        raise NotImplementedError

    def fileflags(self, name):

        """return file flags"""

        return ''

    def archive(self, ui, archiver, prefix):

        files = self.files()

        total = len(files)

        relpath = subrelpath(self)

        ui.progress(_('archiving (%s)') % relpath, 0,

                    unit=_('files'), total=total)

        for i, name in enumerate(files):

            flags = self.fileflags(name)

            mode = 'x' in flags and 0755 or 0644

            symlink = 'l' in flags

            archiver.addfile(os.path.join(prefix, self._path, name),

                             mode, symlink, self.filedata(name))

            ui.progress(_('archiving (%s)') % relpath, i + 1,

                        unit=_('files'), total=total)

        ui.progress(_('archiving (%s)') % relpath, None)

class hgsubrepo(abstractsubrepo):

    def __init__(self, ctx, path, state):

        self._path = path

        self._state = state

        r = ctx._repo

        root = r.wjoin(path)

        create = False

        if not os.path.exists(os.path.join(root, '.hg')):

            create = True

            util.makedirs(root)

        self._repo = hg.repository(r.ui, root, create=create)

        self._initrepo(r, state[0], create)

    def _initrepo(self, parentrepo, source, create):

        self._repo._subparent = parentrepo

        self._repo._subsource = source

        if create:

            fp = self._repo.opener("hgrc", "w", text=True)

            fp.write('[paths]\n')

            def addpathconfig(key, value):

                if value:

                    fp.write('%s = %s\n' % (key, value))

                    self._repo.ui.setconfig('paths', key, value)

            defpath = _abssource(self._repo, abort=False)

            defpushpath = _abssource(self._repo, True, abort=False)

            addpathconfig('default', defpath)

            if defpath != defpushpath:

                addpathconfig('default-push', defpushpath)

            fp.close()

    def add(self, ui, match, dryrun, prefix):

        return cmdutil.add(ui, self._repo, match, dryrun, True,

                           os.path.join(prefix, self._path))

    def status(self, rev2, **opts):

        try:

            rev1 = self._state[1]

            ctx1 = self._repo[rev1]

            ctx2 = self._repo[rev2]

            return self._repo.status(ctx1, ctx2, **opts)

        except error.RepoLookupError, inst:

            self._repo.ui.warn(_('warning: error "%s" in subrepository "%s"\n')

                               % (inst, subrelpath(self)))

            return [], [], [], [], [], [], []

    def diff(self, diffopts, node2, match, prefix, **opts):

        try:

            node1 = node.bin(self._state[1])

            # We currently expect node2 to come from substate and be

            # in hex format

            if node2 is not None:

                node2 = node.bin(node2)

            cmdutil.diffordiffstat(self._repo.ui, self._repo, diffopts,

                                   node1, node2, match,

                                   prefix=os.path.join(prefix, self._path),

                                   listsubrepos=True, **opts)

        except error.RepoLookupError, inst:

            self._repo.ui.warn(_('warning: error "%s" in subrepository "%s"\n')

                               % (inst, subrelpath(self)))

    def archive(self, ui, archiver, prefix):

        self._get(self._state + ('hg',))

        abstractsubrepo.archive(self, ui, archiver, prefix)

        rev = self._state[1]

        ctx = self._repo[rev]

        for subpath in ctx.substate:

            s = subrepo(ctx, subpath)

            s.archive(ui, archiver, os.path.join(prefix, self._path))

    def dirty(self, ignoreupdate=False):

        r = self._state[1]

        if r == '' and not ignoreupdate: # no state recorded

            return True

        w = self._repo[None]

        if r != w.p1().hex() and not ignoreupdate:

            # different version checked out

            return True

        return w.dirty() # working directory changed

    def checknested(self, path):

        return self._repo._checknested(self._repo.wjoin(path))

    def commit(self, text, user, date):

        # don't bother committing in the subrepo if it's only been

        # updated

        if not self.dirty(True):

            return self._repo['.'].hex()

        self._repo.ui.debug("committing subrepo %s\n" % subrelpath(self))

        n = self._repo.commit(text, user, date)

        if not n:

            return self._repo['.'].hex() # different version checked out

        return node.hex(n)

    def remove(self):

        # we can't fully delete the repository as it may contain

        # local-only history

        self._repo.ui.note(_('removing subrepo %s\n') % subrelpath(self))

        hg.clean(self._repo, node.nullid, False)

    def _get(self, state):

        source, revision, kind = state

        if revision not in self._repo:

            self._repo._subsource = source

            srcurl = _abssource(self._repo)

            other = hg.peer(self._repo.ui, {}, srcurl)

            if len(self._repo) == 0:

                self._repo.ui.status(_('cloning subrepo %s from %s\n')

                                     % (subrelpath(self), srcurl))

                parentrepo = self._repo._subparent

                shutil.rmtree(self._repo.path)

                other, self._repo = hg.clone(self._repo._subparent.ui, {}, other,

                                         self._repo.root, update=False)

                self._initrepo(parentrepo, source, create=True)

            else:

                self._repo.ui.status(_('pulling subrepo %s from %s\n')

                                     % (subrelpath(self), srcurl))

                self._repo.pull(other)