From 0fa0c34304e62d91bd07607261d451ee29e27d2e Mon Sep 17 00:00:00 2001
From: Ulf Adams <ulfjack@gmail.com>
Date: Sat, 15 Aug 2020 17:18:50 +0200
Subject: [PATCH] Allow overriding os_family in rbe_autoconfig

Provides a workaround for https://github.com/bazelbuild/bazel-toolchains/issues/895.

Use like this:
rbe_autoconfig(
    name = "engflow_remote_config",
    os_family = "Linux",
    java_home = "/usr/lib/jvm/11.29.3-ca-jdk11.0.2/reduced",
    use_checked_in_confs = "Force",
)
---
 rules/rbe_repo.bzl      | 7 +++++++
 rules/rbe_repo/util.bzl | 2 ++
 2 files changed, 9 insertions(+)

diff --git a/rules/rbe_repo.bzl b/rules/rbe_repo.bzl
index 1b3bd5658..462229043 100644
--- a/rules/rbe_repo.bzl
+++ b/rules/rbe_repo.bzl
@@ -823,6 +823,11 @@ _rbe_autoconfig = repository_rule(
             doc = ("Optional. The Java release version in the container. For " +
                    " example, 11. Should only be set if java_home is set."),
         ),
+        "os_family": attr.string(
+            doc = ("Optional. The os_family to generate the config for. For example, " +
+                   "Linux or Windows (Mac OS X is not supported at this time). The default is " +
+                   "OS Bazel runs on."),
+        ),
         "registry": attr.string(
             doc = ("Optional. The registry to pull the container from. For example, " +
                    "marketplace.gcr.io. The default is the value for the selected " +
@@ -912,6 +917,7 @@ def rbe_autoconfig(
         export_configs = False,
         java_home = None,
         java_version = None,
+        os_family = None,
         tag = None,
         toolchain_config_suite_spec = default_toolchain_config_suite_spec(),
         registry = None,
@@ -1244,6 +1250,7 @@ def rbe_autoconfig(
         export_configs = export_configs,
         java_home = java_home,
         java_version = java_version,
+        os_family = os_family,
         toolchain_config_suite_spec = toolchain_config_suite_spec_stripped,
         registry = registry,
         repository = repository,
diff --git a/rules/rbe_repo/util.bzl b/rules/rbe_repo/util.bzl
index 71a0e9ee3..7174ff148 100644
--- a/rules/rbe_repo/util.bzl
+++ b/rules/rbe_repo/util.bzl
@@ -144,6 +144,8 @@ def os_family(ctx):
     Returns:
       Returns the name of the OS Family
     """
+    if ctx.attr.os_family:
+      return ctx.attr.os_family
     os_name = ctx.os.name.lower()
     if os_name.find("windows") != -1:
         return "Windows"