Skip to content

Commit 99e96d8

Browse files
committed
modify enum generator to take into account protobuf enum style argument
1 parent 908097a commit 99e96d8

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

protoc_plugin/lib/src/enum_generator.dart

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,8 +64,21 @@ class EnumGenerator extends ProtobufContainer {
6464
_aliases.add(EnumAlias(value, canonicalValue));
6565
_originalAliasIndices.add(i);
6666
}
67+
// Generate Dart name for the enum value
68+
String dartName;
69+
final isProtobufEnumStyle =
70+
parent.fileGen?.options.protobufEnumStyle ?? false;
71+
if (isProtobufEnumStyle) {
72+
// Strip enum prefix from protobuf-style enum values
73+
final strippedName = stripEnumPrefix(descriptor.name, value.name);
74+
dartName = avoidInitialUnderscore(strippedName);
75+
} else {
76+
// Use original style
77+
dartName = avoidInitialUnderscore(value.name);
78+
}
79+
6780
dartNames[value.name] = disambiguateName(
68-
avoidInitialUnderscore(value.name),
81+
dartName,
6982
usedNames,
7083
enumSuffixes(),
7184
);

0 commit comments

Comments
 (0)