Avoid duplicates in enum imports (Scala).

kobylynskyi · Apr 29, 2021 · 8d63027 · 8d63027
1 parent 7f93937
commit 8d63027
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/src/main/resources/templates/scala-lang/scalaClassGraphqlType.ftl b/src/main/resources/templates/scala-lang/scalaClassGraphqlType.ftl
@@ -15,23 +15,37 @@ import com.kobylynskyi.graphql.codegen.model.graphql.GraphQLRequestSerializer
 import java.util.Objects
 </#if>
 import scala.collection.JavaConverters._
+<#assign duplicateEnumImports = [] />
+<#assign enumImports = [] />
 <#if fields?has_content>
     <#if enumImportItSelfInScala?has_content>
         <#list fields as field>
             <#list enumImportItSelfInScala as enum>
                 <#if MapperUtil.isScalaCollection(field.type)>
                     <#if enum == MapperUtil.getGenericParameter(field.type)>
-import ${enum}._
+                        <#assign duplicateEnumImports = duplicateEnumImports + [enum] />
                     </#if>
                 <#else >
                     <#if enum == field.type>
-import ${enum}._
+                        <#assign duplicateEnumImports = duplicateEnumImports + [enum] />
                     </#if>
                 </#if>
             </#list>
         </#list>
     </#if>
 </#if>
+<#if duplicateEnumImports?has_content>
+    <#list duplicateEnumImports as duplicateEnumImport>
+        <#if !enumImports?seq_contains(duplicateEnumImport)>
+            <#assign enumImports = enumImports + [duplicateEnumImport]>
+        </#if>
+    </#list>
+</#if>
+<#if enumImports?has_content>
+    <#list enumImports as enumImport>
+import ${enumImport}._
+    </#list>
+</#if>
 <#assign duplicateParentInterfaces = [] />
 <#assign parentInterfaces = [] />
 <#if fields?has_content>