diff mercurial/localrepo.py @ 2230:332950340788

localrepository.addchangegroup: add more source infos to hooks
author Vadim Gelfer <vadim.gelfer@gmail.com>
date Mon, 08 May 2006 16:50:27 -0700
parents 0ff326c2b286
children ef3c039e7ab8
line wrap: on
line diff
--- a/mercurial/localrepo.py	Mon May 08 16:07:56 2006 -0700
+++ b/mercurial/localrepo.py	Mon May 08 16:50:27 2006 -0700
@@ -1079,7 +1079,7 @@
             cg = remote.changegroup(fetch, 'pull')
         else:
             cg = remote.changegroupsubset(fetch, heads, 'pull')
-        return self.addchangegroup(cg)
+        return self.addchangegroup(cg, 'pull')
 
     def push(self, remote, force=False, revs=None):
         lock = remote.lock()
@@ -1116,7 +1116,7 @@
             cg = self.changegroup(update, 'push')
         else:
             cg = self.changegroupsubset(update, revs, 'push')
-        return remote.addchangegroup(cg)
+        return remote.addchangegroup(cg, 'push')
 
     def changegroupsubset(self, bases, heads, source):
         """This function generates a changegroup consisting of all the nodes
@@ -1455,7 +1455,7 @@
 
         return util.chunkbuffer(gengroup())
 
-    def addchangegroup(self, source):
+    def addchangegroup(self, source, srctype):
         """add changegroup to repo.
         returns number of heads modified or added + 1."""
 
@@ -1469,7 +1469,7 @@
         if not source:
             return 0
 
-        self.hook('prechangegroup', throw=True, source=source)
+        self.hook('prechangegroup', throw=True, source=srctype)
 
         changesets = files = revisions = 0
 
@@ -1534,17 +1534,17 @@
                          % (changesets, revisions, files, heads))
 
         self.hook('pretxnchangegroup', throw=True,
-                  node=hex(self.changelog.node(cor+1)), source=source)
+                  node=hex(self.changelog.node(cor+1)), source=srctype)
 
         tr.close()
 
         if changesets > 0:
             self.hook("changegroup", node=hex(self.changelog.node(cor+1)),
-                      source=source)
+                      source=srctype)
 
             for i in range(cor + 1, cnr + 1):
                 self.hook("incoming", node=hex(self.changelog.node(i)),
-                          source=source)
+                          source=srctype)
 
         return newheads - oldheads + 1