scripts/fetch_cipd.py

# 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.

"""Download a file from CIPD.

This is used for downloading a tool executable managed in CIPD.

CIPD URI format is locally defined for this script.
Example:
cipd://chromiumos/infra/tclint/linux-amd64:${version}
chromiumos/infra/tclint/linux-amd64 is the path sent to cipd tool,
and ${version} is the version defined in CIPD.
"""

import os
import urllib.parse

from chromite.lib import commandline
from chromite.lib import constants
from chromite.lib import cros_build_lib


def GetParser():
    """Creates the argparse parser."""
    parser = commandline.ArgumentParser(description=__doc__)
    parser.add_argument(
        "uri", type="cipd", help="CIPD URI of a file to download."
    )
    parser.add_argument(
        "output", type="path", help="Location to store the file."
    )
    return parser


def ParseCipdUri(uri):
    o = urllib.parse.urlparse(uri)
    if o.scheme != "cipd":
        raise ValueError("wrong scheme: ", o.scheme)
    if ":" not in o.path:
        raise ValueError("version not specified")
    pkgpath, version = o.path.rsplit(":", 1)
    return (o.netloc + pkgpath, version)


def main(argv):
    parser = GetParser()
    options = parser.parse_args(argv)
    options.Freeze()

    (pkgpath, version) = ParseCipdUri(options.uri)
    try:
        cros_build_lib.run(
            [
                os.path.join(constants.DEPOT_TOOLS_DIR, "cipd"),
                "pkg-fetch",
                "-out",
                options.output,
                "-version",
                version,
                "-verbose",
                pkgpath,
            ],
            check=True,
        )

    except cros_build_lib.RunCommandError as ex:
        # Hide the stack trace using Die.
        cros_build_lib.Die("%s", ex)