# templatekw.py - common changeset template keywords

#

# Copyright 2005-2009 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

from .i18n import _

from .node import (

    hex,

    nullid,

    short,

)

from . import (

    encoding,

    error,

    hbisect,

    obsutil,

    patch,

    pycompat,

    registrar,

    scmutil,

    util,

)

class _hybrid(object):

    """Wrapper for list or dict to support legacy template

    This class allows us to handle both:

    - "{files}" (legacy command-line-specific list hack) and

    - "{files % '{file}\n'}" (hgweb-style with inlining and function support)

    and to access raw values:

    - "{ifcontains(file, files, ...)}", "{ifcontains(key, extras, ...)}"

    - "{get(extras, key)}"

    - "{files|json}"

    """

    def __init__(self, gen, values, makemap, joinfmt):

        if gen is not None:

            self.gen = gen

        self._values = values

        self._makemap = makemap

        self.joinfmt = joinfmt

    @util.propertycache

    def gen(self):

        return self._defaultgen()

    def _defaultgen(self):

        """Generator to stringify this as {join(self, ' ')}"""

        for i, d in enumerate(self.itermaps()):

            if i > 0:

                yield ' '

            yield self.joinfmt(d)

    def itermaps(self):

        makemap = self._makemap

        for x in self._values:

            yield makemap(x)

    def __contains__(self, x):

        return x in self._values

    def __getitem__(self, key):

        return self._values[key]

    def __len__(self):

        return len(self._values)

    def __iter__(self):

        return iter(self._values)

    def __getattr__(self, name):

        if name not in ('get', 'items', 'iteritems', 'iterkeys', 'itervalues',

                        'keys', 'values'):

            raise AttributeError(name)

        return getattr(self._values, name)

def hybriddict(data, key='key', value='value', fmt='%s=%s', gen=None):

    """Wrap data to support both dict-like and string-like operations"""

    return _hybrid(gen, data, lambda k: {key: k, value: data[k]},

                   lambda d: fmt % (d[key], d[value]))

def hybridlist(data, name, fmt='%s', gen=None):

    """Wrap data to support both list-like and string-like operations"""

    return _hybrid(gen, data, lambda x: {name: x}, lambda d: fmt % d[name])

def unwraphybrid(thing):

    """Return an object which can be stringified possibly by using a legacy

    template"""

    if not util.safehasattr(thing, 'gen'):

        return thing

    return thing.gen

def showdict(name, data, mapping, plural=None, key='key', value='value',

             fmt='%s=%s', separator=' '):

    c = [{key: k, value: v} for k, v in data.iteritems()]

    f = _showlist(name, c, mapping, plural, separator)

    return hybriddict(data, key=key, value=value, fmt=fmt, gen=f)

def showlist(name, values, mapping, plural=None, element=None, separator=' '):

    if not element:

        element = name

    f = _showlist(name, values, mapping, plural, separator)

    return hybridlist(values, name=element, gen=f)

def _showlist(name, values, mapping, plural=None, separator=' '):

    '''expand set of values.

    name is name of key in template map.

    values is list of strings or dicts.

    plural is plural of name, if not simply name + 's'.

    separator is used to join values as a string

    expansion works like this, given name 'foo'.

    if values is empty, expand 'no_foos'.

    if 'foo' not in template map, return values as a string,

    joined by 'separator'.

    expand 'start_foos'.

    for each value, expand 'foo'. if 'last_foo' in template

    map, expand it instead of 'foo' for last key.

    expand 'end_foos'.

    '''

    templ = mapping['templ']

    strmapping = pycompat.strkwargs(mapping)

    if not plural:

        plural = name + 's'

    if not values:

        noname = 'no_' + plural

        if noname in templ:

            yield templ(noname, **strmapping)

        return

    if name not in templ:

        if isinstance(values[0], bytes):

            yield separator.join(values)

        else:

            for v in values:

                yield dict(v, **strmapping)

        return

    startname = 'start_' + plural

    if startname in templ:

        yield templ(startname, **strmapping)

    vmapping = mapping.copy()

    def one(v, tag=name):

        try:

            vmapping.update(v)

        except (AttributeError, ValueError):

            try:

                for a, b in v:

                    vmapping[a] = b

            except ValueError:

                vmapping[name] = v

        return templ(tag, **pycompat.strkwargs(vmapping))

    lastname = 'last_' + name

    if lastname in templ:

        last = values.pop()

    else:

        last = None

    for v in values:

        yield one(v)

    if last is not None:

        yield one(last, tag=lastname)

    endname = 'end_' + plural

    if endname in templ:

        yield templ(endname, **strmapping)

