1# Copyright 2019 The Bazel Authors. All rights reserved.
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7#    http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14
15"""Example showing how to get CcToolchainInfo in a custom rule."""
16
17load("@bazel_tools//tools/cpp:toolchain_utils.bzl", "find_cpp_toolchain", "use_cpp_toolchain")
18
19def _write_cc_toolchain_cpu_impl(ctx):
20    cc_toolchain = find_cpp_toolchain(ctx)
21    output = ctx.actions.declare_file(ctx.label.name + "_cpu")
22    ctx.actions.write(output, cc_toolchain.cpu)
23    return [DefaultInfo(files = depset([output]))]
24
25# This rule does nothing, just writes the target_cpu from the cc_toolchain used for this build.
26write_cc_toolchain_cpu = rule(
27    implementation = _write_cc_toolchain_cpu_impl,
28    attrs = {
29        "_cc_toolchain": attr.label(default = Label("@bazel_tools//tools/cpp:current_cc_toolchain")),
30    },
31    toolchains = use_cpp_toolchain(),
32)
33