Suwayomi-Server/server/src/main/kotlin/suwayomi/tachidesk/global/impl/GlobalMeta.kt

package suwayomi.tachidesk.global.impl

import org.jetbrains.exposed.dao.id.EntityID
import org.jetbrains.exposed.sql.batchInsert
import org.jetbrains.exposed.sql.selectAll
import org.jetbrains.exposed.sql.statements.BatchUpdateStatement
import org.jetbrains.exposed.sql.transactions.transaction
import suwayomi.tachidesk.global.model.table.GlobalMetaTable

/*
 * Copyright (C) Contributors to the Suwayomi project
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */

object GlobalMeta {
    fun modifyMeta(
        key: String,
        value: String,
    ) {
        modifyMetas(mapOf(key to value))
    }

    fun modifyMetas(meta: Map<String, String>) {
        transaction {
            val dbMetaMap =
                GlobalMetaTable
                    .selectAll()
                    .where { GlobalMetaTable.key inList meta.keys }
                    .associateBy { it[GlobalMetaTable.key] }
            val (existingMeta, newMeta) = meta.toList().partition { (key) -> key in dbMetaMap.keys }

            if (existingMeta.isNotEmpty()) {
                BatchUpdateStatement(GlobalMetaTable).apply {
                    existingMeta.forEach { (key, value) ->
                        addBatch(EntityID(dbMetaMap[key]!![GlobalMetaTable.id].value, GlobalMetaTable))
                        this[GlobalMetaTable.value] = value
                    }
                    execute(this@transaction)
                }
            }

            if (newMeta.isNotEmpty()) {
                GlobalMetaTable.batchInsert(newMeta) { (key, value) ->
                    this[GlobalMetaTable.key] = key
                    this[GlobalMetaTable.value] = value
                }
            }
        }
    }

    fun getMetaMap(): Map<String, String> =
        transaction {
            GlobalMetaTable
                .selectAll()
                .associate { it[GlobalMetaTable.key] to it[GlobalMetaTable.value] }
        }
}