package com.bugsnag.android.internal

import com.bugsnag.android.NdkPluginCaller

class InternalMetricsImpl(source: Map<String, Any>? = null) : InternalMetrics {
    private val configDifferences: MutableMap<String, Any>
    private val callbackCounts: MutableMap<String, Int>
    private var metadataStringsTrimmedCount = 0
    private var metadataCharsTruncatedCount = 0
    private var breadcrumbsRemovedCount = 0
    private var breadcrumbBytesRemovedCount = 0

    init {
        if (source != null) {
            @Suppress("UNCHECKED_CAST")
            configDifferences = (source["config"] as MutableMap<String, Any>?) ?: hashMapOf()
            @Suppress("UNCHECKED_CAST")
            callbackCounts = (source["callbacks"] as MutableMap<String, Int>?) ?: hashMapOf()
            @Suppress("UNCHECKED_CAST")
            val system = source["system"] as MutableMap<String, Any>?
            if (system != null) {
                metadataStringsTrimmedCount = (system["stringsTruncated"] as Number?)?.toInt() ?: 0
                metadataCharsTruncatedCount = (system["stringCharsTruncated"] as Number?)?.toInt() ?: 0
                breadcrumbsRemovedCount = (system["breadcrumbsRemovedCount"] as Number?)?.toInt() ?: 0
                breadcrumbBytesRemovedCount = (system["breadcrumbBytesRemoved"] as Number?)?.toInt() ?: 0
            }
        } else {
            configDifferences = hashMapOf()
            callbackCounts = hashMapOf()
        }
    }

    override fun toJsonableMap(): Map<String, Any> {
        val callbacks = allCallbacks()

        val system = listOfNotNull(
            if (metadataStringsTrimmedCount > 0) "stringsTruncated" to metadataStringsTrimmedCount else null,
            if (metadataCharsTruncatedCount > 0) "stringCharsTruncated" to metadataCharsTruncatedCount else null,
            if (breadcrumbsRemovedCount > 0) "breadcrumbsRemoved" to breadcrumbsRemovedCount else null,
            if (breadcrumbBytesRemovedCount > 0) "breadcrumbBytesRemoved" to breadcrumbBytesRemovedCount else null,
        ).toMap()

        return listOfNotNull(
            if (configDifferences.isNotEmpty()) "config" to configDifferences else null,
            if (callbacks.isNotEmpty()) "callbacks" to callbacks else null,
            if (system.isNotEmpty()) "system" to system else null,
        ).toMap()
    }

    override fun setConfigDifferences(differences: Map<String, Any>) {
        configDifferences.clear()
        configDifferences.putAll(differences)
        // This is currently the only place where we set static data.
        // When that changes in future, we'll need a StaticData object to properly merge data
        // coming from multiple sources.
        NdkPluginCaller.setStaticData(mapOf("usage" to mapOf("config" to configDifferences)))
    }

    override fun setCallbackCounts(newCallbackCounts: Map<String, Int>) {
        callbackCounts.clear()
        callbackCounts.putAll(newCallbackCounts)
        NdkPluginCaller.initCallbackCounts(newCallbackCounts)
    }

    override fun notifyAddCallback(callback: String) {
        modifyCallback(callback, 1)
        NdkPluginCaller.notifyAddCallback(callback)
    }

    override fun notifyRemoveCallback(callback: String) {
        modifyCallback(callback, -1)
        NdkPluginCaller.notifyRemoveCallback(callback)
    }

    private fun modifyCallback(callback: String, delta: Int) {
        var currentValue = callbackCounts[callback] ?: 0
        currentValue += delta
        callbackCounts[callback] = currentValue.coerceAtLeast(0)
    }

    private fun allCallbacks(): Map<String, Any> {
        val result = hashMapOf<String, Any>()
        result.putAll(callbackCounts)

        val counts = NdkPluginCaller.getCurrentCallbackSetCounts()
        if (counts != null) {
            // ndkOnError comes from the native side. The rest we already have.
            val ndkOnError = counts["ndkOnError"]
            if (ndkOnError != null) {
                result["ndkOnError"] = ndkOnError
            }
        }

        val usage = NdkPluginCaller.getCurrentNativeApiCallUsage()
        if (usage != null) {
            result.putAll(usage)
        }

        return result
    }

    override fun setMetadataTrimMetrics(stringsTrimmed: Int, charsRemoved: Int) {
        metadataStringsTrimmedCount = stringsTrimmed
        metadataCharsTruncatedCount = charsRemoved
    }

    override fun setBreadcrumbTrimMetrics(breadcrumbsRemoved: Int, bytesRemoved: Int) {
        breadcrumbsRemovedCount = breadcrumbsRemoved
        breadcrumbBytesRemovedCount = bytesRemoved
    }
}