def _formatrevnode(ctx):

    """Format changeset as '{rev}:{node|formatnode}', which is the default

    template provided by cmdutil.changeset_templater"""

    repo = ctx.repo()

    if repo.ui.debugflag:

        hexfunc = hex

    else:

        hexfunc = short

    return '%d:%s' % (scmutil.intrev(ctx), hexfunc(scmutil.binnode(ctx)))

def getfiles(repo, ctx, revcache):

    if 'files' not in revcache:

        revcache['files'] = repo.status(ctx.p1(), ctx)[:3]

    return revcache['files']

def getlatesttags(repo, ctx, cache, pattern=None):

    '''return date, distance and name for the latest tag of rev'''

    cachename = 'latesttags'

    if pattern is not None:

        cachename += '-' + pattern

        match = util.stringmatcher(pattern)[2]

    else:

        match = util.always

    if cachename not in cache:

        # Cache mapping from rev to a tuple with tag date, tag

        # distance and tag name

        cache[cachename] = {-1: (0, 0, ['null'])}

    latesttags = cache[cachename]

    rev = ctx.rev()

    todo = [rev]

    while todo:

        rev = todo.pop()

        if rev in latesttags:

            continue

        ctx = repo[rev]

        tags = [t for t in ctx.tags()

                if (repo.tagtype(t) and repo.tagtype(t) != 'local'

                    and match(t))]

        if tags:

            latesttags[rev] = ctx.date()[0], 0, [t for t in sorted(tags)]

            continue

        try:

            # The tuples are laid out so the right one can be found by

            # comparison.

            pdate, pdist, ptag = max(

                latesttags[p.rev()] for p in ctx.parents())

        except KeyError:

            # Cache miss - recurse

            todo.append(rev)

            todo.extend(p.rev() for p in ctx.parents())

            continue

        latesttags[rev] = pdate, pdist + 1, ptag

    return latesttags[rev]

def getrenamedfn(repo, endrev=None):

    rcache = {}

    if endrev is None:

        endrev = len(repo)

    def getrenamed(fn, rev):

        '''looks up all renames for a file (up to endrev) the first

        time the file is given. It indexes on the changerev and only

        parses the manifest if linkrev != changerev.

        Returns rename info for fn at changerev rev.'''

        if fn not in rcache:

            rcache[fn] = {}

            fl = repo.file(fn)

            for i in fl:

                lr = fl.linkrev(i)

                renamed = fl.renamed(fl.node(i))

                rcache[fn][lr] = renamed

                if lr >= endrev:

                    break

        if rev in rcache[fn]:

            return rcache[fn][rev]

        # If linkrev != rev (i.e. rev not found in rcache) fallback to

        # filectx logic.

        try:

            return repo[rev][fn].renamed()

        except error.LookupError:

            return None

    return getrenamed

# default templates internally used for rendering of lists

defaulttempl = {

    'parent': '{rev}:{node|formatnode} ',

    'manifest': '{rev}:{node|formatnode}',

    'file_copy': '{name} ({source})',

    'envvar': '{key}={value}',

    'extra': '{key}={value|stringescape}'

}

# filecopy is preserved for compatibility reasons

