#260 TreeElementSelection appearance for last and first element of se…

…lection
JetBrains · Dec 1, 2023 · bf31f5d · bf31f5d
1 parent 079b0b5
commit bf31f5d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/foundation/src/main/kotlin/org/jetbrains/jewel/foundation/lazy/tree/BasicLazyTree.kt b/foundation/src/main/kotlin/org/jetbrains/jewel/foundation/lazy/tree/BasicLazyTree.kt
@@ -152,10 +152,10 @@ public fun <T> BasicLazyTree(
             val backgroundShape by derivedStateOf {
                 val hasRoundedTopCorners = flattenedTree.getOrNull(index - 1)?.id?.let {
                     it !in treeState.delegate.selectedKeys
-                } ?: false
+                } ?: true
                 val hasRoundedBottomCorners = flattenedTree.getOrNull(index + 1)?.id?.let {
                     it !in treeState.delegate.selectedKeys
-                } ?: false
+                } ?: true
                 val topCornerSize = computerCornerSize(hasRoundedTopCorners, elementBackgroundCornerSize)
                 val bottomCornerSize = computerCornerSize(hasRoundedBottomCorners, elementBackgroundCornerSize)
                 RoundedCornerShape(