build/dist/zip_helper.py

#!/usr/bin/env fuchsia-vendored-python
# Copyright 2022 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.

# Extracts a zip file and writes a partial distribution manifest.
# See //docs/development/build/build_system/internals/manifest_formats.md

from argparse import ArgumentParser
import json
from os import path
from zipfile import ZipFile

parser = ArgumentParser()
parser.add_argument(
    "zip", help="extracts a zip file and writes a partial distribution manifest"
)
parser.add_argument(
    "--output-dir", help="directory to which to extract", required=True
)
parser.add_argument(
    "--output-manifest",
    help="path to which to output the distribution manifest",
    required=True,
)
parser.add_argument(
    "--output-depfile", help="path to the generated depfile", required=True
)
parser.add_argument(
    "--dest-path-prefix",
    help="path to prefix all entries within the zip when writing the manifest",
)


def main():
    args = parser.parse_args()
    with ZipFile(args.zip) as archive:
        archive.extractall(args.output_dir)
        json_data = []
        depfile_data = []
        for entry in archive.infolist():
            destination = (
                path.join(args.dest_path_prefix, entry.filename)
                if args.dest_path_prefix
                else entry.filename
            )
            source = path.join(args.output_dir, entry.filename)
            json_data.append({"destination": destination, "source": source})
            depfile_data.append(source)
        with open(args.output_manifest, "w") as f:
            json.dump(json_data, f)
        with open(args.output_depfile, "w") as f:
            f.write(" ".join(depfile_data))
            f.write(f": {args.zip}\n")


if __name__ == "__main__":
    main()