Better target source set configuration

README.adoc

@@ -684,6 +684,19 @@ plugins {
 ----
 ====
 
+==== Applying to main source sets
+
+In some cases, you may need to apply the processor to the common-main source-set instead of common-test.
+
+[source,kotlin,subs="verbatim,attributes"]
+.build.gradle.kts
+----
+mockmp {
+    targetSourceSet = CommonMain
+}
+----
+
+
 === With KSP and its incomplete multiplatform support
 
 KSP for multiplatform is in beta, and *https://github.com/google/ksp/issues/567[KSP for common tests is not supported]* (yet).

mockmp-gradle-plugin/src/main/kotlin/org/kodein/mock/gradle/MocKMPGradlePlugin.kt

@@ -14,18 +14,21 @@ import org.jetbrains.kotlin.gradle.plugin.KotlinSourceSet
 // Come on Google, having KSP work in multiplatform correctly must be very important!
 class MocKMPGradlePlugin : Plugin<Project> {
 
+    @Suppress("EnumEntryName")
+    enum class SourceSetTarget {
+        CommonMain,
+        CommonTest_Via_JVM
+    }
+
     @Suppress("PropertyName")
     class Extension(
         var usesHelper: Boolean = false,
         var throwErrors: Boolean = false,
         var public: Boolean = false,
-        var sourceSet: SourceSet = SourceSet.CommonTest_Via_JVM
+        var targetSourceSet: SourceSetTarget = SourceSetTarget.CommonTest_Via_JVM
     ) {
-        @Suppress("ClassName")
-        sealed class SourceSet {
-            object CommonMain : SourceSet()
-            object CommonTest_Via_JVM : SourceSet()
-        }
+        val CommonMain get() = SourceSetTarget.CommonMain
+        val CommonTest_Via_JVM get() = SourceSetTarget.CommonTest_Via_JVM
     }
 
     private val Project.kotlinExtension get() = project.extensions.findByType<KotlinMultiplatformExtension>() ?: throw GradleException("Could not find Kotlin/Multiplatform plugin")
@@ -105,9 +108,9 @@ class MocKMPGradlePlugin : Plugin<Project> {
         target.extensions.add("mockmp", ext)
 
         target.afterEvaluate {
-            when (ext.sourceSet) {
-                Extension.SourceSet.CommonMain -> executeOnMain(target, ext)
-                Extension.SourceSet.CommonTest_Via_JVM -> executeOnTests(target, ext)
+            when (ext.targetSourceSet) {
+                SourceSetTarget.CommonMain -> executeOnMain(target, ext)
+                SourceSetTarget.CommonTest_Via_JVM -> executeOnTests(target, ext)
             }
 
         }