pkg/api/deploymentapi/depresourceapi/kibana.go

// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you 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.

package depresourceapi

import (
	"fmt"

	"github.com/elastic/cloud-sdk-go/pkg/models"
	"github.com/elastic/cloud-sdk-go/pkg/util/ec"
)

const (
	// DefaultKibanaRefID is used when the RefID is not specified.
	DefaultKibanaRefID = "main-kibana"
)

// NewKibana creates a *models.KibanaPayload from the parameters.
// It relies on a simplified single dimension memory size and zone count to
// construct the Kibana's topology.
func NewKibana(params NewStateless) (*models.KibanaPayload, error) {
	params.fillDefaults(DefaultKibanaRefID)
	if err := params.Validate(); err != nil {
		return nil, err
	}

	// When either not set, we obtain the values from the running deployment.
	// Overriding either or both is done at the end of the if.
	if err := getTemplateAndRefID(&params); err != nil {
		return nil, err
	}

	if len(params.DeploymentTemplateInfoV2.DeploymentTemplate.Resources.Kibana) == 0 {
		return nil, fmt.Errorf("deployment: the %s template is not configured for Kibana. Please use another template if you wish to start Kibana instances",
			params.TemplateID)
	}

	var clusterTopology = params.DeploymentTemplateInfoV2.DeploymentTemplate.Resources.Kibana[0].Plan.ClusterTopology
	var topology = models.KibanaClusterTopologyElement{Size: new(models.TopologySize)}
	if len(clusterTopology) > 0 {
		topology = *clusterTopology[0]
	}
	var payload = newKibanaPayload(params, topology)

	return &payload, nil
}

func newKibanaPayload(params NewStateless, topology models.KibanaClusterTopologyElement) models.KibanaPayload {
	if params.Size > 0 {
		topology.Size.Value = ec.Int32(params.Size)
	}
	if params.ZoneCount > 0 {
		topology.ZoneCount = params.ZoneCount
	}

	return models.KibanaPayload{
		ElasticsearchClusterRefID: ec.String(params.ElasticsearchRefID),
		DisplayName:               params.Name,
		Region:                    ec.String(params.Region),
		RefID:                     ec.String(params.RefID),
		Plan: &models.KibanaClusterPlan{
			Kibana:          &models.KibanaConfiguration{Version: params.Version},
			ClusterTopology: []*models.KibanaClusterTopologyElement{&topology},
		},
	}
}