diff -r 55ec13c82ea0 -r 8491845a75b2 mercurial/exchange.py --- a/mercurial/exchange.py Thu Nov 17 00:59:41 2016 -0800 +++ b/mercurial/exchange.py Thu Nov 17 00:59:41 2016 -0800 @@ -1661,6 +1661,17 @@ if chunks: bundler.newpart('hgtagsfnodes', data=''.join(chunks)) +def _getbookmarks(repo, **kwargs): + """Returns bookmark to node mapping. + + This function is primarily used to generate `bookmarks` bundle2 part. + It is a separate function in order to make it easy to wrap it + in extensions. Passing `kwargs` to the function makes it easy to + add new parameters in extensions. + """ + + return dict(bookmod.listbinbookmarks(repo)) + def check_heads(repo, their_heads, context): """check if the heads of a repo have been modified