# Copyright 2023 The Pigweed Authors
#
# 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
#
#     https://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.

"""Bazel transitions for sample project examples.

TODO:  b/301334234 - Use platform-based flags and retire these transitions.
"""

load("@pigweed//pw_build:merge_flags.bzl", "merge_flags_for_transition_impl", "merge_flags_for_transition_outputs")
load("@pigweed//targets/rp2040:transition.bzl", "RP2_SYSTEM_FLAGS")
load("@pigweed//third_party/freertos:flags.bzl", "FREERTOS_FLAGS")

_stm32_overrides = {
    "//command_line_option:copt": ["-DSTM32CUBE_HEADER=\"stm32f4xx.h\"", "-DSTM32F429xx"],
    "//command_line_option:platforms": "//targets/stm32f429i_disc1_stm32cube:platform",
    "@cmsis_device//:cmsis_core": "@cmsis_core",
    "@freertos//:freertos_config": "@pigweed//third_party/freertos:freertos_config",
    "@hal_driver//:cmsis_device": "@cmsis_device",
    "@hal_driver//:cmsis_init": "@cmsis_device//:default_cmsis_init",
    "@hal_driver//:hal_config": "@pigweed//targets/stm32f429i_disc1_stm32cube:hal_config",
    "@pigweed//pw_assert:backend": "@pigweed//pw_assert_basic",
    "@pigweed//pw_assert:backend_impl": "@pigweed//pw_assert_basic:impl",
    "@pigweed//pw_assert:check_backend": "@pigweed//pw_assert_basic",
    "@pigweed//pw_assert:check_backend_impl": "@pigweed//pw_assert_basic:impl",
    "@pigweed//pw_boot:backend": "@pigweed//pw_boot_cortex_m",
    "@pigweed//pw_interrupt:backend": "@pigweed//pw_interrupt_cortex_m:context",
    "@pigweed//pw_log:backend": "@pigweed//pw_log_tokenized",
    "@pigweed//pw_log:backend_impl": "@pigweed//pw_log_tokenized:impl",
    "@pigweed//pw_log_tokenized:handler_backend": "@pigweed//pw_system:log_backend",
    "@pigweed//pw_malloc:backend": "@pigweed//pw_malloc_freelist",
    "@pigweed//pw_sys_io:backend": "@pigweed//pw_sys_io_stm32cube",
    "@pigweed//pw_system:extra_platform_libs": "//targets/stm32f429i_disc1_stm32cube:extra_platform_libs",
    "@pigweed//third_party/stm32cube:hal_driver": "@hal_driver",
}

def _stm32_transition_impl(settings, attr):
    # buildifier: disable=unused-variable
    _ignore = settings, attr

    return merge_flags_for_transition_impl(base = FREERTOS_FLAGS, override = _stm32_overrides)

_stm32_transition = transition(
    implementation = _stm32_transition_impl,
    inputs = [],
    outputs = merge_flags_for_transition_outputs(base = FREERTOS_FLAGS, override = _stm32_overrides),
)

def _stm32_binary_impl(ctx):
    out = ctx.actions.declare_file(ctx.label.name)
    ctx.actions.symlink(output = out, target_file = ctx.executable.binary)
    return [DefaultInfo(files = depset([out]), executable = out)]

stm32_binary = rule(
    _stm32_binary_impl,
    attrs = {
        "binary": attr.label(
            doc = "cc_binary to build for stm32f429i_disc1_stm32cube.",
            cfg = _stm32_transition,
            executable = True,
            mandatory = True,
        ),
        "_allowlist_function_transition": attr.label(
            default = "@bazel_tools//tools/allowlists/function_transition_allowlist",
        ),
    },
    doc = "Builds the specified binary for the stm32f429i_disc1_stm32cube platform.",
)

_rp2040_overrides = {
    "//command_line_option:platforms": "//targets/rp2040:platform",
    "@pigweed//pw_system:extra_platform_libs": "//targets/rp2040:extra_platform_libs",
}

def _rp2040_transition_impl(settings, attr):
    # buildifier: disable=unused-variable
    _ignore = settings, attr

    return merge_flags_for_transition_impl(base = RP2_SYSTEM_FLAGS, override = _rp2040_overrides)

_rp2040_transition = transition(
    implementation = _rp2040_transition_impl,
    inputs = [],
    outputs = merge_flags_for_transition_outputs(base = RP2_SYSTEM_FLAGS, override = _rp2040_overrides),
)

def _rp2040_binary_impl(ctx):
    out = ctx.actions.declare_file(ctx.label.name)
    ctx.actions.symlink(output = out, target_file = ctx.executable.binary)
    return [DefaultInfo(files = depset([out]), executable = out)]

rp2040_binary = rule(
    _rp2040_binary_impl,
    attrs = {
        "binary": attr.label(
            doc = "cc_binary to build for rp2040 using pico-sdk.",
            cfg = _rp2040_transition,
            executable = True,
            mandatory = True,
        ),
        "_allowlist_function_transition": attr.label(
            default = "@bazel_tools//tools/allowlists/function_transition_allowlist",
        ),
    },
    doc = "Builds the specified binary for the pico-sdk rp2040 platform.",
)