Fixed sub-menu path in prompt

rust-embedded-community · Aug 30, 2024 · a7c5623 · a7c5623
1 parent 01e6f16
commit a7c5623
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -281,10 +281,15 @@ impl<'a, I, T> PromptIter<'a, I, T> {
         let state = if newline {
             PromptIterState::Newline
         } else {
-            PromptIterState::Menu(0)
+            Self::first_menu()
         };
         Self { menu_mgr, state }
     }
+
+    const fn first_menu() -> PromptIterState {
+        // Skip the first menu level which is root
+        PromptIterState::Menu(1)
+    }
 }
 
 impl<'a, I, T> Iterator for PromptIter<'a, I, T> {
@@ -294,11 +299,11 @@ impl<'a, I, T> Iterator for PromptIter<'a, I, T> {
         loop {
             match self.state {
                 PromptIterState::Newline => {
-                    self.state = PromptIterState::Menu(0);
+                    self.state = Self::first_menu();
                     break Some("\n");
                 }
                 PromptIterState::Menu(i) => {
-                    if i >= self.menu_mgr.depth() {
+                    if i > self.menu_mgr.depth() {
                         self.state = PromptIterState::Arrow;
                     } else {
                         let menu = self.menu_mgr.get_menu(Some(i));