You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
2.6 KiB
100 lines
2.6 KiB
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
# Copyright 2020 The ChromiumOS Authors
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
"""Chroot helper functions."""
|
|
|
|
|
|
import collections
|
|
import os
|
|
import subprocess
|
|
|
|
|
|
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.
|
|
|
|
Raises:
|
|
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).
|
|
|
|
Args:
|
|
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.
|
|
|
|
Returns:
|
|
A list of chroot paths of the packages' ebuild files.
|
|
|
|
Raises:
|
|
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],
|
|
cwd=chromeos_root,
|
|
encoding="utf-8",
|
|
)
|
|
chroot_paths.append(chroot_path.strip())
|
|
|
|
return chroot_paths
|
|
|
|
|
|
def ConvertChrootPathsToAbsolutePaths(chromeos_root, chroot_paths):
|
|
"""Converts the chroot path(s) to absolute symlink path(s).
|
|
|
|
Args:
|
|
chromeos_root: The absolute path to the chroot.
|
|
chroot_paths: A list of chroot paths to convert to absolute paths.
|
|
|
|
Returns:
|
|
A list of absolute path(s).
|
|
|
|
Raises:
|
|
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)
|
|
|
|
abs_paths.append(abs_path)
|
|
|
|
return abs_paths
|