# localrepo.py - read/write repository class for mercurial

#

# 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 errno

import hashlib

import inspect

import os

import random

import time

import weakref

from .i18n import _

from .node import (

    hex,

    nullid,

    short,

)

from . import (

    bookmarks,

    branchmap,

    bundle2,

    changegroup,

    changelog,

    color,

    context,

    dirstate,

    dirstateguard,

    encoding,

    error,

    exchange,

    extensions,

    filelog,

    hook,

    lock as lockmod,

    manifest,

    match as matchmod,

    merge as mergemod,

    mergeutil,

    namespaces,

    obsolete,

    pathutil,

    peer,

    phases,

    pushkey,

    pycompat,

    repoview,

    revset,

    revsetlang,

    scmutil,

    store,

    subrepo,

    tags as tagsmod,

    transaction,

    txnutil,

    util,

    vfs as vfsmod,

)

release = lockmod.release

urlerr = util.urlerr

urlreq = util.urlreq

# set of (path, vfs-location) tuples. vfs-location is:

# - 'plain for vfs relative paths

# - '' for svfs relative paths

_cachedfiles = set()

class _basefilecache(scmutil.filecache):

    """All filecache usage on repo are done for logic that should be unfiltered

    """

    def __get__(self, repo, type=None):

        if repo is None:

            return self

        return super(_basefilecache, self).__get__(repo.unfiltered(), type)

    def __set__(self, repo, value):

        return super(_basefilecache, self).__set__(repo.unfiltered(), value)

    def __delete__(self, repo):

        return super(_basefilecache, self).__delete__(repo.unfiltered())

class repofilecache(_basefilecache):

    """filecache for files in .hg but outside of .hg/store"""

    def __init__(self, *paths):

        super(repofilecache, self).__init__(*paths)

        for path in paths:

            _cachedfiles.add((path, 'plain'))

    def join(self, obj, fname):

        return obj.vfs.join(fname)

class storecache(_basefilecache):

    """filecache for files in the store"""

    def __init__(self, *paths):

        super(storecache, self).__init__(*paths)

        for path in paths:

            _cachedfiles.add((path, ''))

    def join(self, obj, fname):

        return obj.sjoin(fname)

class unfilteredpropertycache(util.propertycache):

    """propertycache that apply to unfiltered repo only"""

    def __get__(self, repo, type=None):

        unfi = repo.unfiltered()

        if unfi is repo:

            return super(unfilteredpropertycache, self).__get__(unfi)

        return getattr(unfi, self.name)

class filteredpropertycache(util.propertycache):

    """propertycache that must take filtering in account"""

    def cachevalue(self, obj, value):

        object.__setattr__(obj, self.name, value)

def hasunfilteredcache(repo, name):

    """check if a repo has an unfilteredpropertycache value for <name>"""

    return name in vars(repo.unfiltered())

def unfilteredmethod(orig):

    """decorate method that always need to be run on unfiltered version"""

    def wrapper(repo, *args, **kwargs):

        return orig(repo.unfiltered(), *args, **kwargs)

    return wrapper

moderncaps = {'lookup', 'branchmap', 'pushkey', 'known', 'getbundle',

              'unbundle'}

legacycaps = moderncaps.union({'changegroupsubset'})

class localpeer(peer.peerrepository):

    '''peer for a local repo; reflects only the most recent API'''

    def __init__(self, repo, caps=None):

        if caps is None:

            caps = moderncaps.copy()

        peer.peerrepository.__init__(self)

        self._repo = repo.filtered('served')

        self.ui = repo.ui

        self._caps = repo._restrictcapabilities(caps)

        self.requirements = repo.requirements

        self.supportedformats = repo.supportedformats

    def close(self):

        self._repo.close()

    def _capabilities(self):

        return self._caps

    def local(self):

        return self._repo

    def canpush(self):

        return True

    def url(self):

        return self._repo.url()

    def lookup(self, key):

        return self._repo.lookup(key)

    def branchmap(self):

        return self._repo.branchmap()

    def heads(self):

        return self._repo.heads()

    def known(self, nodes):

        return self._repo.known(nodes)

    def getbundle(self, source, heads=None, common=None, bundlecaps=None,

                  **kwargs):

        chunks = exchange.getbundlechunks(self._repo, source, heads=heads,

                                          common=common, bundlecaps=bundlecaps,

                                          **kwargs)

        cb = util.chunkbuffer(chunks)

        if exchange.bundle2requested(bundlecaps):

            # When requesting a bundle2, getbundle returns a stream to make the

            # wire level function happier. We need to build a proper object

            # from it in local peer.

            return bundle2.getunbundler(self.ui, cb)

        else:

            return changegroup.getunbundler('01', cb, None)

    # TODO We might want to move the next two calls into legacypeer and add

    # unbundle instead.

    def unbundle(self, cg, heads, url):

        """apply a bundle on a repo

        This function handles the repo locking itself."""

        try:

            try:

                cg = exchange.readbundle(self.ui, cg, None)

                ret = exchange.unbundle(self._repo, cg, heads, 'push', url)

                if util.safehasattr(ret, 'getchunks'):

                    # This is a bundle20 object, turn it into an unbundler.

                    # This little dance should be dropped eventually when the

                    # API is finally improved.

                    stream = util.chunkbuffer(ret.getchunks())

                    ret = bundle2.getunbundler(self.ui, stream)

                return ret

            except Exception as exc:

                # If the exception contains output salvaged from a bundle2

                # reply, we need to make sure it is printed before continuing

                # to fail. So we build a bundle2 with such output and consume

                # it directly.

                #

                # This is not very elegant but allows a "simple" solution for

                # issue4594

                output = getattr(exc, '_bundle2salvagedoutput', ())

                if output:

                    bundler = bundle2.bundle20(self._repo.ui)

                    for out in output:

                        bundler.addpart(out)

                    stream = util.chunkbuffer(bundler.getchunks())

                    b = bundle2.getunbundler(self.ui, stream)

                    bundle2.processbundle(self._repo, b)

                raise

        except error.PushRaced as exc:

            raise error.ResponseError(_('push failed:'), str(exc))

    def lock(self):

        return self._repo.lock()

    def pushkey(self, namespace, key, old, new):

        return self._repo.pushkey(namespace, key, old, new)

    def listkeys(self, namespace):

        return self._repo.listkeys(namespace)

    def debugwireargs(self, one, two, three=None, four=None, five=None):

        '''used to test argument passing over the wire'''

        return "%s %s %s %s %s" % (one, two, three, four, five)

