comparison mercurial/formatter.py @ 37073:44757e6dad93

templater: introduce resourcemapper class

A couple more functions will be added later to work around nested mapping
bugs such as the issue 5612.

37072:d64ae4fef471

                   for k, v in ui.configitems('templates'))
    if tmpl:
        t.cache[''] = tmpl
    return t

def templateresources(ui, repo=None):
    """Create a dict of template resources designed for the default templatekw
    and function"""
    resmap = {
        'cache': {},  # for templatekw/funcs to store reusable data
        'repo': repo,
        'ui': ui,
    }

    def getsome(context, mapping, key):
        v = mapping.get(key)
        if v is not None:
            return v
        return resmap.get(key)

    def getctx(context, mapping, key):
        ctx = mapping.get('ctx')
        if ctx is not None:
            return ctx
        fctx = mapping.get('fctx')
        if fctx is not None:
            return fctx.changectx()

    def getrepo(context, mapping, key):
        ctx = getctx(context, mapping, 'ctx')
        if ctx is not None:
            return ctx.repo()
        return getsome(context, mapping, key)

    return {
        'cache': getsome,
        'ctx': getctx,
        'fctx': getsome,
        'repo': getrepo,
        'revcache': getsome,  # per-ctx cache; set later
        'ui': getsome,
    }

def formatter(ui, out, topic, opts):
    template = opts.get("template", "")
    if template == "json":
        return jsonformatter(ui, out, topic, opts)

37073:44757e6dad93

                   for k, v in ui.configitems('templates'))
    if tmpl:
        t.cache[''] = tmpl
    return t

class templateresources(templater.resourcemapper):
    """Resource mapper designed for the default templatekw and function"""

    def __init__(self, ui, repo=None):
        self._resmap = {
            'cache': {},  # for templatekw/funcs to store reusable data
            'repo': repo,
            'ui': ui,
        }

    def knownkeys(self):
        return self._knownkeys

    def lookup(self, context, mapping, key):
        get = self._gettermap.get(key)
        if not get:
            return None
        return get(self, context, mapping, key)

    def _getsome(self, context, mapping, key):
        v = mapping.get(key)
        if v is not None:
            return v
        return self._resmap.get(key)

    def _getctx(self, context, mapping, key):
        ctx = mapping.get('ctx')
        if ctx is not None:
            return ctx
        fctx = mapping.get('fctx')
        if fctx is not None:
            return fctx.changectx()

    def _getrepo(self, context, mapping, key):
        ctx = self._getctx(context, mapping, 'ctx')
        if ctx is not None:
            return ctx.repo()
        return self._getsome(context, mapping, key)

    _gettermap = {
        'cache': _getsome,
        'ctx': _getctx,
        'fctx': _getsome,
        'repo': _getrepo,
        'revcache': _getsome,  # per-ctx cache; set later
        'ui': _getsome,
    }
    _knownkeys = set(_gettermap.keys())

def formatter(ui, out, topic, opts):
    template = opts.get("template", "")
    if template == "json":
        return jsonformatter(ui, out, topic, opts)