fix: enums snake case

eagletrt · Apr 3, 2024 · 29ecb28 · 29ecb28
1 parent 66341d5
commit 29ecb28
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/generator/utils.py b/src/generator/utils.py
@@ -111,8 +111,11 @@ def parse_report_tree(key: str, node: dict, parent_config: dict) -> list:
 
     return topics
 
+def role_enum_name(role: str) -> str:
+    return 'role_' + re.sub(r'(?<!^)(?=[A-Z])', '_', role).lower()
+
 def topic_enum_name(topic: str) -> str:
-    return (topic[0].lower() + topic[1:])
+    return re.sub(r'(?<!^)(?=[A-Z])', '_', topic).lower()
 
 def topic_get_name(topic: str) -> str: 
     return ('GetTopic' + topic)

diff --git a/src/templates/topics.h.j2 b/src/templates/topics.h.j2
@@ -9,7 +9,7 @@ namespace MQTTTopics
 enum class Role 
 {
     {%- for role in roles %}
-    role{{ role.capitalize() }} = {{ loop.index - 1}}
+    {{ utils.role_enum_name(role) }} = {{ loop.index - 1}}
     {%- if not loop.last -%} , {%- endif -%}
     {%- endfor %}
 };