Store and restore drafts (#2898)

ElementX/Sources/Services/ComposerDraft/ComposerDraftService.swift

@@ -0,0 +1,74 @@
+//
+// Copyright 2024 New Vector Ltd
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+
+import Foundation
+
+import MatrixRustSDK
+
+final class ComposerDraftService: ComposerDraftServiceProtocol {
+    private let roomProxy: RoomProxyProtocol
+    private let timelineItemfactory: RoomTimelineItemFactoryProtocol
+    
+    init(roomProxy: RoomProxyProtocol, timelineItemfactory: RoomTimelineItemFactoryProtocol) {
+        self.roomProxy = roomProxy
+        self.timelineItemfactory = timelineItemfactory
+    }
+    
+    func saveDraft(_ draft: ComposerDraftProxy) async -> Result<Void, ComposerDraftServiceError> {
+        switch await roomProxy.saveDraft(draft.toRust) {
+        case .success:
+            MXLog.info("Successfully saved draft")
+            return .success(())
+        case .failure(let error):
+            MXLog.info("Failed to save draft: \(error)")
+            return .failure(.failedToSaveDraft)
+        }
+    }
+    
+    func loadDraft() async -> Result<ComposerDraftProxy?, ComposerDraftServiceError> {
+        switch await roomProxy.loadDraft() {
+        case .success(let draft):
+            guard let draft else {
+                return .success(nil)
+            }
+            return .success(ComposerDraftProxy(from: draft))
+        case .failure(let error):
+            MXLog.info("Failed to load draft: \(error)")
+            return .failure(.failedToLoadDraft)
+        }
+    }
+    
+    func getReply(eventID: String) async -> Result<TimelineItemReply, ComposerDraftServiceError> {
+        switch await roomProxy.timeline.getLoadedReplyDetails(eventID: eventID) {
+        case .success(let replyDetails):
+            return await .success(timelineItemfactory.buildReply(details: replyDetails))
+        case .failure(let error):
+            MXLog.error("Could not load reply: \(error)")
+            return .failure(.failedToLoadReply)
+        }
+    }
+    
+    func clearDraft() async -> Result<Void, ComposerDraftServiceError> {
+        switch await roomProxy.clearDraft() {
+        case .success:
+            MXLog.info("Successfully cleared draft")
+            return .success(())
+        case .failure(let error):
+            MXLog.info("Failed to clear draft: \(error)")
+            return .failure(.failedToClearDraft)
+        }
+    }
+}

ElementX/Sources/Services/ComposerDraft/ComposerDraftServiceProtocol.swift

@@ -0,0 +1,80 @@
+//
+// Copyright 2024 New Vector Ltd
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+
+import Foundation
+
+import MatrixRustSDK
+
+struct ComposerDraftProxy: Equatable {
+    enum ComposerDraftType: Equatable {
+        case newMessage
+        case reply(eventID: String)
+        case edit(eventID: String)
+        
+        var toRust: MatrixRustSDK.ComposerDraftType {
+            switch self {
+            case .newMessage:
+                return .newMessage
+            case .edit(let eventID):
+                return .edit(eventId: eventID)
+            case .reply(let eventID):
+                return .reply(eventId: eventID)
+            }
+        }
+        
+        init(from rustDraftType: MatrixRustSDK.ComposerDraftType) {
+            switch rustDraftType {
+            case .newMessage:
+                self = .newMessage
+            case .edit(let eventID):
+                self = .edit(eventID: eventID)
+            case .reply(let eventID):
+                self = .reply(eventID: eventID)
+            }
+        }
+    }
+    
+    let plainText: String
+    let htmlText: String?
+    let draftType: ComposerDraftType
+    
+    var toRust: ComposerDraft {
+        ComposerDraft(plainText: plainText, htmlText: htmlText, draftType: draftType.toRust)
+    }
+}
+
+extension ComposerDraftProxy {
+    init(from rustDraft: ComposerDraft) {
+        plainText = rustDraft.plainText
+        htmlText = rustDraft.htmlText
+        draftType = ComposerDraftType(from: rustDraft.draftType)
+    }
+}
+
+enum ComposerDraftServiceError: Error {
+    case failedToLoadDraft
+    case failedToLoadReply
+    case failedToSaveDraft
+    case failedToClearDraft
+}
+
+// sourcery: AutoMockable
+protocol ComposerDraftServiceProtocol {
+    func saveDraft(_ draft: ComposerDraftProxy) async -> Result<Void, ComposerDraftServiceError>
+    func loadDraft() async -> Result<ComposerDraftProxy?, ComposerDraftServiceError>
+    func clearDraft() async -> Result<Void, ComposerDraftServiceError>
+    func getReply(eventID: String) async -> Result<TimelineItemReply, ComposerDraftServiceError>
+}