From 37f6a0b4c81b56b67a3a19668575742ef3cc78ff Mon Sep 17 00:00:00 2001
From: Shadowfacts <me@shadowfacts.net>
Date: Sat, 9 Feb 2019 21:12:31 -0500
Subject: [PATCH] Add custom emojis

Closes #6
---
 Pachyderm/Model/Emoji.swift           | 11 +++--
 Tusker/Caching/ImageCache.swift       |  1 +
 Tusker/Views/ContentLabel.swift       | 62 ++++++++++++++++++++++++++-
 Tusker/Views/StatusContentLabel.swift |  1 +
 4 files changed, 68 insertions(+), 7 deletions(-)

diff --git a/Pachyderm/Model/Emoji.swift b/Pachyderm/Model/Emoji.swift
index 2ea10b60..bec0ea9d 100644
--- a/Pachyderm/Model/Emoji.swift
+++ b/Pachyderm/Model/Emoji.swift
@@ -9,17 +9,16 @@
 import Foundation
 
 public class Emoji: Decodable {
-    let shortcode: String
-    let url: URL
-    let staticURL: URL
-    // TODO: missing in pleroma
-//    let visibleInPicker: Bool
+    public let shortcode: String
+    public let url: URL
+    public let staticURL: URL
+    public let visibleInPicker: Bool
     
     private enum CodingKeys: String, CodingKey {
         case shortcode
         case url
         case staticURL = "static_url"
-//        case visibleInPicker = "visible_in_picker"
+        case visibleInPicker = "visible_in_picker"
     }
 }
 
diff --git a/Tusker/Caching/ImageCache.swift b/Tusker/Caching/ImageCache.swift
index b86c8a52..c6b8a586 100644
--- a/Tusker/Caching/ImageCache.swift
+++ b/Tusker/Caching/ImageCache.swift
@@ -14,6 +14,7 @@ class ImageCache {
     static let avatars = ImageCache(name: "Avatars", memoryExpiry: .seconds(60 * 60), diskExpiry: .seconds(60 * 60 * 24))
     static let headers = ImageCache(name: "Headers", memoryExpiry: .seconds(60 * 60), diskExpiry: .seconds(60 * 60 * 24))
     static let attachments = ImageCache(name: "Attachments", memoryExpiry: .seconds(60 * 2))
+    static let emojis = ImageCache(name: "Emojis", memoryExpiry: .seconds(60 * 5), diskExpiry: .seconds(60 * 60))
 
     let cache: Cache<Data>
     
diff --git a/Tusker/Views/ContentLabel.swift b/Tusker/Views/ContentLabel.swift
index 9ec06335..75aac6e6 100644
--- a/Tusker/Views/ContentLabel.swift
+++ b/Tusker/Views/ContentLabel.swift
@@ -8,14 +8,74 @@
 
 import UIKit
 import SafariServices
-import TTTAttributedLabel
 import Pachyderm
 import SwiftSoup
 
 class ContentLabel: LinkLabel {
     
+    private static let emojiRegex = try! NSRegularExpression(pattern: ":(\\w+):", options: [])
+    
     var navigationDelegate: TuskerNavigationDelegate?
     
+    // MARK: - Emojis
+    func setEmojis(_ emojis: [Emoji]) {
+        guard !emojis.isEmpty else { return }
+        
+        let group = DispatchGroup()
+
+        let mutAttrString = NSMutableAttributedString(attributedString: self.attributedText!)
+        let string = mutAttrString.string
+        let matches = ContentLabel.emojiRegex.matches(in: string, options: [], range: NSRange(location: 0, length: mutAttrString.length))
+        for match in matches.reversed() {
+            let shortcode = (string as NSString).substring(with: match.range(at: 1))
+            guard let emoji = emojis.first(where: { $0.shortcode == shortcode }) else {
+                continue
+            }
+
+            group.enter()
+            ImageCache.emojis.get(emoji.url) { (data) in
+                guard let data = data, let image = UIImage(data: data) else {
+                    group.leave()
+                    return
+                }
+                DispatchQueue.main.async {
+                    let attachment = self.createEmojiTextAttachment(image: image, index: match.range.location)
+                    mutAttrString.replaceCharacters(in: match.range, with: NSAttributedString(attachment: attachment))
+                    
+                    group.leave()
+                }
+            }
+        }
+
+        group.notify(queue: .main) {
+            self.attributedText = mutAttrString
+            self.setNeedsLayout()
+            self.setNeedsDisplay()
+        }
+    }
+    
+    // Based on https://github.com/ReticentJohn/Amaroq/blob/7c5b7088eb9fd1611dcb0f47d43bf8df093e142c/DireFloof/InlineImageHelpers.m
+    func createEmojiTextAttachment(image: UIImage, index: Int) -> NSTextAttachment {
+        let font = self.font!
+        
+        let adjustedCapHeight = font.capHeight - 1
+        var imageSizeMatchingFontSize = CGSize(width: image.size.width * (adjustedCapHeight / image.size.height), height: adjustedCapHeight)
+        
+        let defaultScale: CGFloat = 1.4
+        imageSizeMatchingFontSize = CGSize(width: imageSizeMatchingFontSize.width * defaultScale, height: imageSizeMatchingFontSize.height * defaultScale)
+        let textColor = self.textColor!
+        
+        UIGraphicsBeginImageContextWithOptions(imageSizeMatchingFontSize, false, 0.0)
+        textColor.set()
+        image.draw(in: CGRect(origin: .zero, size: imageSizeMatchingFontSize))
+        let attachmentImage = UIGraphicsGetImageFromCurrentImageContext()
+        UIGraphicsEndImageContext()
+        
+        let attachment = NSTextAttachment()
+        attachment.image = attachmentImage
+        return attachment
+    }
+    
     // MARK: - HTML Parsing
     func setTextFromHtml(_ html: String) {
         let doc = try! SwiftSoup.parse(html)
diff --git a/Tusker/Views/StatusContentLabel.swift b/Tusker/Views/StatusContentLabel.swift
index a32dfeb4..5c113e74 100644
--- a/Tusker/Views/StatusContentLabel.swift
+++ b/Tusker/Views/StatusContentLabel.swift
@@ -16,6 +16,7 @@ class StatusContentLabel: ContentLabel {
             guard let statusID = statusID else { return }
             guard let status = MastodonCache.status(for: statusID) else { fatalError("Can't set StatusContentLabel text without cached status \(statusID)") }
             setTextFromHtml(status.content)
+            setEmojis(status.emojis)
         }
     }