defaulttempl['filecopy'] = defaulttempl['file_copy']

# keywords are callables like:

# fn(repo, ctx, templ, cache, revcache, **args)

# with:

# repo - current repository instance

# ctx - the changectx being displayed

# templ - the templater instance

# cache - a cache dictionary for the whole templater run

# revcache - a cache dictionary for the current revision

keywords = {}

templatekeyword = registrar.templatekeyword(keywords)

@templatekeyword('author')

def showauthor(repo, ctx, templ, **args):

    """String. The unmodified author of the changeset."""

    return ctx.user()

@templatekeyword('bisect')

def showbisect(repo, ctx, templ, **args):

    """String. The changeset bisection status."""

    return hbisect.label(repo, ctx.node())

@templatekeyword('branch')

def showbranch(**args):

    """String. The name of the branch on which the changeset was

    committed.

    """

    return args[r'ctx'].branch()

@templatekeyword('branches')

def showbranches(**args):

    """List of strings. The name of the branch on which the

    changeset was committed. Will be empty if the branch name was

    default. (DEPRECATED)

    """

    args = pycompat.byteskwargs(args)

    branch = args['ctx'].branch()

    if branch != 'default':

        return showlist('branch', [branch], args, plural='branches')

    return showlist('branch', [], args, plural='branches')

@templatekeyword('bookmarks')

def showbookmarks(**args):

    """List of strings. Any bookmarks associated with the

    changeset. Also sets 'active', the name of the active bookmark.

    """

    args = pycompat.byteskwargs(args)

    repo = args['ctx']._repo

    bookmarks = args['ctx'].bookmarks()

    active = repo._activebookmark

    makemap = lambda v: {'bookmark': v, 'active': active, 'current': active}

    f = _showlist('bookmark', bookmarks, args)

    return _hybrid(f, bookmarks, makemap, lambda x: x['bookmark'])

@templatekeyword('children')

def showchildren(**args):

    """List of strings. The children of the changeset."""

    args = pycompat.byteskwargs(args)

    ctx = args['ctx']

    childrevs = ['%d:%s' % (cctx, cctx) for cctx in ctx.children()]

    return showlist('children', childrevs, args, element='child')

# Deprecated, but kept alive for help generation a purpose.

@templatekeyword('currentbookmark')

def showcurrentbookmark(**args):

    """String. The active bookmark, if it is

    associated with the changeset (DEPRECATED)"""

    return showactivebookmark(**args)

@templatekeyword('activebookmark')

def showactivebookmark(**args):

    """String. The active bookmark, if it is

    associated with the changeset"""

    active = args[r'repo']._activebookmark

    if active and active in args[r'ctx'].bookmarks():

        return active

    return ''

@templatekeyword('date')

def showdate(repo, ctx, templ, **args):

    """Date information. The date when the changeset was committed."""

    return ctx.date()

@templatekeyword('desc')

def showdescription(repo, ctx, templ, **args):

    """String. The text of the changeset description."""

    s = ctx.description()

    if isinstance(s, encoding.localstr):

        # try hard to preserve utf-8 bytes

        return encoding.tolocal(encoding.fromlocal(s).strip())

    else:

        return s.strip()

@templatekeyword('diffstat')

def showdiffstat(repo, ctx, templ, **args):

    """String. Statistics of changes with the following format:

    "modified files: +added/-removed lines"

    """

    stats = patch.diffstatdata(util.iterlines(ctx.diff(noprefix=False)))

    maxname, maxtotal, adds, removes, binary = patch.diffstatsum(stats)

    return '%s: +%s/-%s' % (len(stats), adds, removes)

@templatekeyword('envvars')

def showenvvars(repo, **args):

    """A dictionary of environment variables. (EXPERIMENTAL)"""

    args = pycompat.byteskwargs(args)

    env = repo.ui.exportableenviron()

    env = util.sortdict((k, env[k]) for k in sorted(env))