TokiTrainer/Toki Trainer/ViewModels/TokiDictionaryViewModel.swift
2021-11-06 13:35:56 -04:00

61 lines
1.9 KiB
Swift

//
// TokiDictionaryViewModel.swift
// Toki Trainer
//
// Created by Avery Ada Pace on 11/4/21.
//
import Foundation
class TokiDictionaryViewModel: ObservableObject {
let jsonLoader: TokiJSONLoader = TokiJSONLoader()
private var fullDictionary: [TokiDictEntry] = []
private var fullPartsOfSpeech: [TokiPartOfSpeech] = []
@Published var dictionary: [TokiDictEntry] = []
@Published var partsOfSpeech: [TokiPartOfSpeech] = []
init() {
if let safeDictionary = jsonLoader.loadDictionary() {
dictionary = safeDictionary
fullDictionary = safeDictionary
}
if let safePartsOfSpeech = jsonLoader.loadPOS() {
partsOfSpeech = safePartsOfSpeech
fullPartsOfSpeech = safePartsOfSpeech
}
}
func filterDictionary(_ input: String) {
dictionary = []
if(input.isEmpty) {
dictionary = fullDictionary
} else {
let capturePattern = #"(\w+)"#
let captures = self.searchStringForRegex(string: input, regex: capturePattern)
for capture in captures {
print(capture)
for value in fullDictionary {
if value.word == capture {
dictionary.append(value)
}
}
}
}
}
func searchStringForRegex(string: String, regex: String) -> [String] {
let lowerCaseString = string.lowercased()
let strRange = NSRange(lowerCaseString.startIndex..<lowerCaseString.endIndex, in: lowerCaseString)
let nameRegex = try! NSRegularExpression(pattern: regex, options: [])
let results = nameRegex.matches(in: lowerCaseString, options: [], range: strRange)
return results.map {
String(lowerCaseString[Range($0.range, in: lowerCaseString)!])
}
}
}