# HG changeset patch # User Matt Harbison # Date 1545624111 18000 # Node ID 0840862977c88868d75760147b724c1438d9d3c3 # Parent 4506f801e4921aadbf07df68a8b2690e366cffe4 largefiles: drop the uisetup module This is small enough to live in the __init__ module. diff -r 4506f801e492 -r 0840862977c8 hgext/largefiles/__init__.py --- a/hgext/largefiles/__init__.py Mon Dec 24 17:04:37 2018 +0530 +++ b/hgext/largefiles/__init__.py Sun Dec 23 23:01:51 2018 -0500 @@ -107,10 +107,15 @@ from __future__ import absolute_import from mercurial import ( + cmdutil, configitems, + extensions, exthelper, hg, + httppeer, localrepo, + sshpeer, + wireprotov1server, ) from . import ( @@ -118,7 +123,6 @@ overrides, proto, reposetup, - uisetup as uisetupmod, ) # Note for extension authors: ONLY specify testedwith = 'ships-with-hg-core' for @@ -156,6 +160,36 @@ def _uisetup(ui): localrepo.featuresetupfuncs.add(featuresetup) hg.wirepeersetupfuncs.append(proto.wirereposetup) - uisetupmod.uisetup(ui) + + cmdutil.outgoinghooks.add('largefiles', overrides.outgoinghook) + cmdutil.summaryremotehooks.add('largefiles', overrides.summaryremotehook) + + # create the new wireproto commands ... + wireprotov1server.wireprotocommand('putlfile', 'sha', permission='push')( + proto.putlfile) + wireprotov1server.wireprotocommand('getlfile', 'sha', permission='pull')( + proto.getlfile) + wireprotov1server.wireprotocommand('statlfile', 'sha', permission='pull')( + proto.statlfile) + wireprotov1server.wireprotocommand('lheads', '', permission='pull')( + wireprotov1server.heads) + + extensions.wrapfunction(wireprotov1server.commands['heads'], 'func', + proto.heads) + # TODO also wrap wireproto.commandsv2 once heads is implemented there. + + # can't do this in reposetup because it needs to have happened before + # wirerepo.__init__ is called + proto.ssholdcallstream = sshpeer.sshv1peer._callstream + proto.httpoldcallstream = httppeer.httppeer._callstream + sshpeer.sshv1peer._callstream = proto.sshrepocallstream + httppeer.httppeer._callstream = proto.httprepocallstream + + # override some extensions' stuff as well + for name, module in extensions.extensions(): + if name == 'rebase': + # TODO: teach exthelper to handle this + extensions.wrapfunction(module, 'rebase', + overrides.overriderebase) revsetpredicate = overrides.revsetpredicate diff -r 4506f801e492 -r 0840862977c8 hgext/largefiles/uisetup.py --- a/hgext/largefiles/uisetup.py Mon Dec 24 17:04:37 2018 +0530 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,56 +0,0 @@ -# Copyright 2009-2010 Gregory P. Ward -# Copyright 2009-2010 Intelerad Medical Systems Incorporated -# Copyright 2010-2011 Fog Creek Software -# Copyright 2010-2011 Unity Technologies -# -# This software may be used and distributed according to the terms of the -# GNU General Public License version 2 or any later version. - -'''setup for largefiles extension: uisetup''' -from __future__ import absolute_import - -from mercurial import ( - cmdutil, - extensions, - httppeer, - sshpeer, - wireprotov1server, -) - -from . import ( - overrides, - proto, -) - -def uisetup(ui): - - cmdutil.outgoinghooks.add('largefiles', overrides.outgoinghook) - cmdutil.summaryremotehooks.add('largefiles', overrides.summaryremotehook) - - # create the new wireproto commands ... - wireprotov1server.wireprotocommand('putlfile', 'sha', permission='push')( - proto.putlfile) - wireprotov1server.wireprotocommand('getlfile', 'sha', permission='pull')( - proto.getlfile) - wireprotov1server.wireprotocommand('statlfile', 'sha', permission='pull')( - proto.statlfile) - wireprotov1server.wireprotocommand('lheads', '', permission='pull')( - wireprotov1server.heads) - - extensions.wrapfunction(wireprotov1server.commands['heads'], 'func', - proto.heads) - # TODO also wrap wireproto.commandsv2 once heads is implemented there. - - # can't do this in reposetup because it needs to have happened before - # wirerepo.__init__ is called - proto.ssholdcallstream = sshpeer.sshv1peer._callstream - proto.httpoldcallstream = httppeer.httppeer._callstream - sshpeer.sshv1peer._callstream = proto.sshrepocallstream - httppeer.httppeer._callstream = proto.httprepocallstream - - # override some extensions' stuff as well - for name, module in extensions.extensions(): - if name == 'rebase': - # TODO: teach exthelper to handle this - extensions.wrapfunction(module, 'rebase', - overrides.overriderebase)