From b1bdd20255c65ea8ebbe58d62f3a13acd91db196 Mon Sep 17 00:00:00 2001
From: Vitor Pamplona <vitor@vitorpamplona.com>
Date: Fri, 24 Feb 2023 10:08:35 -0500
Subject: [PATCH] Makes sure the note is always added in the threadView

---
 .../java/com/vitorpamplona/amethyst/model/ThreadAssembler.kt  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/app/src/main/java/com/vitorpamplona/amethyst/model/ThreadAssembler.kt b/app/src/main/java/com/vitorpamplona/amethyst/model/ThreadAssembler.kt
index 5b930415d..208fa2529 100644
--- a/app/src/main/java/com/vitorpamplona/amethyst/model/ThreadAssembler.kt
+++ b/app/src/main/java/com/vitorpamplona/amethyst/model/ThreadAssembler.kt
@@ -6,6 +6,8 @@ import kotlin.time.measureTimedValue
 class ThreadAssembler {
 
   fun searchRoot(note: Note, testedNotes: MutableSet<Note> = mutableSetOf()): Note? {
+    testedNotes.add(note)
+
     if (note.replyTo == null || note.replyTo?.isEmpty() == true) return note
 
     val markedAsRoot = note.event?.tags?.firstOrNull { it[0] == "e" && it.size > 3 && it[3] == "root" }?.getOrNull(1)
@@ -14,8 +16,6 @@ class ThreadAssembler {
     val hasNoReplyTo = note.replyTo?.firstOrNull { it.replyTo?.isEmpty() == true }
     if (hasNoReplyTo != null) return hasNoReplyTo
 
-    testedNotes.add(note)
-
     // recursive
     val roots = note.replyTo?.map {
       if (it !in testedNotes)