From 3cb6d77d8fd32e2e4af6001fc272b3fb1eb2b18a Mon Sep 17 00:00:00 2001 From: kirillk Date: Wed, 27 May 2026 16:18:33 -0400 Subject: [PATCH] fix(jetbrains): address session view review feedback --- .../client/session/views/base/AbstractSessionPartView.kt | 2 +- .../ai/kilocode/client/session/views/base/GenericView.kt | 9 ++++++--- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/packages/kilo-jetbrains/frontend/src/main/kotlin/ai/kilocode/client/session/views/base/AbstractSessionPartView.kt b/packages/kilo-jetbrains/frontend/src/main/kotlin/ai/kilocode/client/session/views/base/AbstractSessionPartView.kt index 86dfe55498c..4141eef1f76 100644 --- a/packages/kilo-jetbrains/frontend/src/main/kotlin/ai/kilocode/client/session/views/base/AbstractSessionPartView.kt +++ b/packages/kilo-jetbrains/frontend/src/main/kotlin/ai/kilocode/client/session/views/base/AbstractSessionPartView.kt @@ -23,7 +23,7 @@ abstract class AbstractSessionPartView( protected val arrow = JBLabel() protected val row = JPanel(BorderLayout(JBUI.scale(SessionUiStyle.View.CARD_LAYOUT_GAP), 0)) - private val bound = mutableListOf() + private val bound = linkedSetOf() private val click = object : MouseAdapter() { override fun mouseClicked(e: MouseEvent) { diff --git a/packages/kilo-jetbrains/frontend/src/main/kotlin/ai/kilocode/client/session/views/base/GenericView.kt b/packages/kilo-jetbrains/frontend/src/main/kotlin/ai/kilocode/client/session/views/base/GenericView.kt index 6b0c0343f30..c0973f9d166 100644 --- a/packages/kilo-jetbrains/frontend/src/main/kotlin/ai/kilocode/client/session/views/base/GenericView.kt +++ b/packages/kilo-jetbrains/frontend/src/main/kotlin/ai/kilocode/client/session/views/base/GenericView.kt @@ -13,12 +13,15 @@ import com.intellij.ui.components.JBLabel * confusing empty gaps), this shows a dim label with the raw type name. * This makes it easy to spot new part types that need a proper renderer. */ -class GenericView(content: Generic) : SecondarySessionPartView(JBLabel("[${content.type}]"), JBLabel()) { +class GenericView private constructor( + content: Generic, + private val label: JBLabel, +) : SecondarySessionPartView(label, JBLabel()) { + + constructor(content: Generic) : this(content, JBLabel("[${content.type}]")) override val contentId: String = content.id - private val label = row.getComponent(0) as JBLabel - init { label.foreground = UiStyle.Colors.weak() applyStyle(SessionEditorStyle.current())