fixup launcher

Author: josvazg

internal/cmd/nextako/main.go

@@ -29,21 +29,32 @@ import (
 	"k8s.io/apimachinery/pkg/runtime"
 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
 	ctrl "sigs.k8s.io/controller-runtime"
+	"sigs.k8s.io/controller-runtime/pkg/client"
 	ctrlog "sigs.k8s.io/controller-runtime/pkg/log"
 	metricsserver "sigs.k8s.io/controller-runtime/pkg/metrics/server"
 
+	"github.com/mongodb/mongodb-atlas-kubernetes/v2/internal/controller/atlas"
+	"github.com/mongodb/mongodb-atlas-kubernetes/v2/internal/kube"
 	akov2next "github.com/mongodb/mongodb-atlas-kubernetes/v2/internal/nextapi/v1"
 )
 
+const (
+	dryRun = false
+)
+
 func main() {
 	akoPath, ok := os.LookupEnv("AKO_BINARY_PATH")
 	if !ok {
 		log.Fatalf("AKO_BINARY_PATH must be set and point to the AKO binary path")
 	}
+	domain := os.Getenv("MCLI_OPS_MANAGER_URL")
+	if domain == "" {
+		domain = "https://cloud-qa.mongodb.com"
+	}
 
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
-	if err := startExperimentalControllers(cancel); err != nil {
+	if err := startExperimentalControllers(cancel, domain); err != nil {
 		log.Fatalf("Failed to launch experimental controllers: %v", err)
 	}
 
@@ -57,7 +68,7 @@ func main() {
 	handleShutdown(ctx, cancel, akoCmd)
 }
 
-func startExperimentalControllers(cancel context.CancelFunc) error {
+func startExperimentalControllers(cancel context.CancelFunc, domain string) error {
 	setupLog := ctrl.Log.WithName("experimental-launcher")
 	setupLog.Info("Starting experimental controllers")
 
@@ -70,9 +81,11 @@ func startExperimentalControllers(cancel context.CancelFunc) error {
 		return fmt.Errorf("failed to get kube config (is cluster running?): %w", err)
 	}
 
-	if err := setupControllerRuntimeLogger(); err != nil {
+	logger, err := newRootLogger()
+	if err != nil {
 		return err
 	}
+	ctrlog.SetLogger(zapr.NewLogger(logger))
 	mgr, err := ctrl.NewManager(kubeCfg, ctrl.Options{
 		Scheme:                 experimentalScheme,
 		Metrics:                metricsserver.Options{BindAddress: ":8090"},
@@ -82,8 +95,10 @@ func startExperimentalControllers(cancel context.CancelFunc) error {
 		return fmt.Errorf("failed to start experimental controller manager: %w", err)
 	}
 
+	atlasProvider := atlas.NewProductionProvider(domain, dryRun)
+	secretRef := operatorGlobalKeySecret()
 	for _, reconciler := range experimentalReconcilers {
-		if err := reconciler.register(mgr); err != nil {
+		if err := reconciler.register(mgr, atlasProvider, logger, secretRef); err != nil {
 			return fmt.Errorf("failed to register experimental %s: %w", reconciler.name, err)
 		}
 		log.Printf("Experimental %s registered", reconciler.name)
@@ -98,13 +113,12 @@ func startExperimentalControllers(cancel context.CancelFunc) error {
 	return nil
 }
 
-func setupControllerRuntimeLogger() error {
+func newRootLogger() (*zap.Logger, error) {
 	zapLog, err := zap.NewDevelopment()
 	if err != nil {
-		return fmt.Errorf("failed to initialize logger: %w", err)
+		return nil, fmt.Errorf("failed to initialize logger: %w", err)
 	}
-	ctrlog.SetLogger(zapr.NewLogger(zapLog))
-	return nil
+	return zapLog, nil
 }
 
 func startAKO(akoPath string) (*exec.Cmd, error) {
@@ -139,3 +153,22 @@ func handleShutdown(ctx context.Context, cancel context.CancelFunc, akoCmd *exec
 	_ = akoCmd.Wait()
 	log.Println("Experimental launcher shut down completely.")
 }
+
+func operatorGlobalKeySecret() client.ObjectKey {
+	operatorPodName := os.Getenv("OPERATOR_POD_NAME")
+	if operatorPodName == "" {
+		log.Fatal(`"OPERATOR_POD_NAME" environment variable must be set!`)
+	}
+	deploymentName, err := kube.ParseDeploymentNameFromPodName(operatorPodName)
+	if err != nil {
+		log.Fatalf(`Failed to get Operator Deployment name from "OPERATOR_POD_NAME" environment variable: %s`, err.Error())
+	}
+	secretName := deploymentName + "-api-key"
+
+	operatorNamespace := os.Getenv("OPERATOR_NAMESPACE")
+	if operatorNamespace == "" {
+		log.Fatal(`"OPERATOR_NAMESPACE" environment variable must be set!`)
+	}
+
+	return client.ObjectKey{Namespace: operatorNamespace, Name: secretName}
+}

internal/cmd/nextako/registrations.go

@@ -15,12 +15,15 @@
 package main
 
 import (
+	"go.uber.org/zap"
 	ctrl "sigs.k8s.io/controller-runtime"
+	"sigs.k8s.io/controller-runtime/pkg/client"
 
+	"github.com/mongodb/mongodb-atlas-kubernetes/v2/internal/controller/atlas"
 	integrations "github.com/mongodb/mongodb-atlas-kubernetes/v2/internal/controller/atlasthirdpartyintegrations"
 )
 
-type registerReconcilerFn func(ctrl.Manager) error
+type registerReconcilerFn func(ctrl.Manager, atlas.Provider, *zap.Logger, client.ObjectKey) error
 
 type experimentalReconcilerInitSpec struct {
 	name     string
@@ -30,8 +33,12 @@ type experimentalReconcilerInitSpec struct {
 var experimentalReconcilers = []experimentalReconcilerInitSpec{
 	{
 		name: "AtlasThirdPartyIntegrationsReconciler",
-		register: func(mgr ctrl.Manager) error {
-			return integrations.NewAtlasThirdPartyIntegrationsReconciler().SetupWithManager(mgr)
+		register: func(mgr ctrl.Manager, provider atlas.Provider, logger *zap.Logger, globalSecretRef client.ObjectKey) error {
+			return integrations.NewAtlasThirdPartyIntegrationsReconciler(
+				provider,
+				logger,
+				globalSecretRef,
+			).SetupWithManager(mgr)
 		},
 	},
 }