Skip analysing for specific compose annotation (#111)

Author: skydoves

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

@@ -154,6 +154,26 @@ internal object StabilityAnalyzer {
    * PSI-based analysis (fallback for K1 mode or when K2 fails).
    */
   private fun analyzePsi(function: KtNamedFunction): ComposableStabilityInfo {
+    // Skip analysis for @NonRestartableComposable / @NonSkippableComposable —
+    // these composables have no caching/comparison code, so stability is irrelevant.
+    val hasNonRestartable = function.hasAnnotation(
+      StabilityConstants.Strings.NON_RESTARTABLE_COMPOSABLE,
+    )
+    val hasNonSkippable = function.hasAnnotation(
+      StabilityConstants.Strings.NON_SKIPPABLE_COMPOSABLE,
+    )
+    if (hasNonRestartable || hasNonSkippable) {
+      return ComposableStabilityInfo(
+        name = function.name ?: StabilityConstants.Strings.UNKNOWN,
+        fqName = function.fqName?.asString() ?: StabilityConstants.Strings.UNKNOWN,
+        isRestartable = !hasNonRestartable,
+        isSkippable = false,
+        isReadonly = function.hasAnnotation(StabilityConstants.Strings.READ_ONLY_COMPOSABLE),
+        parameters = emptyList(),
+        receivers = emptyList(),
+      )
+    }
+
     val parameters = function.valueParameters.mapNotNull { param ->
       analyzeParameter(param)
     }

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

@@ -78,6 +78,7 @@ internal object StabilityConstants {
     const val UNKNOWN = "Unknown"
     const val COMPOSABLE = "Composable"
     const val NON_RESTARTABLE_COMPOSABLE = "NonRestartableComposable"
+    const val NON_SKIPPABLE_COMPOSABLE = "NonSkippableComposable"
     const val READ_ONLY_COMPOSABLE = "ReadOnlyComposable"
   }
 

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

@@ -113,6 +113,8 @@ public class StabilityLineMarkerProvider : LineMarkerProvider {
     val allStable = allParams.all { it.stability == ParameterStability.STABLE }
 
     val color = when {
+      // Non-restartable / non-skippable composables have no stability semantics — show gray
+      !analysis.isSkippable && allParams.isEmpty() -> Color(0x80, 0x80, 0x80)
       analysis.isSkippable && allStable -> Color(settings.stableGutterColorRGB)
       analysis.isSkippable -> Color(settings.stableGutterColorRGB)
       hasUnstable -> Color(settings.unstableGutterColorRGB)
@@ -149,6 +151,15 @@ public class StabilityLineMarkerProvider : LineMarkerProvider {
           isRuntimeOnly ->
             "🟡 Runtime Stability (skippability determined at runtime)"
 
+          // Non-restartable / non-skippable with no params → annotated to skip caching
+          !analysis.isRestartable && totalCount == 0 ->
+            "⏭️ Non-Restartable (@NonRestartableComposable)" +
+              "\nStability analysis is not applicable."
+
+          analysis.isRestartable && !analysis.isSkippable && totalCount == 0 ->
+            "⏭️ Non-Skippable (@NonSkippableComposable)" +
+              "\nStability analysis is not applicable."
+
           analysis.isRestartable ->
             "⚠️ Restartable (not skippable)"
 

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

@@ -74,6 +74,25 @@ internal object StabilityAnalyzerK2 {
     // Get function symbol
     val functionSymbol = function.symbol
 
+    // Skip analysis for @NonRestartableComposable / @NonSkippableComposable —
+    // these composables have no caching/comparison code, so stability is irrelevant.
+    val hasNonRestartable =
+      function.hasAnnotation(StabilityConstants.Strings.NON_RESTARTABLE_COMPOSABLE)
+    val hasNonSkippable =
+      function.hasAnnotation(StabilityConstants.Strings.NON_SKIPPABLE_COMPOSABLE)
+    if (hasNonRestartable || hasNonSkippable) {
+      return ComposableStabilityInfo(
+        name = function.name ?: StabilityConstants.Strings.UNKNOWN,
+        fqName = functionSymbol.callableId?.asSingleFqName()?.asString()
+          ?: StabilityConstants.Strings.UNKNOWN,
+        isRestartable = !hasNonRestartable,
+        isSkippable = false,
+        isReadonly = function.hasAnnotation(StabilityConstants.Strings.READ_ONLY_COMPOSABLE),
+        parameters = emptyList(),
+        receivers = emptyList(),
+      )
+    }
+
     // Get the module containing this composable function (usage site)
     val usageSiteModule = ProjectFileIndex.getInstance(function.project).getModuleForFile(
       function.containingKtFile.virtualFile,