build/assembly/scripts/generate_platform_artifacts.py

#!/usr/bin/env fuchsia-vendored-python
# Copyright 2024 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import argparse
import json
import os
import shutil
import sys

import depfile


def main() -> int:
    parser = argparse.ArgumentParser(
        description="Create a platform artifacts directory"
    )
    parser.add_argument(
        "--aib-list", type=argparse.FileType("r"), required=True
    )
    parser.add_argument("--name", type=str, required=True)
    parser.add_argument("--version", type=str)
    parser.add_argument("--repo", type=str, required=True)
    parser.add_argument("--tools", type=str, nargs="*")
    parser.add_argument("--outdir", type=str, required=True)
    parser.add_argument("--depfile", type=argparse.FileType("w"), required=True)
    args = parser.parse_args()

    shutil.rmtree(args.outdir)
    os.mkdir(args.outdir)
    os.mkdir(os.path.join(args.outdir, "tools"))

    artifacts = json.load(args.aib_list)
    deps = []
    for artifact in artifacts:
        path = artifact["path"]
        deps.append(os.path.join(path, "assembly_config.json"))
        src = os.path.realpath(path)
        dst = os.path.join(args.outdir, os.path.basename(path))
        try:
            os.link(src, dst)
        except OSError:
            shutil.copytree(src, dst)

    if args.tools:
        for tool in args.tools:
            src = os.path.realpath(tool)
            dst = os.path.join(args.outdir, "tools", os.path.basename(tool))
            try:
                os.link(src, dst)
            except OSError:
                shutil.copytree(src, dst)

    version = args.version if args.version else "unversioned"
    config_data = {
        "release_info": {
            "name": args.name,
            "repository": args.repo,
            "version": version,
        },
    }

    config = os.path.join(args.outdir, "platform_artifacts.json")
    with open(config, "w") as f:
        json.dump(config_data, f, indent=2)

    depfile.DepFile.from_deps(config, deps).write_to(args.depfile)

    return 0


if __name__ == "__main__":
    sys.exit(main())