escape_jinja_in_json_leaf_strs

type JsonVal = bool | int | float | str | None | dict[str, JsonVal] | list[JsonVal]


def escape_jinja_in_leaf_strs(val: JsonVal) -> JsonVal:
    """
    Escape Jinja template markers in leaf strings.

    Assuming that any strings at leaf nodes of the given JSON object, sanitize them so that:
    (a) Jinja processing doesn't fail, and
    (b) the result after such processing is equivalent to the input `val`.
    """
    match val:
        case dict():
            return {k: escape_jinja_in_leaf_strs(v) for k, v in val.items()}
        case list():
            return [escape_jinja_in_leaf_strs(v) for v in val]
        case str():
            if "{{" in val or "{%" in val:
                return "{%raw%}" + val + "{%endraw%}"
            else:
                return val
        case _:
            return val