comparison hgext/largefiles/overrides.py @ 41061:98681293c890

largefiles: port commands to exthelper

One subtle change here is that the purge, rebase and transplant extensions are
wrapped in extsetup() instead of uisetup().

41060:7250cbaabde0

from mercurial import (
    archival,
    cmdutil,
    error,
    hg,
    logcmdutil,
    match as matchmod,
    pathutil,
    pycompat,

from . import (
    lfcommands,
    lfutil,
    storefactory,
)

# -- Utility functions: commonly/repeatedly needed functionality ---------------

def composelargefilematcher(match, manifest):
    '''create a matcher that matches only the largefiles in the original

def decodepath(orig, path):
    return lfutil.splitstandin(path) or path

# -- Wrappers: modify existing commands --------------------------------

def overrideadd(orig, ui, repo, *pats, **opts):
    if opts.get(r'normal') and opts.get(r'large'):
        raise error.Abort(_('--normal cannot be used with --large'))
    return orig(ui, repo, *pats, **opts)

        repo._repo.lfstatus = True
        return orig(repo, rev2, **opts)
    finally:
        repo._repo.lfstatus = False

def overridestatus(orig, ui, repo, *pats, **opts):
    try:
        repo.lfstatus = True
        return orig(ui, repo, *pats, **opts)
    finally:

        repo._repo.lfstatus = True
        return orig(repo, ignoreupdate=ignoreupdate, missing=missing)
    finally:
        repo._repo.lfstatus = False

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.

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

def overrideverify(orig, ui, repo, *pats, **opts):
    large = opts.pop(r'large', False)
    all = opts.pop(r'lfa', False)
    contents = opts.pop(r'lfc', False)

    result = orig(ui, repo, *pats, **opts)
    if large or all or contents:
        result = result or lfcommands.verifylfiles(ui, repo, all, contents)
    return result

def overridedebugstate(orig, ui, repo, *pats, **opts):
    large = opts.pop(r'large', False)
    if large:
        class fakerepo(object):
            dirstate = lfutil.openlfdirstate(ui, repo)

        lfcommands.updatelfiles(ui, repo, filelist, printmessage=False,
                                normallookup=True)

# after pulling changesets, we need to take some extra care to get
# largefiles updated remotely
def overridepull(orig, ui, repo, source=None, **opts):
    revsprepull = len(repo)
    if not source:
        source = 'default'
    repo.lfpullsource = source

        finally:
            del repo.firstpulled
        ui.status(_("%d largefiles cached\n") % numcached)
    return result

def overridepush(orig, ui, repo, *args, **kwargs):
    """Override push command and store --lfrev parameters in opargs"""
    lfrevs = kwargs.pop(r'lfrev', None)
    if lfrevs:
        opargs = kwargs.setdefault(r'opargs', {})

        firstpulled = repo.firstpulled
    except AttributeError:
        raise error.Abort(_("pulled() only available in --lfrev"))
    return smartset.baseset([r for r in subset if r >= firstpulled])

def overrideclone(orig, ui, source, dest=None, **opts):
    d = dest
    if d is None:
        d = hg.defaultdest(source)
    if opts.get(r'all_largefiles') and not hg.islocal(d):

            if missing != 0:
                return None

    return result

def overriderebase(orig, ui, repo, **opts):
    if not util.safehasattr(repo, '_largefilesenabled'):
        return orig(ui, repo, **opts)

    resuming = opts.get(r'continue')

        return orig(ui, repo, **opts)
    finally:
        repo._lfstatuswriters.pop()
        repo._lfcommithooks.pop()

def overridearchivecmd(orig, ui, repo, dest, **opts):
    repo.unfiltered().lfstatus = True

    try:
        return orig(ui, repo.unfiltered(), dest, **opts)

            for file in sorted(toupload):
                ui.status(lfutil.splitstandin(file) + '\n')
                showhashes(file)
            ui.status('\n')

def summaryremotehook(ui, repo, opts, changes):
    largeopt = opts.get('large', False)
    if changes is None:
        if largeopt:
            return (False, True) # only outgoing check is needed

        else:
            # i18n: column positioning for "hg summary"
            ui.status(_('largefiles: %d entities for %d files to upload\n')
                      % (len(lfhashes), len(toupload)))

def overridesummary(orig, ui, repo, *pats, **opts):
    try:
        repo.lfstatus = True
        orig(ui, repo, *pats, **opts)
    finally:

    matcher = composenormalfilematcher(matcher, repo[None].manifest(), added)
    return orig(repo, matcher, prefix, opts)

# Calling purge with --all will cause the largefiles to be deleted.
# Override repo.status to prevent this from happening.
def overridepurge(orig, ui, repo, *dirs, **opts):
    # XXX Monkey patching a repoview will not work. The assigned attribute will
    # be set on the unfiltered repo, but we will only lookup attributes in the
    # unfiltered repo if the lookup in the repoview object itself fails. As the
    # monkey patched method exists on the repoview class the lookup will not

                              unknown, ignored, r.clean)
    repo.status = overridestatus
    orig(ui, repo, *dirs, **opts)
    repo.status = oldstatus

def overriderollback(orig, ui, repo, **opts):
    with repo.wlock():
        before = repo.dirstate.parents()
        orphans = set(f for f in repo.dirstate
                      if lfutil.isstandin(f) and repo.dirstate[f] != 'r')

        for lfile in orphans:
            lfdirstate.drop(lfile)
        lfdirstate.write()
    return result

def overridetransplant(orig, ui, repo, *revs, **opts):
    resuming = opts.get(r'continue')
    repo._lfcommithooks.append(lfutil.automatedcommithook(resuming))
    repo._lfstatuswriters.append(lambda *msg, **opts: None)
    try:

    finally:
        repo._lfstatuswriters.pop()
        repo._lfcommithooks.pop()
    return result

def overridecat(orig, ui, repo, file1, *pats, **opts):
    opts = pycompat.byteskwargs(opts)
    ctx = scmutil.revsingle(repo, opts.get('rev'))
    err = 1
    notbad = set()

41061:98681293c890

from mercurial import (
    archival,
    cmdutil,
    error,
    exthelper,
    hg,
    logcmdutil,
    match as matchmod,
    pathutil,
    pycompat,

from . import (
    lfcommands,
    lfutil,
    storefactory,
)

eh = exthelper.exthelper()

# -- Utility functions: commonly/repeatedly needed functionality ---------------

def composelargefilematcher(match, manifest):
    '''create a matcher that matches only the largefiles in the original

def decodepath(orig, path):
    return lfutil.splitstandin(path) or path

# -- Wrappers: modify existing commands --------------------------------

@eh.wrapcommand('add',
    opts=[('', 'large', None, _('add as largefile')),
          ('', 'normal', None, _('add as normal file')),
          ('', 'lfsize', '', _('add all files above this size (in megabytes) '
                               'as largefiles (default: 10)'))])
def overrideadd(orig, ui, repo, *pats, **opts):
    if opts.get(r'normal') and opts.get(r'large'):
        raise error.Abort(_('--normal cannot be used with --large'))
    return orig(ui, repo, *pats, **opts)

        repo._repo.lfstatus = True
        return orig(repo, rev2, **opts)
    finally:
        repo._repo.lfstatus = False

@eh.wrapcommand('status')
def overridestatus(orig, ui, repo, *pats, **opts):
    try:
        repo.lfstatus = True
        return orig(ui, repo, *pats, **opts)
    finally:

        repo._repo.lfstatus = True
        return orig(repo, ignoreupdate=ignoreupdate, missing=missing)
    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.

        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,
                _('verify largefiles in all revisions, not just current')),
          ('', 'lfc', None,
                _('verify local largefile contents, not just existence'))])
