CDRIVER-4730 TLS Backend Cleanup

.evergreen/config_generator/components/earthly.py

@@ -1,41 +1,160 @@
-from typing import Iterable
+from __future__ import annotations
+
+import functools
+import itertools
+from typing import Any, Iterable, Literal, TypeVar, get_args, NamedTuple, get_type_hints
 from shrub.v3.evg_build_variant import BuildVariant
 from shrub.v3.evg_task import EvgTaskRef
 from ..etc.utils import Task
 from shrub.v3.evg_command import subprocess_exec, EvgCommandType
 
-_ENV_PARAM_NAME = "MONGOC_EARTHLY_ENV"
-
+T = TypeVar("T")
 
-class EarthlyTask(Task):
-    def __init__(self, *, suffix: str, target: str, sasl: str) -> None:
-        super().__init__(
-            name=f"earthly-{suffix}",
-            commands=[
-                subprocess_exec(
-                    "bash",
-                    args=[
-                        "tools/earthly.sh",
-                        f"+{target}",
-                        f"--env=${{{_ENV_PARAM_NAME}}}",
-                        f"--enable_sasl={sasl}",
-                    ],
-                    working_dir="mongoc",
-                    command_type=EvgCommandType.TEST,
-                )
-            ],
-            tags=[f"earthly", "pr-merge-gate"],
-            run_on=CONTAINER_RUN_DISTROS,
-        )
+_ENV_PARAM_NAME = "MONGOC_EARTHLY_ENV"
+"The name of the EVG expansion parameter used to key the Earthly build env"
 
+EnvKey = Literal["u22", "alpine3.18", "archlinux"]
+"Identifiers for environments. These correspond to special '*-env' targets in the Earthfile."
 
