Add: support jinja.

Author: ya7010

cmdcomp/app.py

@@ -33,7 +33,7 @@ def run(cls, args: list[str] | None = None) -> None:
             "-f",
             required=True,
             type=FileType("rb"),
-            help="config file ('.json', '.yaml', '.toml' support).",
+            help="config file ('.json', '.yaml', '.toml', '.jinja2' support).",
         )
 
         parser.add_argument(

cmdcomp/config.py

@@ -1,8 +1,10 @@
 import json
 import os
 import tomllib
-from typing import BinaryIO
+from io import StringIO
+from typing import IO
 
+import jinja2
 import yaml
 from pydantic import RootModel
 
@@ -17,8 +19,8 @@ class Config(RootModel):
     root: V1Config | V2Config
 
 
-def load(file: BinaryIO) -> Config:
-    _, extension = os.path.splitext(file.name)
+def load(file: IO) -> Config:
+    root, extension = os.path.splitext(file.name)
     match extension:
         case ".json":
             data = json.load(file)
@@ -29,6 +31,12 @@ def load(file: BinaryIO) -> Config:
         case ".toml":
             data = tomllib.load(file)
 
+        case ".jinja" | "jinja2" | ".j2":
+            data = StringIO(jinja2.Template(file.read()).render(os.environ))
+            data.name = root
+
+            return load(data)
+
         case _:
             raise ConfigFileExtensionError(extension)
 

tests/test_config.py

@@ -0,0 +1,34 @@
+from io import StringIO
+from textwrap import dedent
+
+from cmdcomp.config import load
+
+
+def make_file(filename: str, contents: str) -> StringIO:
+    file = StringIO(contents)
+    file.name = filename
+
+    return file
+
+
+class TestConfig:
+    def test_jinja(self):
+        config = load(
+            make_file(
+                "config.yaml.jinja",
+                dedent(
+                    """
+                    {%- set app_name = "mycli" -%}
+                    cmdcomp:
+                        version: "2"
+                    app:
+                        name: {{ app_name }}
+                    root:
+                        arguments:
+                            --help:
+                    """
+                ),
+            )
+        )
+
+        assert config.root.app.name == "mycli"