NekoX/TMessagesProj/src/main/java/tw/nekomimi/nekogram/transtale/TranslateDb.kt

package tw.nekomimi.nekogram.transtale

import org.dizitart.no2.filters.FluentFilter.where
import org.dizitart.no2.repository.ObjectRepository
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(TransItem::class.java, code)

    companion object {

        val db = mkDatabase("translate_caches")

        val repo = HashMap<Locale, TranslateDb>()
        val chat = db.getRepository(ChatLanguage::class.java, "chat")
        val ccTarget = db.getRepository(ChatCCTarget::class.java, "cc")

        @JvmStatic
        fun getChatLanguage(chatId: Int, default: Locale): Locale {

            return chat.find(where("chatId").eq(chatId)).firstOrNull()?.language?.code2Locale
                    ?: default

        }

        @JvmStatic
        fun saveChatLanguage(chatId: Int, locale: Locale) = UIUtil.runOnIoDispatcher {

            chat.update(ChatLanguage(chatId, locale.locale2code), true)

        }

        @JvmStatic
        fun getChatCCTarget(chatId: Int, default: String?): String? {

            return ccTarget.find(where("chatId").eq(chatId)).firstOrNull()?.ccTarget
                    ?: default

        }

        @JvmStatic
        fun saveChatCCTarget(chatId: Int, target: String) = UIUtil.runOnIoDispatcher {

            ccTarget.update(ChatCCTarget(chatId, target), 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()
                    .filter { it != "chat" }
                    .map { db.getCollection(it) }
                    .forEach { it.drop() }

            repo.clear()

        }

    }

    fun clear() = synchronized(this) {

        conn.drop()

    }

    fun contains(text: String) = synchronized(this) { conn.find(where("text").eq(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(where("text").eq(text)).firstOrNull()?.trans

    }

}