System Initiative

# Copyright (c) Meta Platforms, Inc. and affiliates. # # This source code is licensed under both the MIT license found in the # LICENSE-MIT file in the root directory of this source tree and the Apache # License, Version 2.0 found in the LICENSE-APACHE file in the root directory # of this source tree. load("@prelude//go:coverage.bzl", "GoCoverageMode") load(":tags_helper.bzl", "selects_for_tags", "tag_to_constrant_value") def _cgo_enabled_transition(platform, refs, attrs): constraints = platform.configuration.constraints # Cancel transition if the value already set # to enable using configuration modifiers for overriding this option cgo_enabled_setting = refs.cgo_enabled_true[ConstraintValueInfo].setting if cgo_enabled_setting.label in constraints: return platform if attrs.cgo_enabled == None: return platform elif attrs.cgo_enabled == True: cgo_enabled_ref = refs.cgo_enabled_true else: cgo_enabled_ref = refs.cgo_enabled_false cgo_enabled_value = cgo_enabled_ref[ConstraintValueInfo] constraints[cgo_enabled_value.setting.label] = cgo_enabled_value new_cfg = ConfigurationInfo( constraints = constraints, values = platform.configuration.values, ) return PlatformInfo( label = platform.label, configuration = new_cfg, ) def _race_transition(platform, refs, attrs): constraints = platform.configuration.constraints # Cancel transition if the value already set # to enable using configuration modifiers for overriding this option race_setting = refs.race_false[ConstraintValueInfo].setting if race_setting.label in constraints: return platform # change configuration only when we can't avoid it if attrs.race == True: race_ref = refs.race_true else: return platform race_value = race_ref[ConstraintValueInfo] constraints[race_value.setting.label] = race_value new_cfg = ConfigurationInfo( constraints = constraints, values = platform.configuration.values, ) return PlatformInfo( label = platform.label, configuration = new_cfg, ) def _asan_transition(platform, refs, attrs): constraints = platform.configuration.constraints # Cancel transition if the value already set # to enable using configuration modifiers for overriding this option asan_setting = refs.asan_false[ConstraintValueInfo].setting if asan_setting.label in constraints: return platform # change configuration only when we can't avoid it if attrs.asan == True: asan_ref = refs.asan_true else: return platform asan_value = asan_ref[ConstraintValueInfo] constraints[asan_value.setting.label] = asan_value new_cfg = ConfigurationInfo( constraints = constraints, values = platform.configuration.values, ) return PlatformInfo( label = platform.label, configuration = new_cfg, ) def _coverage_mode_transition(platform, refs, attrs): constraints = platform.configuration.constraints # Cancel transition if the value already set # to enable using configuration modifiers for overriding this option coverage_mode_setting = refs.coverage_mode_set[ConstraintValueInfo].setting if coverage_mode_setting.label in constraints: return platform if attrs.coverage_mode == None: return platform elif attrs.coverage_mode == "set": coverage_mode_ref = refs.coverage_mode_set elif attrs.coverage_mode == "count": coverage_mode_ref = refs.coverage_mode_count elif attrs.coverage_mode == "atomic": coverage_mode_ref = refs.coverage_mode_atomic else: fail("`coverage_mode` can be either: 'set', 'count', 'atomic' or None, got: {}".format(attrs.coverage_mode)) coverage_mode_value = coverage_mode_ref[ConstraintValueInfo] constraints[coverage_mode_value.setting.label] = coverage_mode_value new_cfg = ConfigurationInfo( constraints = constraints, values = platform.configuration.values, ) return PlatformInfo( label = platform.label, configuration = new_cfg, ) def _tags_transition(platform, refs, attrs): constraints = platform.configuration.constraints if not attrs.build_tags: return platform for build_tag in attrs.build_tags: ref_name = "tag_{}__value".format(build_tag) if not hasattr(refs, ref_name): fail("Add build_tag '{}' to .buckconfig attribute `go.allowed_build_tags` to allow using it".format(build_tag)) tag_value = getattr(refs, ref_name)[ConstraintValueInfo] constraints[tag_value.setting.label] = tag_value new_cfg = ConfigurationInfo( constraints = constraints, values = platform.configuration.values, ) return PlatformInfo( label = platform.label, configuration = new_cfg, ) def _force_mingw_on_windows(platform, refs, _): constraints = platform.configuration.constraints abi_gnu_value = refs.abi_gnu[ConstraintValueInfo] if abi_gnu_value.setting.label in constraints and constraints[abi_gnu_value.setting.label] == abi_gnu_value: # Already MinGW/GNU, do nothing return platform os_windows_value = refs.os_windows[ConstraintValueInfo] if os_windows_value.setting.label in constraints and constraints[os_windows_value.setting.label] != os_windows_value: # Non-Windows, do nothing return platform constraints[abi_gnu_value.setting.label] = abi_gnu_value new_cfg = ConfigurationInfo( constraints = constraints, values = platform.configuration.values, ) return PlatformInfo( label = platform.label, configuration = new_cfg, ) def _chain_transitions(transitions): def tr(platform, refs, attrs): for t in transitions: platform = t(platform, refs, attrs) return platform return tr _all_level_tansitions = [_force_mingw_on_windows] _top_level_tansitions = [_asan_transition, _cgo_enabled_transition, _race_transition, _tags_transition] + _all_level_tansitions _all_level_refs = { "abi_gnu": "prelude//abi/constraints:gnu", "os_windows": "config//os/constraints:windows", } _top_level_refs = { "asan_false": "prelude//go/constraints:asan_false", "asan_true": "prelude//go/constraints:asan_true", "cgo_enabled_false": "prelude//go/constraints:cgo_enabled_false", "cgo_enabled_true": "prelude//go/constraints:cgo_enabled_true", "race_false": "prelude//go/constraints:race_false", "race_true": "prelude//go/constraints:race_true", } | { "tag_{}__value".format(tag): constrant_value for tag, constrant_value in tag_to_constrant_value().items() } | _all_level_refs _attrs = ["asan", "cgo_enabled", "race", "build_tags"] go_binary_transition = transition( impl = _chain_transitions(_top_level_tansitions), refs = _top_level_refs, attrs = _attrs, ) go_test_transition = transition( impl = _chain_transitions(_top_level_tansitions + [_coverage_mode_transition]), refs = _top_level_refs | { "coverage_mode_atomic": "prelude//go/constraints:coverage_mode_atomic", "coverage_mode_count": "prelude//go/constraints:coverage_mode_count", "coverage_mode_set": "prelude//go/constraints:coverage_mode_set", }, attrs = _attrs + ["coverage_mode"], ) go_exported_library_transition = transition( impl = _chain_transitions(_top_level_tansitions), refs = _top_level_refs, attrs = _attrs, ) go_library_transition = transition( impl = _chain_transitions(_all_level_tansitions), refs = _all_level_refs, attrs = [], ) go_stdlib_transition = transition( impl = _chain_transitions(_all_level_tansitions), refs = _all_level_refs, attrs = [], ) cgo_enabled_attr = attrs.default_only(attrs.option(attrs.bool(), default = select({ "DEFAULT": None, "prelude//go/constraints:cgo_enabled_false": False, "prelude//go/constraints:cgo_enabled_true": True, }))) race_attr = attrs.default_only(attrs.bool(default = select({ "DEFAULT": False, "prelude//go/constraints:race_false": False, "prelude//go/constraints:race_true": True, }))) asan_attr = attrs.default_only(attrs.bool(default = select({ "DEFAULT": False, "prelude//go/constraints:asan_false": False, "prelude//go/constraints:asan_true": True, }))) coverage_mode_attr = attrs.default_only(attrs.option(attrs.enum(GoCoverageMode.values()), default = select({ "DEFAULT": None, "prelude//go/constraints:coverage_mode_atomic": "atomic", "prelude//go/constraints:coverage_mode_count": "count", "prelude//go/constraints:coverage_mode_set": "set", }))) build_tags_attr = attrs.default_only(attrs.list(attrs.string(), default = selects_for_tags()))