bazel_rules_fuchsia/fuchsia/private/assembly/fuchsia_zbi.bzl

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

load(":providers.bzl", "FuchsiaZbiInfo")

# Define Zbi compression format
ZBI_COMPRESSION = struct(
    ZSTD_MAX = "zstd.max",
    ZSTD = "zstd",
)

def _fuchsia_zbi_impl(ctx):
    return [
        FuchsiaZbiInfo(
            zbi_name = ctx.attr.zbi_name,
            compression = ctx.attr.compression,
            post_processing_script = ctx.file.post_processing_script,
            post_processing_args = ctx.attr.post_processing_args,
        ),
    ]

fuchsia_zbi = rule(
    doc = """Generates a fuchsia zbi image.""",
    implementation = _fuchsia_zbi_impl,
    provides = [FuchsiaZbiInfo],
    attrs = {
        "zbi_name": attr.string(
            doc = "Name of zbi image appeared in image configuration",
            default = "fuchsia",
        ),
        "compression": attr.string(
            doc = "Zbi compression format",
            default = "zstd",
            values = [ZBI_COMPRESSION.ZSTD_MAX, ZBI_COMPRESSION.ZSTD],
        ),
        "post_processing_script": attr.label(
            doc = "Post procesing script",
            allow_single_file = True,
            default = None,
        ),
        "post_processing_args": attr.string_list(
            doc = "Args needed by post processing script",
            default = [],
        ),
    },
)