diff --git a/kotlinpoet/src/main/java/com/squareup/kotlinpoet/FileSpec.kt b/kotlinpoet/src/main/java/com/squareup/kotlinpoet/FileSpec.kt index 7379eef68a..448bd8248b 100644 --- a/kotlinpoet/src/main/java/com/squareup/kotlinpoet/FileSpec.kt +++ b/kotlinpoet/src/main/java/com/squareup/kotlinpoet/FileSpec.kt @@ -344,19 +344,28 @@ public class FileSpec private constructor( constant.name, ) - public fun addImport(`class`: Class<*>, vararg names: String): Builder = apply { - require(names.isNotEmpty()) { "names array is empty" } - addImport(`class`.asClassName(), names.toList()) + public fun addImport(`class`: Class<*>, names: Array): Builder = apply { + addImport(`class`.asClassName(), listOf(*names)) } - public fun addImport(`class`: KClass<*>, vararg names: String): Builder = apply { - require(names.isNotEmpty()) { "names array is empty" } - addImport(`class`.asClassName(), names.toList()) + public fun addImport(`class`: KClass<*>, names: Array): Builder = apply { + addImport(`class`.asClassName(), listOf(*names)) } - public fun addImport(className: ClassName, vararg names: String): Builder = apply { - require(names.isNotEmpty()) { "names array is empty" } - addImport(className, names.toList()) + public fun addImport(className: ClassName, names: Array): Builder = apply { + addImport(`class`.asClassName(), listOf(*names)) + } + + public fun addImport(`class`: Class<*>, names: Array): Builder = apply { + addImport(`class`.asClassName(), listOf(*names)) + } + + public fun addImport(`class`: KClass<*>, name: String, vararg names: String): Builder = apply { + addImport(`class`.asClassName(), listOf(name, *names)) + } + + public fun addImport(className: ClassName, name: String, vararg names: String): Builder = apply { + addImport(className, listOf(name, *names)) } public fun addImport(`class`: Class<*>, names: Iterable): Builder = @@ -372,9 +381,13 @@ public class FileSpec private constructor( } } - public fun addImport(packageName: String, vararg names: String): Builder = apply { + public fun addImport(packageName: String, names: Array): Builder = apply { require(names.isNotEmpty()) { "names array is empty" } - addImport(packageName, names.toList()) + addImport(packageName, listOf(*names)) + } + + public fun addImport(packageName: String, name: String, vararg names: String): Builder = apply { + addImport(packageName, listOf(name, *names)) } public fun addImport(packageName: String, names: Iterable): Builder = apply {