blob: b10ddbac67fe2cead1c6db41a295bc0711eaf067 [file] [log] [blame]
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright 2020 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.
"""Chroot helper functions."""
from __future__ import print_function
import os
import subprocess
import collections
CommitContents = collections.namedtuple('CommitContents', ['url', 'cl_number'])
def InChroot():
"""Returns True if currently in the chroot."""
return 'CROS_WORKON_SRCROOT' in os.environ
def VerifyOutsideChroot():
"""Checks whether the script invoked was executed in the chroot.
AssertionError: The script was run inside the chroot.
assert not InChroot(), 'Script should be run outside the chroot.'
def GetChrootEbuildPaths(chromeos_root, packages):
"""Gets the chroot path(s) of the package(s).
chromeos_root: The absolute path to the chroot to
use for executing chroot commands.
packages: A list of a package/packages to
be used to find their chroot path.
A list of chroot paths of the packages' ebuild files.
ValueError: Failed to get the chroot path of a package.
chroot_paths = []
# Find the chroot path for each package's ebuild.
for package in packages:
chroot_path = subprocess.check_output(
['cros_sdk', '--', 'equery', 'w', package],
return chroot_paths
def ConvertChrootPathsToAbsolutePaths(chromeos_root, chroot_paths):
"""Converts the chroot path(s) to absolute symlink path(s).
chromeos_root: The absolute path to the chroot.
chroot_paths: A list of chroot paths to convert to absolute paths.
A list of absolute path(s).
ValueError: Invalid prefix for the chroot path or
invalid chroot paths were provided.
abs_paths = []
chroot_prefix = '/mnt/host/source/'
# Iterate through the chroot paths.
# For each chroot file path, remove '/mnt/host/source/' prefix
# and combine the chroot path with the result and add it to the list.
for chroot_path in chroot_paths:
if not chroot_path.startswith(chroot_prefix):
raise ValueError('Invalid prefix for the chroot path: %s' % chroot_path)
rel_path = chroot_path[len(chroot_prefix):]
# combine the chromeos root path + '/src/...'
abs_path = os.path.join(chromeos_root, rel_path)
return abs_paths