Improve BubbleStateProvider (and this removes duplicated values!)

features/messages/src/main/kotlin/io/element/android/features/messages/timeline/components/BubbleStateProvider.kt

@@ -22,33 +22,18 @@ import io.element.android.features.messages.timeline.model.TimelineItemGroupPosi
 open class BubbleStateProvider : PreviewParameterProvider<BubbleState> {
     override val values: Sequence<BubbleState>
         get() = sequenceOf(
-            BubbleState(TimelineItemGroupPosition.First, isMine = false, isHighlighted = false),
+            TimelineItemGroupPosition.First,
-            BubbleState(TimelineItemGroupPosition.First, isMine = false, isHighlighted = true),
+            TimelineItemGroupPosition.Middle,
-            BubbleState(TimelineItemGroupPosition.First, isMine = false, isHighlighted = false),
+            TimelineItemGroupPosition.Last,
-            BubbleState(TimelineItemGroupPosition.First, isMine = false, isHighlighted = true),
+        ).map { groupPosition ->
-            BubbleState(TimelineItemGroupPosition.First, isMine = true, isHighlighted = false),
+            sequenceOf(false, true).map { isMine ->
-            BubbleState(TimelineItemGroupPosition.First, isMine = true, isHighlighted = true),
+                sequenceOf(false, true).map { isHighlighted ->
-            BubbleState(TimelineItemGroupPosition.First, isMine = true, isHighlighted = false),
+                    BubbleState(groupPosition, isMine = isMine, isHighlighted = isHighlighted)
-            BubbleState(TimelineItemGroupPosition.First, isMine = true, isHighlighted = true),
+                }
-
+            }
-            BubbleState(TimelineItemGroupPosition.Middle, isMine = false, isHighlighted = false),
+                .flatten()
-            BubbleState(TimelineItemGroupPosition.Middle, isMine = false, isHighlighted = true),
+        }
-            BubbleState(TimelineItemGroupPosition.Middle, isMine = false, isHighlighted = false),
+            .flatten()
-            BubbleState(TimelineItemGroupPosition.Middle, isMine = false, isHighlighted = true),
-            BubbleState(TimelineItemGroupPosition.Middle, isMine = true, isHighlighted = false),
-            BubbleState(TimelineItemGroupPosition.Middle, isMine = true, isHighlighted = true),
-            BubbleState(TimelineItemGroupPosition.Middle, isMine = true, isHighlighted = false),
-            BubbleState(TimelineItemGroupPosition.Middle, isMine = true, isHighlighted = true),
-
-            BubbleState(TimelineItemGroupPosition.Last, isMine = false, isHighlighted = false),
-            BubbleState(TimelineItemGroupPosition.Last, isMine = false, isHighlighted = true),
-            BubbleState(TimelineItemGroupPosition.Last, isMine = false, isHighlighted = false),
-            BubbleState(TimelineItemGroupPosition.Last, isMine = false, isHighlighted = true),
-            BubbleState(TimelineItemGroupPosition.Last, isMine = true, isHighlighted = false),
-            BubbleState(TimelineItemGroupPosition.Last, isMine = true, isHighlighted = true),
-            BubbleState(TimelineItemGroupPosition.Last, isMine = true, isHighlighted = false),
-            BubbleState(TimelineItemGroupPosition.Last, isMine = true, isHighlighted = true),
-        )
 }
 
 fun aBubbleState() = BubbleState(