summaryrefslogtreecommitdiffstats
path: root/scripts/lib/wic/plugin.py
blob: c200822af74216172a510f8b2df606244f03df6f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
#!/usr/bin/env python -tt
#
# Copyright (c) 2011 Intel, Inc.
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the Free
# Software Foundation; version 2 of the License
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
# or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
# for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc., 59
# Temple Place - Suite 330, Boston, MA 02111-1307, USA.

import os
import sys
import logging

from wic import pluginbase, WicError
from wic.utils.misc import get_bitbake_var

PLUGIN_TYPES = ["imager", "source"]

PLUGIN_DIR = "/lib/wic/plugins" # relative to scripts
SCRIPTS_PLUGIN_DIR = "scripts" + PLUGIN_DIR

logger = logging.getLogger('wic')

class PluginMgr:
    plugin_dirs = {}
    wic_path = os.path.dirname(__file__)
    eos = wic_path.rfind('scripts') + len('scripts')
    scripts_path = wic_path[:eos]
    plugin_dir = scripts_path + PLUGIN_DIR
    layers_path = None

    @classmethod
    def _build_plugin_dir_list(cls, plugin_dir, ptype):
        if cls.layers_path is None:
            cls.layers_path = get_bitbake_var("BBLAYERS")
        layer_dirs = []

        if cls.layers_path is not None:
            for layer_path in cls.layers_path.split():
                path = os.path.join(layer_path, SCRIPTS_PLUGIN_DIR, ptype)
                layer_dirs.append(path)

        path = os.path.join(plugin_dir, ptype)
        layer_dirs.append(path)

        return layer_dirs

    @classmethod
    def append_dirs(cls, dirs):
        for path in dirs:
            cls._add_plugindir(path)

        # load all the plugins AGAIN
        cls._load_all()

    @classmethod
    def _add_plugindir(cls, path):
        path = os.path.abspath(os.path.expanduser(path))

        if not os.path.isdir(path):
            return

        if path not in cls.plugin_dirs:
            cls.plugin_dirs[path] = False
            # the value True/False means "loaded"

    @classmethod
    def _load_all(cls):
        for (pdir, loaded) in cls.plugin_dirs.items():
            if loaded:
                continue

            sys.path.insert(0, pdir)
            for mod in [x[:-3] for x in os.listdir(pdir) if x.endswith(".py")]:
                if mod and mod != '__init__':
                    if mod in sys.modules:
                        logger.warning("Module %s already exists, skip", mod)
                    else:
                        try:
                            pymod = __import__(mod)
                            cls.plugin_dirs[pdir] = True
                            logger.debug("Plugin module %s:%s imported",
                                         mod, pymod.__file__)
                        except ImportError as err:
                            logger.warning('Failed to load plugin %s/%s: %s',
                                           os.path.basename(pdir), mod, err)

            del sys.path[0]

    @classmethod
    def get_plugins(cls, ptype):
        """ the return value is dict of name:class pairs """

        if ptype not in PLUGIN_TYPES:
            raise WicError('%s is not valid plugin type' % ptype)

        plugins_dir = cls._build_plugin_dir_list(cls.plugin_dir, ptype)

        cls.append_dirs(plugins_dir)

        return pluginbase.get_plugins(ptype)

    @classmethod
    def get_plugin_methods(cls, ptype, pname, methods):
        """
        The methods param is a dict with the method names to find.  On
        return, the dict values will be filled in with pointers to the
        corresponding methods.  If one or more methods are not found,
        None is returned.
        """
        result = {}
        plugin = cls.get_plugins(ptype).get(pname)
        for method in methods:
            if not hasattr(plugin, method):
                raise WicError("Unimplemented %s plugin interface for: %s" %
                               (method, pname))
            result[method] = getattr(plugin, method)
        return result