# 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 node import bin, hex, nullid, nullrev, short

from i18n import _

import repo, changegroup, subrepo, discovery, pushkey, obsolete

import changelog, dirstate, filelog, manifest, context, bookmarks, phases

import lock, transaction, store, encoding, base85

import scmutil, util, extensions, hook, error, revset

import match as matchmod

import merge as mergemod

import tags as tagsmod

from lock import release

import weakref, errno, os, time, inspect

propertycache = util.propertycache

filecache = scmutil.filecache

class storecache(filecache):

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

    def join(self, obj, fname):

        return obj.sjoin(fname)

class localrepository(repo.repository):

    capabilities = set(('lookup', 'changegroupsubset', 'branchmap', 'pushkey',

                        'known', 'getbundle'))

    supportedformats = set(('revlogv1', 'generaldelta'))

    supported = supportedformats | set(('store', 'fncache', 'shared',

                                        'dotencode'))

    openerreqs = set(('revlogv1', 'generaldelta'))

    requirements = ['revlogv1']

    def _baserequirements(self, create):

        return self.requirements[:]

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

        repo.repository.__init__(self)

        self.wopener = scmutil.opener(path, expand=True)

        self.wvfs = self.wopener

        self.root = self.wvfs.base

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

        self.origroot = path

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

        self.opener = scmutil.opener(self.path)

        self.vfs = self.opener

        self.baseui = baseui

        self.ui = baseui.copy()

        # 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.join("hgrc"), self.root)

            extensions.loadall(self.ui)

        except IOError:

            pass

        if not self.vfs.isdir():

            if create:

                if not self.wvfs.exists():

                    self.wvfs.makedirs()

                self.vfs.makedir(notindexed=True)

                requirements = self._baserequirements(create)

                if self.ui.configbool('format', 'usestore', True):

                    self.vfs.mkdir("store")

                    requirements.append("store")

                    if self.ui.configbool('format', 'usefncache', True):

                        requirements.append("fncache")

                        if self.ui.configbool('format', 'dotencode', True):

                            requirements.append('dotencode')

                    # create an invalid changelog

                    self.vfs.append(

                        "00changelog.i",

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

                        ' dummy changelog to prevent using the old repo layout'

                    )

                if self.ui.configbool('format', 'generaldelta', False):

                    requirements.append("generaldelta")

                requirements = set(requirements)

            else:

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

        elif create:

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

        else:

            try:

                requirements = scmutil.readrequires(self.vfs, self.supported)

            except IOError, inst:

                if inst.errno != errno.ENOENT:

                    raise

                requirements = set()

        self.sharedpath = self.path

        try:

            s = os.path.realpath(self.opener.read("sharedpath").rstrip('\n'))

            if not os.path.exists(s):

                raise error.RepoError(

                    _('.hg/sharedpath points to nonexistent directory %s') % s)

            self.sharedpath = s

        except IOError, inst:

            if inst.errno != errno.ENOENT:

                raise

        self.store = store.store(requirements, self.sharedpath, scmutil.opener)

        self.spath = self.store.path

        self.sopener = self.store.opener

        self.svfs = self.sopener

        self.sjoin = self.store.join

        self.opener.createmode = self.store.createmode

        self._applyrequirements(requirements)

        if create:

            self._writerequirements()

        self._branchcache = None

        self._branchcachetip = None

        self.filterpats = {}

        self._datafilters = {}

        self._transref = self._lockref = self._wlockref = None

        # A cache for various files under .hg/ that tracks file changes,

        # (used by the filecache decorator)

        #

        # Maps a property name to its util.filecacheentry

        self._filecache = {}

    def _applyrequirements(self, requirements):

        self.requirements = requirements

        self.sopener.options = dict((r, 1) for r in requirements

                                           if r in self.openerreqs)

    def _writerequirements(self):

        reqfile = self.opener("requires", "w")

        for r in self.requirements:

            reqfile.write("%s\n" % r)

        reqfile.close()

    def _checknested(self, path):

        """Determine if path is a legal nested repository."""

        if not path.startswith(self.root):

            return False

        subpath = path[len(self.root) + 1:]

        normsubpath = util.pconvert(subpath)

        # XXX: Checking against the current working copy is wrong in

        # the sense that it can reject things like

        #

        #   $ hg cat -r 10 sub/x.txt

        #

        # if sub/ is no longer a subrepository in the working copy

        # parent revision.

        #

        # However, it can of course also allow things that would have

        # been rejected before, such as the above cat command if sub/

        # is a subrepository now, but was a normal directory before.

        # The old path auditor would have rejected by mistake since it

        # panics when it sees sub/.hg/.

        #

        # All in all, checking against the working copy seems sensible

        # since we want to prevent access to nested repositories on

        # the filesystem *now*.

        ctx = self[None]

        parts = util.splitpath(subpath)

        while parts:

            prefix = '/'.join(parts)

            if prefix in ctx.substate:

                if prefix == normsubpath:

                    return True

                else:

                    sub = ctx.sub(prefix)

                    return sub.checknested(subpath[len(prefix) + 1:])

            else:

                parts.pop()

        return False

    @filecache('bookmarks')

    def _bookmarks(self):

        return bookmarks.read(self)

    @filecache('bookmarks.current')

    def _bookmarkcurrent(self):

        return bookmarks.readcurrent(self)

    def _writebookmarks(self, marks):

      bookmarks.write(self)

    def bookmarkheads(self, bookmark):

        name = bookmark.split('@', 1)[0]

        heads = []

        for mark, n in self._bookmarks.iteritems():

            if mark.split('@', 1)[0] == name:

                heads.append(n)

        return heads

    @storecache('phaseroots')

    def _phasecache(self):

        return phases.phasecache(self, self._phasedefaults)

    @storecache('obsstore')

    def obsstore(self):

        store = obsolete.obsstore(self.sopener)

        return store

    @storecache('00changelog.i')

    def changelog(self):

        c = changelog.changelog(self.sopener)

        if 'HG_PENDING' in os.environ:

            p = os.environ['HG_PENDING']

            if p.startswith(self.root):

                c.readpending('00changelog.i.a')

        return c

    @storecache('00manifest.i')

    def manifest(self):

        return manifest.manifest(self.sopener)

    @filecache('dirstate')

    def dirstate(self):

        warned = [0]

        def validate(node):

            try:

                self.changelog.rev(node)

                return node

            except error.LookupError:

                if not warned[0]:

                    warned[0] = True

                    self.ui.warn(_("warning: ignoring unknown"

                                   " working parent %s!\n") % short(node))

                return nullid

        return dirstate.dirstate(self.opener, self.ui, self.root, validate)

    def __getitem__(self, changeid):

        if changeid is None:

            return context.workingctx(self)

        return context.changectx(self, changeid)

    def __contains__(self, changeid):

        try:

            return bool(self.lookup(changeid))

        except error.RepoLookupError:

            return False

    def __nonzero__(self):

        return True

    def __len__(self):

        return len(self.changelog)

    def __iter__(self):

        for i in xrange(len(self)):

            yield i

    def revs(self, expr, *args):

        '''Return a list of revisions matching the given revset'''

        expr = revset.formatspec(expr, *args)

        m = revset.match(None, expr)

        return [r for r in m(self, range(len(self)))]

    def set(self, expr, *args):

        '''

        Yield a context for each matching revision, after doing arg

        replacement via revset.formatspec

        '''

        for r in self.revs(expr, *args):

            yield self[r]

    def url(self):

        return 'file:' + self.root

    def hook(self, name, throw=False, **args):

        return hook.hook(self.ui, self, name, throw, **args)

    tag_disallowed = ':\r\n'

    def _tag(self, names, node, message, local, user, date, extra={}):

        if isinstance(names, str):

            allchars = names

            names = (names,)

        else:

            allchars = ''.join(names)

        for c in self.tag_disallowed:

            if c in allchars:

                raise util.Abort(_('%r cannot be used in a tag name') % c)

        branches = self.branchmap()

        for name in names:

            self.hook('pretag', throw=True, node=hex(node), tag=name,

                      local=local)

            if name in branches:

                self.ui.warn(_("warning: tag %s conflicts with existing"

                " branch name\n") % name)

        def writetags(fp, names, munge, prevtags):

            fp.seek(0, 2)

            if prevtags and prevtags[-1] != '\n':

                fp.write('\n')

            for name in names:

                m = munge and munge(name) or name

                if (self._tagscache.tagtypes and

                    name in self._tagscache.tagtypes):

                    old = self.tags().get(name, nullid)

                    fp.write('%s %s\n' % (hex(old), m))

                fp.write('%s %s\n' % (hex(node), m))

            fp.close()

        prevtags = ''

        if local:

            try:

                fp = self.opener('localtags', 'r+')

            except IOError:

                fp = self.opener('localtags', 'a')

            else:

                prevtags = fp.read()

            # local tags are stored in the current charset

            writetags(fp, names, None, prevtags)

            for name in names:

                self.hook('tag', node=hex(node), tag=name, local=local)

            return

        try:

            fp = self.wfile('.hgtags', 'rb+')

        except IOError, e:

            if e.errno != errno.ENOENT:

                raise

            fp = self.wfile('.hgtags', 'ab')

        else:

            prevtags = fp.read()

        # committed tags are stored in UTF-8

        writetags(fp, names, encoding.fromlocal, prevtags)

        fp.close()

        self.invalidatecaches()

        if '.hgtags' not in self.dirstate:

            self[None].add(['.hgtags'])

        m = matchmod.exact(self.root, '', ['.hgtags'])

        tagnode = self.commit(message, user, date, extra=extra, match=m)

        for name in names:

            self.hook('tag', node=hex(node), tag=name, local=local)

        return tagnode

    def tag(self, names, node, message, local, user, date):

        '''tag a revision with one or more symbolic names.

        names is a list of strings or, when adding a single tag, names may be a

        string.

        if local is True, the tags are stored in a per-repository file.

        otherwise, they are stored in the .hgtags file, and a new

        changeset is committed with the change.