mirror of https://github.com/NekoX-Dev/NekoX.git
72 lines
2.0 KiB
Kotlin
72 lines
2.0 KiB
Kotlin
|
package tw.nekomimi.nekogram.transtale
|
||
|
|
||
|
import org.dizitart.no2.objects.ObjectRepository
|
||
|
import org.dizitart.no2.objects.filters.ObjectFilters
|
||
|
import org.telegram.messenger.LocaleController
|
||
|
import tw.nekomimi.nekogram.NekoConfig
|
||
|
import tw.nekomimi.nekogram.database.mkDatabase
|
||
|
import tw.nekomimi.nekogram.utils.UIUtil
|
||
|
import java.util.*
|
||
|
import kotlin.collections.HashMap
|
||
|
|
||
|
class TranslateDb(val code: String) {
|
||
|
|
||
|
var conn: ObjectRepository<TransItem> = db.getRepository(code, TransItem::class.java)
|
||
|
|
||
|
companion object {
|
||
|
|
||
|
val db = mkDatabase("translate_caches")
|
||
|
|
||
|
val repo = HashMap<Locale, TranslateDb>()
|
||
|
val chat = db.getRepository(ChatLanguage::class.java)
|
||
|
|
||
|
@JvmStatic fun getChatLanguage(chatId: Int, default: Locale): Locale {
|
||
|
|
||
|
return chat.find(ObjectFilters.eq("chatId", chatId)).firstOrDefault()?.language?.code2Locale ?: default
|
||
|
|
||
|
}
|
||
|
|
||
|
@JvmStatic fun saveChatLanguage(chatId: Int, locale: Locale) = UIUtil.runOnIoDispatcher {
|
||
|
|
||
|
chat.update(ChatLanguage(chatId, locale.locale2code), true)
|
||
|
|
||
|
}
|
||
|
|
||
|
@JvmStatic fun currentTarget() = NekoConfig.translateToLang?.transDbByCode ?: LocaleController.getInstance().currentLocale.transDb
|
||
|
|
||
|
@JvmStatic fun forLocale(locale: Locale) = locale.transDb
|
||
|
|
||
|
@JvmStatic fun currentInputTarget() = NekoConfig.translateInputLang.transDbByCode
|
||
|
|
||
|
@JvmStatic fun clearAll() {
|
||
|
|
||
|
db.listRepositories().map { it.transDbByCode }.forEach { it.clear() }
|
||
|
|
||
|
repo.clear()
|
||
|
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
fun clear() = synchronized(this) {
|
||
|
|
||
|
conn.drop()
|
||
|
conn = db.getRepository(code, TransItem::class.java)
|
||
|
|
||
|
}
|
||
|
|
||
|
fun contains(text: String) = synchronized(this) { conn.find(ObjectFilters.eq("text", text)).count() > 0 }
|
||
|
|
||
|
fun save(text: String, trans: String) = synchronized<Unit>(this) {
|
||
|
|
||
|
conn.update(TransItem(text, trans), true)
|
||
|
|
||
|
}
|
||
|
|
||
|
fun query(text: String) = synchronized(this) {
|
||
|
|
||
|
conn.find(ObjectFilters.eq("text", text)).firstOrDefault()?.trans
|
||
|
|
||
|
}
|
||
|
|
||
|
}
|