-#: A mapping from environment keys to the environment name.
-#: These correspond to special "*-env" targets in the Earthfile.
-ENVS = {
+_ENV_NAMES: dict[EnvKey, str] = {
     "u22": "Ubuntu 22.04",
     "alpine3.18": "Alpine 3.18",
     "archlinux": "Arch Linux",
 }
+"A mapping from environment keys to 'pretty' environment names"
+
+# Other options: SSPI (Windows only), AUTO (not reliably test-able without more environments)
+SASLOption = Literal["Cyrus", "off"]
+"Valid options for the SASL configuration parameter"
+TLSOption = Literal["LibreSSL", "OpenSSL", "off"]
+"Options for the TLS backend configuration parameter (AKA 'ENABLE_SSL')"
+CxxVersion = Literal["master", "r3.8.0"]
+"C++ driver refs that are under CI test"
+
+# A Unicode non-breaking space character
+_BULLET = "\N{Bullet}"
+
+
+class Configuration(NamedTuple):
+    """
+    Represent a complete set of argument values to give to the Earthly task
+    execution. Each field name matches the ARG in the Earthfile.
+
+    Adding/removing fields will add/remove dimensions on the task matrix.
+
+    The 'env' parameter is not encoded here, but is managed separately.
+
+    Keep this in sync with the 'PartialConfiguration' class defined below!
+    """
+
+    sasl: SASLOption
+    tls: TLSOption
+    test_mongocxx_ref: CxxVersion
+
+    @classmethod
+    def all(cls) -> Iterable[Configuration]:
+        """
+        Generate all configurations for all options of our parameters.
+        """
+        # Iter each configuration parameter:
+        fields: Iterable[tuple[str, type]] = get_type_hints(Configuration).items()
+        # Generate lists of pairs of parameter names their options:
+        all_pairs: Iterable[Iterable[tuple[str, str]]] = (
+            # Generate a (key, opt) pair for each option in parameter 'key'
+            [(key, opt) for opt in get_args(typ)]
+            # Over each parameter and type thereof:
+            for key, typ in fields
+        )
+        # Now generate the cross product of all alternative for all options:
+        matrix: Iterable[dict[str, Any]] = map(dict, itertools.product(*all_pairs))
+        for items in matrix:
+            # Convert each item to a Configuration:
+            yield Configuration(**items)
+
+    @property
+    def suffix(self) -> str:
+        return f"{_BULLET}".join(f"{k}={v}" for k, v in self._asdict().items())
+
+
+def task_filter(env: EnvKey, conf: Configuration) -> bool:
+    """
+    Control which tasks are actually defined by matching on the platform and
+    configuration values.
+    """
+    match env, conf:
+        # We only need one task with "sasl=off"
+        case "u22", ("off", "OpenSSL", "master"):
+            return True
+        # Other sasl=off tasks we'll just ignore:
+        case _, ("off", _tls, _cxx):
+            return False
+        # Ubuntu does not ship with a LibreSSL package:
+        case e, (_sasl, "LibreSSL", _cxx) if _ENV_NAMES[e].startswith("Ubuntu"):
+            return False
+        # Anything else: Allow it to run:
+        case _:
+            return True
+
+
+def envs_for(config: Configuration) -> Iterable[EnvKey]:
+    """Get all environment keys that are not excluded for the given configuration"""
+    all_envs: tuple[EnvKey, ...] = get_args(EnvKey)
+    allow_env_for_config = functools.partial(task_filter, conf=config)
+    return filter(allow_env_for_config, all_envs)
+
+
+def earthly_task(
+    *,
+    name: str,
+    targets: Iterable[str],
+    config: Configuration,
+) -> Task | None:
+    # Attach "earthly-xyz" tags to the task to allow build variants to select
+    # these tasks by the environment of that variant.
+    env_tags = sorted(f"earthly-{e}" for e in sorted(envs_for(config)))
+    if not env_tags:
+        # All environments have been excluded for this configuration. This means
+        # the task itself should not be run:
+        return
+    # Generate the build-arg arguments based on the configuration options. The
+    # NamedTuple field names must match with the ARG keys in the Earthfile!
+    earthly_args = [f"--{key}={val}" for key, val in config._asdict().items()]
+    return Task(
+        name=name,
+        commands=[
+            # First, just build the "env-warmup" which will prepare the build environment.
+            # This won't generate any output, but allows EVG to track it as a separate build step
+            # for timing and logging purposes. The subequent build step will cache-hit the
+            # warmed-up build environments.
+            subprocess_exec(
+                "bash",
+                args=[
+                    "tools/earthly.sh",
+                    "+env-warmup",
+                    f"--env=${{{_ENV_PARAM_NAME}}}",
+                    *earthly_args,
+                ],
+                working_dir="mongoc",
+                command_type=EvgCommandType.SETUP,
+            ),
+            # Now execute the main tasks:
+            subprocess_exec(
+                "bash",
+                args=[
+                    "tools/earthly.sh",
+                    "+run",
+                    f"--targets={' '.join(targets)}",
+                    f"--env=${{{_ENV_PARAM_NAME}}}",
+                    *earthly_args,
+                ],
+                working_dir="mongoc",
+                command_type=EvgCommandType.TEST,
+            ),
+        ],
+        tags=[f"earthly", "pr-merge-gate", *env_tags],
+        run_on=CONTAINER_RUN_DISTROS,
+    )
+
 
 CONTAINER_RUN_DISTROS = [
     "ubuntu2204-small",
@@ -49,28 +168,28 @@ def __init__(self, *, suffix: str, target: str, sasl: str) -> None:
     "amazon2",
 ]
 
-# Other options: SSPI (Windows only), AUTO (not reliably test-able without more environments)
-_SASL_OPTIONS = ["CYRUS", "OFF"]
-
 
 def tasks() -> Iterable[Task]:
-    for sasl in _SASL_OPTIONS:
-        yield EarthlyTask(
-            suffix=f"build-check-sasl:{sasl}".lower(),
-            target="test-example",
-            sasl=sasl,
+    for conf in Configuration.all():
+        task = earthly_task(
+            name=f"check:{conf.suffix}",
+            targets=("test-example", "test-cxx-driver"),
+            config=conf,
         )
+        if task is not None:
+            yield task
 
 
 def variants() -> list[BuildVariant]:
+    envs: tuple[EnvKey, ...] = get_args(EnvKey)
     return [
         BuildVariant(
-            name=f"earthly-{env_key}",
-            tasks=[EvgTaskRef(name=".earthly")],
-            display_name=env_name,
+            name=f"earthly-{env}",
+            tasks=[EvgTaskRef(name=f".earthly-{env}")],
+            display_name=_ENV_NAMES[env],
             expansions={
-                _ENV_PARAM_NAME: env_key,
+                _ENV_PARAM_NAME: env,
             },
         )
-        for env_key, env_name in ENVS.items()
+        for env in envs
     ]