Support local images in deploy command

Author: Moritz Clasmeier

cmd/deploy.go

@@ -29,7 +29,8 @@ import (
 )
 
 var (
-	sharedNamespace = "stackrox"
+	sharedNamespace     = "stackrox"
+	imagePreLoadCommand string
 )
 
 func newDeployCmd(settings *deployer.Config) *cobra.Command {
@@ -51,6 +52,8 @@ Examples:
 	cmd.Flags().StringVar(&shell, "shell", "", "Shell to spawn after Central deployment")
 
 	cmd.Flags().StringVar(&envrc, "envrc", "", "Write environment to file instead of spawning sub-shell")
+	cmd.Flags().StringVar(&imagePreLoadCommand, "image-preload-command", "",
+		"Use custom command for pre-loading images to local cluster. Image can be referenced as $IMAGE.")
 
 	registerFlag(cmd, settings, "olm", "Deploy operator via OLM (requires OLM installed)",
 		withNoOptDefVal("true"),
@@ -321,6 +324,31 @@ func runDeploy(cmd *cobra.Command, args []string) error {
 	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Minute)
 	defer cancel()
 
+	// If we are deploying to a local cluster and the images exist locally, then we transfer them
+	// to the local cluster.
+	if deploySettings.Roxie.ClusterType.IsLocal() && !deploySettings.Roxie.KonfluxImages {
+		var preLoader deployer.ImagePreLoader
+		if imagePreLoadCommand != "" {
+			preLoader = deployer.NewCustomImagePreloader(ctx, log, imagePreLoadCommand)
+		} else {
+			preLoader, err = d.GetPreLoaderForCluster()
+			if err != nil && !errors.Is(err, deployer.ErrLocalImagesUnsupported) {
+				return fmt.Errorf("obtaining image preloader for cluster: %w", err)
+			}
+		}
+		if preLoader == nil {
+			log.Warningf("Image preloading not supported for cluster %s.", d.GetKubeContext())
+			log.Warningf("Use --image-preload-command for specifying custom image preloading mechanism.")
+		} else {
+			log.Dimf("Using image pre-loader %q", preLoader.Name())
+
+			if err := d.TryTransferLocalImages(ctx, preLoader); err != nil {
+				// Best effort, keep running.
+				log.Warningf("Transferring images to local cluster failed: %v", err)
+			}
+		}
+	}
+
 	if components.IncludesCentral() {
 		d.PrintCentralDeploymentSummary()
 	}
@@ -380,11 +408,15 @@ func configureConfig(log *logger.Logger, components component.Component, deployS
 		profile := clusterdefaults.ResolveAutoResourceProfile(clusterType)
 		log.Dimf("Selecting resource profile %v for Central", profile)
 		deploySettings.Central.ResourceProfile = profile
+	} else if components.IncludesCentral() && deploySettings.Roxie.ClusterType.IsLocal() {
+		log.Warning("You are deploying Central to a local cluster, it is recommended to use --resources=auto.")
 	}
 	if deploySettings.SecuredCluster.ResourceProfile == types.ResourceProfileAuto {
 		profile := clusterdefaults.ResolveAutoResourceProfile(clusterType)
 		log.Dimf("Selecting resource profile %v for SecuredCluster", profile)
 		deploySettings.SecuredCluster.ResourceProfile = profile
+	} else if components.IncludesSensor() && deploySettings.Roxie.ClusterType.IsLocal() {
+		log.Warning("You are deploying SecuredCluster to a local cluster, it is recommended to use --resources=auto.")
 	}
 
 	// We need to do this regardless of whether the operator is deployed or not, because