Skip to content

Commit f4b2f1a

Browse files
committed
fix(protobuf): wrap enums in messages
Signed-off-by: Ahmed Mohamed <[email protected]>
1 parent e0c19bc commit f4b2f1a

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

src/s2dm/exporters/protobuf/templates/proto.j2

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,14 @@ package {{ package }};
88
{% if enum.description %}
99
// {{ enum.description }}
1010
{% endif %}
11-
enum {{ enum.name }} {
12-
{{ enum.name | upper }}_UNSPECIFIED = 0;
11+
message {{ enum.name }} {
12+
enum Enum {
13+
{{ enum.name | upper }}_UNSPECIFIED = 0;
1314
{% for value in enum.enum_values %}
14-
{{ value.name }} = {{ value.number }};{% if value.description %} // {{ value.description }}{% endif %}
15+
{{ value.name }} = {{ value.number }};{% if value.description %} // {{ value.description }}{% endif %}
1516

1617
{% endfor %}
18+
}
1719
}
1820

1921
{% endfor %}

src/s2dm/exporters/protobuf/transformer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -343,7 +343,7 @@ def _get_field_proto_type(self, field_type: GraphQLType) -> str:
343343

344344
if is_enum_type(field_type):
345345
enum_type = cast(GraphQLEnumType, field_type)
346-
return enum_type.name
346+
return f"{enum_type.name}.Enum"
347347

348348
if is_object_type(field_type) or is_interface_type(field_type):
349349
named_type = cast(GraphQLObjectType | GraphQLInterfaceType, field_type)

0 commit comments

Comments
 (0)