| #!/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 |