change PS1 to show that atmos is in the atmos terraform shell mode (#…

…761) * add atmos in PS1 Signed-off-by: Pulak Kanti Bhowmick <[email protected]> * add review suggestions Signed-off-by: Pulak Kanti Bhowmick <[email protected]> * Update internal/exec/shell_utils.go Co-authored-by: Erik Osterman (CEO @ Cloud Posse) <[email protected]> * fix for zsh Signed-off-by: Pulak Kanti Bhowmick <[email protected]> * compare only shell name Signed-off-by: Pulak Kanti Bhowmick <[email protected]> --------- Signed-off-by: Pulak Kanti Bhowmick <[email protected]> Co-authored-by: Erik Osterman (CEO @ Cloud Posse) <[email protected]> Co-authored-by: Andriy Knysh <[email protected]>
cloudposse · Nov 11, 2024 · d502eb0 · d502eb0
1 parent 1250929
commit d502eb0
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/internal/exec/shell_utils.go b/internal/exec/shell_utils.go
@@ -7,6 +7,7 @@ import (
 	"io"
 	"os"
 	"os/exec"
+	"path/filepath"
 	"runtime"
 	"strings"
 
@@ -154,6 +155,7 @@ func execTerraformShellCommand(
 	componentEnvList = append(componentEnvList, fmt.Sprintf("TF_CLI_ARGS_import=-var-file=%s", varFile))
 	componentEnvList = append(componentEnvList, fmt.Sprintf("TF_CLI_ARGS_destroy=-var-file=%s", varFile))
 	componentEnvList = append(componentEnvList, fmt.Sprintf("TF_CLI_ARGS_console=-var-file=%s", varFile))
+	componentEnvList = append(componentEnvList, fmt.Sprintf("PS1=atmos:%s/%s> ", stack, component))
 
 	u.LogDebug(cliConfig, "\nStarting a new interactive shell where you can execute all native Terraform commands (type 'exit' to go back)")
 	u.LogDebug(cliConfig, fmt.Sprintf("Component: %s\n", component))
@@ -183,11 +185,21 @@ func execTerraformShellCommand(
 		if len(shellCommand) == 0 {
 			bashPath, err := exec.LookPath("bash")
 			if err != nil {
-				return err
+				// Try fallback to sh if bash is not available
+				shPath, shErr := exec.LookPath("sh")
+				if shErr != nil {
+					return fmt.Errorf("no suitable shell found: %v", shErr)
+				}
+				shellCommand = shPath
+			} else {
+				shellCommand = bashPath
 			}
-			shellCommand = bashPath
 		}
-		shellCommand = shellCommand + " -l"
+
+		shellName := filepath.Base(shellCommand)
+		if shellName == "zsh" {
+			shellCommand = shellCommand + " -d -f -i"
+		}
 	}
 
 	u.LogDebug(cliConfig, fmt.Sprintf("Starting process: %s\n", shellCommand))