expose custom type naming (#56)

palantir · May 22, 2018 · 8f4eed9 · 8f4eed9
1 parent 4b98917
commit 8f4eed9
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/...src/main/java/com/palantir/code/ts/generator/TypescriptServiceGeneratorConfiguration.java b/...src/main/java/com/palantir/code/ts/generator/TypescriptServiceGeneratorConfiguration.java
@@ -12,6 +12,7 @@
 import java.net.URI;
 import java.util.ArrayList;
 import java.util.HashSet;
+import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
@@ -65,6 +66,13 @@ public Result processType(Type javaType, Context context) {
         };
     }
 
+    @Value.Default
+    public Map<String, String> customTypeNaming() {
+        return new LinkedHashMap<>();
+    }
+
+    public abstract Optional<String> customTypeNamingFunction();
+
     /**
      * Provides a strategy for resolving duplicate method names
      * @return
@@ -210,6 +218,10 @@ public Settings getSettings() {
         typeProcessors.add(getOverridingTypeParser());
         typeProcessors.add(genericTypeProcessor);
         settings.customTypeProcessor = new TypeProcessor.Chain(typeProcessors);
+        settings.customTypeNaming = customTypeNaming();
+        if (customTypeNamingFunction().isPresent()) {
+            settings.customTypeNamingFunction = customTypeNamingFunction().get();
+        }
         settings.addTypeNamePrefix = generatedInterfacePrefix();
         settings.sortDeclarations = true;
         settings.noFileComment = true;