# exchange.py - utility to exchange data between repos.

#

# Copyright 2005-2007 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.

from __future__ import absolute_import

import collections

import hashlib

from .i18n import _

from .node import (

    bin,

    hex,

    nullid,

    nullrev,

)

from .thirdparty import (

    attr,

)

from . import (

    bookmarks as bookmod,

    bundle2,

    changegroup,

    discovery,

    error,

    exchangev2,

    lock as lockmod,

    logexchange,

    narrowspec,

    obsolete,

    phases,

    pushkey,

    pycompat,

    repository,

    scmutil,

    sslutil,

    streamclone,

    url as urlmod,

    util,

    wireprototypes,

)

from .utils import (

    stringutil,

)

urlerr = util.urlerr

urlreq = util.urlreq

_NARROWACL_SECTION = 'narrowacl'

# Maps bundle version human names to changegroup versions.

_bundlespeccgversions = {'v1': '01',

                         'v2': '02',

                         'packed1': 's1',

                         'bundle2': '02', #legacy

                        }

# Maps bundle version with content opts to choose which part to bundle

_bundlespeccontentopts = {

    'v1': {

        'changegroup': True,

        'cg.version': '01',

        'obsolescence': False,

        'phases': False,

        'tagsfnodescache': False,

        'revbranchcache': False

    },

    'v2': {

        'changegroup': True,

        'cg.version': '02',

        'obsolescence': False,

        'phases': False,

        'tagsfnodescache': True,

        'revbranchcache': True

    },

    'packed1' : {

        'cg.version': 's1'

    }

}

_bundlespeccontentopts['bundle2'] = _bundlespeccontentopts['v2']

_bundlespecvariants = {"streamv2": {"changegroup": False, "streamv2": True,

                                    "tagsfnodescache": False,

                                    "revbranchcache": False}}

# Compression engines allowed in version 1. THIS SHOULD NEVER CHANGE.

_bundlespecv1compengines = {'gzip', 'bzip2', 'none'}

@attr.s

class bundlespec(object):

    compression = attr.ib()

    wirecompression = attr.ib()

    version = attr.ib()

    wireversion = attr.ib()

    params = attr.ib()

    contentopts = attr.ib()

