corrected packaged

tieskedh · Oct 23, 2018 · 8493ff8 · 8493ff8
1 parent 84a92d3
commit 8493ff8
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/main/kotlin/nl/devhaan/kotlinpoetdsl/helpers.kt b/src/main/kotlin/nl/devhaan/kotlinpoetdsl/helpers.kt
@@ -4,13 +4,15 @@ package nl.devhaan.kotlinpoetdsl
 import com.squareup.kotlinpoet.*
 
 val EMPTY_CODEBLOCK = CodeBlock.builder().build()
-fun TypeSpec.packaged(pack: String, vararg outerClasses: String) = ClassName.bestGuess((listOf(pack) + outerClasses).joinToString("."+name!!))
-
+fun TypeSpec.packaged(pack: String, vararg outerClasses: String): ClassName {
+    return if (outerClasses.isEmpty()) ClassName(pack, name!!)
+    else ClassName.bestGuess((listOf(pack) + outerClasses+name!!).joinToString("."))
+}
 inline fun <reified T> typeNameFor(): ClassName {
     val typeName = T::class.asTypeName()
     return if (null is T) typeName.asNullable() else typeName.asNonNullable()
 }
 fun String.S() = "\"$this\""
 
 fun TypeSpec.println() = println(this)
-fun FunSpec.println() = println(this)
+fun FunSpec.println() = println(this)