class locallegacypeer(localpeer):

    '''peer extension which implements legacy methods too; used for tests with

    restricted capabilities'''

    def __init__(self, repo):

        localpeer.__init__(self, repo, caps=legacycaps)

    def branches(self, nodes):

        return self._repo.branches(nodes)

    def between(self, pairs):

        return self._repo.between(pairs)

    def changegroup(self, basenodes, source):

        return changegroup.changegroup(self._repo, basenodes, source)

    def changegroupsubset(self, bases, heads, source):

        return changegroup.changegroupsubset(self._repo, bases, heads, source)

# Increment the sub-version when the revlog v2 format changes to lock out old

# clients.

REVLOGV2_REQUIREMENT = 'exp-revlogv2.0'

class localrepository(object):

    supportedformats = {

        'revlogv1',

        'generaldelta',

        'treemanifest',

        'manifestv2',

        REVLOGV2_REQUIREMENT,

    }

    _basesupported = supportedformats | {

        'store',

        'fncache',

        'shared',

        'relshared',

        'dotencode',

    }

    openerreqs = {

        'revlogv1',

        'generaldelta',

        'treemanifest',

        'manifestv2',

    }

    # a list of (ui, featureset) functions.

    # only functions defined in module of enabled extensions are invoked

    featuresetupfuncs = set()

    def __init__(self, baseui, path, create=False):

        self.requirements = set()

        self.filtername = None

        # wvfs: rooted at the repository root, used to access the working copy

        self.wvfs = vfsmod.vfs(path, expandpath=True, realpath=True)

        # vfs: rooted at .hg, used to access repo files outside of .hg/store

        self.vfs = None

        # svfs: usually rooted at .hg/store, used to access repository history

        # If this is a shared repository, this vfs may point to another

        # repository's .hg/store directory.

        self.svfs = None

        self.root = self.wvfs.base

        self.path = self.wvfs.join(".hg")

        self.origroot = path

        # These auditor are not used by the vfs,

        # only used when writing this comment: basectx.match

        self.auditor = pathutil.pathauditor(self.root, self._checknested)

        self.nofsauditor = pathutil.pathauditor(self.root, self._checknested,

                                                realfs=False)

        self.vfs = vfsmod.vfs(self.path)

        self.baseui = baseui

        self.ui = baseui.copy()

        self.ui.copy = baseui.copy # prevent copying repo configuration

        # A list of callback to shape the phase if no data were found.

        # Callback are in the form: func(repo, roots) --> processed root.

        # This list it to be filled by extension during repo setup

        self._phasedefaults = []

        try:

            self.ui.readconfig(self.vfs.join("hgrc"), self.root)

            self._loadextensions()

        except IOError:

            pass

        if self.featuresetupfuncs:

            self.supported = set(self._basesupported) # use private copy

            extmods = set(m.__name__ for n, m

                          in extensions.extensions(self.ui))

            for setupfunc in self.featuresetupfuncs:

                if setupfunc.__module__ in extmods:

                    setupfunc(self.ui, self.supported)

        else:

            self.supported = self._basesupported

        color.setup(self.ui)

        # Add compression engines.

        for name in util.compengines:

            engine = util.compengines[name]

            if engine.revlogheader():

                self.supported.add('exp-compression-%s' % name)

        if not self.vfs.isdir():

            if create:

                self.requirements = newreporequirements(self)

                if not self.wvfs.exists():

                    self.wvfs.makedirs()

                self.vfs.makedir(notindexed=True)

                if 'store' in self.requirements:

                    self.vfs.mkdir("store")

                    # create an invalid changelog

                    self.vfs.append(

                        "00changelog.i",

                        '\0\0\0\2' # represents revlogv2

                        ' dummy changelog to prevent using the old repo layout'

                    )

            else:

                raise error.RepoError(_("repository %s not found") % path)

        elif create:

            raise error.RepoError(_("repository %s already exists") % path)

        else:

            try:

                self.requirements = scmutil.readrequires(

                        self.vfs, self.supported)

            except IOError as inst:

                if inst.errno != errno.ENOENT:

                    raise

        self.sharedpath = self.path

        try:

            sharedpath = self.vfs.read("sharedpath").rstrip('\n')

            if 'relshared' in self.requirements:

                sharedpath = self.vfs.join(sharedpath)

            vfs = vfsmod.vfs(sharedpath, realpath=True)

            s = vfs.base