Replace KotlinTopLevelExtensionConfig with KotlinBaseExtension in Pok…

…oBuildPlugin (#453)
drewhamilton · Dec 16, 2024 · 79e50bf · 79e50bf
1 parent 9281ce8
commit 79e50bf
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/build-support/src/main/java/dev/drewhamilton/poko/build/PokoBuildPlugin.kt b/build-support/src/main/java/dev/drewhamilton/poko/build/PokoBuildPlugin.kt
@@ -8,7 +8,8 @@ import org.gradle.api.Plugin
 import org.gradle.api.Project
 import org.gradle.api.plugins.AppliedPlugin
 import org.gradle.kotlin.dsl.buildConfigField
-import org.jetbrains.kotlin.gradle.dsl.KotlinTopLevelExtensionConfig
+import org.gradle.kotlin.dsl.getByType
+import org.jetbrains.kotlin.gradle.dsl.KotlinBaseExtension
 import org.jetbrains.kotlin.gradle.tasks.KotlinCompilationTask
 
 private val Project.pokoGroupId get() = property("PUBLISH_GROUP") as String
@@ -88,9 +89,7 @@ class PokoBuildPlugin : Plugin<Project> {
 
             // Published modules should be explicit about their API visibility.
             val kotlinPluginHandler = Action<AppliedPlugin> {
-                val kotlin = project.extensions.getByType(
-                    KotlinTopLevelExtensionConfig::class.java
-                )
+                val kotlin = project.extensions.getByType<KotlinBaseExtension>()
                 kotlin.explicitApi()
             }
             project.pluginManager.withPlugin("org.jetbrains.kotlin.jvm", kotlinPluginHandler)