From 3f82b6a47367bea746160ee26a03edb2df3d104b Mon Sep 17 00:00:00 2001
From: rcourtman <courtmanr@gmail.com>
Date: Wed, 8 Oct 2025 18:14:24 +0000
Subject: [PATCH] Include OIDC in exports and refresh guest metadata

---
 internal/config/export.go         | 45 ++++++++++++++++++++-----------
 internal/config/guest_metadata.go | 26 ++++++++++++++++++
 2 files changed, 55 insertions(+), 16 deletions(-)

diff --git a/internal/config/export.go b/internal/config/export.go
index c4b71cccb..41e1c7172 100644
--- a/internal/config/export.go
+++ b/internal/config/export.go
@@ -27,6 +27,7 @@ type ExportData struct {
 	Webhooks      []notifications.WebhookConfig `json:"webhooks"`
 	System        SystemSettings                `json:"system"`
 	GuestMetadata map[string]*GuestMetadata     `json:"guestMetadata,omitempty"`
+	OIDC          *OIDCConfig                   `json:"oidc,omitempty"`
 }
 
 // ExportConfig exports all configuration with passphrase-based encryption
@@ -63,6 +64,14 @@ func (c *ConfigPersistence) ExportConfig(passphrase string) (string, error) {
 	if err != nil {
 		return "", fmt.Errorf("failed to load system settings: %w", err)
 	}
+	if systemSettings == nil {
+		systemSettings = &SystemSettings{}
+	}
+
+	oidcConfig, err := c.LoadOIDCConfig()
+	if err != nil {
+		return "", fmt.Errorf("failed to load oidc configuration: %w", err)
+	}
 
 	// Load guest metadata (stored in data directory)
 	// Use PULSE_DATA_DIR if set, otherwise use /etc/pulse for backwards compatibility
@@ -83,6 +92,7 @@ func (c *ConfigPersistence) ExportConfig(passphrase string) (string, error) {
 		Webhooks:      webhooks,
 		System:        *systemSettings,
 		GuestMetadata: guestMetadata,
+		OIDC:          oidcConfig,
 	}
 
 	// Marshal to JSON
@@ -151,24 +161,27 @@ func (c *ConfigPersistence) ImportConfig(encryptedData string, passphrase string
 		return fmt.Errorf("failed to import system settings: %w", err)
 	}
 
-	// Import guest metadata if present
-	if exportData.GuestMetadata != nil && len(exportData.GuestMetadata) > 0 {
-		// Use PULSE_DATA_DIR if set, otherwise use /etc/pulse for backwards compatibility
-		dataPath := os.Getenv("PULSE_DATA_DIR")
-		if dataPath == "" {
-			dataPath = "/etc/pulse"
+	// Import OIDC configuration
+	if exportData.OIDC != nil {
+		if err := c.SaveOIDCConfig(*exportData.OIDC); err != nil {
+			return fmt.Errorf("failed to import oidc configuration: %w", err)
 		}
-		guestMetadataStore := NewGuestMetadataStore(dataPath)
+	} else {
+		// Remove existing OIDC config if backup did not include one
+		if err := os.Remove(c.oidcFile); err != nil && !os.IsNotExist(err) {
+			return fmt.Errorf("failed to remove existing oidc configuration: %w", err)
+		}
+	}
 
-		// Import each guest metadata entry
-		for guestID, metadata := range exportData.GuestMetadata {
-			if metadata != nil {
-				if err := guestMetadataStore.Set(guestID, metadata); err != nil {
-					// Log warning but don't fail entire import
-					fmt.Printf("Warning: Failed to import guest metadata for %s: %v\n", guestID, err)
-				}
-			}
-		}
+	// Import guest metadata if present
+	// Use PULSE_DATA_DIR if set, otherwise use /etc/pulse for backwards compatibility
+	dataPath := os.Getenv("PULSE_DATA_DIR")
+	if dataPath == "" {
+		dataPath = "/etc/pulse"
+	}
+	guestMetadataStore := NewGuestMetadataStore(dataPath)
+	if err := guestMetadataStore.ReplaceAll(exportData.GuestMetadata); err != nil {
+		fmt.Printf("Warning: Failed to import guest metadata: %v\n", err)
 	}
 
 	return nil
diff --git a/internal/config/guest_metadata.go b/internal/config/guest_metadata.go
index d746ff13d..60a12b5ef 100644
--- a/internal/config/guest_metadata.go
+++ b/internal/config/guest_metadata.go
@@ -91,6 +91,32 @@ func (s *GuestMetadataStore) Delete(guestID string) error {
 	return s.save()
 }
 
+// ReplaceAll replaces all metadata entries and persists them to disk.
+func (s *GuestMetadataStore) ReplaceAll(metadata map[string]*GuestMetadata) error {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+
+	s.metadata = make(map[string]*GuestMetadata)
+
+	if metadata != nil {
+		for guestID, meta := range metadata {
+			if meta == nil {
+				continue
+			}
+
+			clone := *meta
+			clone.ID = guestID
+			// Ensure slice copy is not nil to allow JSON marshalling of empty tags
+			if clone.Tags == nil {
+				clone.Tags = []string{}
+			}
+			s.metadata[guestID] = &clone
+		}
+	}
+
+	return s.save()
+}
+
 // Load reads metadata from disk
 func (s *GuestMetadataStore) Load() error {
 	filePath := filepath.Join(s.dataPath, "guest_metadata.json")