add checking of running container on each exporter host

relies on labels jumpstarter.version and jumpstarter.revision on image, or standard OCI labels
org.opencontainers.image.version and org.opencontainers.image.revision

Author: michalskrivanek

cmd/apply.go

@@ -60,6 +60,7 @@ var applyCmd = &cobra.Command{
 		}
 
 		config_lint.Validate(cfg)
+
 		tapplier, err := templating.NewTemplateApplier(cfg, nil)
 		if err != nil {
 			return fmt.Errorf("error creating template applier %w", err)

internal/config/config.go

@@ -4,15 +4,17 @@ import (
 	"os"
 	"path/filepath"
 
+	"github.com/jumpstarter-dev/jumpstarter-lab-config/internal/container"
 	"gopkg.in/yaml.v3"
 )
 
 // Config represents the structure of the jumpstarter-lab.yaml file.
 type Config struct {
-	Sources   Sources          `yaml:"sources"`
-	Variables []string         `yaml:"variables"`
-	BaseDir   string           `yaml:"-"` // Not serialized, set programmatically
-	Loaded    *LoadedLabConfig `yaml:"-"` // Not serialized, used internally
+	Sources           Sources                           `yaml:"sources"`
+	Variables         []string                          `yaml:"variables"`
+	BaseDir           string                            `yaml:"-"` // Not serialized, set programmatically
+	Loaded            *LoadedLabConfig                  `yaml:"-"` // Not serialized, used internally
+	ContainerVersions map[string]*container.ImageLabels `yaml:"-"` // Not serialized, container versions by image URL
 }
 
 // Sources defines the paths for various configuration files.
@@ -43,6 +45,9 @@ func LoadConfig(filePath string, vaultPassFile string) (*Config, error) {
 	cfg.BaseDir = filepath.Dir(filePath)
 
 	cfg.Loaded, err = LoadAllResources(&cfg, vaultPassFile)
+	if err != nil {
+		return nil, err
+	}
 
-	return &cfg, err
+	return &cfg, nil
 }

internal/config/loader.go

@@ -9,6 +9,7 @@ import (
 
 	jsApi "github.com/jumpstarter-dev/jumpstarter-controller/api/v1alpha1"
 	api "github.com/jumpstarter-dev/jumpstarter-lab-config/api/v1alpha1"
+	"github.com/jumpstarter-dev/jumpstarter-lab-config/internal/container"
 	"github.com/jumpstarter-dev/jumpstarter-lab-config/internal/vars"
 
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@@ -305,9 +306,47 @@ func LoadAllResources(cfg *Config, vaultPassFile string) (*LoadedLabConfig, erro
 		}
 	}
 
+	// Retrieve container versions for all unique container images found in exporters
+	containerVersions := retrieveContainerVersionsFromExporters(loaded)
+
+	// Store the container versions in the config
+	cfg.ContainerVersions = containerVersions
+
 	return loaded, nil
 }
 