def overrideverify(orig, ui, repo, *pats, **opts):
    large = opts.pop(r'large', False)
    all = opts.pop(r'lfa', False)
    contents = opts.pop(r'lfc', False)

    result = orig(ui, repo, *pats, **opts)
    if large or all or contents:
        result = result or lfcommands.verifylfiles(ui, repo, all, contents)
    return result

@eh.wrapcommand('debugstate',
    opts=[('', 'large', None, _('display largefiles dirstate'))])
def overridedebugstate(orig, ui, repo, *pats, **opts):
    large = opts.pop(r'large', False)
    if large:
        class fakerepo(object):
            dirstate = lfutil.openlfdirstate(ui, repo)

        lfcommands.updatelfiles(ui, repo, filelist, printmessage=False,
                                normallookup=True)

# after pulling changesets, we need to take some extra care to get
# largefiles updated remotely
@eh.wrapcommand('pull',
    opts=[('', 'all-largefiles', None,
                _('download all pulled versions of largefiles (DEPRECATED)')),
          ('', 'lfrev', [],
                _('download largefiles for these revisions'), _('REV'))])
def overridepull(orig, ui, repo, source=None, **opts):
    revsprepull = len(repo)
    if not source:
        source = 'default'
    repo.lfpullsource = source

        finally:
            del repo.firstpulled
        ui.status(_("%d largefiles cached\n") % numcached)
    return result

