blob: 754a041d8df725f4e7df005c664198655cfacb3e [file] [log] [blame]
# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Calculate what overlays are needed for a particular board."""
from __future__ import print_function
import os
from chromite.cbuildbot import constants
from chromite.lib import commandline
from chromite.lib import cros_build_lib
from chromite.lib import portage_util
def _ParseArguments(argv):
parser = commandline.ArgumentParser(description=__doc__)
parser.add_argument('--board', default=None, help='Board name')
parser.add_argument('--board_overlay', default=None,
help='Location of the board overlay. Used by '
'./setup_board to allow developers to add custom '
parser.add_argument('--primary_only', default=False, action='store_true',
help='Only return the path to the primary overlay. This '
'only makes sense when --board is specified.')
parser.add_argument('-a', '--all', default=False, action='store_true',
help='Show all overlays (even common ones).')
opts = parser.parse_args(argv)
if opts.primary_only and opts.board is None:
parser.error('--board is required when --primary_only is supplied.')
return opts
def main(argv):
opts = _ParseArguments(argv)
args = (constants.BOTH_OVERLAYS, opts.board)
# Verify that a primary overlay exists.
primary_overlay = portage_util.FindPrimaryOverlay(*args)
except portage_util.MissingOverlayException as ex:
# Get the overlays to print.
if opts.primary_only:
overlays = [primary_overlay]
overlays = portage_util.FindOverlays(*args)
# Exclude any overlays in src/third_party, for backwards compatibility with
# scripts that expected these to not be listed.
if not opts.all:
ignore_prefix = os.path.join(constants.SOURCE_ROOT, 'src', 'third_party')
overlays = [o for o in overlays if not o.startswith(ignore_prefix)]
if opts.board_overlay and os.path.isdir(opts.board_overlay):