summaryrefslogtreecommitdiffstats
path: root/scripts/lib/devtool/build-image.py
blob: 5759bece0436d60ec1199ecf9742950689494b8a (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
# Development tool - build-image plugin
#
# Copyright (C) 2015 Intel Corporation
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
#
# 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.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

"""Devtool plugin containing the build-image subcommand."""

import os
import logging

from bb.process import ExecutionError
from devtool import exec_build_env_command, setup_tinfoil, parse_recipe

logger = logging.getLogger('devtool')

def _get_packages(tinfoil, workspace, config):
    """Get list of packages from recipes in the workspace."""
    result = []
    for recipe in workspace:
        data = parse_recipe(config, tinfoil, recipe, True)
        if 'class-target' in data.getVar('OVERRIDES', True).split(':'):
            if recipe in data.getVar('PACKAGES', True):
                result.append(recipe)
            else:
                logger.warning("Skipping recipe %s as it doesn't produce a "
                               "package with the same name", recipe)
    return result

def build_image(args, config, basepath, workspace):
    """Entry point for the devtool 'build-image' subcommand."""
    image = args.recipe
    appendfile = os.path.join(config.workspace_path, 'appends',
                              '%s.bbappend' % image)

    # remove <image>.bbappend to make sure setup_tinfoil doesn't
    # break because of it
    if os.path.isfile(appendfile):
        os.unlink(appendfile)

    tinfoil = setup_tinfoil()
    if workspace:
        packages = _get_packages(tinfoil, workspace, config)
        if packages:
            with open(appendfile, 'w') as afile:
                # include packages from workspace recipes into the image
                afile.write('IMAGE_INSTALL_append = " %s"\n' % ' '.join(packages))

                # Generate notification callback devtool_warn_image_extended
                afile.write('do_rootfs[prefuncs] += "devtool_warn_image_extended"\n\n')
                afile.write("python devtool_warn_image_extended() {\n")
                afile.write("    bb.plain('NOTE: %%s: building with additional '\n"
                            "             'packages due to \"devtool build-image\"'"
                            "              %% d.getVar('PN', True))\n"
                            "    bb.plain('NOTE: delete %%s to clear this' %% \\\n"
                            "             '%s')\n" % os.path.relpath(appendfile, basepath))
                afile.write("}\n")

                logger.info('Building image %s with the following '
                            'additional packages: %s', image, ' '.join(packages))
        else:
            logger.warning('No packages to add, building image %s unmodified', image)
    else:
        logger.warning('No recipes in workspace, building image %s unmodified', image)

    deploy_dir_image = tinfoil.config_data.getVar('DEPLOY_DIR_IMAGE', True)

    tinfoil.shutdown()

    # run bitbake to build image
    try:
        exec_build_env_command(config.init_path, basepath,
                               'bitbake %s' % image, watch=True)
    except ExecutionError as err:
        return err.exitcode

    logger.info('Successfully built %s. You can find output files in %s'
                % (image, deploy_dir_image))

def register_commands(subparsers, context):
    """Register devtool subcommands from the build-image plugin"""
    parser = subparsers.add_parser('build-image',
                                   help='Build image including workspace recipe packages',
                                   description='Builds an image, extending it to include '
                                   'packages from recipes in the workspace')
    parser.add_argument('recipe', help='Image recipe to build')
    parser.set_defaults(func=build_image)