#
# Copyright (c) 2019 Wind River Systems, Inc.
#
# SPDX-License-Identifier: Apache-2.0
#

from oslo_log import log as logging
from sysinv.common import exception
from sysinv.common import utils
from sysinv.helm import common
from sysinv.helm import elastic

LOG = logging.getLogger(__name__)


class ElasticsearchHelm(elastic.ElasticBaseHelm):
    """Class to encapsulate helm operations for elasticsearch"""

    CHART = common.HELM_CHART_ELASTICSEARCH

    def get_overrides(self, namespace=None):
        overrides = {
            common.HELM_NS_MONITOR: {
                'cluster': self._get_cluster_overrides(),
                'master': self._get_master_overrides(),
                'data': self._get_data_overrides(),
                'client': self._get_client_overrides(),
            }
        }

        if namespace in self.SUPPORTED_NAMESPACES:
            return overrides[namespace]
        elif namespace:
            raise exception.InvalidHelmNamespace(chart=self.CHART,
                                                 namespace=namespace)
        else:
            return overrides

    def _get_cluster_overrides(self):

        env_vars = {'MINIMUM_MASTER_NODES': "1",
                    'EXPECTED_MASTER_NODES': "1",
                    'RECOVER_AFTER_MASTER_NODES': "1"}

        if utils.is_aio_simplex_system(self.dbapi):
            cluster_initial_master_nodes = ['stx-elasticsearch-master-0']
        else:
            cluster_initial_master_nodes = ['stx-elasticsearch-master-0',
                                            'stx-elasticsearch-master-1']

        conf = {
            'env': env_vars,
            'config': {
                'cluster.initial_master_nodes': cluster_initial_master_nodes},
        }
        return conf

    def _get_master_overrides(self):
        if utils.is_aio_system(self.dbapi):
            heap_size = "256m"
        else:
            heap_size = "512m"

        conf = {
            'replicas':
                self._count_hosts_by_label(common.LABEL_MONITOR_CONTROLLER),
            'heapSize': heap_size,
            'nodeSelector': {common.LABEL_MONITOR_CONTROLLER: "enabled"},
        }
        return conf

    def _get_data_overrides(self):
        # Note memory values are to be system engineered.

        if utils.is_aio_system(self.dbapi):
            heap_size = "512m"
            memory_size = "512Mi"
        else:
            heap_size = "1536m"
            memory_size = "1536Mi"

        conf = {
            'replicas':
                self._count_hosts_by_label(common.LABEL_MONITOR_DATA),
            'heapSize': heap_size,
            'resources': {
                'limits': {
                    'cpu': "1"
                },
                'requests': {
                    'cpu': "25m",
                    'memory': memory_size,
                }, },
            'persistence': {'storageClass': 'general',
                            'size': "100Gi"},
            'nodeSelector': {common.LABEL_MONITOR_DATA: "enabled"},
        }
        return conf

    def _get_client_overrides(self):
        if utils.is_aio_system(self.dbapi):
            heap_size = "256m"
        else:
            heap_size = "512m"

        conf = {
            'replicas':
                self._count_hosts_by_label(common.LABEL_MONITOR_CLIENT),
            'heapSize': heap_size,
            'nodeSelector': {common.LABEL_MONITOR_CLIENT: "enabled"},
        }
        return conf