6060import static com .vizor .unreal .tree .CppType .plain ;
6161import static com .vizor .unreal .util .Misc .reorder ;
6262import static com .vizor .unreal .util .Misc .snakeCaseToCamelCase ;
63+ import static com .vizor .unreal .util .Misc .dashToUnderscore ;
6364import static com .vizor .unreal .util .Misc .stringIsNullOrEmpty ;
6465import static com .vizor .unreal .util .Tuple .of ;
6566import static java .io .File .separator ;
@@ -102,8 +103,8 @@ class ProtoProcessor implements Runnable
102103 this .moduleName = requireNonNull (moduleName );
103104
104105 this .wrapperName = removeExtension (pathToProto .toFile ().getName ());
105-
106- this .className = snakeCaseToCamelCase (wrapperName );
106+
107+ this .className = snakeCaseToCamelCase (dashToUnderscore ( wrapperName ) );
107108
108109// if (parse.packageName() == null)
109110// throw new RuntimeException("package filed in proto file is required for cornerstone");
@@ -118,8 +119,8 @@ public void run()
118119
119120 for (final TypeElement t : parse .types ())
120121 {
121- ueProvider .register (t .name (), ueNamedType (className , t ));
122- protoProvider .register (t .name (), cppNamedType (t ));
122+ ueProvider .register (dashToUnderscore ( t .name () ), ueNamedType (className , t ));
123+ protoProvider .register (dashToUnderscore ( t .name () ), cppNamedType (t ));
123124 }
124125
125126 final List <Tuple <CppStruct , CppStruct >> castAssociations = new ArrayList <>();
@@ -332,9 +333,9 @@ private CppEnum extractEnum(final TypesProvider provider, final EnumElement ee)
332333 private CppType ueNamedType (final String serviceName , final TypeElement el )
333334 {
334335 if (el instanceof MessageElement )
335- return plain ("F" + serviceName + "_" + el .name (), Struct );
336+ return plain ("F" + serviceName + "_" + dashToUnderscore ( el .name () ), Struct );
336337 else if (el instanceof EnumElement )
337- return plain ("E" + serviceName + "_" + el .name (), Enum );
338+ return plain ("E" + serviceName + "_" + dashToUnderscore ( el .name () ), Enum );
338339 else
339340 throw new RuntimeException ("Unknown type: '" + el .getClass ().getName () + "'" );
340341 }
0 commit comments