troubleshoot/pkg/controller/preflightjob/analyzers.go

package preflightjob

import (
	"context"
	// "fmt"

	troubleshootv1beta1 "github.com/replicatedhq/troubleshoot/pkg/apis/troubleshoot/v1beta1"
)

type AnalysisResult struct {
	Success    bool
	TextOutput string
}

func (r *ReconcilePreflightJob) reconcilePreflightAnalyzers(instance *troubleshootv1beta1.PreflightJob, preflight *troubleshootv1beta1.Preflight) error {
	for _, analyzer := range preflight.Spec.Analyzers {
		if err := r.reconcileOnePreflightAnalyzer(instance, analyzer); err != nil {
			return err
		}
	}

	return nil
}

func (r *ReconcilePreflightJob) reconcileOnePreflightAnalyzer(instance *troubleshootv1beta1.PreflightJob, analyze *troubleshootv1beta1.Analyze) error {
	if contains(instance.Status.AnalyzersRunning, idForAnalyzer(analyze)) {
		// these are the analyzers we want to attempt to run
		if analyze.ClusterVersion != nil {
			result, err := r.analyzeClusterVersion(instance, analyze.ClusterVersion)
			if err != nil {
				return err
			}

			if result.Success == false {
				return nil // collectors are not yet ready
			}

			instance.Status.AnalyzersSuccessful = append(instance.Status.AnalyzersSuccessful, idForAnalyzer(analyze))
			instance.Status.AnalyzersRunning = remove(instance.Status.AnalyzersRunning, idForAnalyzer(analyze))

			if err := r.Update(context.Background(), instance); err != nil {
				return err
			}

			return nil
		}

		return nil
	}

	return nil
}

func idForAnalyzer(analyzer *troubleshootv1beta1.Analyze) string {
	if analyzer.ClusterVersion != nil {
		return "cluster-version"
	}

	if analyzer.StorageClass != nil {
		return "storage-classes"
	}

	return ""
}