main.go

/*
Copyright 2021.

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 main

import (
	"flag"
	"fmt"
	"os"
	"slices"

	obopo "github.com/rhobs/obo-prometheus-operator/pkg/operator"
	"go.uber.org/zap/zapcore"
	k8sflag "k8s.io/component-base/cli/flag"
	"k8s.io/utils/ptr"
	ctrl "sigs.k8s.io/controller-runtime"
	"sigs.k8s.io/controller-runtime/pkg/log/zap"

	"github.com/rhobs/observability-operator/pkg/operator"
)

// The default values we use. Prometheus and Alertmanager are handled by
// prometheus-operator. For thanos we use the default version from
// prometheus-operator.
var defaultImages = map[string]string{
	"prometheus":                 "",
	"alertmanager":               "",
	"thanos":                     obopo.DefaultThanosImage,
	"ui-dashboards":              "quay.io/openshift-observability-ui/console-dashboards-plugin:v0.4.2",
	"ui-troubleshooting-panel":   "quay.io/openshift-observability-ui/troubleshooting-panel-console-plugin:v0.4.4",
	"ui-distributed-tracing-pf4": "quay.io/openshift-observability-ui/distributed-tracing-console-plugin:v0.3.2",
	"ui-distributed-tracing-pf5": "quay.io/openshift-observability-ui/distributed-tracing-console-plugin:v0.4.2",
	"ui-distributed-tracing":     "quay.io/openshift-observability-ui/distributed-tracing-console-plugin:v1.0.2",
	"ui-logging-pf4":             "quay.io/openshift-observability-ui/logging-view-plugin:v6.0.4",
	"ui-logging":                 "quay.io/openshift-observability-ui/logging-view-plugin:v6.1.5",
<<<<<<< HEAD
	"korrel8r":                   "quay.io/korrel8r/korrel8r:0.8.4",
||||||| 8bb890bd
	"ui-dashboards":              "quay.io/openshift-observability-ui/console-dashboards-plugin:v0.4.1",
	"ui-troubleshooting-panel":   "quay.io/openshift-observability-ui/troubleshooting-panel-console-plugin:v0.4.3",
	"ui-distributed-tracing-pf4": "quay.io/openshift-observability-ui/distributed-tracing-console-plugin:v0.3.1",
	"ui-distributed-tracing-pf5": "quay.io/openshift-observability-ui/distributed-tracing-console-plugin:v0.4.1",
	"ui-distributed-tracing":     "quay.io/openshift-observability-ui/distributed-tracing-console-plugin:v1.0.1",
	"ui-logging-pf4":             "quay.io/openshift-observability-ui/logging-view-plugin:v6.0.1",
	"ui-logging":                 "quay.io/openshift-observability-ui/logging-view-plugin:v6.1.2",
	"korrel8r":                   "quay.io/korrel8r/korrel8r:0.8.4",
=======
	"korrel8r":                   "quay.io/korrel8r/korrel8r:0.9.1",
>>>>>>> main
	"health-analyzer":            "quay.io/openshiftanalytics/cluster-health-analyzer:v1.1.0",
	"ui-monitoring-pf5":          "quay.io/openshift-observability-ui/monitoring-console-plugin:v0.4.4",
	"ui-monitoring":              "quay.io/openshift-observability-ui/monitoring-console-plugin:v0.5.3",
	"perses":                     "quay.io/openshift-observability-ui/perses:v0.53.0-go-1.25",
}

func imagesUsed() []string {
	i := 0
	imgs := make([]string, len(defaultImages))
	for k := range defaultImages {
		imgs[i] = k
		i++
	}
	slices.Sort(imgs)
	return imgs
}

// validateImages merges the passed images with the defaults and checks if any
// unknown image names are passed. If an unknown image is found, this raises an
// error.
func validateImages(images *k8sflag.MapStringString) (map[string]string, error) {
	res := defaultImages
	if images.Empty() {
		return res, nil
	}
	imgs := *images.Map
	for k, v := range imgs {
		if _, ok := res[k]; !ok {
			return nil, fmt.Errorf("image %v is unknown", k)
		}
		res[k] = v
	}
	return res, nil
}

func main() {
	var (
		namespace        string
		metricsAddr      string
		healthProbeAddr  string
		openShiftEnabled bool
		otelCSVName      string
		tempoCSVName     string

		setupLog = ctrl.Log.WithName("setup")
	)
	images := k8sflag.NewMapStringString(ptr.To(make(map[string]string)))

	flag.StringVar(&namespace, "namespace", "default", "The namespace in which the operator runs")
	flag.StringVar(&metricsAddr, "metrics-bind-address", ":8080", "The address the metric endpoint binds to.")
	flag.StringVar(&healthProbeAddr, "health-probe-bind-address", ":8081", "The address the health probe endpoint binds to.")
	flag.Var(images, "images", fmt.Sprintf("Full images refs to use for containers managed by the operator. E.g thanos=quay.io/thanos/thanos:v0.33.0. Images used are %v", imagesUsed()))
	flag.BoolVar(&openShiftEnabled, "openshift.enabled", false, "Enable OpenShift specific features such as Console Plugins.")
	flag.StringVar(&otelCSVName, "opentelemetry-csv", "", "OpenTelemetry Operator starting CSV name. This can be used to install a specific OpenTelemetry Operator version. Empty string means the latest version will be installed.")
	flag.StringVar(&tempoCSVName, "tempo-csv", "", "Tempo Operator starting CSV name. This can be used to install a specific Tempo Operator version. Empty string means the latest version will be installed.")

	opts := zap.Options{
		Development: true,
		TimeEncoder: zapcore.RFC3339TimeEncoder,
	}
	opts.BindFlags(flag.CommandLine)
	flag.Parse()

	ctrl.SetLogger(zap.New(zap.UseFlagOptions(&opts)))

	setupLog.Info("running with arguments",
		"namespace", namespace,
		"metrics-bind-address", metricsAddr,
		"images", images,
		"openshift.enabled", openShiftEnabled,
	)

	imgMap, err := validateImages(images)
	if err != nil {
		setupLog.Error(err, "cannot create a new operator")
		os.Exit(1)
	}

	ctx := ctrl.SetupSignalHandler()

	op, err := operator.New(
		ctx,
		operator.NewOperatorConfiguration(
			operator.WithNamespace(namespace),
			operator.WithMetricsAddr(metricsAddr),
			operator.WithHealthProbeAddr(healthProbeAddr),
			operator.WithPrometheusImage(imgMap["prometheus"]),
			operator.WithAlertmanagerImage(imgMap["alertmanager"]),
			operator.WithThanosSidecarImage(imgMap["thanos"]),
			operator.WithThanosQuerierImage(imgMap["thanos"]),
			operator.WithUIPluginImages(imgMap),
			operator.WithObservabilityInstaller(operator.ObservabilityInstallerConfiguration{
				COONamespace:     os.Getenv("NAMESPACE"),
				OpenTelemetryCSV: otelCSVName,
				TempoCSV:         tempoCSVName,
			}),
			operator.WithFeatureGates(operator.FeatureGates{
				OpenShift: operator.OpenShiftFeatureGates{
					Enabled: openShiftEnabled,
				},
			}),
		))
	if err != nil {
		setupLog.Error(err, "cannot create a new operator")
		os.Exit(1)
	}

	setupLog.Info("starting manager")
	if err := op.Start(ctx); err != nil {
		setupLog.Error(err, "terminating")
		os.Exit(1)
	}
}