// Copyright 2019 ReactiveOps
//
// 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.

package validator

import (
	conf "github.com/reactiveops/fairwinds/pkg/config"
	"github.com/reactiveops/fairwinds/pkg/kube"
	appsv1 "k8s.io/api/apps/v1"
)

// ValidateDeployment validates a single deployment, returns a PodResult.
func ValidateDeployment(conf conf.Configuration, deploy *appsv1.Deployment) ControllerResult {
	pod := deploy.Spec.Template.Spec
	podResult := ValidatePod(conf, &pod)
	return ControllerResult{
		Name:      deploy.Name,
		Type:      "Deployment",
		PodResult: podResult,
	}
}

// ValidateDeployments validates that each deployment conforms to the Fairwinds config,
// returns a list of ResourceResults organized by namespace.
func ValidateDeployments(config conf.Configuration, kubeResources *kube.ResourceProvider) (NamespacedResults, error) {
	nsResults := NamespacedResults{}

	for _, deploy := range kubeResources.Deployments {
		deploymentResult := ValidateDeployment(config, &deploy)
		nsResults = addResult(deploymentResult, nsResults, deploy.Namespace)
	}

	return nsResults, nil
}

func addResult(deploymentResult ControllerResult, nsResults NamespacedResults, nsName string) NamespacedResults {
	nsResult := &NamespaceResult{}

	// If there is already data stored for this namespace name,
	// then append to the ResourceResults to the existing data.
	switch nsResults[nsName] {
	case nil:
		nsResult = &NamespaceResult{
			Summary:           &ResultSummary{},
			DeploymentResults: []ControllerResult{},
		}
		nsResults[nsName] = nsResult
	default:
		nsResult = nsResults[nsName]
	}

	nsResult.DeploymentResults = append(nsResult.DeploymentResults, deploymentResult)
	nsResult.Summary.appendResults(*deploymentResult.PodResult.Summary)

	return nsResults
}