fix: omit also jinja functions

src/rattler_build_conda_compat/jinja.py

@@ -27,7 +27,28 @@ def jinja_env() -> jinja2.Environment:
     """
     Create a `rattler-build` specific Jinja2 environment with modified syntax.
     """
-    return jinja2.Environment(
+
+    def stub_compatible_pin(*args, **kwargs) -> str:  # noqa: ARG001, ANN003, ANN002
+        return f"compatible_pin {args[0]}"
+
+    def stub_subpackage_pin(*args, **kwargs) -> str:  # noqa: ARG001, ANN003, ANN002
+        return f"subpackage_pin {args[0]}"
+
+    def version_to_build_string(some_string: str) -> str:
+        """Converts some version by removing the . character and returning only the first two elements of the version)"""
+        # We first split the string by whitespace and take the first part
+        split = some_string.split()[0] if some_string.split() else some_string
+        # We then split the string by . and take the first two parts
+        parts = split.split(".")
+        major = parts[0] if len(parts) > 0 else ""
+        minor = parts[1] if len(parts) > 1 else ""
+        return f"{major}{minor}"
+
+    def split_filter(s: str, sep: str = " ") -> list[str]:
+        """Filter that split a string by a separator"""
+        return s.split(sep)
+
+    env = jinja2.Environment(
         variable_start_string="${{",
         variable_end_string="}}",
         trim_blocks=True,
@@ -36,6 +57,26 @@ def jinja_env() -> jinja2.Environment:
         undefined=_MissingUndefined,
     )
 
+    # inject rattler-build recipe functions in jinja environment
+    env.globals.update(
+        {
+            "compiler": lambda x: x + "_compiler_stub",
+            "stdlib": lambda x: x + "_stdlib_stub",
+            "pin_subpackage": stub_subpackage_pin,
+            "pin_compatible": stub_compatible_pin,
+            "cdt": lambda *args, **kwargs: "cdt_stub",  # noqa: ARG005
+        }
+    )
+
+    # inject rattler-build recipe filters in jinja environment
+    env.filters.update(
+        {
+            "version_to_buildstring": version_to_build_string,
+            "split": split_filter,
+        }
+    )
+    return env
+
 
 def load_recipe_context(context: dict[str, str], jinja_env: jinja2.Environment) -> dict[str, str]:
     """

tests/__snapshots__/test_jinja.ambr

@@ -2,14 +2,17 @@
 # name: test_render_recipe_with_context
   '''
   build:
-    string: ${{ blas_variant }}${{ hash }}_foo-bla
+    string: ${{ blas_variant }}${{ hash }}_foo-1.2.3
   context:
+    cuda_version: ${{ hash }}_cuda_12
     name: foo
-    name_version: foo-bla
-    version: bla
+    name_version: foo-1.2.3
+    split_version: 1;2;3
+    version: 1.2.3
   package:
+    func_result: c_compiler_stub
     name: foo
-    version: bla
+    version: 1.2.3
 
   '''
 # ---

tests/data/context.yaml

@@ -1,11 +1,14 @@
 context:
   name: "foo"
-  version: "bla"
+  version: "1.2.3"
   name_version: ${{ name }}-${{ version }}
+  cuda_version: ${{ hash }}_cuda_${{ version | version_to_buildstring }}
+  split_version: ${{ version | split('.') | join(';') }}
 
 package:
   name: ${{ name }}
   version: ${{ version }}
+  func_result: ${{ compiler('c') }}
 
 build:
   string: ${{ blas_variant }}${{ hash }}_${{ name_version }}