Mercurial > public > mercurial-scm > hg

diff mercurial/hg.py @ 32379:71e735bd8170

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
dispatch: make request accept additional reposetups chg needs special logic around repo object creation (like, collecting and reporting repo path to the master server). Adding "reposetup" to dispatch.request seems to be an easy and reasonably clean way to allow that.
author Jun Wu <quark@fb.com>
date Sat, 29 Apr 2017 21:39:47 -0700
parents 448ed4d3ee90
children 963de566de2f
line wrap: on
line diff
--- a/mercurial/hg.py	Sat May 20 14:01:05 2017 -0700
+++ b/mercurial/hg.py	Sat Apr 29 21:39:47 2017 -0700
@@ -148,10 +148,12 @@
 # a list of (ui, repo) functions called for wire peer initialization
 wirepeersetupfuncs = []
 
-def _peerorrepo(ui, path, create=False):
+def _peerorrepo(ui, path, create=False, presetupfuncs=None):
     """return a repository object for the specified path"""
     obj = _peerlookup(path).instance(ui, path, create)
     ui = getattr(obj, "ui", ui)
+    for f in presetupfuncs or []:
+        f(ui, obj)
     for name, module in extensions.extensions(ui):
         hook = getattr(module, 'reposetup', None)
         if hook:
@@ -161,9 +163,9 @@
             f(ui, obj)
     return obj
 
-def repository(ui, path='', create=False):
+def repository(ui, path='', create=False, presetupfuncs=None):
     """return a repository object for the specified path"""
-    peer = _peerorrepo(ui, path, create)
+    peer = _peerorrepo(ui, path, create, presetupfuncs=presetupfuncs)
     repo = peer.local()
     if not repo:
         raise error.Abort(_("repository '%s' is not local") %