more knecht

knecht/cmd/update.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/jensbecker/homelab/knecht/portainer"
 	"github.com/jensbecker/homelab/knecht/stack"
+	"github.com/jensbecker/homelab/knecht/tui"
 	"github.com/spf13/cobra"
 )
 
@@ -37,15 +38,19 @@ var updateCmd = &cobra.Command{
 			return err
 		}
 
-		// Check for missing env keys and prompt for values
 		exampleKeys, err := local.EnvExampleKeys()
 		if err != nil {
 			return err
 		}
-		env, err := mergeEnvVars(remote.Env, exampleKeys)
+
+		env, err := mergeEnvVars(name, remote.Env, exampleKeys)
 		if err != nil {
 			return err
 		}
+		if env == nil {
+			fmt.Println("Cancelled.")
+			return nil
+		}
 
 		s, err := client.UpdateStack(remote.ID, compose, env)
 		if err != nil {
@@ -56,20 +61,32 @@ var updateCmd = &cobra.Command{
 	},
 }
 
-// mergeEnvVars takes existing Portainer env vars and prompts for any keys
-// present in .env.example but missing from Portainer.
-func mergeEnvVars(existing []portainer.EnvVar, exampleKeys []string) ([]portainer.EnvVar, error) {
+// mergeEnvVars preserves existing Portainer env vars and prompts via TUI for
+// any keys present in .env.example but missing from Portainer.
+// Returns nil if the user cancelled the form.
+func mergeEnvVars(stackName string, existing []portainer.EnvVar, exampleKeys []string) ([]portainer.EnvVar, error) {
 	envMap := make(map[string]string, len(existing))
 	for _, e := range existing {
 		envMap[e.Name] = e.Value
 	}
 
+	var missing []string
 	for _, key := range exampleKeys {
 		if _, ok := envMap[key]; !ok {
-			fmt.Printf("Missing env var %q — enter value (leave empty to skip): ", key)
-			var val string
-			fmt.Scanln(&val)
-			envMap[key] = val
+			missing = append(missing, key)
+		}
+	}
+
+	if len(missing) > 0 {
+		values, err := tui.PromptMissingEnv(stackName, missing)
+		if err != nil {
+			return nil, err
+		}
+		if values == nil {
+			return nil, nil // cancelled
+		}
+		for k, v := range values {
+			envMap[k] = v
 		}
 	}