comparison mercurial/patch.py @ 24106:9cf9432a505b

trydiff: extract function that generates filename pairs

The code that identifies copies/renames, as well as the filenames
before and after, is now isolated and we can extract it to a function
so it can be overridden by extensions (in particular the narrow clone
extension).

24105:0f8baebcdbea

def diffui(*args, **kw):
    '''like diff(), but yields 2-tuples of (output, label) for ui.write()'''
    return difflabel(diff, *args, **kw)

def trydiff(repo, revs, ctx1, ctx2, modified, added, removed,
            copy, getfilectx, opts, losedatafn, prefix):

    def gitindex(text):
        if not text:
            text = ""
        l = len(text)
        s = util.sha1('blob %d\0' % l)
        s.update(text)
        return s.hexdigest()

    if opts.noprefix:
        aprefix = bprefix = ''
    else:
        aprefix = 'a/'
        bprefix = 'b/'

    def diffline(f, revs):
        revinfo = ' '.join(["-r %s" % rev for rev in revs])
        return 'diff %s %s' % (revinfo, f)

    date1 = util.datestr(ctx1.date())
    date2 = util.datestr(ctx2.date())

    gone = set()
    gitmode = {'l': '120000', 'x': '100755', '': '100644'}

    copyto = dict([(v, k) for k, v in copy.items()])

    addedset, removedset = set(added), set(removed)
    # Fix up  added, since merged-in additions appear as
    # modifications during merges
    for f in modified:
        if f not in ctx1:
            addedset.add(f)
    for f in sorted(modified + added + removed):
        copyop = None
        f1, f2 = f, f
        if f in addedset:
            f1 = None

            if opts.git:
                # have we already reported a copy above?
                if (f in copyto and copyto[f] in addedset
                    and copy[copyto[f]] == f):
                    continue

        content1 = None
        content2 = None
        if f1:
            content1 = getfilectx(f1, ctx1).data()
        if f2:

            binary = util.binary(content1) or util.binary(content2)

        if losedatafn and not opts.git:
            if (binary or
                # copy/rename
                f in copy or
                # empty file creation
                (not f1 and not content2) or
                # empty file deletion
                (not content1 and not f2) or
                # create with flags
                (not f1 and flag2) or
                # change flags
                (f1 and f2 and flag1 != flag2)):
                losedatafn(f)

        path1 = posixpath.join(prefix, f1 or f2)
        path2 = posixpath.join(prefix, f2 or f1)
        header = []
        if opts.git:

24106:9cf9432a505b

def diffui(*args, **kw):
    '''like diff(), but yields 2-tuples of (output, label) for ui.write()'''
    return difflabel(diff, *args, **kw)

def _filepairs(ctx1, modified, added, removed, copy, opts):
    '''generates tuples (f1, f2, copyop), where f1 is the name of the file
    before and f2 is the the name after. For added files, f1 will be None,
    and for removed files, f2 will be None. copyop may be set to None, 'copy'
    or 'rename' (the latter two only if opts.git is set).'''
    gone = set()

    copyto = dict([(v, k) for k, v in copy.items()])

    addedset, removedset = set(added), set(removed)
    # Fix up  added, since merged-in additions appear as
    # modifications during merges
    for f in modified:
        if f not in ctx1:
            addedset.add(f)

    for f in sorted(modified + added + removed):
        copyop = None
        f1, f2 = f, f
        if f in addedset:
            f1 = None

            if opts.git:
                # have we already reported a copy above?
                if (f in copyto and copyto[f] in addedset
                    and copy[copyto[f]] == f):
                    continue
        yield f1, f2, copyop

def trydiff(repo, revs, ctx1, ctx2, modified, added, removed,
            copy, getfilectx, opts, losedatafn, prefix):

    def gitindex(text):
        if not text:
            text = ""
        l = len(text)
        s = util.sha1('blob %d\0' % l)
        s.update(text)
        return s.hexdigest()

    if opts.noprefix:
        aprefix = bprefix = ''
    else:
        aprefix = 'a/'
        bprefix = 'b/'

    def diffline(f, revs):
        revinfo = ' '.join(["-r %s" % rev for rev in revs])
        return 'diff %s %s' % (revinfo, f)

    date1 = util.datestr(ctx1.date())
    date2 = util.datestr(ctx2.date())

    gitmode = {'l': '120000', 'x': '100755', '': '100644'}

    for f1, f2, copyop in _filepairs(
            ctx1, modified, added, removed, copy, opts):
        content1 = None
        content2 = None
        if f1:
            content1 = getfilectx(f1, ctx1).data()
        if f2:

            binary = util.binary(content1) or util.binary(content2)

        if losedatafn and not opts.git:
            if (binary or
                # copy/rename
                f2 in copy or
                # empty file creation
                (not f1 and not content2) or
                # empty file deletion
                (not content1 and not f2) or
                # create with flags
                (not f1 and flag2) or
                # change flags
                (f1 and f2 and flag1 != flag2)):
                losedatafn(f2 or f1)

        path1 = posixpath.join(prefix, f1 or f2)
        path2 = posixpath.join(prefix, f2 or f1)
        header = []
        if opts.git: