mastodon-ios/Mastodon/Service/EmojiService/EmojiService+CustomEmojiViewModel.swift

//
//  EmojiService+CustomEmojiViewModel.swift
//  Mastodon
//
//  Created by MainasuK Cirno on 2021-3-15.
//

import Foundation
import Combine
import GameplayKit
import MastodonSDK

extension EmojiService {
    final class CustomEmojiViewModel {
        
        var disposeBag = Set<AnyCancellable>()
        
        // input
        let domain: String
        weak var service: EmojiService?
        
        // output
        private(set) lazy var stateMachine: GKStateMachine = {
            // exclude timeline middle fetcher state
            let stateMachine = GKStateMachine(states: [
                LoadState.Initial(viewModel: self),
                LoadState.Loading(viewModel: self),
                LoadState.Fail(viewModel: self),
                LoadState.Finish(viewModel: self),
            ])
            stateMachine.enter(LoadState.Initial.self)
            return stateMachine
        }()
        let emojis = CurrentValueSubject<[Mastodon.Entity.Emoji], Never>([])
        let emojiDict = CurrentValueSubject<[String: [Mastodon.Entity.Emoji]], Never>([:])
        
        private var learnedEmoji: Set<String> = Set()
        
        init(domain: String, service: EmojiService) {
            self.domain = domain
            self.service = service
            
            emojis
                .map { Dictionary(grouping: $0, by: { $0.shortcode }) }
                .assign(to: \.value, on: emojiDict)
                .store(in: &disposeBag)
        }
        
        func emoji(shortcode: String) -> Mastodon.Entity.Emoji? {
            if !learnedEmoji.contains(shortcode) {
                learnedEmoji.insert(shortcode)
                
                DispatchQueue.global().async {
                    UITextChecker.learnWord(shortcode)
                    UITextChecker.learnWord(":" + shortcode + ":")
                }
            }

            return emojiDict.value[shortcode]?.first
        }
        
    }
}