mirror of
https://github.com/facebook/sapling.git
synced 2024-10-16 11:52:02 +03:00
9dc21f8d0b
Summary: D13853115 adds `edenscm/` to `sys.path` and code still uses `import mercurial`. That has nasty problems if both `import mercurial` and `import edenscm.mercurial` are used, because Python would think `mercurial.foo` and `edenscm.mercurial.foo` are different modules so code like `try: ... except mercurial.error.Foo: ...`, or `isinstance(x, mercurial.foo.Bar)` would fail to handle the `edenscm.mercurial` version. There are also some module-level states (ex. `extensions._extensions`) that would cause trouble if they have multiple versions in a single process. Change imports to use the `edenscm` so ideally the `mercurial` is no longer imported at all. Add checks in extensions.py to catch unexpected extensions importing modules from the old (wrong) locations when running tests. Reviewed By: phillco Differential Revision: D13868981 fbshipit-source-id: f4e2513766957fd81d85407994f7521a08e4de48
79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
# unamend.py - undo an amend operation
|
|
#
|
|
# Copyright 2016 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.mercurial import error, extensions, node as nodemod, obsolete, obsutil, registrar
|
|
from edenscm.mercurial.i18n import _
|
|
|
|
|
|
cmdtable = {}
|
|
command = registrar.command(cmdtable)
|
|
|
|
|
|
def predecessormarkers(ctx):
|
|
"""yields the obsolete markers marking the given changeset as a successor"""
|
|
for data in ctx.repo().obsstore.predecessors.get(ctx.node(), ()):
|
|
yield obsutil.marker(ctx.repo(), data)
|
|
|
|
|
|
@command("^unamend", [])
|
|
def unamend(ui, repo, **opts):
|
|
"""undo an amend operation on the current commit
|
|
|
|
This command will roll back to the previous version of a changeset,
|
|
leaving working directory in state in which it was before running
|
|
`hg amend` (e.g. files modified as part of an amend will be
|
|
marked as modified `hg status`)"""
|
|
unfi = repo.unfiltered()
|
|
|
|
# identify the commit from which to unamend
|
|
curctx = repo["."]
|
|
|
|
# identify the commit to which to unamend
|
|
markers = list(predecessormarkers(curctx))
|
|
if len(markers) != 1:
|
|
e = _("changeset must have one predecessor, found %i predecessors")
|
|
raise error.Abort(e % len(markers))
|
|
|
|
prednode = markers[0].prednode()
|
|
|
|
if extensions.enabled().get("commitcloud", False):
|
|
repo.revs("cloudremote(%s)" % nodemod.hex(prednode))
|
|
|
|
predctx = unfi[prednode]
|
|
|
|
if curctx.children():
|
|
raise error.Abort(_("cannot unamend in the middle of a stack"))
|
|
|
|
with repo.wlock(), repo.lock():
|
|
ctxbookmarks = curctx.bookmarks()
|
|
changedfiles = []
|
|
wctx = repo[None]
|
|
wm = wctx.manifest()
|
|
cm = predctx.manifest()
|
|
dirstate = repo.dirstate
|
|
diff = cm.diff(wm)
|
|
changedfiles.extend(diff.iterkeys())
|
|
|
|
tr = repo.transaction("unamend")
|
|
with dirstate.parentchange():
|
|
dirstate.rebuild(prednode, cm, changedfiles)
|
|
# we want added and removed files to be shown
|
|
# properly, not with ? and ! prefixes
|
|
for filename, data in diff.iteritems():
|
|
if data[0][0] is None:
|
|
dirstate.add(filename)
|
|
if data[1][0] is None:
|
|
dirstate.remove(filename)
|
|
changes = []
|
|
for book in ctxbookmarks:
|
|
changes.append((book, prednode))
|
|
repo._bookmarks.applychanges(repo, tr, changes)
|
|
obsolete.createmarkers(repo, [(curctx, (predctx,))])
|
|
tr.close()
|