@eh.wrapcommand('push',
    opts=[('', 'lfrev', [],
               _('upload largefiles for these revisions'), _('REV'))])
def overridepush(orig, ui, repo, *args, **kwargs):
    """Override push command and store --lfrev parameters in opargs"""
    lfrevs = kwargs.pop(r'lfrev', None)
    if lfrevs:
        opargs = kwargs.setdefault(r'opargs', {})

        firstpulled = repo.firstpulled
    except AttributeError:
        raise error.Abort(_("pulled() only available in --lfrev"))
    return smartset.baseset([r for r in subset if r >= firstpulled])

@eh.wrapcommand('clone',
    opts=[('', 'all-largefiles', None,
               _('download all versions of all largefiles'))])
def overrideclone(orig, ui, source, dest=None, **opts):
    d = dest
    if d is None:
        d = hg.defaultdest(source)
    if opts.get(r'all_largefiles') and not hg.islocal(d):

            if missing != 0:
                return None

    return result

@eh.wrapcommand('rebase', extension='rebase')
def overriderebase(orig, ui, repo, **opts):
    if not util.safehasattr(repo, '_largefilesenabled'):
        return orig(ui, repo, **opts)

    resuming = opts.get(r'continue')

        return orig(ui, repo, **opts)
    finally:
        repo._lfstatuswriters.pop()
        repo._lfcommithooks.pop()

@eh.wrapcommand('archive')
def overridearchivecmd(orig, ui, repo, dest, **opts):
    repo.unfiltered().lfstatus = True

    try:
        return orig(ui, repo.unfiltered(), dest, **opts)

            for file in sorted(toupload):
                ui.status(lfutil.splitstandin(file) + '\n')
                showhashes(file)
            ui.status('\n')

@eh.wrapcommand('outgoing',
    opts=[('', 'large', None, _('display outgoing largefiles'))])
def _outgoingcmd(orig, *args, **kwargs):
    # Nothing to do here other than add the extra help option- the hook above
    # processes it.
    return orig(*args, **kwargs)

def summaryremotehook(ui, repo, opts, changes):
    largeopt = opts.get('large', False)
    if changes is None:
        if largeopt:
            return (False, True) # only outgoing check is needed

        else:
            # i18n: column positioning for "hg summary"
            ui.status(_('largefiles: %d entities for %d files to upload\n')
                      % (len(lfhashes), len(toupload)))

@eh.wrapcommand('summary',
    opts=[('', 'large', None, _('display outgoing largefiles'))])
def overridesummary(orig, ui, repo, *pats, **opts):
    try:
        repo.lfstatus = True
        orig(ui, repo, *pats, **opts)
    finally:

    matcher = composenormalfilematcher(matcher, repo[None].manifest(), added)
    return orig(repo, matcher, prefix, opts)

# Calling purge with --all will cause the largefiles to be deleted.
# Override repo.status to prevent this from happening.
@eh.wrapcommand('purge', extension='purge')
def overridepurge(orig, ui, repo, *dirs, **opts):
    # XXX Monkey patching a repoview will not work. The assigned attribute will
    # be set on the unfiltered repo, but we will only lookup attributes in the
    # unfiltered repo if the lookup in the repoview object itself fails. As the
    # monkey patched method exists on the repoview class the lookup will not

                              unknown, ignored, r.clean)
    repo.status = overridestatus
    orig(ui, repo, *dirs, **opts)
    repo.status = oldstatus

@eh.wrapcommand('rollback')
def overriderollback(orig, ui, repo, **opts):
    with repo.wlock():
        before = repo.dirstate.parents()
        orphans = set(f for f in repo.dirstate
                      if lfutil.isstandin(f) and repo.dirstate[f] != 'r')

        for lfile in orphans:
            lfdirstate.drop(lfile)
        lfdirstate.write()
    return result

@eh.wrapcommand('transplant', extension='transplant')
def overridetransplant(orig, ui, repo, *revs, **opts):
    resuming = opts.get(r'continue')
    repo._lfcommithooks.append(lfutil.automatedcommithook(resuming))
    repo._lfstatuswriters.append(lambda *msg, **opts: None)
    try:

    finally:
        repo._lfstatuswriters.pop()
        repo._lfcommithooks.pop()
    return result

@eh.wrapcommand('cat')
def overridecat(orig, ui, repo, file1, *pats, **opts):
    opts = pycompat.byteskwargs(opts)
    ctx = scmutil.revsingle(repo, opts.get('rev'))
    err = 1
    notbad = set()