sapling/edenscm/hgext/amend/restack.py
Jun Wu 86e119ea37 restack: allow the callsite to override rebaseopts["rev"]
Summary: This allows the callsite to control what revs to rebase.

Reviewed By: kulshrax

Differential Revision: D15978954

fbshipit-source-id: 73174475411c3986f3a8e93a1a61ce5b857f454b
2019-06-26 11:24:14 -07:00

57 lines
2.2 KiB
Python

# restack.py - rebase to make a stack connected again
#
# Copyright 2013 Facebook, Inc.
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.
from __future__ import absolute_import
from edenscm.hgext import rebase
from edenscm.mercurial import commands, revsetlang
from edenscm.mercurial.i18n import _
def restack(ui, repo, **rebaseopts):
"""Repair a situation in which one or more commits in a stack
have been obsoleted (thereby leaving their descendants in the stack
orphaned) by finding any such commits and rebasing their descendants
onto the latest version of each respective commit.
"""
rebaseopts = rebaseopts.copy()
with repo.wlock(), repo.lock():
# Find drafts connected to the current stack via either changelog or
# obsolete graph. Note: "draft() & ::." is optimized by D441.
if not rebaseopts["rev"]:
# 1. Connect drafts via changelog
revs = list(repo.revs("(draft() & ::.)::"))
if not revs:
# "." is probably public. Check its direct children.
revs = repo.revs("draft() & children(.)")
if not revs:
ui.status(_("nothing to restack\n"))
return 1
# 2. Connect revs via obsolete graph
revs = list(repo.revs("successors(%ld)+predecessors(%ld)", revs, revs))
# 3. Connect revs via changelog again to cover missing revs
revs = list(repo.revs("(draft() & ::%ld)::", revs))
rebaseopts["rev"] = [revsetlang.formatspec("%ld", revs)]
rebaseopts["dest"] = "_destrestack(SRC)"
rebase.rebase(ui, repo, **rebaseopts)
# Ensure that we always end up on the latest version of the
# current changeset. Usually, this will be taken care of
# by the rebase operation. However, in some cases (such as
# if we are on the precursor of the base changeset) the
# rebase will not update to the latest version, so we need
# to do this manually.
successor = repo.revs("successors(.) - .").last()
if successor is not None:
commands.update(ui, repo, rev=successor)