/*
 * Copyright OpenSearch Contributors
 * SPDX-License-Identifier: Apache-2.0
 */

package org.opensearch.performanceanalyzer.net;


import io.grpc.stub.StreamObserver;
import java.util.Objects;
import java.util.concurrent.atomic.AtomicBoolean;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.opensearch.performanceanalyzer.CertificateUtils;
import org.opensearch.performanceanalyzer.commons.config.PluginSettings;
import org.opensearch.performanceanalyzer.grpc.MetricsRequest;
import org.opensearch.performanceanalyzer.grpc.MetricsResponse;

// TestNetServer is a NetServer that clients can check is running or not
public class TestNetServer extends NetServer implements Runnable {
    private static final Logger LOG = LogManager.getLogger(TestNetServer.class);

    public AtomicBoolean isRunning = new AtomicBoolean(false);

    public TestNetServer(final int port, final int numServerThreads, final boolean useHttps) {
        super(port, numServerThreads, useHttps);
        // If we want TLS, use the certs in our test resources directory
        if (useHttps) {
            ClassLoader classLoader = getClass().getClassLoader();
            // TODO (sidnaray) this is not the best practice for overriding settings
            // CertificateUtils should be abstracted into an interface and should have a regular and
            // testing impl
            // The interface should be passed into the NetServer constructor and have the proper
            // impl "injected"
            PluginSettings.instance()
                    .overrideProperty(
                            CertificateUtils.TRUSTED_CAS_FILE_PATH,
                            Objects.requireNonNull(classLoader.getResource("tls/rootca/RootCA.pem"))
                                    .getFile());
            PluginSettings.instance()
                    .overrideProperty(
                            CertificateUtils.CERTIFICATE_FILE_PATH,
                            Objects.requireNonNull(
                                            classLoader.getResource("tls/server/localhost.crt"))
                                    .getFile());
            PluginSettings.instance()
                    .overrideProperty(
                            CertificateUtils.PRIVATE_KEY_FILE_PATH,
                            Objects.requireNonNull(
                                            classLoader.getResource("tls/server/localhost.key"))
                                    .getFile());
        }
    }

    @Override
    public void getMetrics(
            MetricsRequest request, StreamObserver<MetricsResponse> responseObserver) {
        LOG.debug("MetricsRequest received by server! {}", request);
        responseObserver.onNext(MetricsResponse.newBuilder().setMetricsResult("metrics").build());
        responseObserver.onCompleted();
    }

    @Override
    protected void postStartHook() {
        isRunning.set(true);
    }

    @Override
    protected void shutdownHook() {
        isRunning.set(false);
    }
}