comparison hgext/graphlog.py @ 7716:4ad12930a459

graphlog: extract large parts of repeated code from incoming/outgoing

7715:fd3266287b40

    if path: # could be reset in canonpath
        revdag = filerevs(repo, path, start, stop)
    else:
        revdag = revisions(repo, start, stop)

    repo_parents = repo.dirstate.parents()
    displayer = show_changeset(ui, repo, opts, buffered=True)
    def graphabledag():
        for (ctx, parents) in revdag:
            # log_strings is the list of all log strings to draw alongside
            # the graph.
            displayer.show(ctx)
            lines = displayer.hunk.pop(ctx.rev()).split("\n")[:-1]
            char = ctx.node() in repo_parents and '@' or 'o'
            yield (ctx.rev(), parents, char, lines)

    ascii(ui, grapher(graphabledag()))

def outgoing_revs(ui, repo, dest, opts):
    """cset DAG generator yielding (node, [parents]) tuples

    This generator function walks through the revisions not found
    in the destination
    """
    limit = cmdutil.loglimit(opts)
    dest, revs, checkout = hg.parseurl(
        ui.expandpath(dest or 'default-push', dest or 'default'),
        opts.get('rev'))
    cmdutil.setremoteconfig(ui, opts)
    if revs:

    ui.status(_('comparing with %s\n') % url.hidepassword(dest))
    o = repo.findoutgoing(other, force=opts.get('force'))
    if not o:
        ui.status(_("no changes found\n"))
        return
    o = repo.changelog.nodesbetween(o, revs)[0]
    o.reverse()
    revdict = {}
    for n in o:
        revdict[repo.changectx(n).rev()]=True
    count = 0
    for n in o:
        if count >= limit:
            break
        ctx = repo.changectx(n)
        parents = [p.rev() for p in ctx.parents() if p.rev() in revdict]
        parents.sort()
        yield (ctx, parents)
        count += 1

def goutgoing(ui, repo, dest=None, **opts):
    """show the outgoing changesets alongside an ASCII revision graph

    Print the outgoing changesets alongside a revision graph drawn with
    ASCII characters.

    Nodes printed as an @ character are parents of the working
    directory.
    """
    check_unsupported_flags(opts)
    revdag = outgoing_revs(ui, repo, dest, opts)
    repo_parents = repo.dirstate.parents()
    displayer = show_changeset(ui, repo, opts, buffered=True)
    def graphabledag():
        for (ctx, parents) in revdag:
            # log_strings is the list of all log strings to draw alongside
            # the graph.
            displayer.show(ctx)
            lines = displayer.hunk.pop(ctx.rev()).split("\n")[:-1]
            char = ctx.node() in repo_parents and '@' or 'o'
            yield (ctx.rev(), parents, char, lines)

    ascii(ui, grapher(graphabledag()))

def incoming_revs(other, chlist, opts):
    """cset DAG generator yielding (node, [parents]) tuples

    This generator function walks through the revisions of the destination
    not found in repo
    """
    limit = cmdutil.loglimit(opts)
    chlist.reverse()
    revdict = {}
    for n in chlist:
        revdict[other.changectx(n).rev()]=True
    count = 0
    for n in chlist:
        if count >= limit:
            break
        ctx = other.changectx(n)
        parents = [p.rev() for p in ctx.parents() if p.rev() in revdict]
        parents.sort()
        yield (ctx, parents)
        count += 1

