##// END OF EJS Templates
diffhelpers: backport 9e40bc4c1bde from C implementation...
diffhelpers: backport 9e40bc4c1bde from C implementation 9e40bc4c1bde just says "harden testhunk." I don't think this would be the case, but it makes some sense to avoid negative index.

File last commit:

r37659:add12981 default
r37820:090c89a8 stable
Show More
bookmarks.py
917 lines | 30.1 KiB | text/x-python | PythonLexer
Matt Mackall
bookmarks: move basic io to core
r13350 # Mercurial bookmark support code
#
# Copyright 2008 David Soria Parra <dsp@php.net>
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.
Gregory Szorc
bookmarks: use absolute_import
r25917 from __future__ import absolute_import
import errno
Boris Feld
bookmark: add methods to binary encode and decode bookmark values...
r35258 import struct
Gregory Szorc
bookmarks: use absolute_import
r25917
from .i18n import _
from .node import (
bin,
hex,
Sean Farley
commands: move checkconflict to bookmarks module...
r32956 short,
Boris Feld
bookmark: add methods to binary encode and decode bookmark values...
r35258 wdirid,
Gregory Szorc
bookmarks: use absolute_import
r25917 )
from . import (
encoding,
liscju
bookmarks: abort 'push -B .' when no active bookmark
r29354 error,
obsutil: move 'foreground' to the new modules...
r33147 obsutil,
Pulkit Goyal
py3: fix kwargs handling for `hg bookmarks`
r33092 pycompat,
Sean Farley
commands: move checkformat to bookmarks module...
r32955 scmutil,
FUJIWARA Katsunori
bookmarks: check HG_PENDING strictly...
r31052 txnutil,
Gregory Szorc
bookmarks: use absolute_import
r25917 util,
)
Matt Mackall
bookmarks: move basic io to core
r13350
Sean Farley
commands: move activebookmarklabel to bookmarks module...
r33009 # label constants
# until 3.5, bookmarks.current was the advertised name, not
# bookmarks.active, so we must use both to avoid breaking old
# custom styles
activebookmarklabel = 'bookmarks.active bookmarks.current'
Augie Fackler
bookmarks: hoist getbkfile out of bmstore class...
r27186 def _getbkfile(repo):
"""Hook so that extensions that mess with the store can hook bm storage.
For core, this just handles wether we should see pending
bookmarks or the committed ones. Other extensions (like share)
may need to tweak this behavior further.
"""
FUJIWARA Katsunori
bookmarks: check HG_PENDING strictly...
r31052 fp, pending = txnutil.trypending(repo.root, repo.vfs, 'bookmarks')
return fp
Augie Fackler
bookmarks: hoist getbkfile out of bmstore class...
r27186
Augie Fackler
bookmarks: introduce a bmstore to manage bookmark persistence...
r17922 class bmstore(dict):
"""Storage for bookmarks.
Augie Fackler
bmstore: add handling of the active bookmark...
r27698 This object should do all bookmark-related reads and writes, so
that it's fairly simple to replace the storage underlying
bookmarks without having to clone the logic surrounding
bookmarks. This type also should manage the active bookmark, if
any.
Augie Fackler
bookmarks: introduce a bmstore to manage bookmark persistence...
r17922
This particular bmstore implementation stores bookmarks as
{hash}\s{name}\n (the same format as localtags) in
.hg/bookmarks. The mapping is stored as {name: nodeid}.
"""
Matt Mackall
bookmarks: move read methods to core
r13351
Augie Fackler
bookmarks: introduce a bmstore to manage bookmark persistence...
r17922 def __init__(self, repo):
dict.__init__(self)
self._repo = repo
bookmarks: move variable initialization earlier...
r32738 self._clean = True
self._aclean = True
bookmarks: explicitly convert to 'node' during initialization...
r32735 nm = repo.changelog.nodemap
tonode = bin # force local lookup
bookmarks: directly use base dict 'setitem'...
r32737 setitem = dict.__setitem__
Augie Fackler
bookmarks: introduce a bmstore to manage bookmark persistence...
r17922 try:
bookmarks: make sure we close the bookmark file after reading...
r32794 with _getbkfile(repo) as bkfile:
for line in bkfile:
line = line.strip()
if not line:
continue
try:
sha, refspec = line.split(' ', 1)
node = tonode(sha)
if node in nm:
refspec = encoding.tolocal(refspec)
setitem(self, refspec, node)
except (TypeError, ValueError):
# TypeError:
# - bin(...)
# ValueError:
# - node in nm, for non-20-bytes entry
# - split(...), for string without ' '
repo.ui.warn(_('malformed line in .hg/bookmarks: %r\n')
Augie Fackler
bookmarks: fix a repr in a message on Python 3...
r36586 % pycompat.bytestr(line))
Gregory Szorc
global: mass rewrite to use modern exception syntax...
r25660 except IOError as inst:
Augie Fackler
bookmarks: introduce a bmstore to manage bookmark persistence...
r17922 if inst.errno != errno.ENOENT:
raise
Augie Fackler
bmstore: add handling of the active bookmark...
r27698 self._active = _readactive(repo, self)
@property
def active(self):
return self._active
@active.setter
def active(self, mark):
if mark is not None and mark not in self:
raise AssertionError('bookmark %s does not exist!' % mark)
self._active = mark
self._aclean = False
Augie Fackler
bmstore: add basic clean-state tracking...
r27187
def __setitem__(self, *args, **kwargs):
Matt Harbison
bookmarks: drop deprecated methods (API)...
r35924 raise error.ProgrammingError("use 'bookmarks.applychanges' instead")
Boris Feld
bookmark: deprecate direct set of a bookmark value...
r33517
def _set(self, key, value):
Augie Fackler
bmstore: add basic clean-state tracking...
r27187 self._clean = False
Boris Feld
bookmark: deprecate direct set of a bookmark value...
r33517 return dict.__setitem__(self, key, value)
Augie Fackler
bmstore: add basic clean-state tracking...
r27187
def __delitem__(self, key):
Matt Harbison
bookmarks: drop deprecated methods (API)...
r35924 raise error.ProgrammingError("use 'bookmarks.applychanges' instead")
Boris Feld
bookmark: deprecate direct del of a bookmark value...
r33518
def _del(self, key):
Augie Fackler
bmstore: add basic clean-state tracking...
r27187 self._clean = False
return dict.__delitem__(self, key)
Augie Fackler
bookmarks: introduce a bmstore to manage bookmark persistence...
r17922
Boris Feld
bookmark: deprecate direct update of a bookmark value...
r35697 def update(self, *others):
Matt Harbison
bookmarks: drop deprecated methods (API)...
r35924 raise error.ProgrammingError("use 'bookmarks.applychanges' instead")
Boris Feld
bookmark: deprecate direct update of a bookmark value...
r35697
Martin von Zweigbergk
bookmarks: introduce a repo._bookmarks.changectx(mark) method and use it...
r37468 def changectx(self, mark):
return self._repo[self[mark]]
Boris Feld
bookmark: introduce a 'applychanges' function to gather bookmark movement...
r33480 def applychanges(self, repo, tr, changes):
"""Apply a list of changes to bookmarks
"""
Boris Feld
bookmark: track bookmark changes at the transaction level...
r33516 bmchanges = tr.changes.get('bookmarks')
Boris Feld
bookmark: introduce a 'applychanges' function to gather bookmark movement...
r33480 for name, node in changes:
Boris Feld
bookmark: track bookmark changes at the transaction level...
r33516 old = self.get(name)
Boris Feld
bookmark: introduce a 'applychanges' function to gather bookmark movement...
r33480 if node is None:
Boris Feld
bookmark: deprecate direct del of a bookmark value...
r33518 self._del(name)
Boris Feld
bookmark: introduce a 'applychanges' function to gather bookmark movement...
r33480 else:
Boris Feld
bookmark: deprecate direct set of a bookmark value...
r33517 self._set(name, node)
Boris Feld
bookmark: track bookmark changes at the transaction level...
r33516 if bmchanges is not None:
# if a previous value exist preserve the "initial" value
previous = bmchanges.get(name)
if previous is not None:
old = previous[0]
bmchanges[name] = (old, node)
Boris Feld
bookmark: deprecate 'recordchange' in favor of 'applychanges'...
r33515 self._recordchange(tr)
Boris Feld
bookmark: introduce a 'applychanges' function to gather bookmark movement...
r33480
Boris Feld
bookmark: deprecate 'recordchange' in favor of 'applychanges'...
r33515 def _recordchange(self, tr):
Pierre-Yves David
bookmark: add a `bmstore.recordupdate` to plug bookmarks into the transaction...
r22665 """record that bookmarks have been changed in a transaction
The transaction is then responsible for updating the file content."""
tr.addfilegenerator('bookmarks', ('bookmarks',), self._write,
Pierre-Yves David
transaction: use 'location' instead of 'vfs' objects for file generation...
r23317 location='plain')
Pierre-Yves David
bookmarks: inform transaction-related hooks that some bookmarks were moved...
r22941 tr.hookargs['bookmark_moved'] = '1'
Pierre-Yves David
bookmark: add a `bmstore.recordupdate` to plug bookmarks into the transaction...
r22665
Ryan McElroy
bookmarks: factor out repository lookup from writing bookmarks file...
r23469 def _writerepo(self, repo):
"""Factored out for extensibility"""
Augie Fackler
bmstore: add handling of the active bookmark...
r27698 rbm = repo._bookmarks
if rbm.active not in self:
rbm.active = None
rbm._writeactive()
Augie Fackler
bookmarks: introduce a bmstore to manage bookmark persistence...
r17922
Bryan O'Sullivan
with: use context manager for wlock in _writerepo
r27799 with repo.wlock():
FUJIWARA Katsunori
bookmarks: make writing files out avoid ambiguity of file stat...
r29300 file_ = repo.vfs('bookmarks', 'w', atomictemp=True,
checkambig=True)
Augie Fackler
bmstore: close file in a finally block in _writerepo...
r27188 try:
self._write(file_)
except: # re-raises
file_.discard()
raise
finally:
file_.close()
Augie Fackler
bookmarks: introduce a bmstore to manage bookmark persistence...
r17922
Augie Fackler
bmstore: add handling of the active bookmark...
r27698 def _writeactive(self):
if self._aclean:
return
Bryan O'Sullivan
with: use context manager for wlock in _writeactive
r27800 with self._repo.wlock():
Augie Fackler
bmstore: add handling of the active bookmark...
r27698 if self._active is not None:
FUJIWARA Katsunori
bookmarks: make writing files out avoid ambiguity of file stat...
r29300 f = self._repo.vfs('bookmarks.current', 'w', atomictemp=True,
checkambig=True)
Augie Fackler
bmstore: add handling of the active bookmark...
r27698 try:
f.write(encoding.fromlocal(self._active))
finally:
f.close()
else:
Ryan McElroy
bookmarks: use tryunlink
r31544 self._repo.vfs.tryunlink('bookmarks.current')
Augie Fackler
bmstore: add handling of the active bookmark...
r27698 self._aclean = True
Pierre-Yves David
bookmarks: split bookmark serialization and file handling...
r22664 def _write(self, fp):
Gregory Szorc
bookmarks: write bookmarks file deterministically...
r36471 for name, node in sorted(self.iteritems()):
Pierre-Yves David
bookmarks: split bookmark serialization and file handling...
r22664 fp.write("%s %s\n" % (hex(node), encoding.fromlocal(name)))
Augie Fackler
bmstore: add basic clean-state tracking...
r27187 self._clean = True
Augie Fackler
bookmarks: properly invalidate volatile sets when writing bookmarks...
r29066 self._repo.invalidatevolatilesets()
Pierre-Yves David
bookmarks: split bookmark serialization and file handling...
r22664
liscju
bookmarks: add 'hg push -B .' for pushing the active bookmark (issue4917)
r28182 def expandname(self, bname):
if bname == '.':
liscju
bookmarks: abort 'push -B .' when no active bookmark
r29354 if self.active:
return self.active
else:
raise error.Abort(_("no active bookmark"))
liscju
bookmarks: add 'hg push -B .' for pushing the active bookmark (issue4917)
r28182 return bname
Sean Farley
commands: move checkconflict to bookmarks module...
r32956 def checkconflict(self, mark, force=False, target=None):
"""check repo for a potential clash of mark with an existing bookmark,
branch, or hash
If target is supplied, then check that we are moving the bookmark
forward.
If force is supplied, then forcibly move the bookmark to a new commit
regardless if it is a move forward.
Boris Feld
bookmark: use 'divergent2delete' in checkconflict...
r33513
If divergent bookmark are to be deleted, they will be returned as list.
Sean Farley
commands: move checkconflict to bookmarks module...
r32956 """
Martin von Zweigbergk
bookmarks: switch from repo.changectx('.') to repo['.']...
r37317 cur = self._repo['.'].node()
Sean Farley
commands: move checkconflict to bookmarks module...
r32956 if mark in self and not force:
if target:
if self[mark] == target and target == cur:
# re-activating a bookmark
Boris Feld
bookmark: use 'divergent2delete' in checkconflict...
r33513 return []
Sean Farley
commands: move checkconflict to bookmarks module...
r32956 rev = self._repo[target].rev()
anc = self._repo.changelog.ancestors([rev])
Martin von Zweigbergk
bookmarks: introduce a repo._bookmarks.changectx(mark) method and use it...
r37468 bmctx = self.changectx(mark)
divs = [self[b] for b in self
Sean Farley
commands: move checkconflict to bookmarks module...
r32956 if b.split('@', 1)[0] == mark.split('@', 1)[0]]
# allow resolving a single divergent bookmark even if moving
# the bookmark across branches when a revision is specified
# that contains a divergent bookmark
if bmctx.rev() not in anc and target in divs:
Boris Feld
bookmark: use 'divergent2delete' in checkconflict...
r33513 return divergent2delete(self._repo, [target], mark)
Sean Farley
commands: move checkconflict to bookmarks module...
r32956
deletefrom = [b for b in divs
if self._repo[b].rev() in anc or b == target]
Boris Feld
bookmark: use 'divergent2delete' in checkconflict...
r33513 delbms = divergent2delete(self._repo, deletefrom, mark)
Sean Farley
commands: move checkconflict to bookmarks module...
r32956 if validdest(self._repo, bmctx, self._repo[target]):
self._repo.ui.status(
_("moving bookmark '%s' forward from %s\n") %
(mark, short(bmctx.node())))
Boris Feld
bookmark: use 'divergent2delete' in checkconflict...
r33513 return delbms
Sean Farley
commands: move checkconflict to bookmarks module...
r32956 raise error.Abort(_("bookmark '%s' already exists "
"(use -f to force)") % mark)
if ((mark in self._repo.branchmap() or
mark == self._repo.dirstate.branch()) and not force):
raise error.Abort(
_("a bookmark cannot have the name of an existing branch"))
if len(mark) > 3 and not force:
try:
Martin von Zweigbergk
bookmarks: use isrevsymbol() for detecting collision with existing symbol...
r37415 shadowhash = scmutil.isrevsymbol(self._repo, mark)
Sean Farley
commands: move checkconflict to bookmarks module...
r32956 except error.LookupError: # ambiguous identifier
shadowhash = False
if shadowhash:
self._repo.ui.warn(
_("bookmark %s matches a changeset hash\n"
"(did you leave a -r out of an 'hg bookmark' "
"command?)\n")
% mark)
Boris Feld
bookmark: use 'divergent2delete' in checkconflict...
r33513 return []
Sean Farley
commands: move checkconflict to bookmarks module...
r32956
Augie Fackler
bmstore: add handling of the active bookmark...
r27698 def _readactive(repo, marks):
Ryan McElroy
bookmarks: rename readcurrent to readactive (API)...
r24946 """
Get the active bookmark. We can have an active bookmark that updates
itself as we commit. This function returns the name of that bookmark.
It is stored in .hg/bookmarks.current
"""
Matt Mackall
bookmarks: move read methods to core
r13351 mark = None
Benoit Boissinot
bookmarks: be more restrictive in our Exception catching
r14027 try:
Angel Ezquerra
localrepo: remove all external users of localrepo.opener...
r23877 file = repo.vfs('bookmarks.current')
Gregory Szorc
global: mass rewrite to use modern exception syntax...
r25660 except IOError as inst:
Benoit Boissinot
bookmarks: be more restrictive in our Exception catching
r14027 if inst.errno != errno.ENOENT:
raise
return None
try:
Augie Fackler
bookmarks: make _readactive safe when readlines raises ENOENT...
r27685 # No readline() in osutil.posixfile, reading everything is
# cheap.
# Note that it's possible for readlines() here to raise
# IOError, since we might be reading the active mark over
# static-http which only tries to load the file when we try
# to read from it.
David Soria Parra
bookmarks: read current bookmark as utf-8 and convert it to local
r13381 mark = encoding.tolocal((file.readlines() or [''])[0])
Augie Fackler
bmstore: add handling of the active bookmark...
r27698 if mark == '' or mark not in marks:
Matt Mackall
bookmarks: move read methods to core
r13351 mark = None
Augie Fackler
bookmarks: make _readactive safe when readlines raises ENOENT...
r27685 except IOError as inst:
if inst.errno != errno.ENOENT:
raise
return None
Benoit Boissinot
bookmarks: be more restrictive in our Exception catching
r14027 finally:
Matt Mackall
bookmarks: move read methods to core
r13351 file.close()
return mark
Ryan McElroy
bookmarks: rename setcurrent to activate (API)...
r24945 def activate(repo, mark):
"""
Set the given bookmark to be 'active', meaning that this bookmark will
follow new commits that are made.
Matt Mackall
bookmarks: move basic io to core
r13350 The name is recorded in .hg/bookmarks.current
Ryan McElroy
bookmarks: rename setcurrent to activate (API)...
r24945 """
Augie Fackler
bmstore: add handling of the active bookmark...
r27698 repo._bookmarks.active = mark
repo._bookmarks._writeactive()
Matt Mackall
bookmarks: move update into core
r13352
Ryan McElroy
bookmarks: rename unsetcurrent to deactivate (API)...
r24944 def deactivate(repo):
"""
Mads Kiilerich
spelling: trivial spell checking
r26781 Unset the active bookmark in this repository.
Ryan McElroy
bookmarks: rename unsetcurrent to deactivate (API)...
r24944 """
Augie Fackler
bmstore: add handling of the active bookmark...
r27698 repo._bookmarks.active = None
repo._bookmarks._writeactive()
Idan Kamara
update: delete bookmarks.current when explicitly updating to a rev (issue3276)
r16191
Ryan McElroy
bookmarks: simplify iscurrent to isactivewdirparent (API)...
r24986 def isactivewdirparent(repo):
"""
Tell whether the 'active' bookmark (the one that follows new commits)
points to one of the parents of the current working directory (wdir).
Kevin Bullock
update: update to current bookmark if it moved out from under us (issue3682)...
r18471
Ryan McElroy
bookmarks: simplify iscurrent to isactivewdirparent (API)...
r24986 While this is normally the case, it can on occasion be false; for example,
immediately after a pull, the active bookmark can be moved to point
to a place different than the wdir. This is solved by running `hg update`.
"""
mark = repo._activebookmark
Kevin Bullock
update: update to current bookmark if it moved out from under us (issue3682)...
r18471 marks = repo._bookmarks
Ryan McElroy
bookmarks: simplify iscurrent to isactivewdirparent (API)...
r24986 parents = [p.node() for p in repo[None].parents()]
Kevin Bullock
update: update to current bookmark if it moved out from under us (issue3682)...
r18471 return (mark in marks and marks[mark] in parents)
Boris Feld
bookmark: split out target computation from 'deletedivergent'...
r33510 def divergent2delete(repo, deletefrom, bm):
"""find divergent versions of bm on nodes in deletefrom.
the list of bookmark to delete."""
todelete = []
Siddharth Agarwal
bookmarks: factor out delete divergent code...
r18513 marks = repo._bookmarks
divergent = [b for b in marks if b.split('@', 1)[0] == bm.split('@', 1)[0]]
for mark in divergent:
Matt Mackall
bookmarks: avoid deleting primary bookmarks on rebase...
r21843 if mark == '@' or '@' not in mark:
# can't be divergent by definition
continue
Siddharth Agarwal
bookmarks: factor out delete divergent code...
r18513 if mark and marks[mark] in deletefrom:
if mark != bm:
Boris Feld
bookmark: split out target computation from 'deletedivergent'...
r33510 todelete.append(mark)
return todelete
Siddharth Agarwal
bookmarks: factor out delete divergent code...
r18513
Augie Fackler
localrepo: extract bookmarkheads method to bookmarks.py...
r32381 def headsforactive(repo):
"""Given a repo with an active bookmark, return divergent bookmark nodes.
Args:
repo: A repository with an active bookmark.
Returns:
A list of binary node ids that is the full list of other
revisions with bookmarks divergent from the active bookmark. If
there were no divergent bookmarks, then this list will contain
only one entry.
"""
if not repo._activebookmark:
raise ValueError(
'headsforactive() only makes sense with an active bookmark')
name = repo._activebookmark.split('@', 1)[0]
heads = []
for mark, n in repo._bookmarks.iteritems():
if mark.split('@', 1)[0] == name:
heads.append(n)
return heads
Martin von Zweigbergk
bookmarks: drop always-None argument from calculateupdate()...
r37393 def calculateupdate(ui, repo):
Martin von Zweigbergk
bookmarks: calculateupdate() returns a bookmark, not a rev...
r37377 '''Return a tuple (activemark, movemarkfrom) indicating the active bookmark
and where to move the active bookmark from, if needed.'''
Martin von Zweigbergk
bookmarks: drop always-None argument from calculateupdate()...
r37393 checkout, movemarkfrom = None, None
activemark = repo._activebookmark
if isactivewdirparent(repo):
movemarkfrom = repo['.'].node()
elif activemark:
ui.status(_("updating to active bookmark %s\n") % activemark)
checkout = activemark
Kevin Bullock
bookmarks: pull --update updates to active bookmark if it moved (issue4007)...
r19523 return (checkout, movemarkfrom)
Matt Mackall
bookmarks: move update into core
r13352 def update(repo, parents, node):
Sean Farley
bookmarks: resolve divergent bookmarks when moving active bookmark forward...
r19110 deletefrom = parents
Matt Mackall
bookmarks: move update into core
r13352 marks = repo._bookmarks
Augie Fackler
bmstore: add handling of the active bookmark...
r27698 active = marks.active
Ryan McElroy
bookmarks: rename current to active in variables and comments...
r25100 if not active:
David Soria Parra
bookmarks: delete divergent bookmarks on merge
r16706 return False
Boris Feld
bookmarks: use 'applychanges' for bookmark update...
r33491 bmchanges = []
Ryan McElroy
bookmarks: rename current to active in variables and comments...
r25100 if marks[active] in parents:
Siddharth Agarwal
bookmarks: factor out delete divergent code...
r18513 new = repo[node]
Martin von Zweigbergk
bookmarks: introduce a repo._bookmarks.changectx(mark) method and use it...
r37468 divs = [marks.changectx(b) for b in marks
Ryan McElroy
bookmarks: rename current to active in variables and comments...
r25100 if b.split('@', 1)[0] == active.split('@', 1)[0]]
Sean Farley
bookmarks: resolve divergent bookmarks when moving active bookmark forward...
r19110 anc = repo.changelog.ancestors([new.rev()])
deletefrom = [b.node() for b in divs if b.rev() in anc or b == new]
Martin von Zweigbergk
bookmarks: introduce a repo._bookmarks.changectx(mark) method and use it...
r37468 if validdest(repo, marks.changectx(active), new):
Boris Feld
bookmarks: use 'applychanges' for bookmark update...
r33491 bmchanges.append((active, new.node()))
Siddharth Agarwal
bookmarks: factor out delete divergent code...
r18513
Boris Feld
bookmark: use 'divergent2delete' when updating a bookmark
r33512 for bm in divergent2delete(repo, deletefrom, active):
bmchanges.append((bm, None))
Siddharth Agarwal
bookmarks: factor out delete divergent code...
r18513
Boris Feld
bookmark: use 'divergent2delete' when updating a bookmark
r33512 if bmchanges:
Martin von Zweigbergk
bookmarks: use context managers for lock and transaction in update()...
r35594 with repo.lock(), repo.transaction('bookmark') as tr:
Boris Feld
bookmarks: use 'applychanges' for bookmark update...
r33491 marks.applychanges(repo, tr, bmchanges)
Boris Feld
bookmark: use 'divergent2delete' when updating a bookmark
r33512 return bool(bmchanges)
Matt Mackall
bookmarks: move pushkey functions into core
r13353
Stanislau Hlebik
bookmarks: introduce listbinbookmarks()...
r30481 def listbinbookmarks(repo):
# We may try to list bookmarks on a repo type that does not
# support it (e.g., statichttprepository).
marks = getattr(repo, '_bookmarks', {})
hasnode = repo.changelog.hasnode
for k, v in marks.iteritems():
# don't expose local divergent bookmarks
if hasnode(v) and ('@' not in k or k.endswith('@')):
yield k, v
Matt Mackall
bookmarks: move pushkey functions into core
r13353 def listbookmarks(repo):
d = {}
Stanislau Hlebik
bookmarks: use listbinbookmarks() in listbookmarks()
r30482 for book, node in listbinbookmarks(repo):
d[book] = hex(node)
Matt Mackall
bookmarks: move pushkey functions into core
r13353 return d
def pushbookmark(repo, key, old, new):
Martin von Zweigbergk
bookmarks: use context managers for locks and transaction in pushbookmark()...
r35595 with repo.wlock(), repo.lock(), repo.transaction('bookmarks') as tr:
Matt Mackall
bookmarks: move pushkey functions into core
r13353 marks = repo._bookmarks
Durham Goode
bookmarks: allow pushkey if new equals current...
r22364 existing = hex(marks.get(key, ''))
if existing != old and existing != new:
Matt Mackall
bookmarks: move pushkey functions into core
r13353 return False
if new == '':
Boris Feld
bookmark: use 'applychanges' when updating a bookmark through pushkey
r33485 changes = [(key, None)]
Matt Mackall
bookmarks: move pushkey functions into core
r13353 else:
if new not in repo:
return False
Boris Feld
bookmark: use 'applychanges' when updating a bookmark through pushkey
r33485 changes = [(key, repo[new].node())]
marks.applychanges(repo, tr, changes)
Matt Mackall
bookmarks: move pushkey functions into core
r13353 return True
Matt Mackall
bookmarks: move diff to core
r13354
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 def comparebookmarks(repo, srcmarks, dstmarks, targets=None):
FUJIWARA Katsunori
bookmarks: add function to centralize the logic to compare bookmarks...
r20024 '''Compare bookmarks between srcmarks and dstmarks
This returns tuple "(addsrc, adddst, advsrc, advdst, diverge,
differ, invalid)", each are list of bookmarks below:
:addsrc: added on src side (removed on dst side, perhaps)
:adddst: added on dst side (removed on src side, perhaps)
:advsrc: advanced on src side
:advdst: advanced on dst side
:diverge: diverge
:differ: changed, but changeset referred on src is unknown on dst
:invalid: unknown on both side
Gregory Szorc
bookmarks: explicitly track identical bookmarks...
r23081 :same: same on both side
FUJIWARA Katsunori
bookmarks: add function to centralize the logic to compare bookmarks...
r20024
Each elements of lists in result tuple is tuple "(bookmark name,
changeset ID on source side, changeset ID on destination
side)". Each changeset IDs are 40 hexadecimal digit string or
None.
Changeset IDs of tuples in "addsrc", "adddst", "differ" or
"invalid" list may be unknown for repo.
If "targets" is specified, only bookmarks listed in it are
examined.
'''
if targets:
bset = set(targets)
else:
srcmarkset = set(srcmarks)
dstmarkset = set(dstmarks)
Gregory Szorc
bookmarks: explicitly track identical bookmarks...
r23081 bset = srcmarkset | dstmarkset
FUJIWARA Katsunori
bookmarks: add function to centralize the logic to compare bookmarks...
r20024
Gregory Szorc
bookmarks: explicitly track identical bookmarks...
r23081 results = ([], [], [], [], [], [], [], [])
FUJIWARA Katsunori
bookmarks: add function to centralize the logic to compare bookmarks...
r20024 addsrc = results[0].append
adddst = results[1].append
advsrc = results[2].append
advdst = results[3].append
diverge = results[4].append
differ = results[5].append
invalid = results[6].append
Gregory Szorc
bookmarks: explicitly track identical bookmarks...
r23081 same = results[7].append
FUJIWARA Katsunori
bookmarks: add function to centralize the logic to compare bookmarks...
r20024
for b in sorted(bset):
if b not in srcmarks:
if b in dstmarks:
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 adddst((b, None, dstmarks[b]))
FUJIWARA Katsunori
bookmarks: add function to centralize the logic to compare bookmarks...
r20024 else:
invalid((b, None, None))
elif b not in dstmarks:
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 addsrc((b, srcmarks[b], None))
FUJIWARA Katsunori
bookmarks: add function to centralize the logic to compare bookmarks...
r20024 else:
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 scid = srcmarks[b]
dcid = dstmarks[b]
Gregory Szorc
bookmarks: explicitly track identical bookmarks...
r23081 if scid == dcid:
same((b, scid, dcid))
elif scid in repo and dcid in repo:
FUJIWARA Katsunori
bookmarks: add function to centralize the logic to compare bookmarks...
r20024 sctx = repo[scid]
dctx = repo[dcid]
if sctx.rev() < dctx.rev():
if validdest(repo, sctx, dctx):
advdst((b, scid, dcid))
else:
diverge((b, scid, dcid))
else:
if validdest(repo, dctx, sctx):
advsrc((b, scid, dcid))
else:
diverge((b, scid, dcid))
else:
# it is too expensive to examine in detail, in this case
differ((b, scid, dcid))
return results
FUJIWARA Katsunori
bookmarks: reuse @number bookmark, if it refers changeset referred remotely...
r24355 def _diverge(ui, b, path, localmarks, remotenode):
FUJIWARA Katsunori
bookmarks: prevent divergent bookmark from being updated unexpectedly...
r24353 '''Return appropriate diverged bookmark for specified ``path``
This returns None, if it is failed to assign any divergent
bookmark name.
FUJIWARA Katsunori
bookmarks: reuse @number bookmark, if it refers changeset referred remotely...
r24355
This reuses already existing one with "@number" suffix, if it
refers ``remotenode``.
FUJIWARA Katsunori
bookmarks: prevent divergent bookmark from being updated unexpectedly...
r24353 '''
FUJIWARA Katsunori
bookmarks: rewrite "updatefromremote()" by "compare()"...
r20025 if b == '@':
b = ''
# try to use an @pathalias suffix
# if an @pathalias already exists, we overwrite (update) it
Matt Mackall
bookmarks: fix divergent bookmark path normalization
r22629 if path.startswith("file:"):
path = util.url(path).path
FUJIWARA Katsunori
bookmarks: rewrite "updatefromremote()" by "compare()"...
r20025 for p, u in ui.configitems("paths"):
Matt Mackall
bookmarks: fix divergent bookmark path normalization
r22629 if u.startswith("file:"):
u = util.url(u).path
FUJIWARA Katsunori
bookmarks: rewrite "updatefromremote()" by "compare()"...
r20025 if path == u:
FUJIWARA Katsunori
bookmarks: check @pathalias suffix before available @number for efficiency...
r24354 return '%s@%s' % (b, p)
# assign a unique "@number" suffix newly
for x in range(1, 100):
n = '%s@%d' % (b, x)
FUJIWARA Katsunori
bookmarks: reuse @number bookmark, if it refers changeset referred remotely...
r24355 if n not in localmarks or localmarks[n] == remotenode:
FUJIWARA Katsunori
bookmarks: check @pathalias suffix before available @number for efficiency...
r24354 return n
return None
FUJIWARA Katsunori
bookmarks: rewrite "updatefromremote()" by "compare()"...
r20025
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 def unhexlifybookmarks(marks):
binremotemarks = {}
for name, node in marks.items():
binremotemarks[name] = bin(node)
return binremotemarks
Boris Feld
bookmark: add methods to binary encode and decode bookmark values...
r35258 _binaryentry = struct.Struct('>20sH')
def binaryencode(bookmarks):
"""encode a '(bookmark, node)' iterable into a binary stream
the binary format is:
<node><bookmark-length><bookmark-name>
:node: is a 20 bytes binary node,
:bookmark-length: an unsigned short,
:bookmark-name: the name of the bookmark (of length <bookmark-length>)
wdirid (all bits set) will be used as a special value for "missing"
"""
binarydata = []
for book, node in bookmarks:
if not node: # None or ''
node = wdirid
binarydata.append(_binaryentry.pack(node, len(book)))
binarydata.append(book)
return ''.join(binarydata)
def binarydecode(stream):
"""decode a binary stream into an '(bookmark, node)' iterable
the binary format is:
<node><bookmark-length><bookmark-name>
:node: is a 20 bytes binary node,
:bookmark-length: an unsigned short,
:bookmark-name: the name of the bookmark (of length <bookmark-length>))
wdirid (all bits set) will be used as a special value for "missing"
"""
entrysize = _binaryentry.size
books = []
while True:
entry = stream.read(entrysize)
if len(entry) < entrysize:
if entry:
raise error.Abort(_('bad bookmark stream'))
break
node, length = _binaryentry.unpack(entry)
bookmark = stream.read(length)
if len(bookmark) < length:
if entry:
raise error.Abort(_('bad bookmark stream'))
if node == wdirid:
node = None
books.append((bookmark, node))
return books
Pierre-Yves David
pull: perform bookmark updates in the transaction
r22666 def updatefromremote(ui, repo, remotemarks, path, trfunc, explicit=()):
David Soria Parra
bookmarks: separate bookmarks update code from localrepo's pull....
r13646 ui.debug("checking for updated bookmarks\n")
Augie Fackler
bookmarks: introduce a bmstore to manage bookmark persistence...
r17922 localmarks = repo._bookmarks
Gregory Szorc
bookmarks: explicitly track identical bookmarks...
r23081 (addsrc, adddst, advsrc, advdst, diverge, differ, invalid, same
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 ) = comparebookmarks(repo, remotemarks, localmarks)
Matt Mackall
bookmarks: mark divergent bookmarks with book@pathalias when source in [paths]
r15614
Pierre-Yves David
bookmarks: allow `updatefromremote` to be quiet...
r22644 status = ui.status
warn = ui.warn
Jun Wu
codemod: register core configitems using a script...
r33499 if ui.configbool('ui', 'quietbookmarkmove'):
Pierre-Yves David
bookmarks: allow `updatefromremote` to be quiet...
r22644 status = warn = ui.debug
Pierre-Yves David
pull: merge bookmark updates and imports...
r22659 explicit = set(explicit)
FUJIWARA Katsunori
bookmarks: rewrite "updatefromremote()" by "compare()"...
r20025 changed = []
for b, scid, dcid in addsrc:
if scid in repo: # add remote bookmarks for changes we already have
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 changed.append((b, scid, status,
FUJIWARA Katsunori
bookmarks: rewrite "updatefromremote()" by "compare()"...
r20025 _("adding remote bookmark %s\n") % (b)))
Pierre-Yves David
bookmark: informs of failure to upgrade a bookmark...
r25564 elif b in explicit:
explicit.remove(b)
ui.warn(_("remote bookmark %s points to locally missing %s\n")
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 % (b, hex(scid)[:12]))
Pierre-Yves David
bookmark: informs of failure to upgrade a bookmark...
r25564
FUJIWARA Katsunori
bookmarks: rewrite "updatefromremote()" by "compare()"...
r20025 for b, scid, dcid in advsrc:
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 changed.append((b, scid, status,
FUJIWARA Katsunori
bookmarks: rewrite "updatefromremote()" by "compare()"...
r20025 _("updating bookmark %s\n") % (b)))
Pierre-Yves David
pull: merge bookmark updates and imports...
r22659 # remove normal movement from explicit set
explicit.difference_update(d[0] for d in changed)
FUJIWARA Katsunori
bookmarks: rewrite "updatefromremote()" by "compare()"...
r20025 for b, scid, dcid in diverge:
Pierre-Yves David
pull: merge bookmark updates and imports...
r22659 if b in explicit:
explicit.discard(b)
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 changed.append((b, scid, status,
Pierre-Yves David
bookmarks: fix formatting of exchange message (issue4439)...
r23199 _("importing bookmark %s\n") % (b)))
Pierre-Yves David
pull: merge bookmark updates and imports...
r22659 else:
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 db = _diverge(ui, b, path, localmarks, scid)
FUJIWARA Katsunori
bookmarks: prevent divergent bookmark from being updated unexpectedly...
r24353 if db:
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 changed.append((db, scid, warn,
FUJIWARA Katsunori
bookmarks: prevent divergent bookmark from being updated unexpectedly...
r24353 _("divergent bookmark %s stored as %s\n") %
(b, db)))
else:
warn(_("warning: failed to assign numbered name "
"to divergent bookmark %s\n") % (b))
Pierre-Yves David
pull: merge bookmark updates and imports...
r22659 for b, scid, dcid in adddst + advdst:
if b in explicit:
explicit.discard(b)
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 changed.append((b, scid, status,
Pierre-Yves David
bookmarks: fix formatting of exchange message (issue4439)...
r23199 _("importing bookmark %s\n") % (b)))
Pierre-Yves David
bookmark: informs of failure to upgrade a bookmark...
r25564 for b, scid, dcid in differ:
if b in explicit:
explicit.remove(b)
ui.warn(_("remote bookmark %s points to locally missing %s\n")
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 % (b, hex(scid)[:12]))
Pierre-Yves David
pull: merge bookmark updates and imports...
r22659
David Soria Parra
bookmarks: separate bookmarks update code from localrepo's pull....
r13646 if changed:
Pierre-Yves David
pull: perform bookmark updates in the transaction
r22666 tr = trfunc()
Boris Feld
bookmark: use 'applychanges' when updating from a remote
r33484 changes = []
FUJIWARA Katsunori
bookmarks: rewrite "updatefromremote()" by "compare()"...
r20025 for b, node, writer, msg in sorted(changed):
Boris Feld
bookmark: use 'applychanges' when updating from a remote
r33484 changes.append((b, node))
FUJIWARA Katsunori
bookmarks: rewrite "updatefromremote()" by "compare()"...
r20025 writer(msg)
Boris Feld
bookmark: use 'applychanges' when updating from a remote
r33484 localmarks.applychanges(repo, tr, changes)
David Soria Parra
bookmarks: separate bookmarks update code from localrepo's pull....
r13646
Gregory Szorc
bookmarks: use command executor for wire protocol commands...
r37659 def incoming(ui, repo, peer):
FUJIWARA Katsunori
bookmarks: add incoming() to replace diff() for incoming bookmarks...
r24397 '''Show bookmarks incoming from other to repo
'''
ui.status(_("searching for changed bookmarks\n"))
Gregory Szorc
bookmarks: use command executor for wire protocol commands...
r37659 with peer.commandexecutor() as e:
remotemarks = unhexlifybookmarks(e.callcommand('listkeys', {
'namespace': 'bookmarks',
}).result())
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 r = comparebookmarks(repo, remotemarks, repo._bookmarks)
FUJIWARA Katsunori
bookmarks: add incoming() to replace diff() for incoming bookmarks...
r24397 addsrc, adddst, advsrc, advdst, diverge, differ, invalid, same = r
incomings = []
if ui.debugflag:
getid = lambda id: id
else:
getid = lambda id: id[:12]
FUJIWARA Katsunori
bookmarks: show detailed status about incoming bookmarks...
r24660 if ui.verbose:
def add(b, id, st):
incomings.append(" %-25s %s %s\n" % (b, getid(id), st))
else:
def add(b, id, st):
incomings.append(" %-25s %s\n" % (b, getid(id)))
FUJIWARA Katsunori
bookmarks: add incoming() to replace diff() for incoming bookmarks...
r24397 for b, scid, dcid in addsrc:
Wagner Bruna
bookmarks: add i18n hints to bookmark sync states
r24832 # i18n: "added" refers to a bookmark
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 add(b, hex(scid), _('added'))
FUJIWARA Katsunori
bookmarks: show incoming bookmarks more exactly...
r24657 for b, scid, dcid in advsrc:
Wagner Bruna
bookmarks: add i18n hints to bookmark sync states
r24832 # i18n: "advanced" refers to a bookmark
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 add(b, hex(scid), _('advanced'))
FUJIWARA Katsunori
bookmarks: show incoming bookmarks more exactly...
r24657 for b, scid, dcid in diverge:
Wagner Bruna
bookmarks: add i18n hints to bookmark sync states
r24832 # i18n: "diverged" refers to a bookmark
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 add(b, hex(scid), _('diverged'))
FUJIWARA Katsunori
bookmarks: show incoming bookmarks more exactly...
r24657 for b, scid, dcid in differ:
Wagner Bruna
bookmarks: add i18n hints to bookmark sync states
r24832 # i18n: "changed" refers to a bookmark
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 add(b, hex(scid), _('changed'))
FUJIWARA Katsunori
bookmarks: add incoming() to replace diff() for incoming bookmarks...
r24397
if not incomings:
ui.status(_("no changed bookmarks found\n"))
return 1
for s in sorted(incomings):
ui.write(s)
return 0
FUJIWARA Katsunori
bookmarks: add outgoing() to replace diff() for outgoing bookmarks...
r24398 def outgoing(ui, repo, other):
'''Show bookmarks outgoing from repo to other
'''
ui.status(_("searching for changed bookmarks\n"))
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 remotemarks = unhexlifybookmarks(other.listkeys('bookmarks'))
r = comparebookmarks(repo, repo._bookmarks, remotemarks)
FUJIWARA Katsunori
bookmarks: add outgoing() to replace diff() for outgoing bookmarks...
r24398 addsrc, adddst, advsrc, advdst, diverge, differ, invalid, same = r
outgoings = []
if ui.debugflag:
getid = lambda id: id
else:
getid = lambda id: id[:12]
FUJIWARA Katsunori
bookmarks: show detailed status about outgoing bookmarks...
r24661 if ui.verbose:
def add(b, id, st):
outgoings.append(" %-25s %s %s\n" % (b, getid(id), st))
else:
def add(b, id, st):
outgoings.append(" %-25s %s\n" % (b, getid(id)))
FUJIWARA Katsunori
bookmarks: add outgoing() to replace diff() for outgoing bookmarks...
r24398 for b, scid, dcid in addsrc:
Wagner Bruna
bookmarks: add i18n hints to bookmark sync states
r24832 # i18n: "added refers to a bookmark
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 add(b, hex(scid), _('added'))
FUJIWARA Katsunori
bookmarks: show outgoing bookmarks more exactly...
r24658 for b, scid, dcid in adddst:
Wagner Bruna
bookmarks: add i18n hints to bookmark sync states
r24832 # i18n: "deleted" refers to a bookmark
FUJIWARA Katsunori
bookmarks: show detailed status about outgoing bookmarks...
r24661 add(b, ' ' * 40, _('deleted'))
FUJIWARA Katsunori
bookmarks: show outgoing bookmarks more exactly...
r24658 for b, scid, dcid in advsrc:
Wagner Bruna
bookmarks: add i18n hints to bookmark sync states
r24832 # i18n: "advanced" refers to a bookmark
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 add(b, hex(scid), _('advanced'))
FUJIWARA Katsunori
bookmarks: show outgoing bookmarks more exactly...
r24658 for b, scid, dcid in diverge:
Wagner Bruna
bookmarks: add i18n hints to bookmark sync states
r24832 # i18n: "diverged" refers to a bookmark
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 add(b, hex(scid), _('diverged'))
FUJIWARA Katsunori
bookmarks: show outgoing bookmarks more exactly...
r24658 for b, scid, dcid in differ:
Wagner Bruna
bookmarks: add i18n hints to bookmark sync states
r24832 # i18n: "changed" refers to a bookmark
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 add(b, hex(scid), _('changed'))
FUJIWARA Katsunori
bookmarks: add outgoing() to replace diff() for outgoing bookmarks...
r24398
if not outgoings:
ui.status(_("no changed bookmarks found\n"))
return 1
for s in sorted(outgoings):
ui.write(s)
return 0
Gregory Szorc
bookmarks: use command executor for wire protocol commands...
r37659 def summary(repo, peer):
FUJIWARA Katsunori
bookmarks: rewrite comparing bookmarks in commands.summary() by compare()...
r24400 '''Compare bookmarks between repo and other for "hg summary" output
This returns "(# of incoming, # of outgoing)" tuple.
'''
Gregory Szorc
bookmarks: use command executor for wire protocol commands...
r37659 with peer.commandexecutor() as e:
remotemarks = unhexlifybookmarks(e.callcommand('listkeys', {
'namespace': 'bookmarks',
}).result())
Stanislau Hlebik
bookmarks: make bookmarks.comparebookmarks accept binary nodes (API)...
r30583 r = comparebookmarks(repo, remotemarks, repo._bookmarks)
FUJIWARA Katsunori
bookmarks: rewrite comparing bookmarks in commands.summary() by compare()...
r24400 addsrc, adddst, advsrc, advdst, diverge, differ, invalid, same = r
return (len(addsrc), len(adddst))
Pierre-Yves David
bookmarks: extract valid destination logic in a dedicated function...
r17550 def validdest(repo, old, new):
"""Is the new bookmark destination a valid update from the old one"""
Pierre-Yves David
clfilter: `bookmark.validdest` should run on unfiltered repo...
r18008 repo = repo.unfiltered()
Pierre-Yves David
bookmark: take successors into account when updating (issue3561)...
r17551 if old == new:
# Old == new -> nothing to update.
FUJIWARA Katsunori
bookmarks: avoid redundant creation/assignment of "validdests" in "validdest()"
r17625 return False
Pierre-Yves David
bookmark: take successors into account when updating (issue3561)...
r17551 elif not old:
# old is nullrev, anything is valid.
# (new != nullrev has been excluded by the previous check)
FUJIWARA Katsunori
bookmarks: avoid redundant creation/assignment of "validdests" in "validdest()"
r17625 return True
Pierre-Yves David
bookmark: take successors into account when updating (issue3561)...
r17551 elif repo.obsstore:
obsutil: move 'foreground' to the new modules...
r33147 return new.node() in obsutil.foreground(repo, [old.node()])
Pierre-Yves David
bookmark: take successors into account when updating (issue3561)...
r17551 else:
Mads Kiilerich
spelling: fixes from proofreading of spell checker issues
r24180 # still an independent clause as it is lazier (and therefore faster)
FUJIWARA Katsunori
bookmarks: use "changectx.descendant()" for efficient descendant examination...
r17627 return old.descendant(new)
Sean Farley
commands: move checkformat to bookmarks module...
r32955
def checkformat(repo, mark):
"""return a valid version of a potential bookmark name
Raises an abort error if the bookmark name is not valid.
"""
mark = mark.strip()
if not mark:
raise error.Abort(_("bookmark names cannot consist entirely of "
"whitespace"))
scmutil.checknewlabel(repo, mark, 'bookmark')
return mark
Sean Farley
bookmarks: factor out delete logic from commands...
r33005
def delete(repo, tr, names):
"""remove a mark from the bookmark store
Raises an abort error if mark does not exist.
"""
marks = repo._bookmarks
Boris Feld
bookmark: use 'applychanges' for bookmark deletion
r33481 changes = []
Sean Farley
bookmarks: factor out delete logic from commands...
r33005 for mark in names:
if mark not in marks:
raise error.Abort(_("bookmark '%s' does not exist") % mark)
if mark == repo._activebookmark:
deactivate(repo)
Boris Feld
bookmark: use 'applychanges' for bookmark deletion
r33481 changes.append((mark, None))
marks.applychanges(repo, tr, changes)
Sean Farley
bookmarks: factor out rename logic from commands...
r33006
def rename(repo, tr, old, new, force=False, inactive=False):
"""rename a bookmark from old to new
If force is specified, then the new name can overwrite an existing
bookmark.
If inactive is specified, then do not activate the new bookmark.
Raises an abort error if old is not in the bookmark store.
"""
marks = repo._bookmarks
mark = checkformat(repo, new)
if old not in marks:
raise error.Abort(_("bookmark '%s' does not exist") % old)
Boris Feld
bookmark: use 'divergent2delete' in checkconflict...
r33513 changes = []
for bm in marks.checkconflict(mark, force):
changes.append((bm, None))
changes.extend([(mark, marks[old]), (old, None)])
Boris Feld
bookmark: use 'applychanges' for bookmark renaming
r33482 marks.applychanges(repo, tr, changes)
Sean Farley
bookmarks: factor out rename logic from commands...
r33006 if repo._activebookmark == old and not inactive:
activate(repo, mark)
Sean Farley
bookmarks: factor out adding a list of bookmarks logic from commands...
r33007
def addbookmarks(repo, tr, names, rev=None, force=False, inactive=False):
"""add a list of bookmarks
If force is specified, then the new name can overwrite an existing
bookmark.
If inactive is specified, then do not activate any bookmark. Otherwise, the
first bookmark is activated.
Raises an abort error if old is not in the bookmark store.
"""
marks = repo._bookmarks
Martin von Zweigbergk
bookmarks: switch from repo.changectx('.') to repo['.']...
r37317 cur = repo['.'].node()
Sean Farley
bookmarks: factor out adding a list of bookmarks logic from commands...
r33007 newact = None
Boris Feld
bookmark: use 'applychanges' for adding new bookmark
r33483 changes = []
Pulkit Goyal
bookmarks: calculate visibility exceptions only once...
r35665 hiddenrev = None
# unhide revs if any
if rev:
repo = scmutil.unhidehashlikerevs(repo, [rev], 'nowarn')
Sean Farley
bookmarks: factor out adding a list of bookmarks logic from commands...
r33007 for mark in names:
mark = checkformat(repo, mark)
if newact is None:
newact = mark
if inactive and mark == repo._activebookmark:
deactivate(repo)
return
tgt = cur
if rev:
Pulkit Goyal
bookmarks: add bookmarks to hidden revs if directaccess config is set...
r35629 ctx = scmutil.revsingle(repo, rev)
if ctx.hidden():
Pulkit Goyal
bookmarks: calculate visibility exceptions only once...
r35665 hiddenrev = ctx.hex()[:12]
Pulkit Goyal
bookmarks: add bookmarks to hidden revs if directaccess config is set...
r35629 tgt = ctx.node()
Boris Feld
bookmark: use 'divergent2delete' in checkconflict...
r33513 for bm in marks.checkconflict(mark, force, tgt):
changes.append((bm, None))
Boris Feld
bookmark: use 'applychanges' for adding new bookmark
r33483 changes.append((mark, tgt))
Pulkit Goyal
bookmarks: calculate visibility exceptions only once...
r35665
if hiddenrev:
repo.ui.warn(_("bookmarking hidden changeset %s\n") % hiddenrev)
Boris Feld
bookmarks: display the obsfate of hidden revision we create a bookmark on...
r35730
if ctx.obsolete():
msg = obsutil._getfilteredreason(repo, "%s" % hiddenrev, ctx)
repo.ui.warn("(%s)\n" % msg)
Boris Feld
bookmark: use 'applychanges' for adding new bookmark
r33483 marks.applychanges(repo, tr, changes)
Sean Farley
bookmarks: factor out adding a list of bookmarks logic from commands...
r33007 if not inactive and cur == marks[newact] and not rev:
activate(repo, newact)
elif cur != tgt and newact == repo._activebookmark:
deactivate(repo)
Sean Farley
bookmarks: factor out bookmark printing from commands
r33010
Sean Farley
bookmarks: factor method _printer out of for loop in printbookmarks...
r33011 def _printbookmarks(ui, repo, bmarks, **opts):
"""private method to print bookmarks
Provides a way for extensions to control how bookmarks are printed (e.g.
prepend or postpend names)
"""
Pulkit Goyal
py3: fix kwargs handling for `hg bookmarks`
r33092 opts = pycompat.byteskwargs(opts)
Sean Farley
bookmarks: factor method _printer out of for loop in printbookmarks...
r33011 fm = ui.formatter('bookmarks', opts)
hexfn = fm.hexfunc
if len(bmarks) == 0 and fm.isplain():
ui.status(_("no bookmarks set\n"))
for bmark, (n, prefix, label) in sorted(bmarks.iteritems()):
fm.startitem()
if not ui.quiet:
fm.plain(' %s ' % prefix, label=label)
fm.write('bookmark', '%s', bmark, label=label)
pad = " " * (25 - encoding.colwidth(bmark))
fm.condwrite(not ui.quiet, 'rev node', pad + ' %d:%s',
repo.changelog.rev(n), hexfn(n), label=label)
fm.data(active=(activebookmarklabel in label))
fm.plain('\n')
fm.end()
Sean Farley
bookmarks: factor out bookmark printing from commands
r33010 def printbookmarks(ui, repo, **opts):
"""print bookmarks to a formatter
Provides a way for extensions to control how bookmarks are printed.
"""
marks = repo._bookmarks
Sean Farley
bookmarks: factor method _printer out of for loop in printbookmarks...
r33011 bmarks = {}
Sean Farley
bookmarks: factor out bookmark printing from commands
r33010 for bmark, n in sorted(marks.iteritems()):
active = repo._activebookmark
if bmark == active:
prefix, label = '*', activebookmarklabel
else:
prefix, label = ' ', ''
Sean Farley
bookmarks: factor method _printer out of for loop in printbookmarks...
r33011 bmarks[bmark] = (n, prefix, label)
_printbookmarks(ui, repo, bmarks, **opts)
Boris Feld
bookmark: add a dedicated txnclose-bookmark hook...
r34709
def preparehookargs(name, old, new):
if new is None:
new = ''
if old is None:
old = ''
return {'bookmark': name,
'node': hex(new),
'oldnode': hex(old)}