Preven cell editor from being focussed when adding to table (#336)

Proton/Sources/Swift/Core/RichTextView.swift

@@ -36,6 +36,19 @@ class RichTextView: AutogrowingTextView {
 
     private var delegateOverrides = [GestureRecognizerDelegateOverride]()
 
+    private var _canBecomeFirstResponder = true
+    override var canBecomeFirstResponder: Bool {
+        return _canBecomeFirstResponder
+    }
+
+    func disableFirstResponder() {
+        _canBecomeFirstResponder = false
+    }
+
+    func enableFirstResponder() {
+        _canBecomeFirstResponder = true
+    }
+
     var preserveBlockAttachmentNewline: PreserveBlockAttachmentNewline = .none {
         didSet {
             richTextStorage.preserveNewlineBeforeBlock = false

Proton/Sources/Swift/Editor/EditorView.swift

@@ -584,22 +584,19 @@ open class EditorView: UIView {
         richTextView.text
     }
 
-    // Boolean flag to control first responder state
-    private var canBecomeFirstResponderFlag = true
-
     // Override canBecomeFirstResponder property
     open override var canBecomeFirstResponder: Bool {
-        return canBecomeFirstResponderFlag
+        return richTextView.canBecomeFirstResponder
     }
 
     // Method to disable becoming first responder
     func disableFirstResponder() {
-        canBecomeFirstResponderFlag = false
+        richTextView.disableFirstResponder()
     }
 
     // Method to enable becoming first responder
     func enableFirstResponder() {
-        canBecomeFirstResponderFlag = true
+        richTextView.enableFirstResponder()
     }
 
     public var selectedRange: NSRange {

Proton/Sources/Swift/Table/TableCell.swift

@@ -188,6 +188,12 @@ public class TableCell {
         contentView?.showEditor()
     }
 
+    func performWithoutChangingFirstResponder(_ closure: () -> Void) {
+        editor?.disableFirstResponder()
+        closure()
+        editor?.enableFirstResponder()
+    }
+
     func removeContentView() {
         attributedText = contentView?.editor.attributedText
         frame = contentView?.frame ?? frame

Proton/Sources/Swift/Table/TableContentView.swift

@@ -534,7 +534,9 @@ class TableContentView: UIScrollView {
 
 extension TableContentView: TableCellDelegate {
     func cell(_ cell: TableCell, didAddContentView view: TableCellContentView) {
-        addSubview(view)
+        cell.performWithoutChangingFirstResponder { [weak self] in
+            self?.addSubview(view)
+        }
         tableContentViewDelegate?.tableContentView(self, didAddCellToViewport: cell)
     }