| import fnmatch |
| import logging |
| import os |
| import shutil |
| import sys |
| import tempfile |
| |
| import bb.utils |
| |
| from bblayers.common import LayerPlugin |
| |
| logger = logging.getLogger('bitbake-layers') |
| |
| |
| def plugin_init(plugins): |
| return ActionPlugin() |
| |
| |
| class ActionPlugin(LayerPlugin): |
| def do_add_layer(self, args): |
| """Add one or more layers to bblayers.conf.""" |
| layerdirs = [os.path.abspath(ldir) for ldir in args.layerdir] |
| |
| for layerdir in layerdirs: |
| if not os.path.exists(layerdir): |
| sys.stderr.write("Specified layer directory %s doesn't exist\n" % layerdir) |
| return 1 |
| |
| layer_conf = os.path.join(layerdir, 'conf', 'layer.conf') |
| if not os.path.exists(layer_conf): |
| sys.stderr.write("Specified layer directory %s doesn't contain a conf/layer.conf file\n" % layerdir) |
| return 1 |
| |
| bblayers_conf = os.path.join('conf', 'bblayers.conf') |
| if not os.path.exists(bblayers_conf): |
| sys.stderr.write("Unable to find bblayers.conf\n") |
| return 1 |
| |
| # Back up bblayers.conf to tempdir before we add layers |
| tempdir = tempfile.mkdtemp() |
| backup = tempdir + "/bblayers.conf.bak" |
| shutil.copy2(bblayers_conf, backup) |
| |
| try: |
| notadded, _ = bb.utils.edit_bblayers_conf(bblayers_conf, layerdirs, None) |
| if not (args.force or notadded): |
| try: |
| self.tinfoil.parseRecipes() |
| except bb.tinfoil.TinfoilUIException: |
| # Restore the back up copy of bblayers.conf |
| shutil.copy2(backup, bblayers_conf) |
| bb.fatal("Parse failure with the specified layer added") |
| else: |
| for item in notadded: |
| sys.stderr.write("Specified layer %s is already in BBLAYERS\n" % item) |
| finally: |
| # Remove the back up copy of bblayers.conf |
| shutil.rmtree(tempdir) |
| |
| def do_remove_layer(self, args): |
| """Remove one or more layers from bblayers.conf.""" |
| bblayers_conf = os.path.join('conf', 'bblayers.conf') |
| if not os.path.exists(bblayers_conf): |
| sys.stderr.write("Unable to find bblayers.conf\n") |
| return 1 |
| |
| layerdirs = [] |
| for item in args.layerdir: |
| if item.startswith('*'): |
| layerdir = item |
| elif not '/' in item: |
| layerdir = '*/%s' % item |
| else: |
| layerdir = os.path.abspath(item) |
| layerdirs.append(layerdir) |
| (_, notremoved) = bb.utils.edit_bblayers_conf(bblayers_conf, None, layerdirs) |
| if notremoved: |
| for item in notremoved: |
| sys.stderr.write("No layers matching %s found in BBLAYERS\n" % item) |
| return 1 |
| |
| def do_flatten(self, args): |
| """flatten layer configuration into a separate output directory. |
| |
| Takes the specified layers (or all layers in the current layer |
| configuration if none are specified) and builds a "flattened" directory |
| containing the contents of all layers, with any overlayed recipes removed |
| and bbappends appended to the corresponding recipes. Note that some manual |
| cleanup may still be necessary afterwards, in particular: |
| |
| * where non-recipe files (such as patches) are overwritten (the flatten |
| command will show a warning for these) |
| * where anything beyond the normal layer setup has been added to |
| layer.conf (only the lowest priority number layer's layer.conf is used) |
| * overridden/appended items from bbappends will need to be tidied up |
| * when the flattened layers do not have the same directory structure (the |
| flatten command should show a warning when this will cause a problem) |
| |
| Warning: if you flatten several layers where another layer is intended to |
| be used "inbetween" them (in layer priority order) such that recipes / |
| bbappends in the layers interact, and then attempt to use the new output |
| layer together with that other layer, you may no longer get the same |
| build results (as the layer priority order has effectively changed). |
| """ |
| if len(args.layer) == 1: |
| logger.error('If you specify layers to flatten you must specify at least two') |
| return 1 |
| |
| outputdir = args.outputdir |
| if os.path.exists(outputdir) and os.listdir(outputdir): |
| logger.error('Directory %s exists and is non-empty, please clear it out first' % outputdir) |
| return 1 |
| |
| layers = self.bblayers |
| if len(args.layer) > 2: |
| layernames = args.layer |
| found_layernames = [] |
| found_layerdirs = [] |
| for layerdir in layers: |
| layername = self.get_layer_name(layerdir) |
| if layername in layernames: |
| found_layerdirs.append(layerdir) |
| found_layernames.append(layername) |
| |
| for layername in layernames: |
| if not layername in found_layernames: |
| logger.error('Unable to find layer %s in current configuration, please run "%s show-layers" to list configured layers' % (layername, os.path.basename(sys.argv[0]))) |
| return |
| layers = found_layerdirs |
| else: |
| layernames = [] |
| |
| # Ensure a specified path matches our list of layers |
| def layer_path_match(path): |
| for layerdir in layers: |
| if path.startswith(os.path.join(layerdir, '')): |
| return layerdir |
| return None |
| |
| applied_appends = [] |
| for layer in layers: |
| overlayed = [] |
| for f in self.tinfoil.cooker.collection.overlayed.keys(): |
| for of in self.tinfoil.cooker.collection.overlayed[f]: |
| if of.startswith(layer): |
| overlayed.append(of) |
| |
| logger.plain('Copying files from %s...' % layer ) |
| for root, dirs, files in os.walk(layer): |
| if '.git' in dirs: |
| dirs.remove('.git') |
| if '.hg' in dirs: |
| dirs.remove('.hg') |
| |
| for f1 in files: |
| f1full = os.sep.join([root, f1]) |
| if f1full in overlayed: |
| logger.plain(' Skipping overlayed file %s' % f1full ) |
| else: |
| ext = os.path.splitext(f1)[1] |
| if ext != '.bbappend': |
| fdest = f1full[len(layer):] |
| fdest = os.path.normpath(os.sep.join([outputdir,fdest])) |
| bb.utils.mkdirhier(os.path.dirname(fdest)) |
| if os.path.exists(fdest): |
| if f1 == 'layer.conf' and root.endswith('/conf'): |
| logger.plain(' Skipping layer config file %s' % f1full ) |
| continue |
| else: |
| logger.warning('Overwriting file %s', fdest) |
| bb.utils.copyfile(f1full, fdest) |
| if ext == '.bb': |
| for append in self.tinfoil.cooker.collection.get_file_appends(f1full): |
| if layer_path_match(append): |
| logger.plain(' Applying append %s to %s' % (append, fdest)) |
| self.apply_append(append, fdest) |
| applied_appends.append(append) |
| |
| # Take care of when some layers are excluded and yet we have included bbappends for those recipes |
| for b in self.tinfoil.cooker.collection.bbappends: |
| (recipename, appendname) = b |
| if appendname not in applied_appends: |
| first_append = None |
| layer = layer_path_match(appendname) |
| if layer: |
| if first_append: |
| self.apply_append(appendname, first_append) |
| else: |
| fdest = appendname[len(layer):] |
| fdest = os.path.normpath(os.sep.join([outputdir,fdest])) |
| bb.utils.mkdirhier(os.path.dirname(fdest)) |
| bb.utils.copyfile(appendname, fdest) |
| first_append = fdest |
| |
| # Get the regex for the first layer in our list (which is where the conf/layer.conf file will |
| # have come from) |
| first_regex = None |
| layerdir = layers[0] |
| for layername, pattern, regex, _ in self.tinfoil.cooker.bbfile_config_priorities: |
| if regex.match(os.path.join(layerdir, 'test')): |
| first_regex = regex |
| break |
| |
| if first_regex: |
| # Find the BBFILES entries that match (which will have come from this conf/layer.conf file) |
| bbfiles = str(self.tinfoil.config_data.getVar('BBFILES')).split() |
| bbfiles_layer = [] |
| for item in bbfiles: |
| if first_regex.match(item): |
| newpath = os.path.join(outputdir, item[len(layerdir)+1:]) |
| bbfiles_layer.append(newpath) |
| |
| if bbfiles_layer: |
| # Check that all important layer files match BBFILES |
| for root, dirs, files in os.walk(outputdir): |
| for f1 in files: |
| ext = os.path.splitext(f1)[1] |
| if ext in ['.bb', '.bbappend']: |
| f1full = os.sep.join([root, f1]) |
| entry_found = False |
| for item in bbfiles_layer: |
| if fnmatch.fnmatch(f1full, item): |
| entry_found = True |
| break |
| if not entry_found: |
| logger.warning("File %s does not match the flattened layer's BBFILES setting, you may need to edit conf/layer.conf or move the file elsewhere" % f1full) |
| |
| def get_file_layer(self, filename): |
| layerdir = self.get_file_layerdir(filename) |
| if layerdir: |
| return self.get_layer_name(layerdir) |
| else: |
| return '?' |
| |
| def get_file_layerdir(self, filename): |
| layer = bb.utils.get_file_layer(filename, self.tinfoil.config_data) |
| return self.bbfile_collections.get(layer, None) |
| |
| def apply_append(self, appendname, recipename): |
| with open(appendname, 'r') as appendfile: |
| with open(recipename, 'a') as recipefile: |
| recipefile.write('\n') |
| recipefile.write('##### bbappended from %s #####\n' % self.get_file_layer(appendname)) |
| recipefile.writelines(appendfile.readlines()) |
| |
| def register_commands(self, sp): |
| parser_add_layer = self.add_command(sp, 'add-layer', self.do_add_layer, parserecipes=False) |
| parser_add_layer.add_argument('layerdir', nargs='+', help='Layer directory/directories to add') |
| |
| parser_remove_layer = self.add_command(sp, 'remove-layer', self.do_remove_layer, parserecipes=False) |
| parser_remove_layer.add_argument('layerdir', nargs='+', help='Layer directory/directories to remove (wildcards allowed, enclose in quotes to avoid shell expansion)') |
| parser_remove_layer.set_defaults(func=self.do_remove_layer) |
| |
| parser_flatten = self.add_command(sp, 'flatten', self.do_flatten) |
| parser_flatten.add_argument('layer', nargs='*', help='Optional layer(s) to flatten (otherwise all are flattened)') |
| parser_flatten.add_argument('outputdir', help='Output directory') |