comparison hgext/keyword.py @ 7216:292fb2ad2846

extensions: use new wrapper functions

7215:0ab5f21c390b

Expansions spanning more than one line and incremental expansions,
like CVS' $Log$, are not supported. A keyword template map
"Log = {desc}" expands to the first line of the changeset description.
'''

from mercurial import commands, cmdutil, dispatch, filelog, revlog
from mercurial import patch, localrepo, templater, templatefilters, util
from mercurial.hgweb import webcommands
from mercurial.node import nullid, hex
from mercurial.i18n import _
import re, shutil, tempfile, time

            kwtools['inc'].append(pat)
        else:
            kwtools['exc'].append(pat)

    if kwtools['inc']:
        def kwdispatch_parse(ui, args):
            '''Monkeypatch dispatch._parse to obtain running hg command.'''
            cmd, func, args, options, cmdoptions = dispatch_parse(ui, args)
            kwtools['hgcmd'] = cmd
            return cmd, func, args, options, cmdoptions

        dispatch_parse = dispatch._parse
        dispatch._parse = kwdispatch_parse

def reposetup(ui, repo):
    '''Sets up repo as kwrepo for keyword substitution.
    Overrides file method to return kwfilelog instead of filelog
    if file matches user configuration.

                return n
            finally:
                del wlock, lock

    # monkeypatches
    def kwpatchfile_init(self, ui, fname, missing=False):
        '''Monkeypatch/wrap patch.patchfile.__init__ to avoid
        rejects or conflicts due to expanded keywords in working dir.'''
        patchfile_init(self, ui, fname, missing)
        # shrink keywords read from working dir
        self.lines = kwt.shrinklines(self.fname, self.lines)

    def kw_diff(repo, node1=None, node2=None, match=None,
                fp=None, changes=None, opts=None):
        '''Monkeypatch patch.diff to avoid expansion except when
        comparing against working dir.'''
        if node2 is not None:
            kwt.matcher = util.never
        elif node1 is not None and node1 != repo['.'].node():
            kwt.restrict = True
        patch_diff(repo, node1, node2, match, fp, changes, opts)

    def kwweb_annotate(web, req, tmpl):
        '''Wraps webcommands.annotate turning off keyword expansion.'''
        kwt.matcher = util.never
        return webcommands_annotate(web, req, tmpl)

    def kwweb_changeset(web, req, tmpl):
        '''Wraps webcommands.changeset turning off keyword expansion.'''
        kwt.matcher = util.never
        return webcommands_changeset(web, req, tmpl)

    def kwweb_filediff(web, req, tmpl):
        '''Wraps webcommands.filediff turning off keyword expansion.'''
        kwt.matcher = util.never
        return webcommands_filediff(web, req, tmpl)

    repo.__class__ = kwrepo

    patchfile_init = patch.patchfile.__init__
    patch_diff = patch.diff
    webcommands_annotate = webcommands.annotate
    webcommands_changeset = webcommands.changeset
    webcommands_filediff = webcommands.filediff

    patch.patchfile.__init__ = kwpatchfile_init
    patch.diff = kw_diff
    webcommands.annotate = kwweb_annotate
    webcommands.changeset = webcommands.rev = kwweb_changeset
    webcommands.filediff = webcommands.diff = kwweb_filediff


cmdtable = {
    'kwdemo':
        (demo,
         [('d', 'default', None, _('show default keyword template maps')),

7216:292fb2ad2846

Expansions spanning more than one line and incremental expansions,
like CVS' $Log$, are not supported. A keyword template map
"Log = {desc}" expands to the first line of the changeset description.
'''

from mercurial import commands, cmdutil, dispatch, filelog, revlog, extensions
from mercurial import patch, localrepo, templater, templatefilters, util
from mercurial.hgweb import webcommands
from mercurial.node import nullid, hex
from mercurial.i18n import _
import re, shutil, tempfile, time

            kwtools['inc'].append(pat)
        else:
            kwtools['exc'].append(pat)

    if kwtools['inc']:
        def kwdispatch_parse(orig, ui, args):
            '''Monkeypatch dispatch._parse to obtain running hg command.'''
            cmd, func, args, options, cmdoptions = orig(ui, args)
            kwtools['hgcmd'] = cmd
            return cmd, func, args, options, cmdoptions

        extensions.wrapfunction(dispatch, '_parse', kwdispatch_parse)

def reposetup(ui, repo):
    '''Sets up repo as kwrepo for keyword substitution.
    Overrides file method to return kwfilelog instead of filelog
    if file matches user configuration.

                return n
            finally:
                del wlock, lock

    # monkeypatches
    def kwpatchfile_init(orig, self, ui, fname, missing=False):
        '''Monkeypatch/wrap patch.patchfile.__init__ to avoid
        rejects or conflicts due to expanded keywords in working dir.'''
        orig(self, ui, fname, missing)
        # shrink keywords read from working dir
        self.lines = kwt.shrinklines(self.fname, self.lines)

    def kw_diff(orig, repo, node1=None, node2=None, match=None,
                fp=None, changes=None, opts=None):
        '''Monkeypatch patch.diff to avoid expansion except when
        comparing against working dir.'''
        if node2 is not None:
            kwt.matcher = util.never
        elif node1 is not None and node1 != repo['.'].node():
            kwt.restrict = True
        orig(repo, node1, node2, match, fp, changes, opts)

    def kwweb_skip(orig, web, req, tmpl):
        '''Wraps webcommands.x turning off keyword expansion.'''
        kwt.matcher = util.never
        return orig(web, req, tmpl)

    repo.__class__ = kwrepo

    extensions.wrapfunction(patch.patchfile, '__init__', kwpatchfile_init)
    extensions.wrapfunction(patch, 'diff', kw_diff)
    for c in 'annotate changeset rev filediff diff'.split():
        extensions.wrapfunction(webcommands, c, kwweb_skip)

cmdtable = {
    'kwdemo':
        (demo,
         [('d', 'default', None, _('show default keyword template maps')),