+// retrieveContainerVersionsFromExporters retrieves container versions for all unique container images found in exporters
+func retrieveContainerVersionsFromExporters(loaded *LoadedLabConfig) map[string]*container.ImageLabels {
+	containerVersions := make(map[string]*container.ImageLabels)
+	uniqueImages := make(map[string]bool)
+
+	// Collect all unique container images from exporter config templates
+	for _, template := range loaded.ExporterConfigTemplates {
+		if template.Spec.ContainerImage != "" {
+			uniqueImages[template.Spec.ContainerImage] = true
+		}
+	}
+
+	// Retrieve version information for each unique image
+	for imageURL := range uniqueImages {
+		fmt.Printf("🔍 Checking container version for %s...\n", imageURL)
+
+		imageLabels, err := container.GetImageLabelsFromRegistry(imageURL)
+		if err != nil {
+			fmt.Printf("Latest container version of %s: unavailable (%v)\n", imageURL, err)
+			containerVersions[imageURL] = &container.ImageLabels{} // Store empty labels
+		} else if imageLabels.IsEmpty() {
+			fmt.Printf("Latest container version of %s: no version info available\n", imageURL)
+			containerVersions[imageURL] = imageLabels
+		} else {
+			fmt.Printf("Latest container version of %s: %s %s\n", imageURL, imageLabels.Version, imageLabels.Revision)
+			containerVersions[imageURL] = imageLabels
+		}
+	}
+
+	return containerVersions
+}
+
 func ReportLoading(cfg *Config) {
 
 	fmt.Println("Reading files from:")

internal/exporter/ssh/ssh.go

@@ -14,6 +14,7 @@ import (
 
 	"github.com/google/go-cmp/cmp"
 	"github.com/jumpstarter-dev/jumpstarter-lab-config/api/v1alpha1"
+	"github.com/jumpstarter-dev/jumpstarter-lab-config/internal/container"
 	"github.com/pkg/sftp"
 	"golang.org/x/crypto/ssh"
 	"golang.org/x/crypto/ssh/agent"
@@ -29,6 +30,10 @@ const (
 	BOOTC_NOT_MANAGED
 )
 
+const (
+	noValuePlaceholder = "<no value>"
+)
+
 type HostManager interface {
 	Status() (string, error)
 	NeedsUpdate() (bool, error)
@@ -186,12 +191,12 @@ func (m *SSHHostManager) Apply(exporterConfig *v1alpha1.ExporterConfigTemplate,
 		if !dryRun {
 			_, enableErr := m.runCommand("systemctl restart " + fmt.Sprintf("%q", serviceName))
 			if enableErr != nil {
-				fmt.Printf("            ❌ Failed to start service %s: %v\n", serviceName, enableErr)
+				fmt.Printf("        ❌ Failed to start service %s: %v\n", serviceName, enableErr)
 			} else {
-				fmt.Printf("            ✅ Service %s started\n", serviceName)
+				fmt.Printf("        ✅ Service %s started\n", serviceName)
 			}
 		} else {
-			fmt.Printf("            📄 Would restart service %s\n", serviceName)
+			fmt.Printf("        📄 Would restart service %s\n", serviceName)
 		}
 	}
 
@@ -217,9 +222,9 @@ func (m *SSHHostManager) Apply(exporterConfig *v1alpha1.ExporterConfigTemplate,
 
 	if m.GetBootcStatus() == BOOTC_UPDATING {
 		if dryRun {
-			fmt.Printf("            📄 Bootc upgrade in progress, would skip exporter service restarts/container updates\n")
+			fmt.Printf("        📄 Bootc upgrade in progress, would skip exporter service restarts/container updates\n")
 		} else {
-			fmt.Printf("            ⚠️ Bootc upgrade in progress, skipping exporter service restarts/container updates\n")
+			fmt.Printf("        ⚠️ Bootc upgrade in progress, skipping exporter service restarts/container updates\n")
 			return nil
 		}
 	}
@@ -245,39 +250,130 @@ func (m *SSHHostManager) Apply(exporterConfig *v1alpha1.ExporterConfigTemplate,
 	} else {
 		// Check if service is running and start if needed
 		statusResult, err := m.runCommand("systemctl is-active " + fmt.Sprintf("%q", svcName))
-		if err != nil || statusResult.Stdout != "active\n" {
-			fmt.Printf("            ⚠️ Service %s is not running...\n", svcName)
+		serviceRunning := err == nil && statusResult.Stdout == "active\n"
+
+		if !serviceRunning {
+			fmt.Printf("        ⚠️ Service %s is not running...\n", svcName)
 			restartService(svcName, dryRun)
+		} else {
+			// Only check container version if service is running
+			err = m.checkContainerVersion(exporterConfig, svcName, dryRun, restartService)
+			if err != nil {
+				return fmt.Errorf("container version check failed: %w", err)
+			}
 		}
+	}
 
-		// Check if container needs updating using podman auto-update (if podman exists)
-		autoUpdateResult, err := m.runCommand(fmt.Sprintf("command -v podman >/dev/null 2>&1 && podman auto-update --dry-run --format json | jq -r '.[] | select(.ContainerName == %q) | .Updated'", svcName))
-		if err != nil {
-			fmt.Printf("            ℹ️ Podman not available, skipping auto-update check\n")
+	return nil
+}
+
+// checkContainerVersion checks if container needs updating using detailed version comparison
+func (m *SSHHostManager) checkContainerVersion(exporterConfig *v1alpha1.ExporterConfigTemplate, svcName string, dryRun bool, restartService func(string, bool)) error {
+	// Only check version for container-based exporters
+	if exporterConfig.Spec.SystemdContainerTemplate == "" {
+		return nil
+	}
+
+	// Check detailed version comparison (if we have container image info)
+	if exporterConfig.Spec.ContainerImage != "" {
+		return m.checkDetailedContainerVersion(exporterConfig.Spec.ContainerImage, svcName, dryRun, restartService)
+	}
+
+	// No container image specified, nothing to check
+	return nil
+}
+
+// checkDetailedContainerVersion performs detailed version comparison using skopeo and podman inspect
+func (m *SSHHostManager) checkDetailedContainerVersion(containerImage, svcName string, dryRun bool, restartService func(string, bool)) error {
+	// Get expected version from registry
+	expectedLabels, err := container.GetImageLabelsFromRegistry(containerImage)
+	if err != nil {
+		fmt.Printf("        ⚠️ Could not check container version: %v\n", err)
+		return nil // Don't fail the entire operation, just skip version check
+	}
+
+	if expectedLabels.IsEmpty() {
+		fmt.Printf("        ℹ️ No version info available for image %s\n", containerImage)
+		return nil // Don't fail, just skip version check
+	}
+
+	// Get running container version
+	runningLabels, err := m.getRunningContainerLabels(svcName)
+	if err != nil {
+		fmt.Printf("        ⚠️ Could not check running container version: %v\n", err)
+		return nil // Container might not be running yet, which is fine
+	}
+
+	// Compare versions
+	if expectedLabels.Matches(runningLabels) {
+		if dryRun {
+			fmt.Printf("        ✅ Exporter container image running latest version\n")
+		}
+		// In non-dry-run mode, print nothing for matching versions as requested
+	} else {
+		if dryRun {
+			fmt.Printf("        🔄 Would restart service for container update (running: %s, latest: %s)\n",
+				runningLabels.String(), expectedLabels.String())
 		} else {
-			updatedOutput := strings.TrimSpace(autoUpdateResult.Stdout)
-			switch updatedOutput {
-			case "":
-				fmt.Printf("            ⚠️ Container %s not found in auto-update check\n", svcName)
-			case "false":
-				if dryRun {
-					fmt.Printf("            ✅ Exporter container image is up to date\n")
-				}
-			case "pending":
-				if dryRun {
-					fmt.Printf("            📄 Would update container %s\n", svcName)
-				} else {
-					restartService(svcName, dryRun)
-				}
-			default:
-				return fmt.Errorf("unexpected auto-update result: %s", updatedOutput)
-			}
+			fmt.Printf("        🔄 Restarting service for container update (running: %s, latest: %s)\n",
+				runningLabels.String(), expectedLabels.String())
+			restartService(svcName, dryRun)
 		}
 	}
 
 	return nil
 }
 
+// getRunningContainerLabels gets container labels from running container
+func (m *SSHHostManager) getRunningContainerLabels(serviceName string) (*container.ImageLabels, error) {
+	// Try jumpstarter labels first, then fall back to OCI standard labels
+	result, err := m.runCommand(fmt.Sprintf("podman inspect --format '{{index .Config.Labels \"jumpstarter.version\"}} {{index .Config.Labels \"jumpstarter.revision\"}} {{index .Config.Labels \"org.opencontainers.image.version\"}} {{index .Config.Labels \"org.opencontainers.image.revision\"}}' %s", serviceName))
+	if err != nil {
+		return nil, fmt.Errorf("failed to inspect container %s: %w", serviceName, err)
+	}
+
+	parts := strings.Fields(strings.TrimSpace(result.Stdout))
+	// Pad with empty strings if we got fewer parts
+	for len(parts) < 4 {
+		parts = append(parts, "")
+	}
+
+	jumpstarterVersion := parts[0]  // jumpstarter.version
+	jumpstarterRevision := parts[1] // jumpstarter.revision
+	ociVersion := parts[2]          // org.opencontainers.image.version
+	ociRevision := parts[3]         // org.opencontainers.image.revision
+
+	// Clean up "<no value>" to empty string
+	if jumpstarterVersion == noValuePlaceholder {
+		jumpstarterVersion = ""
+	}
+	if jumpstarterRevision == noValuePlaceholder {
+		jumpstarterRevision = ""
+	}
+	if ociVersion == noValuePlaceholder {
+		ociVersion = ""
+	}
+	if ociRevision == noValuePlaceholder {
+		ociRevision = ""
+	}
+
+	// Use jumpstarter labels if both exist, otherwise fall back to OCI labels
+	var version, revision string
+	if jumpstarterVersion != "" && jumpstarterRevision != "" {
+		version = jumpstarterVersion
+		revision = jumpstarterRevision
+	} else {
+		version = ociVersion
+		revision = ociRevision
+	}
+
+	// Return labels even if only partially available (version OR revision can be empty)
+	return &container.ImageLabels{
+		Version:  version,
+		Revision: revision,
+	}, nil
+}
+
 // RunHostCommand implements the HostManager interface by exposing runCommand
 func (m *SSHHostManager) RunHostCommand(command string) (*CommandResult, error) {
 	return m.runCommand(command)