/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the "Elastic License
 * 2.0", the "GNU Affero General Public License v3.0 only", and the "Server Side
 * Public License v 1"; you may not use this file except in compliance with, at
 * your election, the "Elastic License 2.0", the "GNU Affero General Public
 * License v3.0 only", or the "Server Side Public License, v 1".
 */

package org.elasticsearch.telemetry;

import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.plugins.Plugin;
import org.elasticsearch.plugins.TelemetryPlugin;
import org.elasticsearch.telemetry.metric.Instrument;
import org.elasticsearch.telemetry.metric.MeterRegistry;
import org.elasticsearch.telemetry.tracing.Tracer;

import java.util.ArrayList;
import java.util.List;

/**
 * TelemetryPlugin that uses RecordingMeterRegistry to record meter calls
 * and exposes measurement getters.
 */
public class TestTelemetryPlugin extends Plugin implements TelemetryPlugin {

    protected final RecordingMeterRegistry meter = new RecordingMeterRegistry();

    Registration getRegistration(Instrument instrument) {
        return meter.getRecorder().getRegistration(instrument);
    }

    public List<Measurement> getMetrics(Instrument instrument) {
        return meter.getRecorder().getMeasurements(instrument);
    }

    public List<Measurement> getDoubleCounterMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.DOUBLE_COUNTER, name);
    }

    public List<Measurement> getLongCounterMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.LONG_COUNTER, name);
    }

    public List<Measurement> getLongAsyncCounterMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.LONG_ASYNC_COUNTER, name);
    }

    public List<Measurement> getDoubleUpDownCounterMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.DOUBLE_UP_DOWN_COUNTER, name);
    }

    public List<Measurement> getLongUpDownCounterMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.LONG_UP_DOWN_COUNTER, name);
    }

    public List<Measurement> getDoubleGaugeMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.DOUBLE_GAUGE, name);
    }

    public List<Measurement> getLongGaugeMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.LONG_GAUGE, name);
    }

    public List<Measurement> getDoubleHistogramMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.DOUBLE_HISTOGRAM, name);
    }

    public List<Measurement> getLongHistogramMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.LONG_HISTOGRAM, name);
    }

    public void collect() {
        meter.getRecorder().collect();
    }

    public void resetMeter() {
        meter.getRecorder().resetCalls();
    }

    public ArrayList<String> getRegisteredMetrics(InstrumentType instrumentType) {
        return meter.getRecorder().getRegisteredMetrics(instrumentType);
    }

    @Override
    public TelemetryProvider getTelemetryProvider(Settings settings) {
        return new TelemetryProvider() {
            @Override
            public Tracer getTracer() {
                return Tracer.NOOP;
            }

            @Override
            public MeterRegistry getMeterRegistry() {
                return meter;
            }
        };
    }
}