Skip to content

Commit fa8ca68

Browse files
Added '-' conversion for class names
1 parent aa4f611 commit fa8ca68

File tree

2 files changed

+12
-6
lines changed

2 files changed

+12
-6
lines changed

src/main/java/com/vizor/unreal/convert/ProtoProcessor.java

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@
6060
import static com.vizor.unreal.tree.CppType.plain;
6161
import static com.vizor.unreal.util.Misc.reorder;
6262
import static com.vizor.unreal.util.Misc.snakeCaseToCamelCase;
63+
import static com.vizor.unreal.util.Misc.dashToUnderscore;
6364
import static com.vizor.unreal.util.Misc.stringIsNullOrEmpty;
6465
import static com.vizor.unreal.util.Tuple.of;
6566
import 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
}

src/main/java/com/vizor/unreal/util/Misc.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -244,6 +244,11 @@ public static <T> void reorder(final List<T> list, final int[] indices)
244244
}
245245
}
246246

247+
public static String dashToUnderscore(final String dashedString)
248+
{
249+
return dashedString.replace("-", "_");
250+
}
251+
247252
/**
248253
* Transforms snake_case_string into camelCaseString.
249254
* @param snakeCaseString String in snake_case.

0 commit comments

Comments
 (0)