113 lines
3.6 KiB
Python
113 lines
3.6 KiB
Python
# 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."""
|
|
|
|
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],
|
|
ctx.attr.src[OutputGroupInfo],
|
|
]
|
|
|
|
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],
|
|
),
|
|
),
|
|
)
|