Implement the heatmap window (#121)

Author: skydoves

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

@@ -32,23 +32,10 @@ import com.intellij.openapi.ui.SimpleToolWindowPanel
 import com.intellij.openapi.vfs.LocalFileSystem
 import com.intellij.psi.PsiDocumentManager
 import com.intellij.psi.PsiManager
-import com.intellij.ui.OnePixelSplitter
 import com.intellij.ui.ScrollPaneFactory
-import com.intellij.ui.components.JBLabel
-import com.intellij.ui.components.JBPanel
-import com.intellij.ui.components.JBScrollPane
 import com.intellij.ui.treeStructure.Tree
-import com.intellij.util.ui.JBUI
-import com.intellij.util.ui.UIUtil
-import com.skydoves.compose.stability.idea.settings.StabilitySettingsState
-import com.skydoves.compose.stability.runtime.ParameterStability
 import org.jetbrains.kotlin.psi.KtNamedFunction
-import java.awt.Color
-import java.awt.Font
-import java.awt.GridBagConstraints
-import java.awt.GridBagLayout
 import javax.swing.JComponent
-import javax.swing.JPanel
 import javax.swing.tree.DefaultMutableTreeNode
 import javax.swing.tree.DefaultTreeModel
 
@@ -58,13 +45,9 @@ import javax.swing.tree.DefaultTreeModel
  */
 internal class CascadePanel(private val project: Project) {
 
-  private val settings: StabilitySettingsState
-    get() = StabilitySettingsState.getInstance()
-
   private val tree: Tree
   private val treeModel: DefaultTreeModel
   private val rootNode: DefaultMutableTreeNode
-  private val detailsPanel: JPanel
   private var currentFunction: KtNamedFunction? = null
 
   init {
@@ -76,14 +59,6 @@ internal class CascadePanel(private val project: Project) {
     tree.isRootVisible = false
     tree.showsRootHandles = true
 
-    detailsPanel = createDetailsPanel()
-
-    // Selection listener for details
-    tree.addTreeSelectionListener { event ->
-      val node = event.path.lastPathComponent as? DefaultMutableTreeNode
-      updateDetailsPanel(node)
-    }
-
     // Double-click to navigate to source
     tree.addMouseListener(object : java.awt.event.MouseAdapter() {
       override fun mouseClicked(e: java.awt.event.MouseEvent) {
@@ -108,12 +83,7 @@ internal class CascadePanel(private val project: Project) {
     val toolbar = createToolbar()
     mainPanel.toolbar = toolbar
 
-    // Create split pane with tree and details
-    val splitter = OnePixelSplitter(false, 0.65f)
-    splitter.firstComponent = ScrollPaneFactory.createScrollPane(tree)
-    splitter.secondComponent = JBScrollPane(detailsPanel)
-
-    mainPanel.setContent(splitter)
+    mainPanel.setContent(ScrollPaneFactory.createScrollPane(tree))
 
     return mainPanel
   }
@@ -232,200 +202,6 @@ internal class CascadePanel(private val project: Project) {
     }
   }
 
-  private fun createDetailsPanel(): JPanel {
-    val panel = JBPanel<JBPanel<*>>(GridBagLayout())
-    panel.border = JBUI.Borders.empty(10)
-
-    val gbc = GridBagConstraints()
-    gbc.gridx = 0
-    gbc.gridy = 0
-    gbc.anchor = GridBagConstraints.NORTHWEST
-    gbc.fill = GridBagConstraints.HORIZONTAL
-    gbc.weightx = 1.0
-
-    val label = JBLabel("Select a composable to view details")
-    label.foreground = UIUtil.getInactiveTextColor()
-    panel.add(label, gbc)
-
-    return panel
-  }
-
-  private fun updateDetailsPanel(node: DefaultMutableTreeNode?) {
-    detailsPanel.removeAll()
-    detailsPanel.layout = GridBagLayout()
-
-    val gbc = GridBagConstraints()
-    gbc.gridx = 0
-    gbc.gridy = 0
-    gbc.anchor = GridBagConstraints.NORTHWEST
-    gbc.fill = GridBagConstraints.HORIZONTAL
-    gbc.weightx = 1.0
-    gbc.insets = JBUI.insets(5)
-
-    when (val data = node?.userObject) {
-      is CascadeTreeNodeData.Composable -> {
-        val info = data.node.stabilityInfo
-        val hasUnstable = info.parameters.any {
-          it.stability == ParameterStability.UNSTABLE
-        }
-        val hasRuntime = info.parameters.any {
-          it.stability == ParameterStability.RUNTIME
-        }
-        val isRuntimeOnly = !info.isSkippable && !hasUnstable && hasRuntime
-
-        // Title
-        addDetailRow(detailsPanel, gbc, "Composable:", info.name, bold = true)
-
-        // Stability status
-        val statusText = when {
-          info.isSkippable -> "STABLE (Skippable)"
-          isRuntimeOnly -> "RUNTIME (Determined at runtime)"
-          else -> "UNSTABLE (Non-skippable)"
-        }
-        val statusColor = when {
-          info.isSkippable -> Color(settings.stableGutterColorRGB)
-          isRuntimeOnly -> Color(settings.runtimeGutterColorRGB)
-          else -> Color(settings.unstableGutterColorRGB)
-        }
-        addDetailRow(detailsPanel, gbc, "Status:", statusText, color = statusColor)
-
-        // Restartable
-        addDetailRow(
-          detailsPanel,
-          gbc,
-          "Restartable:",
-          if (info.isRestartable) "Yes" else "No",
-        )
-
-        // Depth
-        addDetailRow(detailsPanel, gbc, "Depth:", "${data.node.depth}")
-
-        // Parameters
-        if (info.parameters.isNotEmpty()) {
-          gbc.gridy++
-          val headerLabel = JBLabel("Parameters:")
-          headerLabel.font = headerLabel.font.deriveFont(Font.BOLD)
-          detailsPanel.add(headerLabel, gbc)
-
-          info.parameters.forEach { param ->
-            gbc.gridy++
-            val paramText = "${param.name}: ${param.type}"
-            val paramStatus = when (param.stability) {
-              ParameterStability.STABLE -> "Stable"
-              ParameterStability.RUNTIME -> "Runtime"
-              ParameterStability.UNSTABLE -> "Unstable"
-            }
-            val paramColor = when (param.stability) {
-              ParameterStability.STABLE -> Color(settings.stableHintColorRGB)
-              ParameterStability.RUNTIME -> Color(settings.runtimeHintColorRGB)
-              ParameterStability.UNSTABLE -> Color(settings.unstableHintColorRGB)
-            }
-            addDetailRow(detailsPanel, gbc, "  $paramText", paramStatus, color = paramColor)
-
-            // Show reason if available
-            if (param.reason != null) {
-              gbc.gridy++
-              val reasonLabel = JBLabel("    Reason: ${param.reason}")
-              reasonLabel.foreground = UIUtil.getInactiveTextColor()
-              detailsPanel.add(reasonLabel, gbc)
-            }
-          }
-        }
-
-        // Location
-        gbc.gridy++
-        val fileName = data.node.filePath.substringAfterLast('/')
-        addDetailRow(detailsPanel, gbc, "File:", "$fileName:${data.node.line}")
-
-        // Navigation hint
-        gbc.gridy++
-        val navigateLabel = JBLabel("Double-click to navigate to source")
-        navigateLabel.foreground = UIUtil.getInactiveTextColor()
-        detailsPanel.add(navigateLabel, gbc)
-      }
-
-      is CascadeTreeNodeData.Summary -> {
-        addDetailRow(
-          detailsPanel,
-          gbc,
-          "Root Composable:",
-          data.composableName,
-          bold = true,
-        )
-        addDetailRow(
-          detailsPanel,
-          gbc,
-          "Total Downstream:",
-          "${data.summary.totalCount}",
-        )
-        addDetailRow(
-          detailsPanel,
-          gbc,
-          "Skippable:",
-          "${data.summary.skippableCount}",
-          color = Color(settings.stableGutterColorRGB),
-        )
-        addDetailRow(
-          detailsPanel,
-          gbc,
-          "Unskippable:",
-          "${data.summary.unskippableCount}",
-          color = Color(settings.unstableGutterColorRGB),
-        )
-        addDetailRow(
-          detailsPanel,
-          gbc,
-          "Max Depth:",
-          "${data.summary.maxDepth}",
-        )
-        if (data.summary.hasTruncatedBranches) {
-          addDetailRow(
-            detailsPanel,
-            gbc,
-            "Note:",
-            "Some branches were truncated",
-            color = UIUtil.getInactiveTextColor(),
-          )
-        }
-      }
-
-      else -> {
-        val label = JBLabel("Select a composable to view details")
-        label.foreground = UIUtil.getInactiveTextColor()
-        detailsPanel.add(label, gbc)
-      }
-    }
-
-    // Filler at the bottom
-    gbc.gridy++
-    gbc.weighty = 1.0
-    detailsPanel.add(JPanel(), gbc)
-
-    detailsPanel.revalidate()
-    detailsPanel.repaint()
-  }
-
-  private fun addDetailRow(
-    panel: JPanel,
-    gbc: GridBagConstraints,
-    label: String,
-    value: String,
-    bold: Boolean = false,
-    color: Color? = null,
-  ) {
-    gbc.gridy++
-
-    val fullLabel = JBLabel("$label $value")
-    if (bold) {
-      fullLabel.font = fullLabel.font.deriveFont(Font.BOLD)
-    }
-    if (color != null) {
-      fullLabel.foreground = color
-    }
-
-    panel.add(fullLabel, gbc)
-  }
-
   /**
    * Refresh action - re-analyzes the current function.
    */
@@ -457,7 +233,6 @@ internal class CascadePanel(private val project: Project) {
     override fun actionPerformed(e: AnActionEvent) {
       currentFunction = null
       showEmptyState()
-      updateDetailsPanel(null)
     }
   }
 }

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

@@ -25,8 +25,11 @@ import com.intellij.openapi.editor.EditorCustomElementRenderer
 import com.intellij.openapi.editor.EditorFactory
 import com.intellij.openapi.editor.Inlay
 import com.intellij.openapi.editor.colors.EditorFontType
+import com.intellij.openapi.editor.event.EditorMouseEvent
+import com.intellij.openapi.editor.event.EditorMouseListener
 import com.intellij.openapi.editor.markup.TextAttributes
 import com.intellij.openapi.project.Project
+import com.intellij.openapi.wm.ToolWindowManager
 import com.intellij.psi.PsiDocumentManager
 import com.intellij.psi.util.PsiTreeUtil
 import com.intellij.util.ui.JBUI
@@ -74,7 +77,41 @@ internal class HeatmapInlayManager(
   @Volatile
   private var refreshTask: ScheduledFuture<*>? = null
 
+  @Volatile
+  private var clickListenerRegistered = false
+
+  /**
+   * Mouse listener that detects clicks on heatmap block inlays
+   * and opens the Heatmap tab to show recomposition logs.
+   */
+  private val clickListener = object : EditorMouseListener {
+    override fun mouseClicked(event: EditorMouseEvent) {
+      if (event.mouseEvent.clickCount != 1) return
+      val editor = event.editor
+      if (editor.project != project) return
+
+      val editorKey = System.identityHashCode(editor)
+      val entries = editorState[editorKey] ?: return
+
+      val point = event.mouseEvent.point
+      for ((name, entry) in entries) {
+        if (!entry.inlay.isValid) continue
+        val bounds = entry.inlay.bounds ?: continue
+        if (bounds.contains(point)) {
+          openHeatmapPanel(name)
+          return
+        }
+      }
+    }
+  }
+
   fun start() {
+    if (!clickListenerRegistered) {
+      clickListenerRegistered = true
+      EditorFactory.getInstance().eventMulticaster
+        .addEditorMouseListener(clickListener, this)
+    }
+
     refreshTask = com.intellij.util.concurrency.AppExecutorUtil
       .getAppScheduledExecutorService()
       .scheduleWithFixedDelay(
@@ -264,7 +301,7 @@ internal class HeatmapInlayManager(
       val editorFont = editor.colorsScheme.getFont(EditorFontType.PLAIN)
       val smallFont = editorFont.deriveFont(
         Font.PLAIN,
-        editor.colorsScheme.editorFontSize2D.toFloat(),
+        editor.colorsScheme.editorFontSize2D,
       )
       g.font = smallFont
       val fm = g.fontMetrics
@@ -279,6 +316,25 @@ internal class HeatmapInlayManager(
     }
   }
 
+  /**
+   * Opens the Heatmap tab in the tool window and displays
+   * recomposition data for the given composable.
+   */
+  private fun openHeatmapPanel(composableName: String) {
+    ApplicationManager.getApplication().invokeLater {
+      val toolWindow = ToolWindowManager.getInstance(project)
+        .getToolWindow("Compose Stability Analyzer") ?: return@invokeLater
+      toolWindow.show {
+        val heatmapContent = toolWindow.contentManager
+          .findContent("Heatmap") ?: return@show
+        toolWindow.contentManager.setSelectedContent(heatmapContent)
+        val panel = heatmapContent.component
+          .getClientProperty(HeatmapPanel::class.java) as? HeatmapPanel ?: return@show
+        panel.showComposableData(composableName)
+      }
+    }
+  }
+
   companion object {
     private const val REFRESH_INTERVAL_MS = 1000L
     fun getInstance(project: Project): HeatmapInlayManager =