/*
 * Copyright 2014-2021 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
 */

package io.ktor.server.metrics.dropwizard

import com.codahale.metrics.*
import com.codahale.metrics.MetricRegistry.*
import com.codahale.metrics.jvm.*
import io.ktor.server.application.*
import io.ktor.server.application.hooks.*
import io.ktor.server.request.*
import io.ktor.server.routing.*
import io.ktor.utils.io.*
import java.util.concurrent.*

/**
 * A configuration for the [DropwizardMetrics] plugin.
 *
 * [Report a problem](https://ktor.io/feedback/?fqname=io.ktor.server.metrics.dropwizard.DropwizardMetricsConfig)
 */
@KtorDsl
public class DropwizardMetricsConfig {
    /**
     * Specifies the base name (prefix) of Ktor metrics used for monitoring HTTP requests.
     *
     * [Report a problem](https://ktor.io/feedback/?fqname=io.ktor.server.metrics.dropwizard.DropwizardMetricsConfig.baseName)
     *
     * @see [DropwizardMetrics]
     */
    public var baseName: String = name("ktor.calls")

    /**
     * Specifies the meter registry for your monitoring system.
     *
     * [Report a problem](https://ktor.io/feedback/?fqname=io.ktor.server.metrics.dropwizard.DropwizardMetricsConfig.registry)
     *
     * @see [DropwizardMetrics]
     */
    public var registry: MetricRegistry = MetricRegistry()

    /**
     * Allows you to configure a set of metrics for monitoring the JVM.
     * You can disable these metrics by setting this property to `false`.
     *
     * [Report a problem](https://ktor.io/feedback/?fqname=io.ktor.server.metrics.dropwizard.DropwizardMetricsConfig.registerJvmMetricSets)
     *
     * @see [DropwizardMetrics]
     */
    public var registerJvmMetricSets: Boolean = true
}

/**
 * A plugin that lets you configure the `Metrics` library to get
 * useful information about the server and incoming requests.
 *
 * You can learn more from [Dropwizard metrics](https://ktor.io/docs/dropwizard-metrics.html).
 *
 * [Report a problem](https://ktor.io/feedback/?fqname=io.ktor.server.metrics.dropwizard.DropwizardMetrics)
 */
public val DropwizardMetrics: ApplicationPlugin<DropwizardMetricsConfig> =
    createApplicationPlugin("DropwizardMetrics", ::DropwizardMetricsConfig) {
        val registry = pluginConfig.registry
        val baseName = pluginConfig.baseName
        val duration = registry.timer(name(baseName, "duration"))
        val active = registry.counter(name(baseName, "active"))
        val exceptions = registry.meter(name(baseName, "exceptions"))
        val httpStatus = ConcurrentHashMap<Int, Meter>()

        if (pluginConfig.registerJvmMetricSets) {
            listOf<Pair<String, () -> Metric>>(
                "jvm.memory" to ::MemoryUsageGaugeSet,
                "jvm.garbage" to ::GarbageCollectorMetricSet,
                "jvm.threads" to ::ThreadStatesGaugeSet,
                "jvm.files" to ::FileDescriptorRatioGauge,
                "jvm.attributes" to ::JvmAttributeGaugeSet
            ).filter { (name, _) ->
                !registry.names.any { existingName -> existingName.startsWith(name) }
            }.forEach { (name, metric) -> registry.register(name, metric()) }
        }

        on(CallFailed) { _, _ ->
            exceptions.mark()
        }

        on(MonitoringEvent(RoutingRoot.RoutingCallStarted)) { call ->
            val name = call.route.toString()
            val meter = registry.meter(name(baseName, name, "meter"))
            val timer = registry.timer(name(baseName, name, "timer"))
            meter.mark()
            val context = timer.time()
            call.attributes.put(
                routingMetricsKey,
                RoutingMetrics(name, context)
            )
        }

        @OptIn(InternalAPI::class)
        on(Metrics) { call ->
            active.inc()
            call.attributes.put(measureKey, CallMeasure(duration.time()))
        }

        on(ResponseSent) { call ->
            val routingMetrics = call.attributes.takeOrNull(routingMetricsKey)
            val name = routingMetrics?.name ?: call.request.routeName
            val status = call.response.status()?.value ?: 0
            val statusMeter =
                registry.meter(name(baseName, name, status.toString()))
            statusMeter.mark()
            routingMetrics?.context?.stop()

            active.dec()
            val meter = httpStatus.computeIfAbsent(call.response.status()?.value ?: 0) {
                registry.meter(name(baseName, "status", it.toString()))
            }
            meter.mark()
            call.attributes.getOrNull(measureKey)?.apply {
                timer.stop()
            }
        }
    }

private val ApplicationRequest.routeName: String
    get() {
        val metricUri = uri.ifEmpty { "/" }.let { if (it.endsWith('/')) it else "$it/" }
        return "$metricUri(method:${httpMethod.value})"
    }
