Refactor Kotlin Multiplatform Plugin

Refactored the Kotlin Multiplatform Plugin and added plugin parameters in a generated static class. Changes include modifications and relocations of several utilities methods related to KMP attributes and conditions. Environment variables have been introduced to control runtime behavior.

.github/workflows/publish-snapshot.yml

@@ -26,3 +26,4 @@ jobs:
           MAVEN_SPACE_USERNAME: ${{ secrets.MAVEN_SPACE_USERNAME }}
           RUN_NUMBER: ${{ github.run_number }}
           RUN_ATTEMPT: ${{ github.run_attempt }}
+          KMP: true

.run/Build Plugin.run.xml → .run/[KMP] Package Search Plugin.run.xml

@@ -1,6 +1,12 @@
 <component name="ProjectRunConfigurationManager">
-  <configuration default="false" name="Build Plugin" type="GradleRunConfiguration" factoryName="Gradle">
+  <configuration default="false" name="[KMP] Package Search Plugin" type="GradleRunConfiguration" factoryName="Gradle">
+    <log_file alias="idea.log" path="$PROJECT_DIR$/plugin/build/idea-sandbox/system/log/idea.log"/>
     <ExternalSystemSettings>
+      <option name="env">
+        <map>
+          <entry key="KMP" value="true"/>
+        </map>
+      </option>
       <option name="executionName" />
       <option name="externalProjectPath" value="$PROJECT_DIR$" />
       <option name="externalSystemIdString" value="GRADLE" />
@@ -10,7 +16,7 @@
       </option>
       <option name="taskNames">
         <list>
-          <option value=":plugin:buildShadowPlugin" />
+          <option value=":plugin:runIde"/>
         </list>
       </option>
       <option name="vmOptions" />

buildSrc/src/main/kotlin/org/jetbrains/packagesearch/gradle/GeneratePackageSearchObject.kt

@@ -6,14 +6,14 @@ import com.squareup.kotlinpoet.KModifier
 import com.squareup.kotlinpoet.PropertySpec
 import com.squareup.kotlinpoet.TypeSpec
 import javax.inject.Inject
+import kotlin.math.max
 import org.gradle.api.DefaultTask
 import org.gradle.api.file.DirectoryProperty
 import org.gradle.api.model.ObjectFactory
 import org.gradle.api.tasks.Input
 import org.gradle.api.tasks.OutputDirectory
 import org.gradle.api.tasks.TaskAction
 import org.gradle.kotlin.dsl.property
-import org.gradle.kotlin.dsl.the
 
 open class GeneratePackageSearchObject @Inject constructor(objects: ObjectFactory) : DefaultTask() {
 
@@ -22,11 +22,21 @@ open class GeneratePackageSearchObject @Inject constructor(objects: ObjectFactor
 
     @get:Input
     val pluginVersion = objects.property<String>()
-        .convention(project.version.toString())
+        .convention(project.provider {
+            val runNumber = System.getenv("RUN_NUMBER")?.toInt() ?: 0
+            val runAttempt = System.getenv("RUN_ATTEMPT")?.toInt() ?: 0
+            val snapshotMinorVersion = max(0, runNumber + runAttempt - 1)
+            val versionString = project.version.toString()
+            versionString.replace("-SNAPSHOT", ".$snapshotMinorVersion")
+        })
 
     @get:Input
     val deleteCachesOnStartup = objects.property<Boolean>()
-        .convention(System.getenv("CI") != "true")
+        .convention(project.provider { System.getenv("CI") != "true" })
+
+    @get:Input
+    val KMPEnabled = objects.property<Boolean>()
+        .convention(project.provider { System.getenv("KMP") == "true" })
 
     @get:Input
     val packageName = objects.property<String>()
@@ -71,6 +81,15 @@ open class GeneratePackageSearchObject @Inject constructor(objects: ObjectFactor
                             )
                             .build()
                     )
+                    .addProperty(
+                        PropertySpec.builder("isKMPEnabled", Boolean::class)
+                            .getter(
+                                FunSpec.getterBuilder()
+                                    .addStatement("return ${KMPEnabled.get()}")
+                                    .build()
+                            )
+                            .build()
+                    )
                     .build()
             )
             .build()

plugin/core/build.gradle.kts

