14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18
17
"""Support for plugin hooking logic."""
19
from bzrlib import registry
20
from bzrlib.lazy_import import lazy_import
23
from .lazy_import import lazy_import
21
24
lazy_import(globals(), """
25
_format_version_tuple,
28
from bzrlib.help_topics import help_as_plain_text
28
_format_version_tuple,
31
from breezy.i18n import gettext
32
known_hooks = registry.Registry()
33
# known_hooks registry contains
34
# tuple of (module, member name) which is the hook point
35
# module where the specific hooks are defined
36
# callable to get the empty specific Hooks for that attribute
37
known_hooks.register_lazy(('bzrlib.branch', 'Branch.hooks'), 'bzrlib.branch',
39
known_hooks.register_lazy(('bzrlib.bzrdir', 'BzrDir.hooks'), 'bzrlib.bzrdir',
41
known_hooks.register_lazy(('bzrlib.commands', 'Command.hooks'),
42
'bzrlib.commands', 'CommandHooks')
43
known_hooks.register_lazy(('bzrlib.info', 'hooks'),
44
'bzrlib.info', 'InfoHooks')
45
known_hooks.register_lazy(('bzrlib.lock', 'Lock.hooks'), 'bzrlib.lock',
47
known_hooks.register_lazy(('bzrlib.merge', 'Merger.hooks'), 'bzrlib.merge',
49
known_hooks.register_lazy(('bzrlib.msgeditor', 'hooks'), 'bzrlib.msgeditor',
51
known_hooks.register_lazy(('bzrlib.mutabletree', 'MutableTree.hooks'),
52
'bzrlib.mutabletree', 'MutableTreeHooks')
53
known_hooks.register_lazy(('bzrlib.smart.client', '_SmartClient.hooks'),
54
'bzrlib.smart.client', 'SmartClientHooks')
55
known_hooks.register_lazy(('bzrlib.smart.server', 'SmartTCPServer.hooks'),
56
'bzrlib.smart.server', 'SmartServerHooks')
57
known_hooks.register_lazy(
58
('bzrlib.version_info_formats.format_rio', 'RioVersionInfoBuilder.hooks'),
59
'bzrlib.version_info_formats.format_rio', 'RioVersionInfoBuilderHooks')
60
known_hooks.register_lazy(
61
('bzrlib.merge_directive', 'BaseMergeDirective.hooks'),
62
'bzrlib.merge_directive', 'MergeDirectiveHooks')
65
def known_hooks_key_to_object((module_name, member_name)):
66
"""Convert a known_hooks key to a object.
68
:param key: A tuple (module_name, member_name) as found in the keys of
69
the known_hooks registry.
70
:return: The object this specifies.
72
return registry._LazyObjectGetter(module_name, member_name).get_obj()
75
def known_hooks_key_to_parent_and_attribute((module_name, member_name)):
76
"""Convert a known_hooks key to a object.
78
:param key: A tuple (module_name, member_name) as found in the keys of
79
the known_hooks registry.
80
:return: The object this specifies.
82
member_list = member_name.rsplit('.', 1)
83
if len(member_list) == 2:
84
parent_name, attribute = member_list
87
attribute = member_name
88
parent = known_hooks_key_to_object((module_name, parent_name))
89
return parent, attribute
35
class UnknownHook(errors.BzrError):
37
_fmt = "The %(type)s hook '%(hook)s' is unknown in this version of breezy."
39
def __init__(self, hook_type, hook_name):
40
errors.BzrError.__init__(self)
45
class KnownHooksRegistry(registry.Registry):
46
# known_hooks registry contains
47
# tuple of (module, member name) which is the hook point
48
# module where the specific hooks are defined
49
# callable to get the empty specific Hooks for that attribute
51
def register_lazy_hook(self, hook_module_name, hook_member_name,
52
hook_factory_member_name):
53
self.register_lazy((hook_module_name, hook_member_name),
54
hook_module_name, hook_factory_member_name)
56
def iter_parent_objects(self):
57
"""Yield (hook_key, (parent_object, attr)) tuples for every registered
58
hook, where 'parent_object' is the object that holds the hook
61
This is useful for resetting/restoring all the hooks to a known state,
62
as is done in breezy.tests.TestCase._clear_hooks.
64
for key in self.keys():
65
yield key, self.key_to_parent_and_attribute(key)
67
def key_to_parent_and_attribute(self, key):
68
"""Convert a known_hooks key to a (parent_obj, attr) pair.
70
:param key: A tuple (module_name, member_name) as found in the keys of
71
the known_hooks registry.
72
:return: The parent_object of the hook and the name of the attribute on
73
that parent object where the hook is kept.
75
parent_mod, parent_member, attr = pyutils.calc_parent_name(*key)
76
return pyutils.get_named_object(parent_mod, parent_member), attr
79
_builtin_known_hooks = (
80
('breezy.branch', 'Branch.hooks', 'BranchHooks'),
81
('breezy.controldir', 'ControlDir.hooks', 'ControlDirHooks'),
82
('breezy.commands', 'Command.hooks', 'CommandHooks'),
83
('breezy.config', 'ConfigHooks', '_ConfigHooks'),
84
('breezy.info', 'hooks', 'InfoHooks'),
85
('breezy.lock', 'Lock.hooks', 'LockHooks'),
86
('breezy.merge', 'Merger.hooks', 'MergeHooks'),
87
('breezy.msgeditor', 'hooks', 'MessageEditorHooks'),
88
('breezy.mutabletree', 'MutableTree.hooks', 'MutableTreeHooks'),
89
('breezy.bzr.smart.client', '_SmartClient.hooks', 'SmartClientHooks'),
90
('breezy.bzr.smart.server', 'SmartTCPServer.hooks', 'SmartServerHooks'),
91
('breezy.status', 'hooks', 'StatusHooks'),
92
('breezy.transport', 'Transport.hooks', 'TransportHooks'),
93
('breezy.version_info_formats.format_rio', 'RioVersionInfoBuilder.hooks',
94
'RioVersionInfoBuilderHooks'),
95
('breezy.merge_directive', 'BaseMergeDirective.hooks',
96
'MergeDirectiveHooks'),
99
known_hooks = KnownHooksRegistry()
100
for (_hook_module, _hook_attribute, _hook_class) in _builtin_known_hooks:
101
known_hooks.register_lazy_hook(_hook_module, _hook_attribute, _hook_class)
102
del _builtin_known_hooks, _hook_module, _hook_attribute, _hook_class
105
def known_hooks_key_to_object(key):
106
"""Convert a known_hooks key to a object.
108
:param key: A tuple (module_name, member_name) as found in the keys of
109
the known_hooks registry.
110
:return: The object this specifies.
112
return pyutils.get_named_object(*key)
92
115
class Hooks(dict):
96
119
FOO hook is triggered.
122
def __init__(self, module=None, member_name=None):
123
"""Create a new hooks dictionary.
125
:param module: The module from which this hooks dictionary should be loaded
126
(used for lazy hooks)
127
:param member_name: Name under which this hooks dictionary should be loaded.
128
(used for lazy hooks)
100
130
dict.__init__(self)
101
131
self._callable_names = {}
103
def create_hook(self, hook):
104
"""Create a hook which can have callbacks registered for it.
106
:param hook: The hook to create. An object meeting the protocol of
107
bzrlib.hooks.HookPoint. It's name is used as the key for future
132
self._lazy_callable_names = {}
133
self._module = module
134
self._member_name = member_name
136
def add_hook(self, name, doc, introduced, deprecated=None):
137
"""Add a hook point to this dictionary.
139
:param name: The name of the hook, for clients to use when registering.
140
:param doc: The docs for the hook.
141
:param introduced: When the hook was introduced (e.g. (0, 15)).
142
:param deprecated: When the hook was deprecated, None for
110
if hook.name in self:
111
raise errors.DuplicateKey(hook.name)
112
self[hook.name] = hook
146
raise errors.DuplicateKey(name)
148
callbacks = _lazy_hooks.setdefault(
149
(self._module, self._member_name, name), [])
152
hookpoint = HookPoint(name=name, doc=doc, introduced=introduced,
153
deprecated=deprecated, callbacks=callbacks)
154
self[name] = hookpoint
115
157
"""Generate the documentation for this Hooks instance.
145
187
the code names are rarely meaningful for end users and this is not
146
188
intended for debugging.
148
return self._callable_names.get(a_callable, "No hook name")
190
name = self._callable_names.get(a_callable, None)
191
if name is None and a_callable is not None:
192
name = self._lazy_callable_names.get((a_callable.__module__,
193
a_callable.__name__),
196
return 'No hook name'
199
def install_named_hook_lazy(self, hook_name, callable_module,
200
callable_member, name):
201
"""Install a_callable in to the hook hook_name lazily, and label it.
203
:param hook_name: A hook name. See the __init__ method for the complete
205
:param callable_module: Name of the module in which the callable is
207
:param callable_member: Member name of the callable.
208
:param name: A name to associate the callable with, to show users what
212
hook = self[hook_name]
214
raise UnknownHook(self.__class__.__name__, hook_name)
216
hook_lazy = getattr(hook, "hook_lazy")
217
except AttributeError:
218
raise errors.UnsupportedOperation(self.install_named_hook_lazy,
221
hook_lazy(callable_module, callable_member, name)
223
self.name_hook_lazy(callable_module, callable_member, name)
150
225
def install_named_hook(self, hook_name, a_callable, name):
151
226
"""Install a_callable in to the hook hook_name, and label it name.
153
:param hook_name: A hook name. See the __init__ method of BranchHooks
154
for the complete list of hooks.
228
:param hook_name: A hook name. See the __init__ method for the complete
155
230
:param a_callable: The callable to be invoked when the hook triggers.
156
231
The exact signature will depend on the hook - see the __init__
157
method of BranchHooks for details on each hook.
232
method for details on each hook.
158
233
:param name: A name to associate a_callable with, to show users what is
162
237
hook = self[hook_name]
164
raise errors.UnknownHook(self.__class__.__name__, hook_name)
239
raise UnknownHook(self.__class__.__name__, hook_name)
166
241
# list hooks, old-style, not yet deprecated but less useful.
167
242
hook.append(a_callable)
170
245
if name is not None:
171
246
self.name_hook(a_callable, name)
248
def uninstall_named_hook(self, hook_name, label):
249
"""Uninstall named hooks.
251
:param hook_name: Hook point name
252
:param label: Label of the callable to uninstall
255
hook = self[hook_name]
257
raise UnknownHook(self.__class__.__name__, hook_name)
259
uninstall = getattr(hook, "uninstall")
260
except AttributeError:
261
raise errors.UnsupportedOperation(self.uninstall_named_hook, self)
173
265
def name_hook(self, a_callable, name):
174
266
"""Associate name with a_callable to show users what is running."""
175
267
self._callable_names[a_callable] = name
269
def name_hook_lazy(self, callable_module, callable_member, callable_name):
270
self._lazy_callable_names[(callable_module, callable_member)] = \
178
274
class HookPoint(object):
179
275
"""A single hook that clients can register to be called back when it fires.
181
277
:ivar name: The name of the hook.
278
:ivar doc: The docs for using the hook.
182
279
:ivar introduced: A version tuple specifying what version the hook was
183
280
introduced in. None indicates an unknown version.
184
281
:ivar deprecated: A version tuple specifying what version the hook was
185
282
deprecated or superseded in. None indicates that the hook is not
186
283
superseded or deprecated. If the hook is superseded then the doc
187
284
should describe the recommended replacement hook to register for.
188
:ivar doc: The docs for using the hook.
191
def __init__(self, name, doc, introduced, deprecated):
287
def __init__(self, name, doc, introduced, deprecated=None, callbacks=None):
192
288
"""Create a HookPoint.
194
290
:param name: The name of the hook, for clients to use when registering.
213
311
strings.append(self.name)
214
strings.append('~'*len(self.name))
312
strings.append('~' * len(self.name))
215
313
strings.append('')
216
314
if self.introduced:
217
315
introduced_string = _format_version_tuple(self.introduced)
219
317
introduced_string = 'unknown'
220
strings.append('Introduced in: %s' % introduced_string)
318
strings.append(gettext('Introduced in: %s') % introduced_string)
221
319
if self.deprecated:
222
320
deprecated_string = _format_version_tuple(self.deprecated)
223
strings.append('Deprecated in: %s' % deprecated_string)
321
strings.append(gettext('Deprecated in: %s') % deprecated_string)
224
322
strings.append('')
225
323
strings.extend(textwrap.wrap(self.__doc__,
226
break_long_words=False))
324
break_long_words=False))
227
325
strings.append('')
228
326
return '\n'.join(strings)
230
328
def __eq__(self, other):
231
return (type(other) == type(self) and
232
other.__dict__ == self.__dict__)
329
return (isinstance(other, type(self)) and other.__dict__ == self.__dict__)
331
def hook_lazy(self, callback_module, callback_member, callback_label):
332
"""Lazily register a callback to be called when this HookPoint fires.
334
:param callback_module: Module of the callable to use when this
336
:param callback_member: Member name of the callback.
337
:param callback_label: A label to show in the UI while this callback is
340
obj_getter = registry._LazyObjectGetter(callback_module,
342
self._callbacks.append((obj_getter, callback_label))
234
344
def hook(self, callback, callback_label):
235
345
"""Register a callback to be called when this HookPoint fires.