load("@heir//bazel/cheddar:config.bzl", "requires_cheddar")
load("@heir//tools:heir-opt.bzl", "heir_opt")
load("@heir//tools:heir-translate.bzl", "heir_translate")
load("@rules_cc//cc:cc_library.bzl", "cc_library")
load("@rules_cc//cc:cc_test.bzl", "cc_test")

package(
    default_applicable_licenses = ["@heir//:license"],
    default_visibility = ["//visibility:public"],
)

# End-to-end: additive grab-bag (ct+ct, ct-ct, ct+const) on two secret 1024-vectors. The cheddar-level IR is a full module (client
# encrypt/decrypt helpers + compute), so the harness just calls the generated
# functions; lower it the rest of the way to emitc.
heir_opt(
    name = "add_e2e_emitc",
    src = "add_e2e.mlir",
    generated_filename = "add_e2e_emitc.mlir",
    pass_flags = ["--cheddar-to-emitc"],
)

heir_translate(
    name = "add_e2e_cpp_raw",
    src = ":add_e2e_emitc.mlir",
    generated_filename = "add_e2e_raw.cc",
    pass_flags = ["--mlir-to-cpp"],
)

genrule(
    name = "add_e2e_lib_src",
    srcs = [":add_e2e_raw.cc"],
    outs = ["add_e2e_lib.cc"],
    cmd = """cat > $@ <<'PRELUDE_EOF'
// AUTO-GENERATED: do not edit. See tests/Examples/cheddar/add_e2e/BUILD.
#include <array>
#include <complex>
#include <cstddef>
#include <cstdint>
#include <utility>
#include <vector>
#include "core/Context.h"
#include "core/Encode.h"
#include "core/Parameter.h"
#include "UserInterface.h"
using namespace cheddar;
using word = uint64_t;
using Complex = std::complex<double>;
PRELUDE_EOF
cat $(location :add_e2e_raw.cc) >> $@
""",
)

cc_library(
    name = "add_e2e_lib",
    srcs = [":add_e2e_lib_src"],
    target_compatible_with = requires_cheddar(),
    deps = ["@cheddar"],
)

cc_test(
    name = "add_e2e_test",
    timeout = "long",
    srcs = ["add_e2e_test.cpp"],
    linkstatic = False,
    tags = [
        "exclusive",
        "notap",
    ],
    target_compatible_with = requires_cheddar(),
    deps = [
        ":add_e2e_lib",
        "@cheddar",
        "@googletest//:gtest_main",
    ],
)
