hgext/largefiles/proto.py
changeset 19917 cff331cbb5ee
parent 19009 07e40d589b64
child 19947 2a03faf8b5fe
--- a/hgext/largefiles/proto.py	Mon Apr 01 20:01:16 2013 -0700
+++ b/hgext/largefiles/proto.py	Thu Oct 10 04:28:39 2013 +0200
@@ -5,6 +5,7 @@
 
 import os
 import urllib2
+import re
 
 from mercurial import error, httppeer, util, wireproto
 from mercurial.wireproto import batchable, future
@@ -166,9 +167,11 @@
         args['cmds'] = args['cmds'].replace('heads ', 'lheads ')
     return ssholdcallstream(self, cmd, **args)
 
+headsre = re.compile(r'(^|;)heads\b')
+
 def httprepocallstream(self, cmd, **args):
     if cmd == 'heads' and self.capable('largefiles'):
         cmd = 'lheads'
     if cmd == 'batch' and self.capable('largefiles'):
-        args['cmds'] = args['cmds'].replace('heads ', 'lheads ')
+        args['cmds'] = headsre.sub('lheads', args['cmds'])
     return httpoldcallstream(self, cmd, **args)