Define SemConv predefined JQ functions

crates/weaver_codegen_test/build.rs

@@ -16,7 +16,7 @@ use weaver_common::{in_memory, Logger};
 use weaver_forge::config::{Params, WeaverConfig};
 use weaver_forge::file_loader::FileSystemFileLoader;
 use weaver_forge::registry::ResolvedRegistry;
-use weaver_forge::{OutputDirective, TemplateEngine};
+use weaver_forge::{OutputDirective, TemplateEngine, SEMCONV_JQ};
 use weaver_resolver::SchemaResolver;
 use weaver_semconv::path::RegistryPath;
 use weaver_semconv::registry::SemConvRegistry;
@@ -53,7 +53,10 @@ fn main() {
         .unwrap_or_else(|e| process_error(&logger, e));
     let config = WeaverConfig::try_from_path("./templates/registry/rust")
         .unwrap_or_else(|e| process_error(&logger, e));
-    let engine = TemplateEngine::new(config, loader, Params::default());
+    let mut engine = TemplateEngine::new(config, loader, Params::default());
+    engine
+        .import_jq_package(SEMCONV_JQ)
+        .unwrap_or_else(|e| process_error(&logger, e));
     let template_registry = ResolvedRegistry::try_from_resolved_registry(
         schema
             .registry(REGISTRY_ID)

crates/weaver_codegen_test/templates/registry/alt_weaver.yaml

@@ -2,4 +2,4 @@
 params:
   attributes: true
   metrics: false    # With this alternate configuration, the metrics are disabled
-  registry_prefix: "registry."
+  exclude_deprecated: true

crates/weaver_codegen_test/templates/registry/rust/attribute_macros.j2

@@ -1,60 +1,60 @@
 {%- macro comments(attribute, prefix) -%}
-{%- if attribute.brief %}
+    {% if attribute.brief %}
 {{ attribute.brief | comment_with_prefix(prefix ~ " ") }}
-{%- endif %}
-{%- if attribute.note %}
+    {% endif %}
+    {% if attribute.note %}
 {{ prefix }}
 {{ prefix }} Notes:
 {{ attribute.note | comment_with_prefix(prefix ~ "   ") }}
-{%- endif %}
-{%- if attribute.examples %}
-{%- if attribute.examples is sequence %}
+    {% endif %}
+    {% if attribute.examples %}
+        {% if attribute.examples is sequence %}
 {{ prefix }}
 {{ prefix }} Examples:
-{%- for example in attribute.examples %}
+            {% for example in attribute.examples %}
 {{ example | comment_with_prefix(prefix ~ " - ") }}
-{%- endfor %}
-{%- else %}
+            {% endfor %}
+        {% else %}
 {{ prefix }}
 {{ prefix }} Example: {{ attribute.examples | trim }}
-{%- endif %}
-{%- endif %}
-{%- endmacro %}
+        {% endif %}
+    {% endif %}
+{% endmacro %}
 
 {%- macro attributes_to_key_values(required_attributes, not_required_attributes) -%}
         let mut attributes = vec![
-            {%- for attribute in required_attributes | attribute_sort %}
-            {%- if attribute is experimental %}
+            {% for attribute in required_attributes | attribute_sort %}
+            {% if attribute is experimental %}
             #[cfg(feature = "semconv_experimental")]
-            {%- endif %}
-            {%- if attribute.type.members is defined %}
+            {% endif %}
+            {% if attribute.type.members is defined %}
             crate::attributes::{{ attribute.name | attribute_namespace }}::{{ attribute.name | screaming_snake_case }}.value(&required_attributes.{{ attribute.name | snake_case }}),
-            {%- elif attribute.type == "string" %}
+            {% elif attribute.type == "string" %}
             crate::attributes::{{ attribute.name | attribute_namespace }}::{{ attribute.name | screaming_snake_case }}.value(required_attributes.{{ attribute.name | snake_case }}.to_owned().into()),
-            {%- else %}
+            {% else %}
             crate::attributes::{{ attribute.name | attribute_namespace }}::{{ attribute.name | screaming_snake_case }}.value(required_attributes.{{ attribute.name | snake_case }}),
-            {%- endif %}
-            {%- endfor %}
+            {% endif %}
+            {% endfor %}
         ];
 
         if let Some(value) = &not_required_attributes {
-            {%- for attribute in not_required_attributes | attribute_sort %}
-            {%- if attribute is experimental %}
+            {% for attribute in not_required_attributes | attribute_sort %}
+            {% if attribute is experimental %}
             #[cfg(feature = "semconv_experimental")]
-            {%- endif %}
-            {%- if attribute.type.members is defined %}
+            {% endif %}
+            {% if attribute.type.members is defined %}
             if let Some({{ attribute.name | snake_case }}) = &value.{{ attribute.name | snake_case }} {
                 attributes.push(crate::attributes::{{ attribute.name | attribute_namespace }}::{{ attribute.name | screaming_snake_case }}.value({{ attribute.name | snake_case }}));
             }
-            {%- elif attribute.type == "string" %}
+            {% elif attribute.type == "string" %}
             if let Some({{ attribute.name | snake_case }}) = &value.{{ attribute.name | snake_case }} {
                 attributes.push(crate::attributes::{{ attribute.name | attribute_namespace }}::{{ attribute.name | screaming_snake_case }}.value({{ attribute.name | snake_case }}.to_owned().into()));
             }
-            {%- else %}
+            {% else %}
             if let Some({{ attribute.name | snake_case }}) = value.{{ attribute.name | snake_case }} {
                 attributes.push(crate::attributes::{{ attribute.name | attribute_namespace }}::{{ attribute.name | screaming_snake_case }}.value({{ attribute.name | snake_case }}));
             }
-            {%- endif %}
-            {%- endfor %}
+            {% endif %}
+            {% endfor %}
         }
 {%- endmacro %}

crates/weaver_codegen_test/templates/registry/rust/attributes/attributes.rs.j2

@@ -1,4 +1,4 @@
-{%- set file_name = ctx.id | attribute_registry_namespace | snake_case -%}
+{%- set file_name = ctx.namespace | snake_case -%}
 {{- template.set_file_name("attributes/" ~ file_name ~ ".rs") -}}
 {%- import 'attribute_macros.j2' as attribute_macros -%}
 
@@ -7,42 +7,41 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-{{ ctx.brief | comment_with_prefix("//! ") }}
-{%- if ctx.note %}
-//!
-//! Notes:
-{{ ctx.note | comment_with_prefix("//!   ") }}
-{%- endif %}
 //! DO NOT EDIT, THIS FILE HAS BEEN GENERATED BY WEAVER
 
-{%- for attribute in ctx.attributes | attribute_sort %}
-{{ attribute_macros.comments(attribute, "///") }}
-{%- if attribute is experimental %}
+{% for attribute in ctx.attributes | attribute_sort %}
+{{ attribute_macros.comments(attribute, "///") -}}
+    {% if attribute is experimental %}
 #[cfg(feature = "semconv_experimental")]
-{%- endif %}
-{%- if attribute is deprecated %}
+    {% endif %}
+    {% if attribute is deprecated %}
 #[deprecated(note="{{ attribute.deprecated }}")]
-{%- endif %}
-{%- if attribute.type.allow_custom_values is defined %}
+    {% endif %}
+    {% if attribute.type.allow_custom_values is defined %}
 pub const {{ attribute.name | screaming_snake_case }}: crate::attributes::AttributeKey<{{ attribute.name | pascal_case }}> = crate::attributes::AttributeKey::new("{{ attribute.name }}");
-{%- elif attribute.type == "string" %}
+    {% elif attribute.type == "string" %}
 pub const {{ attribute.name | screaming_snake_case }}: crate::attributes::AttributeKey<opentelemetry::{{ attribute.type | map_text("rust_types") }}Value> = crate::attributes::AttributeKey::new("{{ attribute.name }}");
-{%- else %}
+    {% else %}
 pub const {{ attribute.name | screaming_snake_case }}: crate::attributes::AttributeKey<{{ attribute.type | map_text("rust_types") }}> = crate::attributes::AttributeKey::new("{{ attribute.name }}");
-{%- endif %}
-{%- if attribute.type.members is defined %}
+    {% endif %}
 
-{% if attribute.brief %}{{ attribute.brief | comment_with_prefix("/// ") }}{%- endif %}
+    {% if attribute.type.members is defined %}
+        {% if attribute.brief %}
+{{ attribute.brief | comment_with_prefix("/// ") }}
+        {% endif %}
 #[derive(Debug, Clone)]
 #[non_exhaustive]
 pub enum {{ attribute.name | pascal_case }} {
-{%- for variant in attribute.type.members %}
-    {{ variant.brief | default("No brief") | comment_with_prefix("    /// ") }}
-    {%- if variant.note %}{{ variant.note | comment_with_prefix("    /// ") }}{% endif %}
-    {%- if variant is experimental %}
-    #[cfg(feature = "semconv_experimental")] {% endif %}
+        {% for variant in attribute.type.members %}
+    {{ variant.brief | default("No brief") | comment_with_prefix("/// ") }}
+            {% if variant.note %}
+    {{ variant.note | comment_with_prefix("/// ") }}
+            {% endif %}
+            {% if variant is experimental %}
+    #[cfg(feature = "semconv_experimental")]
+            {% endif %}
     {{ variant.id | pascal_case }},
-{%- endfor %}
+        {% endfor %}
     /// This variant allows defining a custom entry in the enum.
     _Custom(String),
 }
@@ -52,11 +51,12 @@ impl {{ attribute.name | pascal_case }} {
     #[must_use]
     pub fn as_str(&self) -> &str {
         match self {
-        {%- for variant in attribute.type.members %}
-            {%- if variant is experimental %}
-            #[cfg(feature = "semconv_experimental")] {% endif %}
+        {% for variant in attribute.type.members %}
+            {% if variant is experimental %}
+            #[cfg(feature = "semconv_experimental")]
+            {% endif %}
             {{ attribute.name | pascal_case }}::{{ variant.id | pascal_case }} => "{{ variant.value }}",
-        {%- endfor %}
+        {% endfor %}
             {{ attribute.name | pascal_case }}::_Custom(v) => v.as_str(),
             // Without this default case, the match expression would not
             // contain any variants if all variants are annotated with the
@@ -81,5 +81,6 @@ impl crate::attributes::AttributeKey<{{ attribute.name | pascal_case }}> {
         opentelemetry::KeyValue::new(self.key.clone(), v.to_string())
     }
 }
-{%- endif %}
-{%- endfor %}
+
+    {% endif %}
+{% endfor %}

crates/weaver_codegen_test/templates/registry/rust/attributes/mod.rs.j2

@@ -12,10 +12,10 @@ use opentelemetry::{Key, KeyValue, StringValue};
 
 pub const ATTRIBUTES_PARAM: bool = {{ params.attributes }};
 
-{% for group in ctx %}
-/// Attributes for the `{{ group.id | attribute_registry_namespace }}` namespace.
-pub mod {{ group.id | attribute_registry_namespace | snake_case }};
-{%- endfor %}
+{% for attributes_namespace in ctx %}
+/// Attributes for the `{{ attributes_namespace.namespace }}` namespace.
+pub mod {{ attributes_namespace.namespace | snake_case }};
+{% endfor %}
 
 /// A typed attribute key.
 pub struct AttributeKey<T> {