# 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, incorporated herein by reference.

from node import *

from i18n import _

import repo, changegroup

import changelog, dirstate, filelog, manifest, context

import re, lock, transaction, tempfile, stat, mdiff, errno, ui

import os, revlog, time, util, extensions, hook

class localrepository(repo.repository):

    capabilities = ('lookup', 'changegroupsubset')

    supported = ('revlogv1', 'store')

    def __del__(self):

        self.transhandle = None

    def __init__(self, parentui, path=None, create=0):

        repo.repository.__init__(self)

        self.path = path

        self.root = os.path.realpath(path)

        self.path = os.path.join(self.root, ".hg")

        self.origroot = path

        self.opener = util.opener(self.path)

        self.wopener = util.opener(self.root)

        if not os.path.isdir(self.path):

            if create:

                if not os.path.exists(path):

                    os.mkdir(path)

                os.mkdir(self.path)

                requirements = ["revlogv1"]

                if parentui.configbool('format', 'usestore', True):

                    os.mkdir(os.path.join(self.path, "store"))

                    requirements.append("store")

                    # create an invalid changelog

                    self.opener("00changelog.i", "a").write(

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

                        ' dummy changelog to prevent using the old repo layout'

                    )

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

                for r in requirements:

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

                reqfile.close()

            else:

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

        elif create:

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

        else:

            # find requirements

            try:

                requirements = self.opener("requires").read().splitlines()

            except IOError, inst:

                if inst.errno != errno.ENOENT:

                    raise

                requirements = []

        # check them

        for r in requirements:

            if r not in self.supported:

                raise repo.RepoError(_("requirement '%s' not supported") % r)

        # setup store

        if "store" in requirements:

            self.encodefn = util.encodefilename

            self.decodefn = util.decodefilename

            self.spath = os.path.join(self.path, "store")

        else:

            self.encodefn = lambda x: x

            self.decodefn = lambda x: x

            self.spath = self.path

        self.sopener = util.encodedopener(util.opener(self.spath), self.encodefn)

        self.ui = ui.ui(parentui=parentui)

        try:

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

            extensions.loadall(self.ui)

        except IOError:

            pass

        self.tagscache = None

        self.branchcache = None

        self.nodetagscache = None

        self.filterpats = {}

        self.transhandle = None

    def __getattr__(self, name):

        if name == 'changelog':

            self.changelog = changelog.changelog(self.sopener)

            self.sopener.defversion = self.changelog.version

            return self.changelog

        if name == 'manifest':

            self.changelog

            self.manifest = manifest.manifest(self.sopener)

            return self.manifest

        if name == 'dirstate':

            self.dirstate = dirstate.dirstate(self.opener, self.ui, self.root)

            return self.dirstate

        else:

            raise AttributeError, name

    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, name, node, message, local, user, date, parent=None):

        use_dirstate = parent is None

        for c in self.tag_disallowed:

            if c in name:

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

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

        if local:

            # local tags are stored in the current charset

            self.opener('localtags', 'a').write('%s %s\n' % (hex(node), name))

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

            return

        # committed tags are stored in UTF-8

        line = '%s %s\n' % (hex(node), util.fromlocal(name))

        if use_dirstate:

            self.wfile('.hgtags', 'ab').write(line)

        else:

            ntags = self.filectx('.hgtags', parent).data()

            self.wfile('.hgtags', 'ab').write(ntags + line)

        if use_dirstate and self.dirstate.state('.hgtags') == '?':

            self.add(['.hgtags'])

        tagnode = self.commit(['.hgtags'], message, user, date, p1=parent)

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

        return tagnode

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

        '''tag a revision with a symbolic name.

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

        otherwise, it is stored in the .hgtags file, and a new

        changeset is committed with the change.

        keyword arguments:

        local: whether to store tag in non-version-controlled file

        (default False)

        message: commit message to use if committing

        user: name of user to use if committing

        date: date tuple to use if committing'''

        for x in self.status()[:5]:

            if '.hgtags' in x:

                raise util.Abort(_('working copy of .hgtags is changed '

                                   '(please commit .hgtags manually)'))

        self._tag(name, node, message, local, user, date)

    def tags(self):

        '''return a mapping of tag to node'''

        if self.tagscache:

            return self.tagscache

        globaltags = {}

        def readtags(lines, fn):

            filetags = {}

            count = 0

            def warn(msg):

                self.ui.warn(_("%s, line %s: %s\n") % (fn, count, msg))

            for l in lines:

                count += 1

                if not l:

                    continue

                s = l.split(" ", 1)

                if len(s) != 2:

                    warn(_("cannot parse entry"))

                    continue

                node, key = s

                key = util.tolocal(key.strip()) # stored in UTF-8

                try:

                    bin_n = bin(node)

                except TypeError:

                    warn(_("node '%s' is not well formed") % node)

                    continue

                if bin_n not in self.changelog.nodemap:

                    warn(_("tag '%s' refers to unknown node") % key)

                    continue

                h = []

                if key in filetags:

                    n, h = filetags[key]

                    h.append(n)

                filetags[key] = (bin_n, h)

            for k, nh in filetags.items():

                if k not in globaltags:

                    globaltags[k] = nh

                    continue

                # we prefer the global tag if:

                #  it supercedes us OR

                #  mutual supercedes and it has a higher rank

                # otherwise we win because we're tip-most

                an, ah = nh

                bn, bh = globaltags[k]

                if (bn != an and an in bh and

                    (bn not in ah or len(bh) > len(ah))):

                    an = bn

                ah.extend([n for n in bh if n not in ah])

                globaltags[k] = an, ah

        # read the tags file from each head, ending with the tip

        f = None

        for rev, node, fnode in self._hgtagsnodes():

            f = (f and f.filectx(fnode) or

                 self.filectx('.hgtags', fileid=fnode))

            readtags(f.data().splitlines(), f)

        try:

            data = util.fromlocal(self.opener("localtags").read())

            # localtags are stored in the local character set

            # while the internal tag table is stored in UTF-8

            readtags(data.splitlines(), "localtags")

        except IOError:

            pass

        self.tagscache = {}

        for k,nh in globaltags.items():

            n = nh[0]

            if n != nullid:

                self.tagscache[k] = n

        self.tagscache['tip'] = self.changelog.tip()

        return self.tagscache

    def _hgtagsnodes(self):

        heads = self.heads()

        heads.reverse()

        last = {}

        ret = []

        for node in heads:

            c = self.changectx(node)

            rev = c.rev()

            try:

                fnode = c.filenode('.hgtags')

            except revlog.LookupError:

                continue

            ret.append((rev, node, fnode))

            if fnode in last:

                ret[last[fnode]] = None

            last[fnode] = len(ret) - 1

        return [item for item in ret if item]

    def tagslist(self):

        '''return a list of tags ordered by revision'''

        l = []

        for t, n in self.tags().items():

            try:

                r = self.changelog.rev(n)

            except:

                r = -2 # sort to the beginning of the list if unknown

            l.append((r, t, n))

        l.sort()

        return [(t, n) for r, t, n in l]

    def nodetags(self, node):

        '''return the tags associated with a node'''

        if not self.nodetagscache:

            self.nodetagscache = {}

            for t, n in self.tags().items():

                self.nodetagscache.setdefault(n, []).append(t)

        return self.nodetagscache.get(node, [])

    def _branchtags(self):

        partial, last, lrev = self._readbranchcache()

        tiprev = self.changelog.count() - 1

        if lrev != tiprev:

            self._updatebranchcache(partial, lrev+1, tiprev+1)

            self._writebranchcache(partial, self.changelog.tip(), tiprev)

        return partial

    def branchtags(self):

        if self.branchcache is not None:

            return self.branchcache

        self.branchcache = {} # avoid recursion in changectx

        partial = self._branchtags()

        # the branch cache is stored on disk as UTF-8, but in the local

        # charset internally

        for k, v in partial.items():

            self.branchcache[util.tolocal(k)] = v

        return self.branchcache

    def _readbranchcache(self):

        partial = {}

        try:

            f = self.opener("branch.cache")

            lines = f.read().split('\n')

            f.close()

        except (IOError, OSError):

            return {}, nullid, nullrev

        try:

            last, lrev = lines.pop(0).split(" ", 1)

            last, lrev = bin(last), int(lrev)

            if not (lrev < self.changelog.count() and

                    self.changelog.node(lrev) == last): # sanity check

                # invalidate the cache

                raise ValueError('Invalid branch cache: unknown tip')

            for l in lines:

                if not l: continue

                node, label = l.split(" ", 1)

                partial[label.strip()] = bin(node)

        except (KeyboardInterrupt, util.SignalInterrupt):

            raise

        except Exception, inst:

            if self.ui.debugflag:

                self.ui.warn(str(inst), '\n')

            partial, last, lrev = {}, nullid, nullrev

        return partial, last, lrev

    def _writebranchcache(self, branches, tip, tiprev):

        try:

            f = self.opener("branch.cache", "w", atomictemp=True)

            f.write("%s %s\n" % (hex(tip), tiprev))

            for label, node in branches.iteritems():

                f.write("%s %s\n" % (hex(node), label))

            f.rename()

        except (IOError, OSError):

            pass

    def _updatebranchcache(self, partial, start, end):

        for r in xrange(start, end):

            c = self.changectx(r)

            b = c.branch()

            partial[b] = c.node()

    def lookup(self, key):

        if key == '.':

            key, second = self.dirstate.parents()

            if key == nullid:

                raise repo.RepoError(_("no revision checked out"))

            if second != nullid:

                self.ui.warn(_("warning: working directory has two parents, "

                               "tag '.' uses the first\n"))

        elif key == 'null':

            return nullid

        n = self.changelog._match(key)

        if n:

            return n

        if key in self.tags():

            return self.tags()[key]

        if key in self.branchtags():

            return self.branchtags()[key]

        n = self.changelog._partialmatch(key)

        if n:

            return n

        raise repo.RepoError(_("unknown revision '%s'") % key)

    def dev(self):

        return os.lstat(self.path).st_dev

    def local(self):

        return True

    def join(self, f):

        return os.path.join(self.path, f)

    def sjoin(self, f):

        f = self.encodefn(f)

        return os.path.join(self.spath, f)

    def wjoin(self, f):

        return os.path.join(self.root, f)

    def file(self, f):

        if f[0] == '/':

            f = f[1:]

        return filelog.filelog(self.sopener, f)

    def changectx(self, changeid=None):

        return context.changectx(self, changeid)

    def workingctx(self):

        return context.workingctx(self)

    def parents(self, changeid=None):

        '''

        get list of changectxs for parents of changeid or working directory

        '''

        if changeid is None:

            pl = self.dirstate.parents()

        else:

            n = self.changelog.lookup(changeid)

            pl = self.changelog.parents(n)

        if pl[1] == nullid:

            return [self.changectx(pl[0])]

        return [self.changectx(pl[0]), self.changectx(pl[1])]

    def filectx(self, path, changeid=None, fileid=None):

        """changeid can be a changeset revision, node, or tag.

           fileid can be a file revision or node."""

        return context.filectx(self, path, changeid, fileid)

    def getcwd(self):

        return self.dirstate.getcwd()

    def pathto(self, f, cwd=None):

        return self.dirstate.pathto(f, cwd)

    def wfile(self, f, mode='r'):

        return self.wopener(f, mode)

    def _link(self, f):

        return os.path.islink(self.wjoin(f))

    def _filter(self, filter, filename, data):

        if filter not in self.filterpats:

            l = []

            for pat, cmd in self.ui.configitems(filter):

                mf = util.matcher(self.root, "", [pat], [], [])[1]

                l.append((mf, cmd))

            self.filterpats[filter] = l

        for mf, cmd in self.filterpats[filter]:

            if mf(filename):

                self.ui.debug(_("filtering %s through %s\n") % (filename, cmd))

                data = util.filter(data, cmd)

                break

        return data

    def wread(self, filename):

        if self._link(filename):

            data = os.readlink(self.wjoin(filename))

        else:

            data = self.wopener(filename, 'r').read()

        return self._filter("encode", filename, data)

    def wwrite(self, filename, data, flags):

        data = self._filter("decode", filename, data)

        if "l" in flags:

            self.wopener.symlink(data, filename)

        else:

            try:

                if self._link(filename):

                    os.unlink(self.wjoin(filename))

            except OSError:

                pass

            self.wopener(filename, 'w').write(data)

            util.set_exec(self.wjoin(filename), "x" in flags)

    def wwritedata(self, filename, data):

        return self._filter("decode", filename, data)

    def transaction(self):

        tr = self.transhandle

        if tr != None and tr.running():

            return tr.nest()

        # save dirstate for rollback

        try:

            ds = self.opener("dirstate").read()

        except IOError:

            ds = ""

        self.opener("journal.dirstate", "w").write(ds)

        renames = [(self.sjoin("journal"), self.sjoin("undo")),