Blank Images problem fix when using Streamoji in TableViewCell

Streamoji/Classes/Extensions/UITextView+Emojis.swift

@@ -27,6 +27,21 @@ extension UITextView {
             self?.applyEmojis(emojis, rendering: rendering)
         }
     }
+
+    /// - Parameter emojis: A dictionary of emoji keyed by its shortcode.
+    /// - Parameter rendering: The rendering options. Defaults to `.highQuality`.
+    /// - Parameter delay: num of seconds to delay async operation `(example : 1.0,  0.1,  0.3, 2.0)`
+    public func configureEmojisWithDelay(_ emojis: [String: EmojiSource], rendering: EmojiRendering = .highQuality, delay : Double?) {
+        self.applyEmojisWithDelay(emojis, rendering: rendering, delay : delay)
+
+        NotificationCenter.default.addObserver(
+            forName: UITextView.textDidChangeNotification,
+            object: self,
+            queue: .main
+        ) { [weak self] _ in
+            self?.applyEmojisWithDelay(emojis, rendering: rendering, delay : delay)
+        }
+    }
 }
 
 // MARK: Private
@@ -47,6 +62,16 @@ extension UITextView {
         selectedRange = NSRange(location: range.location - (count - newCount), length: range.length)
     }
 
+    private func applyEmojisWithDelay(_ emojis: [String: EmojiSource], rendering: EmojiRendering, delay : Double?) {
+        let range = selectedRange
+        let count = attributedText?.string.count ?? 0
+        self.attributedText = attributedText.insertingEmojis(emojis, rendering: rendering)
+        let newCount = attributedText.string.count
+        customEmojiViews.forEach { $0.removeFromSuperview() }
+        addEmojiImagesIfNeededWithDelay(rendering: rendering, delay : delay)
+        selectedRange = NSRange(location: range.location - (count - newCount), length: range.length)
+    }
+
     private func addEmojiImagesIfNeeded(rendering: EmojiRendering) {
         attributedText.enumerateAttributes(in: NSRange(location: 0, length: attributedText.length), options: [], using: { attributes, crange, _ in
             DispatchQueue.main.async {
@@ -104,4 +129,63 @@ extension UITextView {
             }
         })
     }
+
+    private func addEmojiImagesIfNeededWithDelay(rendering: EmojiRendering, delay : Double?) {
+        attributedText.enumerateAttributes(in: NSRange(location: 0, length: attributedText.length), options: [], using: { attributes, crange, _ in
+
+            DispatchQueue.main.asyncAfter(deadline: .now() + (delay ?? 0.0), execute: {
+                guard
+                    let emojiAttachment = attributes[NSAttributedString.Key.attachment] as? NSTextAttachment,
+                    let position1 = self.position(from: self.beginningOfDocument, offset: crange.location),
+                    let position2 = self.position(from: position1, offset: crange.length),
+                    let range = self.textRange(from: position1, to: position2),
+                    let emojiData = emojiAttachment.contents,
+                    let emoji = try? JSONDecoder().decode(EmojiSource.self, from: emojiData)
+                else {
+                    return
+                }
+
+                let rect = self.firstRect(for: range)
+
+                let emojiView = EmojiView(frame: rect)
+                emojiView.backgroundColor = self.backgroundColor
+                emojiView.isUserInteractionEnabled = false
+
+                switch emoji {
+                case let .character(character):
+                    emojiView.label.text = character
+                case let .imageUrl(imageUrl):
+                    guard renderViews[emoji] == nil else {
+                        break
+                    }
+
+                    if let url = URL(string: imageUrl) {
+                        let renderView = UIImageView(frame: rect)
+                        renderView.setFromURL(url, rendering: rendering)
+                        renderViews[emoji] = renderView
+                        self.window?.addSubview(renderView)
+                        renderView.alpha = 0
+                    }
+                case let .imageAsset(imageAsset):
+                    guard renderViews[emoji] == nil else {
+                        break
+                    }
+
+                    let renderView = UIImageView(frame: rect)
+                    renderView.setFromAsset(imageAsset, rendering: rendering)
+                    renderViews[emoji] = renderView
+                    self.window?.addSubview(renderView)
+                    renderView.alpha = 0
+                case .alias:
+                    break
+                }
+
+                if let view = renderViews[emoji] {
+                    emojiView.setFromRenderView(view)
+                }
+
+                self.textContainerView.addSubview(emojiView)
+            })
+        })
+    }
 }