"""
Copyright (C) 2022 The Android Open Source Project

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""

"""A macro to handle build number stamping."""

load(":stripped_cc_common.bzl", "StrippedCcBinaryInfo")
load("@rules_cc//examples:experimental_cc_shared_library.bzl", "CcSharedLibraryInfo")

def stamp_build_number(ctx, prefix = "", extension = ""):
    if len(ctx.files.src) != 1:
        fail("Expected only one input file for build number stamping")

    out_file = ctx.actions.declare_file(prefix + ctx.attr.name + extension)
    android_constraint = ctx.attr._android_constraint[platform_common.ConstraintValueInfo]

    # TODO(b/228461735): We need to dist the output for device target.
    if ctx.target_platform_has_constraint(android_constraint) or not ctx.attr.stamp_build_number:
        ctx.actions.symlink(
            output = out_file,
            target_file = ctx.files.src[0],
        )
        return out_file

    ctx.actions.run_shell(
        inputs = ctx.files.src + [ctx.version_file],
        outputs = [out_file],
        command = """
            build_number=$(cat {file} | grep "BUILD_NUMBER" | cut -f2 -d' ');
            {build_number_stamper} -i {input} -o {output} -s soong_build_number -v $build_number
        """.format(
            file = ctx.version_file.path,
            input = ctx.files.src[0].path,
            output = out_file.path,
            build_number_stamper = ctx.executable._build_number_stamper.path,
        ),
        tools = [ctx.executable._build_number_stamper],
        mnemonic = "StampBuildNumber",
    )

    return out_file

common_attrs = {
    "stamp_build_number": attr.bool(
        default = False,
        doc = "Whether to stamp the build number",
    ),
    "_build_number_stamper": attr.label(
        cfg = "exec",
        doc = "The build number stamp tool.",
        executable = True,
        default = "//prebuilts/build-tools:linux-x86/bin/symbol_inject",
        allow_single_file = True,
    ),
    "_android_constraint": attr.label(
        default = Label("//build/bazel/platforms/os:android"),
    ),
}

def _versioned_binary_impl(ctx):
    common_providers = [
        ctx.attr.src[CcInfo],
        ctx.attr.src[InstrumentedFilesInfo],
        ctx.attr.src[DebugPackageInfo],
        ctx.attr.src[OutputGroupInfo],
    ]

    out_file = stamp_build_number(ctx)

    return [
        DefaultInfo(
            files = depset([out_file]),
            executable = out_file,
        ),
    ] + common_providers

versioned_binary = rule(
    implementation = _versioned_binary_impl,
    attrs = dict(
        common_attrs,
        src = attr.label(mandatory = True, allow_single_file = True, providers = [CcInfo]),
    ),
)

def _versioned_shared_library_impl(ctx):
    out_file = stamp_build_number(ctx, "lib", ".so")

    return [
        DefaultInfo(files = depset([out_file])),
        ctx.attr.src[CcSharedLibraryInfo],
    ]

versioned_shared_library = rule(
    implementation = _versioned_shared_library_impl,
    attrs = dict(
        common_attrs,
        src = attr.label(
            mandatory = True,
            # TODO(b/217908237): reenable allow_single_file
            # allow_single_file = True,
            providers = [CcSharedLibraryInfo],
        ),
    ),
)