/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/plugin.py

  • Committer: Richard Wilbur
  • Date: 2016-02-04 19:07:28 UTC
  • mto: This revision was merged to the branch mainline in revision 6618.
  • Revision ID: richard.wilbur@gmail.com-20160204190728-p0zvfii6zase0fw7
Update COPYING.txt from the original http://www.gnu.org/licenses/gpl-2.0.txt  (Only differences were in whitespace.)  Thanks to Petr Stodulka for pointing out the discrepancy.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2011 Canonical Ltd, 2017 Breezy developers
 
1
# Copyright (C) 2005-2011 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
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
16
16
 
17
 
"""Breezy plugin support.
18
 
 
19
 
Which plugins to load can be configured by setting these environment variables:
20
 
 
21
 
- BRZ_PLUGIN_PATH: Paths to look for plugins in.
22
 
- BRZ_DISABLE_PLUGINS: Plugin names to block from being loaded.
23
 
- BRZ_PLUGINS_AT: Name and paths for plugins to load from specific locations.
24
 
 
25
 
The interfaces this module exports include:
26
 
 
27
 
- disable_plugins: Load no plugins and stop future automatic loading.
28
 
- load_plugins: Load all plugins that can be found in configuration.
29
 
- describe_plugins: Generate text for each loaded (or failed) plugin.
30
 
- extend_path: Mechanism by which the plugins package path is set.
31
 
- plugin_name: Gives unprefixed name of a plugin module.
 
17
"""bzr python plugin support.
 
18
 
 
19
When load_plugins() is invoked, any python module in any directory in
 
20
$BZR_PLUGIN_PATH will be imported.  The module will be imported as
 
21
'bzrlib.plugins.$BASENAME(PLUGIN)'.  In the plugin's main body, it should
 
22
update any bzrlib registries it wants to extend.
32
23
 
33
24
See the plugin-api developer documentation for information about writing
34
25
plugins.
 
26
 
 
27
BZR_PLUGIN_PATH is also honoured for any plugins imported via
 
28
'import bzrlib.plugins.PLUGINNAME', as long as set_plugins_path has been
 
29
called.
35
30
"""
36
31
 
37
32
from __future__ import absolute_import
38
33
 
39
34
import os
40
 
import re
41
35
import sys
42
36
 
43
 
import breezy
44
 
from . import osutils
 
37
from bzrlib import osutils
45
38
 
46
 
from .lazy_import import lazy_import
 
39
from bzrlib.lazy_import import lazy_import
47
40
lazy_import(globals(), """
48
41
import imp
49
 
import importlib
50
 
from importlib import util as importlib_util
 
42
import re
 
43
import types
51
44
 
52
 
