diff --git a/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/misc/BlockExtractor.kt b/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/misc/BlockExtractor.kt index 922ad515..7e30198f 100644 --- a/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/misc/BlockExtractor.kt +++ b/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/misc/BlockExtractor.kt @@ -60,8 +60,10 @@ class BlockExtractor(val regexp: Pattern, val opening: Char, val closing: Char) result.append(line).append("\n") } else { val allowedImports = listOf("com.beust", "java") + val disallowedImports = listOf("com.beust.kobalt.plugin") if (! line.startsWith("import") || - (line.startsWith("import") && allowedImports.any { line.contains(it) })) { + (line.startsWith("import") && allowedImports.any { line.contains(it) } + && ! disallowedImports.any { line.contains(it) })) { topLines.add(line) } updateCount(line)