package analysis

import (
	"github.com/go-logr/logr"
	metricsapi "github.com/keptn/lifecycle-toolkit/metrics-operator/api/v1"
	metricstypes "github.com/keptn/lifecycle-toolkit/metrics-operator/controllers/common/analysis/types"
	"github.com/keptn/lifecycle-toolkit/metrics-operator/controllers/common/providers"
	"golang.org/x/net/context"
	"sigs.k8s.io/controller-runtime/pkg/client"
)

//go:generate moq -pkg fake -skip-ensure -out ./fake/evaluator_mock.go . IObjectivesEvaluator
type IObjectivesEvaluator interface {
	Evaluate(ctx context.Context, metricsProvider *metricsapi.KeptnMetricsProvider, obj chan metricstypes.ProviderRequest)
}

type ObjectivesEvaluator struct {
	*metricsapi.Analysis
	providers.ProviderFactory
	client.Client
	log     logr.Logger
	results chan metricsapi.ProviderResult
	cancel  context.CancelFunc
}

func (oe ObjectivesEvaluator) Evaluate(ctx context.Context, metricsProvider *metricsapi.KeptnMetricsProvider, obj chan metricstypes.ProviderRequest) {
	provider, err := oe.ProviderFactory(metricsProvider, oe.log, oe.Client)
	if err != nil {
		oe.log.Error(err, "Failed to get the correct Provider")
		oe.cancel()
		return
	}
	for o := range obj {
		value := ""
		var strErr string
		value, err = provider.FetchAnalysisValue(ctx, o.Query, *oe.Analysis, o.Provider)

		if err != nil {
			strErr = err.Error()
		}
		result := metricsapi.ProviderResult{
			Objective: o.Objective.AnalysisValueTemplateRef,
			Query:     o.Query,
			Value:     value,
			ErrMsg:    strErr,
		}
		oe.log.Info("provider", "id:", metricsProvider.Spec.Type, "finished job:", o.Objective.AnalysisValueTemplateRef.Name, "result:", result)
		oe.results <- result
	}
}