hgext/largefiles/overrides.py

41591:5f827e9ce870

    archival,
    cmdutil,
    copies as copiesmod,
    error,
    exchange,
    exthelper,
    filemerge,
    hg,
    logcmdutil,
    match as matchmod,

    was called.  no-op if scmutil.match is its original function.

    Note that n calls to installmatchfn will require n calls to
    restore the original matchfn.'''
    scmutil.match = getattr(scmutil.match, 'oldmatch')

def installmatchandpatsfn(f):
    oldmatchandpats = scmutil.matchandpats
    setattr(f, 'oldmatchandpats', oldmatchandpats)
    scmutil.matchandpats = f
    return oldmatchandpats

def restorematchandpatsfn():
    '''restores scmutil.matchandpats to what it was before
    installmatchandpatsfn was called. No-op if scmutil.matchandpats
    is its original function.

    Note that n calls to installmatchandpatsfn will require n calls
    to restore the original matchfn.'''
    scmutil.matchandpats = getattr(scmutil.matchandpats, 'oldmatchandpats',
            scmutil.matchandpats)

def addlargefiles(ui, repo, isaddremove, matcher, **opts):
    large = opts.get(r'large')
    lfsize = lfutil.getminsize(
        ui, lfutil.islfilesrepo(repo), opts.get(r'lfsize'))

    finally:
        repo._repo.lfstatus = False

@eh.wrapcommand('log')
def overridelog(orig, ui, repo, *pats, **opts):
    def overridematchandpats(ctx, pats=(), opts=None, globbed=False,
            default='relpath', badfn=None):
        """Matcher that merges root directory with .hglf, suitable for log.
        It is still possible to match .hglf directly.
        For any listed files run log on the standin too.
        matchfn tries both the given filename and with .hglf stripped.
        """
        if opts is None:
            opts = {}
        matchandpats = oldmatchandpats(ctx, pats, opts, globbed, default,
                                       badfn=badfn)
        m, p = copy.copy(matchandpats)

        if m.always():
            # We want to match everything anyway, so there's no benefit trying
            # to add standins.

    # For hg log --patch, the match object is used in two different senses:
    # (1) to determine what revisions should be printed out, and
    # (2) to determine what files to print out diffs for.
    # The magic matchandpats override should be used for case (1) but not for
    # case (2).
    def overridemakefilematcher(repo, pats, opts, badfn=None):
        wctx = repo[None]
        match, pats = oldmatchandpats(wctx, pats, opts, badfn=badfn)
        return lambda ctx: match

    oldmatchandpats = installmatchandpatsfn(overridematchandpats)
    oldmakefilematcher = logcmdutil._makenofollowfilematcher
    setattr(logcmdutil, '_makenofollowfilematcher', overridemakefilematcher)

    try:
        return orig(ui, repo, *pats, **opts)
    finally:
        restorematchandpatsfn()
        setattr(logcmdutil, '_makenofollowfilematcher', oldmakefilematcher)

@eh.wrapcommand('verify',
    opts=[('', 'large', None,
                _('verify that all largefiles in current revision exists')),
          ('', 'lfa', None,

41592:8d1dc380b026

    archival,
    cmdutil,
    copies as copiesmod,
    error,
    exchange,
    extensions,
    exthelper,
    filemerge,
    hg,
    logcmdutil,
    match as matchmod,

    was called.  no-op if scmutil.match is its original function.

    Note that n calls to installmatchfn will require n calls to
    restore the original matchfn.'''
    scmutil.match = getattr(scmutil.match, 'oldmatch')

def addlargefiles(ui, repo, isaddremove, matcher, **opts):
    large = opts.get(r'large')
    lfsize = lfutil.getminsize(
        ui, lfutil.islfilesrepo(repo), opts.get(r'lfsize'))

    finally:
        repo._repo.lfstatus = False

@eh.wrapcommand('log')
def overridelog(orig, ui, repo, *pats, **opts):
    def overridematchandpats(orig, ctx, pats=(), opts=None, globbed=False,
            default='relpath', badfn=None):
        """Matcher that merges root directory with .hglf, suitable for log.
        It is still possible to match .hglf directly.
        For any listed files run log on the standin too.
        matchfn tries both the given filename and with .hglf stripped.
        """
        if opts is None:
            opts = {}
        matchandpats = orig(ctx, pats, opts, globbed, default, badfn=badfn)
        m, p = copy.copy(matchandpats)

        if m.always():
            # We want to match everything anyway, so there's no benefit trying
            # to add standins.

    # For hg log --patch, the match object is used in two different senses:
    # (1) to determine what revisions should be printed out, and
    # (2) to determine what files to print out diffs for.
    # The magic matchandpats override should be used for case (1) but not for
    # case (2).
    oldmatchandpats = scmutil.matchandpats
    def overridemakefilematcher(orig, repo, pats, opts, badfn=None):
        wctx = repo[None]
        match, pats = oldmatchandpats(wctx, pats, opts, badfn=badfn)
        return lambda ctx: match

    wrappedmatchandpats = extensions.wrappedfunction(scmutil, 'matchandpats',
                                                     overridematchandpats)
    wrappedmakefilematcher = extensions.wrappedfunction(
        logcmdutil, '_makenofollowfilematcher', overridemakefilematcher)
    with wrappedmatchandpats, wrappedmakefilematcher:
        return orig(ui, repo, *pats, **opts)

@eh.wrapcommand('verify',
    opts=[('', 'large', None,
                _('verify that all largefiles in current revision exists')),
          ('', 'lfa', None,