comparison hgext/largefiles/overrides.py @ 25811:7699d3212994

largefiles: allow the archiving of largefiles to be disabled

There are currently no users of this, but it is a necessary step before
converting extdiff to use archive.  It may be useful to add an argument to
extdiff in the future and allow largefiles to be diffed, but archiving
largefiles can have significant overhead and may not be very diffable, so
archiving them by default seems wrong.

It is a mystery to me why the lfstatus attribute needs to be set on the
unfiltered repo.  However if it is set on the filtered repo instead (and the
filtered repo is passed to the original command), the lfstatus attribute is
False in the overrides for archival.archive() and hgsubrepo.archive() when
invoking the archive command.  This smells like the buggy status behavior (see
67d63ec85eb7, which was reverted in df463ca0adef).  Neither the status nor
summary commands have this weird behavior in their respective overrides.

25810:82d6a35cf432

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

def overridearchive(orig, repo, dest, node, kind, decode=True, matchfn=None,
            prefix='', mtime=None, subrepos=None):
    # No need to lock because we are only reading history and
    # largefile caches, neither of which are modified.
    if node is not None:
        lfcommands.cachelfiles(repo.ui, repo, node)

    if subrepos:
        for subpath in sorted(ctx.substate):
            sub = ctx.workingsub(subpath)
            submatch = match_.narrowmatcher(subpath, matchfn)
            sub.archive(archiver, prefix, submatch)

    archiver.done()

def hgsubrepoarchive(orig, repo, archiver, prefix, match=None):
    repo._get(repo._state + ('hg',))
    rev = repo._state[1]
    ctx = repo._repo[rev]

    if ctx.node() is not None:

        write(f, 'x' in ff and 0o755 or 0o644, 'l' in ff, getdata)

    for subpath in sorted(ctx.substate):
        sub = ctx.workingsub(subpath)
        submatch = match_.narrowmatcher(subpath, match)
        sub.archive(archiver, prefix + repo._path + '/', submatch)

# If a largefile is modified, the change is not reflected in its
# standin until a commit. cmdutil.bailifchanged() raises an exception
# if the repo has uncommitted changes. Wrap it to also check if

25811:7699d3212994

        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)
    finally:
        repo.unfiltered().lfstatus = False

def overridearchive(orig, repo, dest, node, kind, decode=True, matchfn=None,
            prefix='', mtime=None, subrepos=None):
    if not repo.lfstatus:
        return orig(repo, dest, node, kind, decode, matchfn, prefix, mtime,
                    subrepos)

    # No need to lock because we are only reading history and
    # largefile caches, neither of which are modified.
    if node is not None:
        lfcommands.cachelfiles(repo.ui, repo, node)

    if subrepos:
        for subpath in sorted(ctx.substate):
            sub = ctx.workingsub(subpath)
            submatch = match_.narrowmatcher(subpath, matchfn)
            sub._repo.lfstatus = True
            sub.archive(archiver, prefix, submatch)

    archiver.done()

def hgsubrepoarchive(orig, repo, archiver, prefix, match=None):
    if not repo._repo.lfstatus:
        return orig(repo, archiver, prefix, match)

    repo._get(repo._state + ('hg',))
    rev = repo._state[1]
    ctx = repo._repo[rev]

    if ctx.node() is not None:

        write(f, 'x' in ff and 0o755 or 0o644, 'l' in ff, getdata)

    for subpath in sorted(ctx.substate):
        sub = ctx.workingsub(subpath)
        submatch = match_.narrowmatcher(subpath, match)
        sub._repo.lfstatus = True
        sub.archive(archiver, prefix + repo._path + '/', submatch)

# If a largefile is modified, the change is not reflected in its
# standin until a commit. cmdutil.bailifchanged() raises an exception
# if the repo has uncommitted changes. Wrap it to also check if