def parsebundlespec(repo, spec, strict=True):

    """Parse a bundle string specification into parts.

    Bundle specifications denote a well-defined bundle/exchange format.

    The content of a given specification should not change over time in

    order to ensure that bundles produced by a newer version of Mercurial are

    readable from an older version.

    The string currently has the form:

       <compression>-<type>[;<parameter0>[;<parameter1>]]

    Where <compression> is one of the supported compression formats

    and <type> is (currently) a version string. A ";" can follow the type and

    all text afterwards is interpreted as URI encoded, ";" delimited key=value

    pairs.

    If ``strict`` is True (the default) <compression> is required. Otherwise,

    it is optional.

    Returns a bundlespec object of (compression, version, parameters).

    Compression will be ``None`` if not in strict mode and a compression isn't

    defined.

    An ``InvalidBundleSpecification`` is raised when the specification is

    not syntactically well formed.

    An ``UnsupportedBundleSpecification`` is raised when the compression or

    bundle type/version is not recognized.

    Note: this function will likely eventually return a more complex data

    structure, including bundle2 part information.

    """

    def parseparams(s):

        if ';' not in s:

            return s, {}

        params = {}

        version, paramstr = s.split(';', 1)

        for p in paramstr.split(';'):

            if '=' not in p:

                raise error.InvalidBundleSpecification(

                    _('invalid bundle specification: '

                      'missing "=" in parameter: %s') % p)

            key, value = p.split('=', 1)

            key = urlreq.unquote(key)

            value = urlreq.unquote(value)

            params[key] = value

        return version, params

    if strict and '-' not in spec:

        raise error.InvalidBundleSpecification(

                _('invalid bundle specification; '

                  'must be prefixed with compression: %s') % spec)

    if '-' in spec:

        compression, version = spec.split('-', 1)

        if compression not in util.compengines.supportedbundlenames:

            raise error.UnsupportedBundleSpecification(

                    _('%s compression is not supported') % compression)

        version, params = parseparams(version)

        if version not in _bundlespeccgversions:

            raise error.UnsupportedBundleSpecification(

                    _('%s is not a recognized bundle version') % version)

    else:

        # Value could be just the compression or just the version, in which

        # case some defaults are assumed (but only when not in strict mode).

        assert not strict

        spec, params = parseparams(spec)

        if spec in util.compengines.supportedbundlenames:

            compression = spec

            version = 'v1'

            # Generaldelta repos require v2.

            if 'generaldelta' in repo.requirements:

                version = 'v2'

            # Modern compression engines require v2.

            if compression not in _bundlespecv1compengines:

                version = 'v2'

        elif spec in _bundlespeccgversions:

            if spec == 'packed1':

                compression = 'none'

            else:

                compression = 'bzip2'

            version = spec

        else:

            raise error.UnsupportedBundleSpecification(

                    _('%s is not a recognized bundle specification') % spec)

    # Bundle version 1 only supports a known set of compression engines.

    if version == 'v1' and compression not in _bundlespecv1compengines:

        raise error.UnsupportedBundleSpecification(

            _('compression engine %s is not supported on v1 bundles') %

            compression)

    # The specification for packed1 can optionally declare the data formats

    # required to apply it. If we see this metadata, compare against what the

    # repo supports and error if the bundle isn't compatible.

    if version == 'packed1' and 'requirements' in params:

        requirements = set(params['requirements'].split(','))

        missingreqs = requirements - repo.supportedformats

        if missingreqs:

            raise error.UnsupportedBundleSpecification(

                    _('missing support for repository features: %s') %

                      ', '.join(sorted(missingreqs)))

    # Compute contentopts based on the version

    contentopts = _bundlespeccontentopts.get(version, {}).copy()

    # Process the variants

    if "stream" in params and params["stream"] == "v2":

        variant = _bundlespecvariants["streamv2"]

        contentopts.update(variant)

    engine = util.compengines.forbundlename(compression)

    compression, wirecompression = engine.bundletype()

    wireversion = _bundlespeccgversions[version]

    return bundlespec(compression, wirecompression, version, wireversion,

                      params, contentopts)

def readbundle(ui, fh, fname, vfs=None):

    header = changegroup.readexactly(fh, 4)

    alg = None

    if not fname:

        fname = "stream"

        if not header.startswith('HG') and header.startswith('\0'):

            fh = changegroup.headerlessfixup(fh, header)

            header = "HG10"

            alg = 'UN'

    elif vfs:

        fname = vfs.join(fname)

    magic, version = header[0:2], header[2:4]

    if magic != 'HG':

        raise error.Abort(_('%s: not a Mercurial bundle') % fname)

    if version == '10':

        if alg is None:

            alg = changegroup.readexactly(fh, 2)

        return changegroup.cg1unpacker(fh, alg)

    elif version.startswith('2'):

        return bundle2.getunbundler(ui, fh, magicstring=magic + version)

    elif version == 'S1':

        return streamclone.streamcloneapplier(fh)

    else:

        raise error.Abort(_('%s: unknown bundle version %s') % (fname, version))

