mastodon-ios/Mastodon/Service/EmojiService/EmojiService+CustomEmojiVie...

91 lines
3.0 KiB
Swift
Raw Normal View History

2021-03-15 07:40:10 +01:00
//
// EmojiService+CustomEmojiViewModel.swift
2021-03-15 07:40:10 +01:00
// Mastodon
//
// Created by MainasuK Cirno on 2021-3-15.
//
import Foundation
import Combine
import GameplayKit
import MastodonSDK
extension EmojiService {
final class CustomEmojiViewModel {
2021-03-15 07:40:10 +01:00
var disposeBag = Set<AnyCancellable>()
// input
let domain: String
weak var service: EmojiService?
2021-03-15 07:40:10 +01:00
// 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>([:])
let emojiMapping = CurrentValueSubject<[String: String], Never>([:])
let emojiTrie = CurrentValueSubject<Trie<Character>?, Never>(nil)
private var learnedEmoji: Set<String> = Set()
2021-03-15 07:40:10 +01:00
init(domain: String, service: EmojiService) {
2021-03-15 07:40:10 +01:00
self.domain = domain
self.service = service
emojis
.map { Dictionary(grouping: $0, by: { $0.shortcode }) }
.assign(to: \.value, on: emojiDict)
.store(in: &disposeBag)
emojiDict
.map { dict in
var mapping: [String: String] = [:]
for (key, values) in dict {
guard let emoji = values.first else { continue }
mapping[key] = emoji.url
}
return mapping
}
.assign(to: \.value, on: emojiMapping)
.store(in: &disposeBag)
emojis
.map { emojis -> Trie<Character>? in
guard !emojis.isEmpty else { return nil }
var trie: Trie<Character> = Trie()
for emoji in emojis {
let key = emoji.shortcode.lowercased()
trie.inserted(Array(key).slice, value: emoji)
}
return trie
}
.assign(to: \.value, on: emojiTrie)
.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
2021-03-15 07:40:10 +01:00
}
}
}