/*
* Copyright 2015-2017 floragunn GmbH
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/

/*
* SPDX-License-Identifier: Apache-2.0
*
* The OpenSearch Contributors require contributions made to
* this file be licensed under the Apache-2.0 license or a
* compatible open source license.
*
* Modifications Copyright OpenSearch Contributors. See
* GitHub history for details.
*/

package org.opensearch.test.framework.cluster;

import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.EnumSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;

import org.opensearch.index.reindex.ReindexModulePlugin;
import org.opensearch.join.ParentJoinModulePlugin;
import org.opensearch.percolator.PercolatorModulePlugin;
import org.opensearch.plugins.Plugin;
import org.opensearch.search.aggregations.matrix.MatrixAggregationModulePlugin;
import org.opensearch.security.OpenSearchSecurityPlugin;
import org.opensearch.transport.Netty4ModulePlugin;

import static java.util.Collections.unmodifiableList;
import static org.opensearch.test.framework.cluster.NodeType.CLIENT;
import static org.opensearch.test.framework.cluster.NodeType.CLUSTER_MANAGER;
import static org.opensearch.test.framework.cluster.NodeType.DATA;

public enum ClusterManager {
    // 3 nodes (1m, 2d)
    DEFAULT(new NodeSettings(NodeRole.CLUSTER_MANAGER), new NodeSettings(NodeRole.DATA), new NodeSettings(NodeRole.DATA)),

    // 1 node (1md)
    SINGLENODE(new NodeSettings(NodeRole.CLUSTER_MANAGER, NodeRole.DATA)),

    SINGLE_REMOTE_CLIENT(new NodeSettings(NodeRole.CLUSTER_MANAGER, NodeRole.DATA, NodeRole.REMOTE_CLUSTER_CLIENT)),

    // 4 node (1m, 2d, 1c)
    CLIENTNODE(
        new NodeSettings(NodeRole.CLUSTER_MANAGER),
        new NodeSettings(NodeRole.DATA),
        new NodeSettings(NodeRole.DATA),
        new NodeSettings()
    ),

    THREE_CLUSTER_MANAGERS(
        new NodeSettings(NodeRole.CLUSTER_MANAGER),
        new NodeSettings(NodeRole.CLUSTER_MANAGER),
        new NodeSettings(NodeRole.CLUSTER_MANAGER),
        new NodeSettings(NodeRole.DATA),
        new NodeSettings(NodeRole.DATA)
    );

    private List<NodeSettings> nodeSettings = new LinkedList<>();

    private ClusterManager(NodeSettings... settings) {
        nodeSettings.addAll(Arrays.asList(settings));
    }

    public List<NodeSettings> getNodeSettings() {
        return unmodifiableList(nodeSettings);
    }

    public List<NodeSettings> getClusterManagerNodeSettings() {
        return unmodifiableList(nodeSettings.stream().filter(a -> a.containRole(NodeRole.CLUSTER_MANAGER)).collect(Collectors.toList()));
    }

    public List<NodeSettings> getNonClusterManagerNodeSettings() {
        return unmodifiableList(nodeSettings.stream().filter(a -> !a.containRole(NodeRole.CLUSTER_MANAGER)).collect(Collectors.toList()));
    }

    public int getNodes() {
        return nodeSettings.size();
    }

    public int getClusterManagerNodes() {
        return (int) nodeSettings.stream().filter(a -> a.containRole(NodeRole.CLUSTER_MANAGER)).count();
    }

    public int getDataNodes() {
        return (int) nodeSettings.stream().filter(a -> a.containRole(NodeRole.DATA)).count();
    }

    public int getClientNodes() {
        return (int) nodeSettings.stream().filter(a -> a.isClientNode()).count();
    }

    public static class NodeSettings {

        private final static List<Class<? extends Plugin>> DEFAULT_PLUGINS = List.of(
            Netty4ModulePlugin.class,
            OpenSearchSecurityPlugin.class,
            MatrixAggregationModulePlugin.class,
            ParentJoinModulePlugin.class,
            PercolatorModulePlugin.class,
            ReindexModulePlugin.class
        );

        private final Set<NodeRole> roles;
        public final List<Class<? extends Plugin>> plugins;

        public NodeSettings(NodeRole... roles) {
            this(roles.length == 0 ? Collections.emptySet() : EnumSet.copyOf(Arrays.asList(roles)), Collections.emptyList());
        }

        public NodeSettings(Set<NodeRole> roles, List<Class<? extends Plugin>> additionalPlugins) {
            super();
            this.roles = Objects.requireNonNull(roles, "Node roles set must not be null");
            this.plugins = mergePlugins(additionalPlugins, DEFAULT_PLUGINS);
        }

        public boolean containRole(NodeRole nodeRole) {
            return roles.contains(nodeRole);
        }

        public boolean isClientNode() {
            return (roles.contains(NodeRole.DATA) == false) && (roles.contains(NodeRole.CLUSTER_MANAGER));
        }

        NodeType recognizeNodeType() {
            if (roles.contains(NodeRole.CLUSTER_MANAGER)) {
                return CLUSTER_MANAGER;
            } else if (roles.contains(NodeRole.DATA)) {
                return DATA;
            } else {
                return CLIENT;
            }
        }

        private List<Class<? extends Plugin>> mergePlugins(Collection<Class<? extends Plugin>>... plugins) {
            List<Class<? extends Plugin>> mergedPlugins = Arrays.stream(plugins)
                .filter(Objects::nonNull)
                .flatMap(Collection::stream)
                .collect(Collectors.toList());
            return unmodifiableList(mergedPlugins);
        }

        @SuppressWarnings("unchecked")
        public Class<? extends Plugin>[] getPlugins() {
            return plugins.toArray(new Class[0]);
        }

        public Class<? extends Plugin>[] pluginsWithAddition(List<Class<? extends Plugin>> additionalPlugins) {
            return mergePlugins(plugins, additionalPlugins).toArray(Class[]::new);
        }
    }
}