fix: add vector header

refactor: nits
tony-go · Feb 26, 2024 · 3ed6626 · 3ed6626
1 parent b9be595
commit 3ed6626
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/prompt/include/qupp/prompt/prompt.h b/src/prompt/include/qupp/prompt/prompt.h
@@ -4,6 +4,7 @@
 
 #include <optional>
 #include <string>
+#include <vector>
 
 namespace qupp::prompt {
 

diff --git a/src/prompt/unix/prompt.cc b/src/prompt/unix/prompt.cc
@@ -1,5 +1,4 @@
 #include "qupp/prompt/prompt.h"
-
 #include <iostream> // std::cout, std::endl
 #include <string>   // std::string
 
@@ -25,13 +24,14 @@ StringResult Prompt::select_from_list(const std::string &question,
   terminal_.enable_raw_mode();
 
   for (const auto &option : options) {
-    bool is_last_option = options.back() == option;
     bool is_first_option = options.front() == option;
     if (is_first_option) {
       terminal_.print_with_prefix("[*]", option, true);
     } else {
       terminal_.print_with_prefix("[ ]", option, true);
     }
+
+    bool is_last_option = options.back() == option;
     if (is_last_option) {
       // Move cursor back to the first option.
       terminal_.move_cursor_up(options.size());