def getbundlespec(ui, fh):

    """Infer the bundlespec from a bundle file handle.

    The input file handle is seeked and the original seek position is not

    restored.

    """

    def speccompression(alg):

        try:

            return util.compengines.forbundletype(alg).bundletype()[0]

        except KeyError:

            return None

    b = readbundle(ui, fh, None)

    if isinstance(b, changegroup.cg1unpacker):

        alg = b._type

        if alg == '_truncatedBZ':

            alg = 'BZ'

        comp = speccompression(alg)

        if not comp:

            raise error.Abort(_('unknown compression algorithm: %s') % alg)

        return '%s-v1' % comp

    elif isinstance(b, bundle2.unbundle20):

        if 'Compression' in b.params:

            comp = speccompression(b.params['Compression'])

            if not comp:

                raise error.Abort(_('unknown compression algorithm: %s') % comp)

        else:

            comp = 'none'

        version = None

        for part in b.iterparts():

            if part.type == 'changegroup':

                version = part.params['version']

                if version in ('01', '02'):

                    version = 'v2'

                else:

                    raise error.Abort(_('changegroup version %s does not have '

                                        'a known bundlespec') % version,

                                      hint=_('try upgrading your Mercurial '

                                              'client'))

            elif part.type == 'stream2' and version is None:

                # A stream2 part requires to be part of a v2 bundle

                requirements = urlreq.unquote(part.params['requirements'])

                splitted = requirements.split()

                params = bundle2._formatrequirementsparams(splitted)

                return 'none-v2;stream=v2;%s' % params

        if not version:

            raise error.Abort(_('could not identify changegroup version in '

                                'bundle'))

        return '%s-%s' % (comp, version)

    elif isinstance(b, streamclone.streamcloneapplier):

        requirements = streamclone.readbundle1header(fh)[2]

        formatted = bundle2._formatrequirementsparams(requirements)

        return 'none-packed1;%s' % formatted

    else:

        raise error.Abort(_('unknown bundle type: %s') % b)

def _computeoutgoing(repo, heads, common):

    """Computes which revs are outgoing given a set of common

    and a set of heads.

    This is a separate function so extensions can have access to

    the logic.

    Returns a discovery.outgoing object.

    """

    cl = repo.changelog

    if common:

        hasnode = cl.hasnode

        common = [n for n in common if hasnode(n)]

    else:

        common = [nullid]

    if not heads:

        heads = cl.heads()

    return discovery.outgoing(repo, common, heads)

def _checkpublish(pushop):

    repo = pushop.repo

    ui = repo.ui

    behavior = ui.config('experimental', 'auto-publish')

    if pushop.publish or behavior not in ('warn', 'confirm', 'abort'):

        return

    remotephases = listkeys(pushop.remote, 'phases')

    if not remotephases.get('publishing', False):

        return

    if pushop.revs is None:

        published = repo.filtered('served').revs('not public()')

    else:

        published = repo.revs('::%ln - public()', pushop.revs)

    if published:

        if behavior == 'warn':

            ui.warn(_('%i changesets about to be published\n')

                    % len(published))

        elif behavior == 'confirm':

            if ui.promptchoice(_('push and publish %i changesets (yn)?'

                                 '$$ &Yes $$ &No') % len(published)):

                raise error.Abort(_('user quit'))

        elif behavior == 'abort':

            msg = _('push would publish %i changesets') % len(published)

            hint = _("use --publish or adjust 'experimental.auto-publish'"

                     " config")

            raise error.Abort(msg, hint=hint)

def _forcebundle1(op):

    """return true if a pull/push must use bundle1

    This function is used to allow testing of the older bundle version"""

    ui = op.repo.ui

    # The goal is this config is to allow developer to choose the bundle

    # version used during exchanged. This is especially handy during test.

    # Value is a list of bundle version to be picked from, highest version

    # should be used.

    #

    # developer config: devel.legacy.exchange

    exchange = ui.configlist('devel', 'legacy.exchange')

    forcebundle1 = 'bundle2' not in exchange and 'bundle1' in exchange

    return forcebundle1 or not op.remote.capable('bundle2')

class pushoperation(object):

    """A object that represent a single push operation

    Its purpose is to carry push related state and very common operations.

    A new pushoperation should be created at the beginning of each push and

    discarded afterward.

    """

    def __init__(self, repo, remote, force=False, revs=None, newbranch=False,