common: use get_nand_dev_by_index()

[oweals/u-boot.git] / common / cli_hush.c

diff --git a/common/cli_hush.c b/common/cli_hush.c
index cbaf22e9129a7078a0912067e92c536b84eadd87..00861e2d9e4b56c114a4dcae71fdbb0f89a996aa 100644 (file)
--- a/common/cli_hush.c
+++ b/common/cli_hush.c
@@ -978,12 +978,22 @@ static inline void setup_prompt_string(int promptmode, char **prompt_str)
 static int uboot_cli_readline(struct in_str *i)
 {
        char *prompt;
+       char __maybe_unused *ps_prompt = NULL;
 
        if (i->promptmode == 1)
                prompt = CONFIG_SYS_PROMPT;
        else
                prompt = CONFIG_SYS_PROMPT_HUSH_PS2;
 
+#ifdef CONFIG_CMDLINE_PS_SUPPORT
+       if (i->promptmode == 1)
+               ps_prompt = getenv("PS1");
+       else
+               ps_prompt = getenv("PS2");
+       if (ps_prompt)
+               prompt = ps_prompt;
+#endif
+
        return cli_readline(prompt);
 }
 #endif