--- a/hgext/remotenames.py Sun Mar 11 16:17:51 2018 +0530
+++ b/hgext/remotenames.py Mon Mar 12 18:38:26 2018 +0530
@@ -32,6 +32,8 @@
bin,
)
from mercurial import (
+ bookmarks,
+ extensions,
logexchange,
namespaces,
pycompat,
@@ -228,6 +230,24 @@
self._nodetohoists.setdefault(node[0], []).append(name)
return self._nodetohoists
+def wrapprintbookmarks(orig, ui, repo, bmarks, **opts):
+ if 'remotebookmarks' not in repo.names:
+ return
+ ns = repo.names['remotebookmarks']
+
+ for name in ns.listnames(repo):
+ nodes = ns.nodes(repo, name)
+ if not nodes:
+ continue
+ node = nodes[0]
+
+ bmarks[name] = (node, ' ', '')
+
+ return orig(ui, repo, bmarks, **opts)
+
+def extsetup(ui):
+ extensions.wrapfunction(bookmarks, '_printbookmarks', wrapprintbookmarks)
+
def reposetup(ui, repo):
if not repo.local():
return