mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-22 03:30:35 +00:00
71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
load("@io_bazel_rules_go//go/private/rules:library.bzl", _go_library = "go_library")
|
|
load("@io_bazel_rules_go//go/private/rules:test.bzl", "go_test_kwargs")
|
|
load("@bazel_gazelle//:deps.bzl", _go_repository = "go_repository")
|
|
|
|
def _go_test_transition_impl(settings, attr):
|
|
"""Edge transition to add minimal or mainnet build tags"""
|
|
settings = dict(settings)
|
|
|
|
if attr.eth_network == "minimal":
|
|
settings["//proto:network"] = "minimal"
|
|
settings["@io_bazel_rules_go//go/config:tags"] = ["minimal"] + settings["@io_bazel_rules_go//go/config:tags"]
|
|
elif attr.eth_network == "mainnet": # Default / optional
|
|
settings["//proto:network"] = "mainnet"
|
|
settings["@io_bazel_rules_go//go/config:tags"] = ["mainnet"] + settings["@io_bazel_rules_go//go/config:tags"]
|
|
|
|
if attr.gotags:
|
|
settings["@io_bazel_rules_go//go/config:tags"] = attr.gotags + settings["@io_bazel_rules_go//go/config:tags"]
|
|
|
|
if str(settings["//command_line_option:compilation_mode"]) == "dbg":
|
|
settings["@io_bazel_rules_go//go/config:debug"] = True
|
|
return settings
|
|
|
|
go_test_transition = transition(
|
|
implementation = _go_test_transition_impl,
|
|
inputs = [
|
|
"@io_bazel_rules_go//go/config:tags",
|
|
"//proto:network",
|
|
"//command_line_option:compilation_mode",
|
|
"@io_bazel_rules_go//go/config:debug",
|
|
],
|
|
outputs = [
|
|
"@io_bazel_rules_go//go/config:tags",
|
|
"//proto:network",
|
|
"//command_line_option:compilation_mode",
|
|
"@io_bazel_rules_go//go/config:debug",
|
|
],
|
|
)
|
|
|
|
def _go_test_transition_rule(**kwargs):
|
|
"""A wrapper around go_test to add an eth_network attribute and incoming edge transition to support compile time configuration"""
|
|
kwargs = dict(kwargs)
|
|
attrs = dict(kwargs["attrs"])
|
|
attrs.update({
|
|
"eth_network": attr.string(values = ["mainnet", "minimal"]),
|
|
})
|
|
kwargs["attrs"] = attrs
|
|
kwargs["cfg"] = go_test_transition
|
|
return rule(**kwargs)
|
|
|
|
go_test = _go_test_transition_rule(**go_test_kwargs)
|
|
|
|
# Alias retained for future ease of use.
|
|
go_library = _go_library
|
|
|
|
# Maybe download a repository rule, if it doesn't exist already.
|
|
def maybe(repo_rule, name, **kwargs):
|
|
if name not in native.existing_rules():
|
|
repo_rule(name = name, **kwargs)
|
|
|
|
# A wrapper around go_repository to add gazelle directives.
|
|
def go_repository(name, **kwargs):
|
|
directives = []
|
|
if "build_directives" in kwargs:
|
|
directives = kwargs["build_directives"]
|
|
|
|
directives += [
|
|
"gazelle:map_kind go_library go_library @prysm//tools/go:def.bzl",
|
|
]
|
|
kwargs["build_directives"] = directives
|
|
maybe(_go_repository, name, **kwargs)
|