From 011630a185285862b92e2f2da7959b3d55ade68c Mon Sep 17 00:00:00 2001
From: Danila Fedorin <danila.fedorin@gmail.com>
Date: Fri, 15 Mar 2019 18:45:55 -0700
Subject: [PATCH] Remove messages once their ID is received.

---
 src/Main.elm        | 17 +++++++++++++++--
 src/Scylla/Sync.elm |  3 +++
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/src/Main.elm b/src/Main.elm
index 098e9dc..bf22009 100644
--- a/src/Main.elm
+++ b/src/Main.elm
@@ -87,7 +87,7 @@ update msg model = case msg of
     ReceiveUserData s r -> updateUserData model s r
     ChangeRoomText r t -> updateChangeRoomText model r t
     SendRoomText r -> updateSendRoomText model r
-    SendRoomTextResponse t r -> ({ model | sending = Dict.remove t model.sending }, Cmd.none)
+    SendRoomTextResponse t r -> updateSendRoomTextResponse model t r
     ReceiveCompletedReadMarker r -> (model, Cmd.none)
     ReceiveCompletedTypingIndicator r -> (model, Cmd.none)
     ReceiveStoreData d -> updateStoreData model d
@@ -109,6 +109,17 @@ update msg model = case msg of
 requestScrollCmd : Cmd Msg
 requestScrollCmd = Task.attempt ViewportAfterMessage (Browser.Dom.getViewportOf "messages-wrapper")
 
+updateSendRoomTextResponse : Model -> Int -> Result Http.Error String -> (Model, Cmd Msg)
+updateSendRoomTextResponse m t r =
+    let
+        updateFunction newId msg = case msg of
+            Just (rid, { body, id }) -> Just (rid, { body = body, id = Just newId })
+            Nothing -> Nothing
+    in
+        case r of
+            Ok s -> ({ m | sending = Dict.update t (updateFunction s) m.sending }, Cmd.none)
+            Err e -> ({ m | sending = Dict.remove t m.sending }, Cmd.none) 
+
 updateDismissError : Model -> Int -> (Model, Cmd Msg)
 updateDismissError m i = ({ m | errors = (List.take i m.errors) ++ (List.drop (i+1) m.errors)}, Cmd.none)
 
@@ -328,9 +339,11 @@ updateSyncResponse model r notify =
         setReadReceiptCmd sr = case (room, List.head <| List.reverse <| roomMessages sr) of
             (Just rid, Just re) -> setReadMarkers model.apiUrl token rid re.eventId <| Just re.eventId
             _ -> Cmd.none
+        receivedEvents sr = List.map Just <| allTimelineEventIds sr
+        sending sr = Dict.filter (\_ (rid, { body, id }) -> not <| List.member id <| receivedEvents sr) model.sending
     in
         case r of
-            Ok sr -> ({ model | sync = mergeSyncResponse model.sync sr }, Cmd.batch
+            Ok sr -> ({ model | sync = mergeSyncResponse model.sync sr, sending = sending (mergeSyncResponse model.sync sr) }, Cmd.batch
                 [ syncCmd
                 , newUserCmd sr
                 , notificationCmd sr
diff --git a/src/Scylla/Sync.elm b/src/Scylla/Sync.elm
index 40a412b..e084dbc 100644
--- a/src/Scylla/Sync.elm
+++ b/src/Scylla/Sync.elm
@@ -476,6 +476,9 @@ allRoomDictTimelineEvents dict = List.concatMap (Maybe.withDefault [] << .events
     <| List.filterMap .timeline
     <| Dict.values dict
 
+allTimelineEventIds : SyncResponse -> List String
+allTimelineEventIds s = List.map .eventId <| allTimelineEvents s
+
 allTimelineEvents : SyncResponse -> List RoomEvent
 allTimelineEvents s =
     let