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

    discovery,

    encoding,

    error,

    exchange,

    extensions,

    filelog,

    hook,

    lock as lockmod,

    manifest,

    match as matchmod,

    merge as mergemod,

    mergeutil,

    namespaces,

    obsolete,

    pathutil,

    peer,

    phases,

    pushkey,

    pycompat,

    repository,

    repoview,

    revset,

    revsetlang,

    scmutil,

    sparse,

    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)

def isfilecached(repo, name):

    """check if a repo has already cached "name" filecache-ed property

    This returns (cachedobj-or-None, iscached) tuple.

    """

    cacheentry = repo.unfiltered()._filecache.get(name, None)

    if not cacheentry:

        return None, False

    return cacheentry.obj, True

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(repository.peer):

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

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

        super(localpeer, self).__init__()

        if caps is None:

            caps = moderncaps.copy()

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

        self._ui = repo.ui

        self._caps = repo._restrictcapabilities(caps)

    # Begin of _basepeer interface.

    @util.propertycache

    def ui(self):

        return self._ui

    def url(self):

        return self._repo.url()

    def local(self):

        return self._repo

    def peer(self):

        return self

    def canpush(self):

        return True

    def close(self):

        self._repo.close()

    # End of _basepeer interface.

    # Begin of _basewirecommands interface.

    def branchmap(self):

        return self._repo.branchmap()

    def capabilities(self):

        return self._caps

    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)

    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)

    def heads(self):

        return self._repo.heads()

    def known(self, nodes):

        return self._repo.known(nodes)

    def listkeys(self, namespace):

        return self._repo.listkeys(namespace)

    def lookup(self, key):

        return self._repo.lookup(key)

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

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

    def stream_out(self):

        raise error.Abort(_('cannot perform stream clone against local '

                            'peer'))

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

    # End of _basewirecommands interface.

    # Begin of peer interface.

    def iterbatch(self):

        return peer.localiterbatcher(self)

    # End of peer interface.

class locallegacypeer(repository.legacypeer, localpeer):

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

    restricted capabilities'''

    def __init__(self, repo):

        super(locallegacypeer, self).__init__(repo, caps=legacycaps)

    # Begin of baselegacywirecommands interface.

    def between(self, pairs):

        return self._repo.between(pairs)

    def branches(self, nodes):

        return self._repo.branches(nodes)

    def changegroup(self, basenodes, source):

        outgoing = discovery.outgoing(self._repo, missingroots=basenodes,

                                      missingheads=self._repo.heads())

        return changegroup.makechangegroup(self._repo, outgoing, '01', source)

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

        outgoing = discovery.outgoing(self._repo, missingroots=bases,

                                      missingheads=heads)

        return changegroup.makechangegroup(self._repo, outgoing, '01', source)

    # End of baselegacywirecommands interface.

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

        'exp-sparse',

    }

    openerreqs = {

        'revlogv1',

        'generaldelta',

        'treemanifest',

        'manifestv2',

    }

    # a list of (ui, featureset) functions.

    # only functions defined in module of enabled extensions are invoked

    featuresetupfuncs = set()

    # list of prefix for file which can be written without 'wlock'

    # Extensions should extend this list when needed

    _wlockfreeprefix = {

        # We migh consider requiring 'wlock' for the next

        # two, but pretty much all the existing code assume

        # wlock is not needed so we keep them excluded for

        # now.

        'hgrc',

        'requires',

        # XXX cache is a complicatged business someone

        # should investigate this in depth at some point

        'cache/',

        # XXX shouldn't be dirstate covered by the wlock?

        'dirstate',

        # XXX bisect was still a bit too messy at the time

        # this changeset was introduced. Someone should fix

        # the remainig bit and drop this line

        'bisect.state',

    }

    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

        # This is only used by context.workingctx.match in order to

        # detect files in subrepos.

        self.auditor = pathutil.pathauditor(

            self.root, callback=self._checknested)

        # This is only used by context.basectx.match in order to detect

        # files in subrepos.

        self.nofsauditor = pathutil.pathauditor(

            self.root, callback=self._checknested, realfs=False, cached=True)

        self.baseui = baseui

        self.ui = baseui.copy()

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

        self.vfs = vfsmod.vfs(self.path, cacheaudited=True)

        if (self.ui.configbool('devel', 'all-warnings') or

            self.ui.configbool('devel', 'check-locks')):

            self.vfs.audit = self._getvfsward(self.vfs.audit)

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