load("@io_bazel_rules_docker//container:providers.bzl", "ImageInfo")

# Supported pairs of system and architecture. (GOOS, GOARCH).
binary_targets = [
    ("linux", "amd64"),
    ("linux", "arm64"),
    # TODO(2849): Enable after cross OS compilation is supported.
    # https://github.com/prysmaticlabs/prysm/issues/2849
    #    ("darwin", "amd64"),
    #    ("windows", "amd64"),
]

# Determine binary targets from a supported pair. These targets are part of the
# distributable bundle.
def determine_targets(pair, common_files):
    targets = {
        "//beacon-chain:beacon-chain-{}-{}".format(
            pair[0],
            pair[1],
        ): "beacon-chain",
        "//validator:validator-{}-{}".format(
            pair[0],
            pair[1],
        ): "validator",
    }
    targets.update(common_files)
    return targets

# Defines the debug transition implementation to enforce dbg mode.
def _debug_transition_impl(settings, attr):
    return {
        "//command_line_option:compilation_mode": "dbg",
    }

# Defines a starlark transition which enforces dbg compilation mode.
build_in_debug_mode = transition(
    implementation = _debug_transition_impl,
    inputs = [],
    outputs = ["//command_line_option:compilation_mode"],
)

# Defines a rule implementation that essentially returns all of the providers from the image attr.
def _go_image_debug_impl(ctx):
    img = ctx.attr.image[0]

    return [
        img[ImageInfo],
        img[OutputGroupInfo],
    ]

# Defines a rule that specifies a starlark transition to enforce debug compilation mode for debug
# images.
go_image_debug = rule(
    implementation = _go_image_debug_impl,
    attrs = {
        "image": attr.label(
            cfg = build_in_debug_mode,
            executable = True,
        ),
        # Whitelist is required or bazel complains.
        "_whitelist_function_transition": attr.label(default = "@bazel_tools//tools/whitelists/function_transition_whitelist"),
    },
)