from breezy import (
 
45
from bzrlib import (
 
46
    _format_version_tuple,
53
47
    config,
54
48
    debug,
55
49
    errors,
56
 
    help_topics,
57
50
    trace,
58
51
    )
 
52
from bzrlib.i18n import gettext
 
53
from bzrlib import plugins as _mod_plugins
59
54
""")
60
55
 
61
56
 
62
 
_MODULE_PREFIX = "breezy.plugins."
63
 
 
64
 
if __debug__ or sys.version_info > (3,):
65
 
    COMPILED_EXT = ".pyc"
66
 
else:
67
 
    COMPILED_EXT = ".pyo"
68
 
 
69
 
 
70
 
def disable_plugins(state=None):
 
57
DEFAULT_PLUGIN_PATH = None
 
58
_loaded = False
 
59
_plugins_disabled = False
 
60
 
 
61
 
 
62
plugin_warnings = {}
 
63
# Map from plugin name, to list of string warnings about eg plugin
 
64
# dependencies.
 
65
 
 
66
 
 
67
def are_plugins_disabled():
 
68
    return _plugins_disabled
 
69
 
 
70
 
 
71
def disable_plugins():
71
72
    """Disable loading plugins.
72
73
 
73
74
    Future calls to load_plugins() will be ignored.
74
 
 
75
 
    :param state: The library state object that records loaded plugins.
76
 
    """
77
 
    if state is None:
78
 
        state = breezy.get_global_state()
79
 
    state.plugins = {}
80
 
 
81
 
 
82
 
def load_plugins(path=None, state=None):
83
 
    """Load breezy plugins.
84
 
 
85
 
    The environment variable BRZ_PLUGIN_PATH is considered a delimited
86
 
    set of paths to look through. Each entry is searched for `*.py`
87
 
    files (and whatever other extensions are used in the platform,
88
 
    such as `*.pyd`).
89
 
 
90
 
    :param path: The list of paths to search for plugins.  By default,
91
 
        it is populated from the __path__ of the breezy.plugins package.
92
 
    :param state: The library state object that records loaded plugins.
93
 
    """
94
 
    if state is None:
95
 
        state = breezy.get_global_state()
96
 
    if getattr(state, 'plugins', None) is not None:
97
 
        # People can make sure plugins are loaded, they just won't be twice
98
 
        return
99
 
 
100
 
    if path is None:
101
 
        # Calls back into extend_path() here
102
 
        from breezy.plugins import __path__ as path
103
 
 
104
 
    state.plugin_warnings = {}
105
 
    _load_plugins(state, path)
106
 
    state.plugins = plugins()
107
 
 
108
 
 
109
 
def plugin_name(module_name):
110
 
    """Gives unprefixed name from module_name or None."""
111
 
    if module_name.startswith(_MODULE_PREFIX):
112
 
        parts = module_name.split(".")
113
 
        if len(parts) > 2:
114
 
            return parts[2]
115
 
    return None
116
 
 
117
 
 
118
 
def extend_path(path, name):
119
 
    """Helper so breezy.plugins can be a sort of namespace package.
120
 
 
121
 
    To be used in similar fashion to pkgutil.extend_path:
122
 
 
123
 
        from breezy.plugins import extend_path
124
 
        __path__ = extend_path(__path__, __name__)
125
 
 
126
 
    Inspects the BRZ_PLUGIN* envvars, sys.path, and the filesystem to find
127
 
    plugins. May mutate sys.modules in order to block plugin loading, and may
128
 
    append a new meta path finder to sys.meta_path for plugins@ loading.
129
 
 
130
 
    Returns a list of paths to import from, as an enhanced object that also
131
 
    contains details of the other configuration used.
132
 
    """
133
 
    blocks = _env_disable_plugins()
134
 
    _block_plugins(blocks)
135
 
 
136
 
    extra_details = _env_plugins_at()
137
 
    _install_importer_if_needed(extra_details)
138
 
 
139
 
    paths = _iter_plugin_paths(_env_plugin_path(), path)
140
 
 
141
 
    return _Path(name, blocks, extra_details, paths)
142
 
 
143
 
 
144
 
class _Path(list):
145
 
    """List type to use as __path__ but containing additional details.
146
 
 
147
 
    Python 3 allows any iterable for __path__ but Python 2 is more fussy.
148
 
    """
149
 
 
150
 
    def __init__(self, package_name, blocked, extra, paths):
151
 
        super(_Path, self).__init__(paths)
152
 
        self.package_name = package_name
153
 
        self.blocked_names = blocked
154
 
        self.extra_details = extra
155
 
 
156
 
    def __repr__(self):
157
 
        return "%s(%r, %r, %r, %s)" % (
158
 
            self.__class__.__name__, self.package_name, self.blocked_names,
159
 
            self.extra_details, list.__repr__(self))
160
 
 
161
 
 
162
 
def _expect_identifier(name, env_key, env_value):
163
 
    """Validate given name from envvar is usable as a Python identifier.
164
 
 
165
 
    Returns the name as a native str, or None if it was invalid.
166
 
 
167
 
    Per PEP 3131 this is no longer strictly correct for Python 3, but as MvL
168
 
    didn't include a neat way to check except eval, this enforces ascii.
169
 
    """
170
 
    if re.match(r"^[A-Za-z_][A-Za-z0-9_]*$", name) is None:
171
 
        trace.warning("Invalid name '%s' in %s='%s'", name, env_key, env_value)
172
 
        return None
173
 
    return str(name)
174
 
 
175
 
 
176
 
def _env_disable_plugins(key='BRZ_DISABLE_PLUGINS'):
177
 
    """Gives list of names for plugins to disable from environ key."""
178
 
    disabled_names = []
179
 
    env = osutils.path_from_environ(key)
180
 
    if env:
181
 
        for name in env.split(os.pathsep):
182
 
            name = _expect_identifier(name, key, env)
183
 
            if name is not None:
184
 
                disabled_names.append(name)
185
 
    return disabled_names
186
 
 
187
 
 
188
 
def _env_plugins_at(key='BRZ_PLUGINS_AT'):
189
 
    """Gives list of names and paths of specific plugins from environ key."""
190
 
    plugin_details = []
191
 
    env = osutils.path_from_environ(key)
192
 
    if env:
193
 
        for pair in env.split(os.pathsep):
194
 
            if '@' in pair:
195
 
                name, path = pair.split('@', 1)
196
 
            else:
197
 
                path = pair
198
 
                name = osutils.basename(path).split('.', 1)[0]
199
 
            name = _expect_identifier(name, key, env)
200
 
            if name is not None:
201
 
                plugin_details.append((name, path))
202
 
    return plugin_details
203
 
 
204
 
 
205
 
def _env_plugin_path(key='BRZ_PLUGIN_PATH'):
206
 
    """Gives list of paths and contexts for plugins from environ key.
207
 
 
208
 
    Each entry is either a specific path to load plugins from and the value
209
 
    'path', or None and one of the three values 'user', 'core', 'site'.
210
 
    """
211
 
    path_details = []
212
 
    env = osutils.path_from_environ(key)
213
 
    defaults = {"user": not env, "core": True, "site": True}
214
 
    if env:
215
 
        # Add paths specified by user in order
216
 
        for p in env.split(os.pathsep):
217
 
            flag, name = p[:1], p[1:]
218
 
            if flag in ("+", "-") and name in defaults:
219
 
                if flag == "+" and defaults[name] is not None:
220
 
                    path_details.append((None, name))
221
 
                defaults[name] = None
222
 
            else:
223
 
                path_details.append((p, 'path'))
224
 
 
225
 
    # Add any remaining default paths
226
 
    for name in ('user', 'core', 'site'):
227
 
        if defaults[name]:
228
 
            path_details.append((None, name))
229
 
 
230
 
    return path_details
231
 
 
232
 
 
233
 
def _iter_plugin_paths(paths_from_env, core_paths):
234
 
    """Generate paths using paths_from_env and core_paths."""
235
 
    # GZ 2017-06-02: This is kinda horrid, should make better.
236
 
    for path, context in paths_from_env:
237
 
        if context == 'path':
238
 
            yield path
239
 
        elif context == 'user':
240
 
            path = get_user_plugin_path()
241
 
            if os.path.isdir(path):
242
 
                yield path
243
 
        elif context == 'core':
244
 
            for path in _get_core_plugin_paths(core_paths):
245
 
                yield path
246
 
        elif context == 'site':
247
 
            for path in _get_site_plugin_paths(sys.path):
248
 
                if os.path.isdir(path):
249
 
                    yield path
250
 
 
251
 
 
252
 
def _install_importer_if_needed(plugin_details):
253
 
    """Install a meta path finder to handle plugin_details if any."""
254
 
    if plugin_details:
255
 
        finder = _PluginsAtFinder(_MODULE_PREFIX, plugin_details)
256
 
        # For Python 3, must insert before default PathFinder to override.
257
 
        sys.meta_path.insert(2, finder)
258
 
 
259
 
 
260
 
def _load_plugins(state, paths):
261
 
    """Do the importing all plugins from paths."""
262
 
    imported_names = set()
263
 
    for name, path in _iter_possible_plugins(paths):
264
 
        if name not in imported_names:
265
 
            msg = _load_plugin_module(name, path)
266
 
            if msg is not None:
267
 
                state.plugin_warnings.setdefault(name, []).append(msg)
268
 
            imported_names.add(name)
269
 
 
270
 
 
271
 
def _block_plugins(names):
272
 
    """Add names to sys.modules to block future imports."""
273
 
    for name in names:
274
 
        package_name = _MODULE_PREFIX + name
275
 
        if sys.modules.get(package_name) is not None:
276
 
            trace.mutter("Blocked plugin %s already loaded.", name)
277
 
        sys.modules[package_name] = None
278
 
 
279
 
 
280
 
def _get_package_init(package_path):
281
 
    """Get path of __init__ file from package_path or None if not a package."""
282
 
    init_path = osutils.pathjoin(package_path, "__init__.py")
283
 
    if os.path.exists(init_path):
284
 
        return init_path
285
 
    init_path = init_path[:-3] + COMPILED_EXT
286
 
    if os.path.exists(init_path):
287
 
        return init_path
288
 
    return None
289
 
 
290
 
 
291
 
def _iter_possible_plugins(plugin_paths):
292
 
    """Generate names and paths of possible plugins from plugin_paths."""
293
 
    # Inspect any from BRZ_PLUGINS_AT first.
294
 
    for name, path in getattr(plugin_paths, "extra_details", ()):
295
 
        yield name, path
296
 
    # Then walk over files and directories in the paths from the package.
297
 
    for path in plugin_paths:
298
 
        if os.path.isfile(path):
299
 
            if path.endswith(".zip"):
300
 
                trace.mutter("Don't yet support loading plugins from zip.")
301
 
        else:
302
 
            for name, path in _walk_modules(path):
303
 
                yield name, path
304
 
 
305
 
 
306
 
def _walk_modules(path):
307
 
    """Generate name and path of modules and packages on path."""
308
 
    for root, dirs, files in os.walk(path):
309
 
        files.sort()
310
 
        for f in files:
311
 
            if f[:2] != "__":
312
 
                if f.endswith((".py", COMPILED_EXT)):
313
 
                    yield f.rsplit(".", 1)[0], root
314
 
        dirs.sort()
315
 
        for d in dirs:
316
 
            if d[:2] != "__":
317
 
                package_dir = osutils.pathjoin(root, d)
318
 
                fullpath = _get_package_init(package_dir)
319
 
                if fullpath is not None:
320
 
                    yield d, package_dir
321
 
        # Don't descend into subdirectories
322
 
        del dirs[:]
323
 
 
324
 
 
325
 
def describe_plugins(show_paths=False, state=None):
 
75
    """
 
76
    global _plugins_disabled
 
77
    _plugins_disabled = True
 
78
    load_plugins([])
 
79
 
 
80
 
 
81
def describe_plugins(show_paths=False):
326
82
    """Generate text description of plugins.
327
83
 
328
 
    Includes both those that have loaded, and those that failed to load.
 
84
    Includes both those that have loaded, and those that failed to 
 
85
    load.
329
86
 
330
 
    :param show_paths: If true, include the plugin path.
331
 
    :param state: The library state object to inspect.
 
87
    :param show_paths: If true,
332
88
    :returns: Iterator of text lines (including newlines.)
333
89
    """
334
 
    if state is None:
335
 
        state = breezy.get_global_state()
336
 
    loaded_plugins = getattr(state, 'plugins', {})
337
 
    plugin_warnings = set(getattr(state, 'plugin_warnings', []))
338
 
    all_names = sorted(set(loaded_plugins.keys()).union(plugin_warnings))
 
90
    from inspect import getdoc
 
91
    loaded_plugins = plugins()
 
92
    all_names = sorted(list(set(
 
93
        loaded_plugins.keys() + plugin_warnings.keys())))
339
94
    for name in all_names:
340
95
        if name in loaded_plugins:
341
96
            plugin = loaded_plugins[name]
343
98
            if version == 'unknown':
344
99
                version = ''
345
100
            yield '%s %s\n' % (name, version)
346
 
            d = plugin.module.__doc__
 
101
            d = getdoc(plugin.module)
347
102
            if d:
348
103
                doc = d.split('\n')[0]
349
104
            else:
351
106
            yield ("  %s\n" % doc)
352
107
            if show_paths:
353
108
                yield ("   %s\n" % plugin.path())
 
109
            del plugin
354
110
        else:
355
111
            yield "%s (failed to load)\n" % name
356
 
        if name in state.plugin_warnings:
357
 
            for line in state.plugin_warnings[name]:
 
112
        if name in plugin_warnings:
 
113
            for line in plugin_warnings[name]:
358
114
                yield "  ** " + line + '\n'
359
115
        yield '\n'
360
116
 
361
117
 
362
 
def _get_core_plugin_paths(existing_paths):
363
 
    """Generate possible locations for plugins based on existing_paths."""
364
 
    if getattr(sys, 'frozen', False):
 
118
def _strip_trailing_sep(path):
 
119
    return path.rstrip("\\/")
 
120
 
 
121
 
 
122
def _get_specific_plugin_paths(paths):
 
123
    """Returns the plugin paths from a string describing the associations.
 
124
 
 
125
    :param paths: A string describing the paths associated with the plugins.
 
126
 
 
127
    :returns: A list of (plugin name, path) tuples.
 
128
 
 
129
    For example, if paths is my_plugin@/test/my-test:her_plugin@/production/her,
 
130
    [('my_plugin', '/test/my-test'), ('her_plugin', '/production/her')] 
 
131
    will be returned.
 
132
 
 
133
    Note that ':' in the example above depends on the os.
 
134
    """
 
135
    if not paths:
 
136
        return []
 
137
    specs = []
 
138
    for spec in paths.split(os.pathsep):
 
139
        try:
 
140
            name, path = spec.split('@')
 
141
        except ValueError:
 
142
            raise errors.BzrCommandError(gettext(
 
143
                '"%s" is not a valid <plugin_name>@<plugin_path> description ')
 
144
                % spec)
 
145
        specs.append((name, path))
 
146
    return specs
 
147
 
 
148
 
 
149
def set_plugins_path(path=None):
 
150
    """Set the path for plugins to be loaded from.
 
151
 
 
152
    :param path: The list of paths to search for plugins.  By default,
 
153
        path will be determined using get_standard_plugins_path.
 
154
        if path is [], no plugins can be loaded.
 
155
    """
 
156
    if path is None:
 
157
        path = get_standard_plugins_path()
 
158
    _mod_plugins.__path__ = path
 
159
    PluginImporter.reset()
 
160
    # Set up a blacklist for disabled plugins
 
161
    disabled_plugins = os.environ.get('BZR_DISABLE_PLUGINS', None)
 
162
    if disabled_plugins is not None:
 
163
        for name in disabled_plugins.split(os.pathsep):
 
164
            PluginImporter.blacklist.add('bzrlib.plugins.' + name)
 
165
    # Set up a the specific paths for plugins
 
166
    for plugin_name, plugin_path in _get_specific_plugin_paths(os.environ.get(
 
167
            'BZR_PLUGINS_AT', None)):
 
168
            PluginImporter.specific_paths[
 
169
                'bzrlib.plugins.%s' % plugin_name] = plugin_path
 
170
    return path
 
171
 
 
172
 
 
173
def _append_new_path(paths, new_path):
 
174
    """Append a new path if it set and not already known."""
 
175
    if new_path is not None and new_path not in paths:
 
176
        paths.append(new_path)
 
177
    return paths
 
178
 
 
179
 
 
180
def get_core_plugin_path():
 
181
    core_path = None
 
182
    bzr_exe = bool(getattr(sys, 'frozen', None))
 
183
    if bzr_exe:    # expand path for bzr.exe
365
184
        # We need to use relative path to system-wide plugin
366
 
        # directory because breezy from standalone brz.exe
 
185
        # directory because bzrlib from standalone bzr.exe
367
186
        # could be imported by another standalone program
368
 
        # (e.g. brz-config; or TortoiseBzr/Olive if/when they
 
187
        # (e.g. bzr-config; or TortoiseBzr/Olive if/when they
369
188
        # will become standalone exe). [bialix 20071123]
370
189
        # __file__ typically is
371
 
        # C:\Program Files\Bazaar\lib\library.zip\breezy\plugin.pyc
 
190
        # C:\Program Files\Bazaar\lib\library.zip\bzrlib\plugin.pyc
372
191
        # then plugins directory is
373
192
        # C:\Program Files\Bazaar\plugins
374
193
        # so relative path is ../../../plugins
375
 
        yield osutils.abspath(osutils.pathjoin(
376
 
            osutils.dirname(__file__), '../../../plugins'))
 
194
        core_path = osutils.abspath(osutils.pathjoin(
 
195
                osutils.dirname(__file__), '../../../plugins'))
377
196
    else:     # don't look inside library.zip
378
 
        for path in existing_paths:
379
 
            yield path
380
 
 
381
 
 
382
 
def _get_site_plugin_paths(sys_paths):
383
 
    """Generate possible locations for plugins from given sys_paths."""
384
 
    for path in sys_paths:
385
 
        if os.path.basename(path) in ('dist-packages', 'site-packages'):
386
 
            yield osutils.pathjoin(path, 'breezy', 'plugins')
 
197
        # search the plugin path before the bzrlib installed dir
 
198
        core_path = os.path.dirname(_mod_plugins.__file__)
 
199
    return core_path
 
200
 
 
201
 
 
202
def get_site_plugin_path():
 
203
    """Returns the path for the site installed plugins."""
 
204
    if sys.platform == 'win32':
 
205
        # We don't have (yet) a good answer for windows since that is certainly
 
206
        # related to the way we build the installers. -- vila20090821
 
207
        return None
 
208
    site_path = None
 
209
    try:
 
210
        from distutils.sysconfig import get_python_lib
 
211
    except ImportError:
 
212
        # If distutuils is not available, we just don't know where they are
 
213
        pass
 
214
    else:
 
215
        site_path = osutils.pathjoin(get_python_lib(), 'bzrlib', 'plugins')
 
216
    return site_path
387
217
 
388
218
 
389
219
def get_user_plugin_path():
390
220
    return osutils.pathjoin(config.config_dir(), 'plugins')
391
221
 
392
222
 
393
 
def record_plugin_warning(warning_message):
 
223
def get_standard_plugins_path():
 
224
    """Determine a plugin path suitable for general use."""
 
225
    # Ad-Hoc default: core is not overriden by site but user can overrides both
 
226
    # The rationale is that:
 
227
    # - 'site' comes last, because these plugins should always be available and
 
228
    #   are supposed to be in sync with the bzr installed on site.
 
229
    # - 'core' comes before 'site' so that running bzr from sources or a user
 
230
    #   installed version overrides the site version.
 
231
    # - 'user' comes first, because... user is always right.
 
232
    # - the above rules clearly defines which plugin version will be loaded if
 
233
    #   several exist. Yet, it is sometimes desirable to disable some directory
 
234
    #   so that a set of plugins is disabled as once. This can be done via
 
235
    #   -site, -core, -user.
 
236
 
 
237
    env_paths = os.environ.get('BZR_PLUGIN_PATH', '+user').split(os.pathsep)
 
238
    defaults = ['+core', '+site']
 
239
 
 
240
    # The predefined references
 
241
    refs = dict(core=get_core_plugin_path(),
 
242
                site=get_site_plugin_path(),
 
243
                user=get_user_plugin_path())
 
244
 
 
245
    # Unset paths that should be removed
 
246
    for k,v in refs.iteritems():
 
247
        removed = '-%s' % k
 
248
        # defaults can never mention removing paths as that will make it
 
249
        # impossible for the user to revoke these removals.
 
250
        if removed in env_paths:
 
251
            env_paths.remove(removed)
 
252
            refs[k] = None
 
253
 
 
254
    # Expand references
 
255
    paths = []
 
256
    for p in env_paths + defaults:
 
257
        if p.startswith('+'):
 
258
            # Resolve references if they are known
 
259
            try:
 
260
                p = refs[p[1:]]
 
261
            except KeyError:
 
262
                # Leave them untouched so user can still use paths starting
 
263
                # with '+'
 
264
                pass
 
265
        _append_new_path(paths, p)
 
266
 
 
267
    # Get rid of trailing slashes, since Python can't handle them when
 
268
    # it tries to import modules.
 
269
    paths = map(_strip_trailing_sep, paths)
 
270
    return paths
 
271
 
 
272
 
 
273
def load_plugins(path=None):
 
274
    """Load bzrlib plugins.
 
275
 
 
276
    The environment variable BZR_PLUGIN_PATH is considered a delimited
 
277
    set of paths to look through. Each entry is searched for `*.py`
 
278
    files (and whatever other extensions are used in the platform,
 
279
    such as `*.pyd`).
 
280
 
 
281
    load_from_path() provides the underlying mechanism and is called with
 
282
    the default directory list to provide the normal behaviour.
 
283
 
 
284
    :param path: The list of paths to search for plugins.  By default,
 
285
        path will be determined using get_standard_plugins_path.
 
286
        if path is [], no plugins can be loaded.
 
287
    """
 
288
    global _loaded
 
289
    if _loaded:
 
290
        # People can make sure plugins are loaded, they just won't be twice
 
291
        return
 
292
    _loaded = True
 
293
 
 
294
    # scan for all plugins in the path.
 
295
    load_from_path(set_plugins_path(path))
 
296
 
 
297
 
 
298
def load_from_path(dirs):
 
299
    """Load bzrlib plugins found in each dir in dirs.
 
300
 
 
301
    Loading a plugin means importing it into the python interpreter.
 
302
    The plugin is expected to make calls to register commands when
 
303
    it's loaded (or perhaps access other hooks in future.)
 
304
 
 
305
    Plugins are loaded into bzrlib.plugins.NAME, and can be found there
 
306
    for future reference.
 
307
 
 
308
    The python module path for bzrlib.plugins will be modified to be 'dirs'.
 
309
    """
 
310
    # Explicitly load the plugins with a specific path
 
311
    for fullname, path in PluginImporter.specific_paths.iteritems():
 
312
        name = fullname[len('bzrlib.plugins.'):]
 
313
        _load_plugin_module(name, path)
 
314
 
 
315
    # We need to strip the trailing separators here as well as in the
 
316
    # set_plugins_path function because calling code can pass anything in to
 
317
    # this function, and since it sets plugins.__path__, it should set it to
 
318
    # something that will be valid for Python to use (in case people try to
 
319
    # run "import bzrlib.plugins.PLUGINNAME" after calling this function).
 
320
    _mod_plugins.__path__ = map(_strip_trailing_sep, dirs)
 
321
    for d in dirs:
 
322
        if not d:
 
323
            continue
 
324
        trace.mutter('looking for plugins in %s', d)
 
325
        if os.path.isdir(d):
 
326
            load_from_dir(d)
 
327
 
 
328
 
 
329
# backwards compatability: load_from_dirs was the old name
 
330
# This was changed in 0.15
 
331
load_from_dirs = load_from_path
 
332
 
 
333
 
 
334
def _find_plugin_module(dir, name):
 
335
    """Check if there is a valid python module that can be loaded as a plugin.
 
336
 
 
337
    :param dir: The directory where the search is performed.
 
338
    :param path: An existing file path, either a python file or a package
 
339
        directory.
 
340
 
 
341
    :return: (name, path, description) name is the module name, path is the
 
342
        file to load and description is the tuple returned by
 
343
        imp.get_suffixes().
 
344
    """
 
345
    path = osutils.pathjoin(dir, name)
 
346
    if os.path.isdir(path):
 
347
        # Check for a valid __init__.py file, valid suffixes depends on -O and
 
348
        # can be .py, .pyc and .pyo
 
349
        for suffix, mode, kind in imp.get_suffixes():
 
350
            if kind not in (imp.PY_SOURCE, imp.PY_COMPILED):
 
351
                # We don't recognize compiled modules (.so, .dll, etc)
 
352
                continue
 
353
            init_path = osutils.pathjoin(path, '__init__' + suffix)
 
354
            if os.path.isfile(init_path):
 
355
                return name, init_path, (suffix, mode, kind)
 
356
    else:
 
357
        for suffix, mode, kind in imp.get_suffixes():
 
358
            if name.endswith(suffix):
 
359
                # Clean up the module name
 
360
                name = name[:-len(suffix)]
 
361
                if kind == imp.C_EXTENSION and name.endswith('module'):
 
362
                    name = name[:-len('module')]
 
363
                return name, path, (suffix, mode, kind)
 
364
    # There is no python module here
 
365
    return None, None, (None, None, None)
 
366
 
 
367
 
 
368
def record_plugin_warning(plugin_name, warning_message):
394
369
    trace.mutter(warning_message)
395
 
    return warning_message
 
370
    plugin_warnings.setdefault(plugin_name, []).append(warning_message)
396
371
 
397
372
 
398
373
def _load_plugin_module(name, dir):
399
 
    """Load plugin by name.
 
374
    """Load plugin name from dir.
400
375
 
401
 
    :param name: The plugin name in the breezy.plugins namespace.
 
376
    :param name: The plugin name in the bzrlib.plugins namespace.
402
377
    :param dir: The directory the plugin is loaded from for error messages.
403
378
    """
404
 
    if _MODULE_PREFIX + name in sys.modules:
 
379
    if ('bzrlib.plugins.%s' % name) in PluginImporter.blacklist:
405
380
        return
406
381
    try:
407
 
        __import__(_MODULE_PREFIX + name)
408
 
    except errors.IncompatibleVersion as e:
 
382
        exec "import bzrlib.plugins.%s" % name in {}
 
383
    except KeyboardInterrupt:
 
384
        raise
 
385
    except errors.IncompatibleAPI, e:
409
386
        warning_message = (
410
 
            "Unable to load plugin %r. It supports %s "
411
 
            "versions %r but the current version is %s" %
412
 
            (name, e.api.__name__, e.wanted, e.current))
413
 
        return record_plugin_warning(warning_message)
414
 
    except Exception as e:
415
 
        trace.log_exception_quietly()
416
 
        if 'error' in debug.debug_flags:
417
 
            trace.print_exception(sys.exc_info(), sys.stderr)
418
 
        # GZ 2017-06-02: Move this name checking up a level, no point trying
419
 
        # to import things with bad names.
420
 
        if re.search('\\.|-| ', name):
 
387
            "Unable to load plugin %r. It requested API version "
 
388
            "%s of module %s but the minimum exported version is %s, and "
 
389
            "the maximum is %s" %
 
390
            (name, e.wanted, e.api, e.minimum, e.current))
 
391
        record_plugin_warning(name, warning_message)
 
392
    except Exception, e:
 
393
        trace.warning("%s" % e)
 
394
        if re.search('\.|-| ', name):
421
395
            sanitised_name = re.sub('[-. ]', '_', name)
422
 
            if sanitised_name.startswith('brz_'):
423
 
                sanitised_name = sanitised_name[len('brz_'):]
 
396
            if sanitised_name.startswith('bzr_'):
 
397
                sanitised_name = sanitised_name[len('bzr_'):]
424
398
            trace.warning("Unable to load %r in %r as a plugin because the "
425
399
                    "file path isn't a valid module name; try renaming "
426
400
                    "it to %r." % (name, dir, sanitised_name))
427
401
        else:
428
 
            return record_plugin_warning(
429
 
                'Unable to load plugin %r from %r: %s' % (name, dir, e))
 
402
            record_plugin_warning(
 
403
                name,
 
404
                'Unable to load plugin %r from %r' % (name, dir))
 
405
        trace.log_exception_quietly()
 
406
        if 'error' in debug.debug_flags:
 
407
            trace.print_exception(sys.exc_info(), sys.stderr)
 
408
 
 
409
 
 
410
def load_from_dir(d):
 
411
    """Load the plugins in directory d.
 
412
 
 
413
    d must be in the plugins module path already.
 
414
    This function is called once for each directory in the module path.
 
415
    """
 
416
    plugin_names = set()
 
417
    for p in os.listdir(d):
 
418
        name, path, desc = _find_plugin_module(d, p)
 
419
        if name is not None:
 
420
            if name == '__init__':
 
421
                # We do nothing with the __init__.py file in directories from
 
422
                # the bzrlib.plugins module path, we may want to, one day
 
423
                # -- vila 20100316.
 
424
                continue # We don't load __init__.py in the plugins dirs
 
425
            elif getattr(_mod_plugins, name, None) is not None:
 
426
                # The module has already been loaded from another directory
 
427
                # during a previous call.
 
428
                # FIXME: There should be a better way to report masked plugins
 
429
                # -- vila 20100316
 
430
                trace.mutter('Plugin name %s already loaded', name)
 
431
            else:
 
432
                plugin_names.add(name)
 
433
 
 
434
    for name in plugin_names:
 
435
        _load_plugin_module(name, d)
430
436
 
431
437
 
432
438
def plugins():
435
441
    Each item in the dictionary is a PlugIn object.
436
442
    """
437
443
    result = {}
438
 
    for fullname in sys.modules:
439
 
        if fullname.startswith(_MODULE_PREFIX):
440
 
            name = fullname[len(_MODULE_PREFIX):]
441
 
            if not "." in name and sys.modules[fullname] is not None:
442
 
                result[name] = PlugIn(name, sys.modules[fullname])
 
444
    for name, plugin in _mod_plugins.__dict__.items():
 
445
        if isinstance(plugin, types.ModuleType):
 
446
            result[name] = PlugIn(name, plugin)
443
447
    return result
444
448
 
445
449
 
446
 
def get_loaded_plugin(name):
447
 
    """Retrieve an already loaded plugin.
448
 
 
449
 
    Returns None if there is no such plugin loaded
450
 
    """
451
 
    try:
452
 
        module = sys.modules[_MODULE_PREFIX + name]
453
 
    except KeyError:
454
 
        return None
455
 
    if module is None:
456
 
        return None
457
 
    return PlugIn(name, module)
458
 
 
459
 
 
460
 
def format_concise_plugin_list(state=None):
 
450
def format_concise_plugin_list():
461
451
    """Return a string holding a concise list of plugins and their version.
462
452
    """
463
 
    if state is None:
464
 
        state = breezy.get_global_state()
465
453
    items = []
466
 
    for name, a_plugin in sorted(state.plugins.items()):
 
454
    for name, a_plugin in sorted(plugins().items()):
467
455
        items.append("%s[%s]" %
468
456
            (name, a_plugin.__version__))
469
457
    return ', '.join(items)
470
458
 
471
459
 
 
460
 
472
461
class PluginsHelpIndex(object):
473
462
    """A help index that returns help topics for plugins."""
474
463
 
488
477
            return []
489
478
        if topic.startswith(self.prefix):
490
479
            topic = topic[len(self.prefix):]
491
 
        plugin_module_name = _MODULE_PREFIX + topic
 
480
        plugin_module_name = 'bzrlib.plugins.%s' % topic
492
481
        try:
493
482
            module = sys.modules[plugin_module_name]
494
483
        except KeyError:
519
508
            result = self.module.__doc__
520
509
        if result[-1] != '\n':
521
510
            result += '\n'
 
511
        from bzrlib import help_topics
522
512
        result += help_topics._format_see_also(additional_see_also)
523
513
        return result
524
514
 
525
515
    def get_help_topic(self):
526
516
        """Return the module help topic: its basename."""
527
 
        return self.module.__name__[len(_MODULE_PREFIX):]
 
517
        return self.module.__name__[len('bzrlib.plugins.'):]
528
518
 
529
519
 
530
520
class PlugIn(object):
531
 
    """The breezy representation of a plugin.
 
521
    """The bzrlib representation of a plugin.
532
522
 
533
523
    The PlugIn object provides a way to manipulate a given plugin module.
534
524
    """
544
534
            return os.path.abspath(self.module.__path__[0])
545
535
        elif getattr(self.module, '__file__', None) is not None:
546
536
            path = os.path.abspath(self.module.__file__)
547
 
            if path[-4:] == COMPILED_EXT:
 
537
            if path[-4:] in ('.pyc', '.pyo'):
548
538
                pypath = path[:-4] + '.py'
549
539
                if os.path.isfile(pypath):
550
540
                    path = pypath
552
542
        else:
553
543
            return repr(self.module)
554
544
 
555
 
    def __repr__(self):
556
 
        return "<%s.%s name=%s, module=%s>" % (
557
 
            self.__class__.__module__, self.__class__.__name__,
 
545
    def __str__(self):
 
546
        return "<%s.%s object at %s, name=%s, module=%s>" % (
 
547
            self.__class__.__module__, self.__class__.__name__, id(self),
558
548
            self.name, self.module)
559
549
 
 
550
    __repr__ = __str__
 
551
 
560
552
    def test_suite(self):
561
553
        """Return the plugin's test suite."""
562
554
        if getattr(self.module, 'test_suite', None) is not None:
569
561
 
570
562
        :param loader: The custom loader that should be used to load additional
571
563
            tests.
 
564
 
572
565
        """
573
566
        if getattr(self.module, 'load_tests', None) is not None:
574
567
            return loader.loadTestsFromModule(self.module)
580
573
        version_info = getattr(self.module, 'version_info', None)
581
574
        if version_info is not None:
582
575
            try:
583
 
                if isinstance(version_info, str):
 
576
                if isinstance(version_info, types.StringType):
584
577
                    version_info = version_info.split('.')
585
578
                elif len(version_info) == 3:
586
579
                    version_info = tuple(version_info) + ('final', 0)
587
 
            except TypeError:
 
580
            except TypeError, e:
588
581
                # The given version_info isn't even iteratible
589
582
                trace.log_exception_quietly()
590
583
                version_info = (version_info,)
591
584
        return version_info
592
585
 
593
 
    @property
594
 
    def __version__(self):
 
586
    def _get__version__(self):
595
587
        version_info = self.version_info()
596
588
        if version_info is None or len(version_info) == 0:
597
589
            return "unknown"
598
590
        try:
599
 
            version_string = breezy._format_version_tuple(version_info)
600
 
        except (ValueError, TypeError, IndexError):
 
591
            version_string = _format_version_tuple(version_info)
 
592
        except (ValueError, TypeError, IndexError), e:
601
593
            trace.log_exception_quietly()
602
 
            # Try to show something for the version anyway
 
594
            # try to return something usefull for bad plugins, in stead of
 
595
            # stack tracing.
603
596
            version_string = '.'.join(map(str, version_info))
604
597
        return version_string
605
598
 
606
 
 
607
 
class _PluginsAtFinder(object):
608
 
    """Meta path finder to support BRZ_PLUGINS_AT configuration."""
609
 
 
610
 
    def __init__(self, prefix, names_and_paths):
611
 
        self.prefix = prefix
612
 
        self.names_to_path = dict((prefix + n, p) for n, p in names_and_paths)
613
 
 
614
 
    def __repr__(self):
615
 
        return "<%s %r>" % (self.__class__.__name__, self.prefix)
616
 
 
617
 
    def find_spec(self, fullname, paths, target=None):
618
 
        """New module spec returning find method."""
619
 
        if fullname not in self.names_to_path:
620
 
            return None
621
 
        path = self.names_to_path[fullname]
622
 
        if os.path.isdir(path):
623
 
            path = _get_package_init(path)
624
 
            if path is None:
625
 
                # GZ 2017-06-02: Any reason to block loading of the name from
626
 
                # further down the path like this?
627
 
                raise ImportError("Not loading namespace package %s as %s" % (
628
 
                    path, fullname))
629
 
        return importlib_util.spec_from_file_location(fullname, path)
630
 
 
631
 
    def find_module(self, fullname, path):
632
 
        """Old PEP 302 import hook find_module method."""
633
 
        if fullname not in self.names_to_path:
634
 
            return None
635
 
        return _LegacyLoader(self.names_to_path[fullname])
636
 
 
637
 
 
638
 
class _LegacyLoader(object):
639
 
    """Source loader implementation for Python versions without importlib."""
640
 
 
641
 
    def __init__(self, filepath):
642
 
        self.filepath = filepath
643
 
 
644
 
    def __repr__(self):
645
 
        return "<%s %r>" % (self.__class__.__name__, self.filepath)
 
599
    __version__ = property(_get__version__)
 
600
 
 
601
 
 
602
class _PluginImporter(object):
 
603
    """An importer tailored to bzr specific needs.
 
604
 
 
605
    This is a singleton that takes care of:
 
606
    - disabled plugins specified in 'blacklist',
 
607
    - plugins that needs to be loaded from specific directories.
 
608
    """
 
609
 
 
610
    def __init__(self):
 
611
        self.reset()
 
612
 
 
613
    def reset(self):
 
614
        self.blacklist = set()
 
615
        self.specific_paths = {}
 
616
 
 
617
    def find_module(self, fullname, parent_path=None):
 
618
        """Search a plugin module.
 
619
 
 
620
        Disabled plugins raise an import error, plugins with specific paths
 
621
        returns a specific loader.
 
622
 
 
623
        :return: None if the plugin doesn't need special handling, self
 
624
            otherwise.
 
625
        """
 
626
        if not fullname.startswith('bzrlib.plugins.'):
 
627
            return None
 
628
        if fullname in self.blacklist:
 
629
            raise ImportError('%s is disabled' % fullname)
 
630
        if fullname in self.specific_paths:
 
631
            return self
 
632
        return None
646
633
 
647
634
    def load_module(self, fullname):
648
635
        """Load a plugin from a specific directory (or file)."""
649
 
        plugin_path = self.filepath
 
636
        # We are called only for specific paths
 
637
        plugin_path = self.specific_paths[fullname]
650
638
        loading_path = None
651
639
        if os.path.isdir(plugin_path):
652
 
            init_path = _get_package_init(plugin_path)
653
 
            if init_path is not None:
654
 
                loading_path = plugin_path
655
 
                suffix = ''
656
 
                mode = ''
657
 
                kind = imp.PKG_DIRECTORY
 
640
            for suffix, mode, kind in imp.get_suffixes():
 
641
                if kind not in (imp.PY_SOURCE, imp.PY_COMPILED):
 
642
                    # We don't recognize compiled modules (.so, .dll, etc)
 
643
                    continue
 
644
                init_path = osutils.pathjoin(plugin_path, '__init__' + suffix)
 
645
                if os.path.isfile(init_path):
 
646
                    # We've got a module here and load_module needs specific
 
647
                    # parameters.
 
648
                    loading_path = plugin_path
 
649
                    suffix = ''
 
650
                    mode = ''
 
651
                    kind = imp.PKG_DIRECTORY
 
652
                    break
658
653
        else:
659
654
            for suffix, mode, kind in imp.get_suffixes():
660
655
                if plugin_path.endswith(suffix):
675
670
        finally:
676
671
            if f is not None:
677
672
                f.close()
 
673
 
 
674
 
 
675
# Install a dedicated importer for plugins requiring special handling
 
676
PluginImporter = _PluginImporter()
 
677
sys.meta_path.append(PluginImporter)