Now removing NoResults from the prompt

jline · Dec 28, 2024 · 125be8c · 125be8c
1 parent 5d7543e
commit 125be8c
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/console-ui/src/main/java/org/jline/consoleui/prompt/ConsolePrompt.java b/console-ui/src/main/java/org/jline/consoleui/prompt/ConsolePrompt.java
@@ -130,7 +130,7 @@ public Map<String, PromptResultItemIF> prompt(
         try {
             Map<String, PromptResultItemIF> resultMap = new HashMap<>();
             prompt(header, promptableElementList, resultMap);
-            return resultMap;
+            return removeNoResults(resultMap);
         } finally {
             close();
         }
@@ -203,7 +203,7 @@ public Map<String, PromptResultItemIF> prompt(
                     peResult = new HashMap<>();
                 }
             }
-            return resultMap;
+            return removeNoResults(resultMap);
         } finally {
             // Restore the original state of cancellable
             config.setCancellableFirstPrompt(cancellable);
@@ -403,6 +403,12 @@ private void removePreviousResult(PromptableElementIF pe) {
         }
     }
 
+    private Map<String, PromptResultItemIF> removeNoResults(Map<String, PromptResultItemIF> resultMap) {
+        return resultMap.entrySet().stream()
+                .filter(e -> !(e.getValue() instanceof NoResult))
+                .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
+    }
+
     /**
      * Creates a {@link PromptBuilder}.
      *