diff --git a/app/src/main/java/com/vitorpamplona/amethyst/ui/note/NoteCompose.kt b/app/src/main/java/com/vitorpamplona/amethyst/ui/note/NoteCompose.kt index e0950d17b..727687e6e 100644 --- a/app/src/main/java/com/vitorpamplona/amethyst/ui/note/NoteCompose.kt +++ b/app/src/main/java/com/vitorpamplona/amethyst/ui/note/NoteCompose.kt @@ -687,6 +687,7 @@ fun NoteBody( if (showSecondRow) { SecondUserInfoRow( baseNote, + editState, accountViewModel, nav, ) @@ -1035,6 +1036,7 @@ private fun ReplyNoteComposition( @Composable fun SecondUserInfoRow( note: Note, + editState: State>, accountViewModel: AccountViewModel, nav: (String) -> Unit, ) { @@ -1069,7 +1071,22 @@ fun SecondUserInfoRow( DisplayPoW(pow) } - DisplayOts(note, accountViewModel) + DisplayOtsIfInOriginal(note, editState, accountViewModel) + } +} + +@Composable +fun DisplayOtsIfInOriginal( + note: Note, + editState: State>, + accountViewModel: AccountViewModel, +) { + val editState = (editState.value as? GenericLoadable.Loaded)?.loaded?.modificationToShow?.value + + if (editState == null) { + DisplayOts(note = note, accountViewModel = accountViewModel) + } else { + DisplayOts(note = editState, accountViewModel = accountViewModel) } } diff --git a/app/src/main/java/com/vitorpamplona/amethyst/ui/screen/ThreadFeedView.kt b/app/src/main/java/com/vitorpamplona/amethyst/ui/screen/ThreadFeedView.kt index 052ae3802..e347f3e45 100644 --- a/app/src/main/java/com/vitorpamplona/amethyst/ui/screen/ThreadFeedView.kt +++ b/app/src/main/java/com/vitorpamplona/amethyst/ui/screen/ThreadFeedView.kt @@ -92,6 +92,7 @@ import com.vitorpamplona.amethyst.ui.components.ObserveDisplayNip05Status import com.vitorpamplona.amethyst.ui.components.mockAccountViewModel import com.vitorpamplona.amethyst.ui.navigation.routeToMessage import com.vitorpamplona.amethyst.ui.note.BlankNote +import com.vitorpamplona.amethyst.ui.note.DisplayOtsIfInOriginal import com.vitorpamplona.amethyst.ui.note.HiddenNote import com.vitorpamplona.amethyst.ui.note.LoadAddressableNote import com.vitorpamplona.amethyst.ui.note.NoteAuthorPicture @@ -105,7 +106,6 @@ import com.vitorpamplona.amethyst.ui.note.elements.DisplayEditStatus import com.vitorpamplona.amethyst.ui.note.elements.DisplayFollowingCommunityInPost import com.vitorpamplona.amethyst.ui.note.elements.DisplayFollowingHashtagsInPost import com.vitorpamplona.amethyst.ui.note.elements.DisplayLocation -import com.vitorpamplona.amethyst.ui.note.elements.DisplayOts import com.vitorpamplona.amethyst.ui.note.elements.DisplayPoW import com.vitorpamplona.amethyst.ui.note.elements.DisplayReward import com.vitorpamplona.amethyst.ui.note.elements.DisplayZapSplits @@ -468,7 +468,7 @@ fun NoteMaster( DisplayPoW(pow) } - DisplayOts(note, accountViewModel) + DisplayOtsIfInOriginal(note, editState, accountViewModel) } } }