package net.taehui.twilight.system

import com.fasterxml.jackson.core.type.TypeReference
import com.fasterxml.jackson.databind.ObjectMapper
import net.taehui.twilight.Avatar
import net.taehui.twilight.Logger
import java.io.IOException
import java.nio.file.Files
import java.nio.file.Paths
import java.util.concurrent.ConcurrentHashMap
import java.util.concurrent.CopyOnWriteArraySet

object History : Logger {
    private var remoteAvatarIDsMap = ConcurrentHashMap<String, CopyOnWriteArraySet<String>>()

    fun loadHistory() {
        try {
            remoteAvatarIDsMap.putAll(
                ObjectMapper().readValue(
                    Paths.get("History.json").toFile(),
                    object : TypeReference<ConcurrentHashMap<String, CopyOnWriteArraySet<String>>>() {})
            )
            logInfo("Loaded History")
            saveHistory()
        } catch (e: IOException) {
            logFault(e)
        }
    }

    fun saveHistory() {
        Files.newOutputStream(Paths.get("History.json"))
            .use { ObjectMapper().writerWithDefaultPrettyPrinter().writeValue(it, remoteAvatarIDsMap) }
        logInfo("Saved History")
    }

    fun putHistory(avatar: Avatar, avatarID: String) {
        if (!avatar.isAwilight) {
            remoteAvatarIDsMap.computeIfAbsent(avatar.remote) {
                CopyOnWriteArraySet()
            }.add(avatarID)
        }
    }

    fun getAvatarID(avatarIP: String): String {
        return (remoteAvatarIDsMap[avatarIP] ?: emptyList<String>()).joinToString()
    }
}