From 4f1112f838fbd4b17e9d1e77cd36c72e32d0ae7f Mon Sep 17 00:00:00 2001
From: aozima <wuyangyong@gmail.com>
Date: Mon, 2 Jul 2018 16:23:00 +0800
Subject: [shell] add finsh_set_prompt().

---
 components/finsh/shell.c | 35 ++++++++++++++++++++++++++++++++---
 components/finsh/shell.h |  6 ++----
 2 files changed, 34 insertions(+), 7 deletions(-)

(limited to 'components')

diff --git a/components/finsh/shell.c b/components/finsh/shell.c
index 667e7ac12a..0109a761bc 100644
--- a/components/finsh/shell.c
+++ b/components/finsh/shell.c
@@ -56,11 +56,35 @@ ALIGN(RT_ALIGN_SIZE)
 static char finsh_thread_stack[FINSH_THREAD_STACK_SIZE];
 #endif
 struct finsh_shell *shell;
+static char *finsh_prompt_custom = RT_NULL;
+
+#ifdef RT_USING_HEAP
+int finsh_set_prompt(const char * prompt)
+{
+    if(finsh_prompt_custom)
+    {
+        rt_free(finsh_prompt_custom);
+        finsh_prompt_custom = RT_NULL;
+    }
+
+    /* strdup */
+    if(prompt)
+    {
+        finsh_prompt_custom = rt_malloc(strlen(prompt)+1);
+        if(finsh_prompt_custom)
+        {
+            strcpy(finsh_prompt_custom, prompt);
+        }
+    }
+
+    return 0;
+}
+#endif /* RT_USING_HEAP */
 
-#if defined(FINSH_USING_MSH) || (defined(RT_USING_DFS) && defined(DFS_USING_WORKDIR))
 #if defined(RT_USING_DFS)
 #include <dfs_posix.h>
-#endif
+#endif /* RT_USING_DFS */
+
 const char *finsh_get_prompt()
 {
 #define _MSH_PROMPT "msh "
@@ -74,6 +98,12 @@ const char *finsh_get_prompt()
         return finsh_prompt;
     }
 
+    if(finsh_prompt_custom)
+    {
+        strncpy(finsh_prompt, finsh_prompt_custom, sizeof(finsh_prompt)-1);
+        return finsh_prompt;
+    }
+
 #ifdef FINSH_USING_MSH
     if (msh_is_used()) strcpy(finsh_prompt, _MSH_PROMPT);
     else
@@ -89,7 +119,6 @@ const char *finsh_get_prompt()
 
     return finsh_prompt;
 }
-#endif
 
 /**
  * @ingroup finsh
diff --git a/components/finsh/shell.h b/components/finsh/shell.h
index e01458adc5..d0fd66ee78 100644
--- a/components/finsh/shell.h
+++ b/components/finsh/shell.h
@@ -44,12 +44,10 @@
 #endif
 
 #define FINSH_OPTION_ECHO   0x01
-#if defined(FINSH_USING_MSH) || (defined(RT_USING_DFS) && defined(DFS_USING_WORKDIR))
+
 #define FINSH_PROMPT        finsh_get_prompt()
 const char* finsh_get_prompt(void);
-#else
-#define FINSH_PROMPT        "finsh>>"
-#endif
+int finsh_set_prompt(const char * prompt);
 
 #ifdef FINSH_USING_HISTORY
     #ifndef FINSH_HISTORY_LINES
-- 
cgit v1.3.1