Merge pull request #24 from skydoves/feature/StabilityInferred

Handle StabilityInferred annotated composables

Author: skydoves

compose-stability-analyzer-idea/CHANGELOG.md

@@ -7,12 +7,15 @@ All notable changes to the IntelliJ IDEA plugin will be documented in this file.
 ### Added
 - New setting: "Show in test source sets" for gutter icons (Issue #21)
 - Gutter icons are now hidden in test directories by default (can be enabled in settings)
+- Support for reading @StabilityInferred annotation parameters for cross-module stability detection (Issue #18)
 
 ### Fixed
 - Fixed typealias detection for Composable function types (Issue #16)
 - Typealiases like `typealias SettingsButtons = @Composable (PlayerUiState) -> Unit` now correctly expand to their underlying function types before stability analysis
 - Fixed ImmutableList/ImmutableSet/ImmutableMap showing as unstable in test code (Issue #21)
 - Added fallback type resolution by simple name for immutable collections when FQN resolution fails in test source sets
+- Improved cross-module stability detection by reading @StabilityInferred(parameters) annotation (Issue #18)
+- Classes from other modules now correctly marked as UNSTABLE unless annotated with @Stable/@Immutable or @StabilityInferred(parameters=0)
 
 ---
 

compose-stability-analyzer-idea/src/main/kotlin/com/skydoves/compose/stability/idea/k2/KtStabilityInferencer.kt

@@ -598,12 +598,44 @@ internal class KtStabilityInferencer {
   }
 
   /**
-   * TODO: Read @StabilityInferred parameters field using K2 Analysis API.
-   * Returns null (conservative RUNTIME) until reliable API is found.
+   * Reads the @StabilityInferred annotation's parameters field.
+   *
+   * @StabilityInferred is added by the Compose compiler to classes from other modules
+   * to indicate their stability:
+   * - parameters = 0: Class is stable
+   * - parameters > 0: Class needs runtime stability check
+   * - null: Annotation not present
+   *
+   * This is crucial for cross-module stability: classes from other modules should be
+   * UNSTABLE unless annotated with @Stable/@Immutable or @StabilityInferred(parameters=0).
    */
   context(KaSession)
   private fun KaClassSymbol.getStabilityInferredParameters(): Int? {
-    return null
+    val stabilityInferredFqName = "androidx.compose.runtime.internal.StabilityInferred"
+    val annotation = annotations.firstOrNull { annotation ->
+      annotation.classId?.asSingleFqName()?.asString() == stabilityInferredFqName
+    } ?: return null
+
+    val parametersArgument = annotation.arguments.firstOrNull { arg ->
+      arg.name.asString() == "parameters"
+    }
+
+    // Extract the Int value from the constant expression
+    return try {
+      when (val expression = parametersArgument?.expression) {
+        is org.jetbrains.kotlin.analysis.api.annotations.KaAnnotationValue.ConstantValue -> {
+          // Get the constant value as Int
+          (expression.value.value as? Int) ?: run {
+            null
+          }
+        }
+        else -> {
+          null
+        }
+      }
+    } catch (e: Exception) {
+      null
+    }
   }
 
   /**