# HG changeset patch
# User Edouard Gomez <ed.gomez@free.fr>
# Date 1278454019 -7200
# Node ID d80b768545cba3e7831fec266c2fc996139228f5
# Parent  9ff22f600c6c7c304d5f56eb788babc8eec27c25
convert: add bookmark support to the hg sink

diff -r 9ff22f600c6c -r d80b768545cb hgext/convert/hg.py
--- a/hgext/convert/hg.py	Wed Jul 07 00:06:59 2010 +0200
+++ b/hgext/convert/hg.py	Wed Jul 07 00:06:59 2010 +0200
@@ -21,7 +21,7 @@
 import os, time, cStringIO
 from mercurial.i18n import _
 from mercurial.node import bin, hex, nullid
-from mercurial import hg, util, context, error
+from mercurial import hg, util, context, bookmarks, error
 
 from common import NoRepo, commit, converter_source, converter_sink
 
@@ -214,6 +214,16 @@
     def setfilemapmode(self, active):
         self.filemapmode = active
 
+    def putbookmarks(self, updatedbookmark):
+        if not len(updatedbookmark):
+            return
+
+        self.ui.status(_("updating bookmarks\n"))
+        for bookmark in updatedbookmark:
+            self.repo._bookmarks[bookmark] = bin(updatedbookmark[bookmark])
+            bookmarks.write(self.repo)
+
+
 class mercurial_source(converter_source):
     def __init__(self, ui, path, rev=None):
         converter_source.__init__(self, ui, path, rev)