blob: 058750d19f7b3aa302f3c224a06b2e52112ad75f [file] [log] [blame]
# -*- coding: utf-8 -*-
# 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.
"""Script for calculating compatible binhosts.
Generates a file that sets the specified board's binhosts to include all of the
other compatible boards in this buildroot.
from __future__ import print_function
import collections
import glob
import os
import sys
from chromite.lib import commandline
from chromite.lib import portage_util
def FindCandidateBoards():
"""Find candidate local boards to grab prebuilts from."""
portageq_prefix = '/usr/local/bin/portageq-'
for path in sorted(glob.glob('%s*' % portageq_prefix)):
# Strip off the portageq prefix, leaving only the board.
yield path.replace(portageq_prefix, '')
def SummarizeCompatibility(board):
"""Returns a string that will be the same for compatible boards."""
result = portage_util.PortageqEnvvars(['ARCH', 'CFLAGS'], board=board)
return '%s %s' % (result['ARCH'], result['CFLAGS'])
def GenerateBinhostLine(build_root, compatible_boards):
"""Generate a binhost line pulling binaries from the specified boards."""
# TODO(davidjames): Prioritize binhosts with more matching use flags.
local_binhosts = ' '.join([
'file://localhost' + os.path.join(build_root, x, 'packages')
for x in sorted(compatible_boards)])
return "LOCAL_BINHOST='%s'" % local_binhosts
def GetParser():
"""Return a command line parser."""
parser = commandline.ArgumentParser(description=__doc__)
parser.add_argument('--build_root', default='/build',
help='Location of boards (normally %(default)s)')
parser.add_argument('--board', required=True,
help='Board name')
return parser
def main(argv):
parser = GetParser()
flags = parser.parse_args(argv)
by_compatibility = collections.defaultdict(set)
compatible_boards = None
for other_board in FindCandidateBoards():
compat_id = SummarizeCompatibility(other_board)
if other_board == flags.board:
compatible_boards = by_compatibility[compat_id]
if compatible_boards is None:
print('Missing portageq wrapper for %s' % flags.board, file=sys.stderr)
print('# Generated by cros_generate_local_binhosts.')
print(GenerateBinhostLine(flags.build_root, compatible_boards))