Fix jvmMain not found

Author: Mr3zee

gradle-plugin/src/main/kotlin/kotlinx/rpc/grpc/DefaultGrpcExtension.kt

@@ -59,7 +59,7 @@ internal open class DefaultGrpcExtension @Inject constructor(
 
         createDefaultProtocPlugins()
 
-        project.configureProtoExtensions { _, protoSourceSet ->
+        project.protoSourceSets.forEach { protoSourceSet ->
             protoSourceSet.protocPlugin(protocPlugins.protobufJava)
             protoSourceSet.protocPlugin(protocPlugins.grpcJava)
             protoSourceSet.protocPlugin(protocPlugins.grpcKotlin)
@@ -310,7 +310,7 @@ internal open class DefaultGrpcExtension @Inject constructor(
             }
 
             name.lowercase().endsWith("test") -> {
-                project.protoSourceSets.getByName(correspondingMainName()) as DefaultProtoSourceSet
+                project.protoSourceSets.findByName(correspondingMainName()) as? DefaultProtoSourceSet
             }
 
             else -> {

gradle-plugin/src/main/kotlin/kotlinx/rpc/proto/DefaultProtoSourceSet.kt

@@ -62,12 +62,7 @@ internal open class DefaultProtoSourceSet @Inject constructor(
     }
 }
 
-internal fun Project.configureProtoExtensions(
-    configure: Project.(
-        languageSourceSet: Any,
-        protoSourceSet: DefaultProtoSourceSet,
-    ) -> Unit
-) {
+internal fun Project.createProtoExtensions() {
     fun findOrCreateAndConfigure(languageSourceSetName: String, languageSourceSet: Any) {
         val container = project.findOrCreate(PROTO_SOURCE_SETS) {
             val container = objects.domainObjectContainer(
@@ -82,7 +77,7 @@ internal fun Project.configureProtoExtensions(
 
         val protoSourceSet = container.maybeCreate(languageSourceSetName) as DefaultProtoSourceSet
 
-        configure(languageSourceSet, protoSourceSet)
+        protoSourceSet.languageSourceSets.add(languageSourceSet)
     }
 
     project.withKotlinJvmExtension {
@@ -110,12 +105,6 @@ internal fun Project.configureProtoExtensions(
     }
 }
 
-internal fun Project.createProtoExtensions() {
-    configureProtoExtensions { languageSourceSet, sourceSet ->
-        sourceSet.initExtension(languageSourceSet)
-    }
-}
-
 private fun DefaultProtoSourceSet.initExtension(languageSourceSet: Any) {
     this.languageSourceSets.add(languageSourceSet)
 }