@@ -43,19 +43,14 @@ kotlin.sourceSets.main {
 
 val pkgsPluginId: String by project
 
-val runNumber = System.getenv("RUN_NUMBER")?.toInt() ?: 0
-val runAttempt = System.getenv("RUN_ATTEMPT")?.toInt() ?: 0
-val snapshotMinorVersion = max(0, runNumber + runAttempt - 1)
-val versionString = project.version.toString()
-
 tasks {
     withType<Test> {
         environment("DB_PATH", layout.buildDirectory.file("tests/cache.db").get().asFile.absolutePath)
     }
     val generatePluginDataSources by registering(GeneratePackageSearchObject::class) {
+        group = "generate"
         pluginId = pkgsPluginId
         outputDir = generatedDir
-        pluginVersion = versionString.replace("-SNAPSHOT", ".$snapshotMinorVersion")
         packageName = "com.jetbrains.packagesearch.plugin.core"
     }
     sourcesJar {

plugin/core/src/main/kotlin/com/jetbrains/packagesearch/plugin/core/utils/KMPAttributes.kt

@@ -0,0 +1,41 @@
+package com.jetbrains.packagesearch.plugin.core.utils
+
+import com.jetbrains.packagesearch.plugin.core.data.PackageSearchModuleVariant
+
+
+data class KmpAttributesGroups(
+    val displayName: String,
+    val aggregationKeyword: String = displayName,
+)
+
+val KMP_ATTRIBUTES_GROUPS = listOf(
+    KmpAttributesGroups("iOS"),
+    KmpAttributesGroups("macOS"),
+    KmpAttributesGroups("tvOS"),
+    KmpAttributesGroups("watchOS"),
+    KmpAttributesGroups("JS"),
+    KmpAttributesGroups("JVM"),
+    KmpAttributesGroups("Linux"),
+    KmpAttributesGroups("Android"),
+    KmpAttributesGroups("WASM"),
+    KmpAttributesGroups("Windows", "mingw"),
+)
+
+fun Set<String>.parseAttributesFromRawStrings() = buildList {
+    var queue = this@parseAttributesFromRawStrings.toList()
+    for (attributeTitle in KMP_ATTRIBUTES_GROUPS) {
+        val (targets, rest) = queue
+            .partition { it.contains(attributeTitle.aggregationKeyword, true) }
+
+        if (targets.isEmpty()) continue
+        this.add(
+            PackageSearchModuleVariant.Attribute.NestedAttribute(
+                attributeTitle.displayName,
+                targets.map { PackageSearchModuleVariant.Attribute.StringAttribute(it) })
+        )
+        queue = rest
+    }
+    addAll(queue.map { PackageSearchModuleVariant.Attribute.StringAttribute(it) })
+
+}
+

plugin/gradle/base/src/main/kotlin/com/jetbrains/packagesearch/plugin/gradle/GradleModuleProvider.kt

@@ -3,6 +3,7 @@
 package com.jetbrains.packagesearch.plugin.gradle
 
 import com.intellij.openapi.module.Module
+import com.jetbrains.packagesearch.plugin.core.PackageSearch
 import com.jetbrains.packagesearch.plugin.core.data.PackageSearchModule
 import com.jetbrains.packagesearch.plugin.core.extensions.PackageSearchModuleBuilderContext
 import com.jetbrains.packagesearch.plugin.gradle.utils.getDeclaredDependencies
@@ -20,7 +21,7 @@ class GradleModuleProvider : AbstractGradleModuleProvider() {
         module: Module,
         model: PackageSearchGradleModel,
     ) {
-        if (!model.isKotlinMultiplatformApplied) {
+        if (!PackageSearch.isKMPEnabled || !model.isKotlinMultiplatformApplied) {
             val availableKnownRepositories =
                 model.repositories.toSet().let { availableGradleRepositories ->
                     knownRepositories.filterValues {

plugin/gradle/kmp/src/main/kotlin/com/jetbrains/packagesearch/plugin/gradle/KMPUtils.kt

@@ -5,6 +5,7 @@ import com.intellij.packageSearch.mppDependencyUpdater.resolved.MppCompilationIn
 import com.jetbrains.packagesearch.plugin.core.data.EditModuleContext
 import com.jetbrains.packagesearch.plugin.core.data.PackageSearchDeclaredPackage
 import com.jetbrains.packagesearch.plugin.core.data.PackageSearchModuleVariant
+import com.jetbrains.packagesearch.plugin.core.utils.parseAttributesFromRawStrings
 import kotlin.contracts.contract
 
 fun Set<MppCompilationInfoModel.Compilation>.buildAttributes(): List<PackageSearchModuleVariant.Attribute> {
@@ -37,25 +38,6 @@ fun List<PackageSearchModuleVariant.Attribute>.mergeAttributes() =
         .toSet()
         .parseAttributesFromRawStrings()
 
-fun Set<String>.parseAttributesFromRawStrings() = buildList {
-    var queue = this@parseAttributesFromRawStrings.toList()
-    for (attributeTitle in KMP_ATTRIBUTES_GROUPS) {
-        val (targets, rest) = queue
-            .partition { it.contains(attributeTitle.aggregationKeyword, true) }
-
-        if (targets.isEmpty()) continue
-        this.add(
-            PackageSearchModuleVariant.Attribute.NestedAttribute(
-                attributeTitle.displayName,
-                targets.map { PackageSearchModuleVariant.Attribute.StringAttribute(it) })
-        )
-        queue = rest
-    }
-    addAll(queue.map { PackageSearchModuleVariant.Attribute.StringAttribute(it) })
-
-}
-
-
 operator fun Set<String>.contains(attribute: PackageSearchModuleVariant.Attribute.NestedAttribute): Boolean =
     attribute.flatten().all { it in this }
 

plugin/gradle/kmp/src/main/kotlin/com/jetbrains/packagesearch/plugin/gradle/KotlinMultiplatformModuleProvider.kt

@@ -12,6 +12,7 @@ import com.intellij.packageSearch.mppDependencyUpdater.resolved.MppCompilationIn
 import com.intellij.packageSearch.mppDependencyUpdater.resolved.MppCompilationInfoModel.Jvm
 import com.intellij.packageSearch.mppDependencyUpdater.resolved.MppCompilationInfoModel.Native
 import com.intellij.packageSearch.mppDependencyUpdater.resolved.MppCompilationInfoProvider
+import com.jetbrains.packagesearch.plugin.core.PackageSearch
 import com.jetbrains.packagesearch.plugin.core.data.IconProvider.Icons
 import com.jetbrains.packagesearch.plugin.core.data.PackageSearchModule
 import com.jetbrains.packagesearch.plugin.core.extensions.PackageSearchModuleBuilderContext
@@ -34,7 +35,7 @@ class KotlinMultiplatformModuleProvider : AbstractGradleModuleProvider() {
         module: Module,
         model: PackageSearchGradleModel,
     ) {
-        if (model.isKotlinMultiplatformApplied && !model.isAmperApplied)
+        if (PackageSearch.isKMPEnabled && model.isKotlinMultiplatformApplied && !model.isAmperApplied)
             MppCompilationInfoProvider.sourceSetsMap(project, model.projectDir)
                 .collect { compilationModel ->
                     val variants = module.getKMPVariants(

plugin/gradle/src/main/kotlin/com/jetbrains/packagesearch/plugin/gradle/utils/GradleUtils.kt

@@ -26,6 +26,7 @@ import kotlinx.coroutines.flow.flatMapConcat
 import kotlinx.coroutines.flow.flow
 import kotlinx.coroutines.flow.map
 import kotlinx.coroutines.flow.merge
+import org.jetbrains.packagesearch.api.v3.ApiMavenPackage
 import org.jetbrains.packagesearch.api.v3.ApiPackage
 import org.jetbrains.packagesearch.api.v3.ApiRepository
 import org.jetbrains.packagesearch.packageversionutils.normalization.NormalizedVersion
@@ -98,11 +99,12 @@ suspend fun Module.getDeclaredDependencies(): List<PackageSearchGradleDeclaredPa
     val remoteInfo = getPackageInfoByIdHashes(distinctIds.map { ApiPackage.hashPackageId(it) }.toSet())
 
     return declaredDependencies
-        .map { declaredDependency ->
+        .mapNotNull { declaredDependency ->
             PackageSearchGradleDeclaredPackage(
                 id = declaredDependency.packageId,
                 declaredVersion = declaredDependency.version?.let { NormalizedVersion.fromStringOrNull(it) },
-                remoteInfo = remoteInfo[declaredDependency.packageId]?.asMavenApiPackage(),
+                remoteInfo = remoteInfo[declaredDependency.packageId] as? ApiMavenPackage
+                    ?: return@mapNotNull null,
                 icon = remoteInfo[declaredDependency.packageId]?.icon
                     ?: IconProvider.Icons.MAVEN,
                 module = declaredDependency.groupId,

plugin/src/main/kotlin/com/jetbrains/packagesearch/plugin/ui/model/infopanel/asPanelContent.kt

@@ -6,7 +6,7 @@ import com.jetbrains.packagesearch.plugin.core.data.PackageSearchDeclaredPackage
 import com.jetbrains.packagesearch.plugin.core.data.listKMPAttributesNames
 import com.jetbrains.packagesearch.plugin.core.extensions.PackageSearchKnownRepositoriesContext
 import com.jetbrains.packagesearch.plugin.core.utils.icon
-import com.jetbrains.packagesearch.plugin.gradle.parseAttributesFromRawStrings
+import com.jetbrains.packagesearch.plugin.core.utils.parseAttributesFromRawStrings
 import com.jetbrains.packagesearch.plugin.ui.model.getLatestVersion
 import org.jetbrains.packagesearch.api.v3.ApiMavenPackage
 import org.jetbrains.packagesearch.api.v3.ApiPackage