def gincoming(ui, repo, source="default", **opts):
    """show the incoming changesets alongside an ASCII revision graph

    Print the incoming changesets alongside a revision graph drawn with

        ui.status(_("no changes found\n"))
        return

    cleanup = None
    try:
        fname = opts["bundle"]
        if fname or not other.local():
            # create a bundle (uncompressed if other repo is not local)
            if revs is None:
                cg = other.changegroup(incoming, "incoming")

            if not other.local():
                # use the created uncompressed bundlerepo
                other = bundlerepo.bundlerepository(ui, repo.root, fname)

        chlist = other.changelog.nodesbetween(incoming, revs)[0]
        revdag = incoming_revs(other, chlist, opts)
        other_parents = []
        displayer = show_changeset(ui, other, opts, buffered=True)
        def graphabledag():
            for (ctx, parents) in revdag:
                # log_strings is the list of all log strings to draw alongside
                # the graph.
                displayer.show(ctx)
                lines = displayer.hunk.pop(ctx.rev()).split("\n")[:-1]
                char = ctx.node() in other_parents and '@' or 'o'
                yield (ctx.rev(), parents, char, lines)

        ascii(ui, grapher(graphabledag()))
    finally:
        if hasattr(other, 'close'):
            other.close()
        if cleanup:
            os.unlink(cleanup)

7716:4ad12930a459

    if path: # could be reset in canonpath
        revdag = filerevs(repo, path, start, stop)
    else:
        revdag = revisions(repo, start, stop)

    graphdag = graphabledag(ui, repo, revdag, opts)
    ascii(ui, grapher(graphdag))

def graphrevs(repo, nodes, opts):
    nodes.reverse()
    include = util.set(nodes)
    limit = cmdutil.loglimit(opts)
    count = 0
    for node in nodes:
        if count >= limit:
            break
        ctx = repo[node]
        parents = [p.rev() for p in ctx.parents() if p.node() in include]
        parents.sort()
        yield (ctx, parents)
        count += 1

def graphabledag(ui, repo, revdag, opts):
    showparents = [ctx.node() for ctx in repo[None].parents()]
    displayer = show_changeset(ui, repo, opts, buffered=True)
    for (ctx, parents) in revdag:
        displayer.show(ctx)
        lines = displayer.hunk.pop(ctx.rev()).split('\n')[:-1]
        char = ctx.node() in showparents and '@' or 'o'
        yield (ctx.rev(), parents, char, lines)

def goutgoing(ui, repo, dest=None, **opts):
    """show the outgoing changesets alongside an ASCII revision graph

    Print the outgoing changesets alongside a revision graph drawn with
    ASCII characters.

    Nodes printed as an @ character are parents of the working
    directory.
    """

    check_unsupported_flags(opts)
    dest, revs, checkout = hg.parseurl(
        ui.expandpath(dest or 'default-push', dest or 'default'),
        opts.get('rev'))
    cmdutil.setremoteconfig(ui, opts)
    if revs:

    ui.status(_('comparing with %s\n') % url.hidepassword(dest))
    o = repo.findoutgoing(other, force=opts.get('force'))
    if not o:
        ui.status(_("no changes found\n"))
        return

    o = repo.changelog.nodesbetween(o, revs)[0]
    revdag = graphrevs(repo, o, opts)
    graphdag = graphabledag(ui, repo, revdag, opts)
    ascii(ui, grapher(graphdag))

def gincoming(ui, repo, source="default", **opts):
    """show the incoming changesets alongside an ASCII revision graph

    Print the incoming changesets alongside a revision graph drawn with

        ui.status(_("no changes found\n"))
        return

    cleanup = None
    try:

        fname = opts["bundle"]
        if fname or not other.local():
            # create a bundle (uncompressed if other repo is not local)
            if revs is None:
                cg = other.changegroup(incoming, "incoming")

            if not other.local():
                # use the created uncompressed bundlerepo
                other = bundlerepo.bundlerepository(ui, repo.root, fname)

        chlist = other.changelog.nodesbetween(incoming, revs)[0]
        revdag = graphrevs(other, chlist, opts)
        other_parents = []
        displayer = show_changeset(ui, other, opts, buffered=True)
        graphdag = graphabledag(ui, repo, revdag, opts)
        ascii(ui, grapher(graphdag))

    finally:
        if hasattr(other, 'close'):
            other.close()
        if cleanup:
            os.unlink(cleanup)