transformTsType() should also transform GenericReferenceType

vojtechhabarta · Apr 21, 2017 · 4a21cec · 4a21cec
1 parent 80424eb
commit 4a21cec
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/typescript-generator-core/src/main/java/cz/habarta/typescript/generator/TsType.java b/typescript-generator-core/src/main/java/cz/habarta/typescript/generator/TsType.java
@@ -239,6 +239,14 @@ public String format(Settings settings) {
 
     public static TsType transformTsType(TsType tsType, Transformer transformer) {
         final TsType type = transformer.transform(tsType);
+        if (type instanceof TsType.GenericReferenceType) {
+            final GenericReferenceType genericReferenceType = (TsType.GenericReferenceType) type;
+            final List<TsType> typeArguments = new ArrayList<>();
+            for (TsType typeArgument : genericReferenceType.typeArguments) {
+                typeArguments.add(transformTsType(typeArgument, transformer));
+            }
+            return new TsType.GenericReferenceType(genericReferenceType.symbol, typeArguments);
+        }
         if (type instanceof TsType.OptionalType) {
             final TsType.OptionalType optionalType = (TsType.OptionalType) type;
             return new TsType.OptionalType(transformTsType(optionalType.type, transformer));