Changes in src/UIElements/TextUI/Query/FileTextQuery.cpp [8df74d:d754bb]

File:

• src/UIElements/TextUI/Query/FileTextQuery.cpp (modified) (2 diffs)

src/UIElements/TextUI/Query/FileTextQuery.cpp ¶

@@ -21 +21 @@
 
 #include <iostream>
+#include <string>
 
 #include <boost/filesystem.hpp>
 
-#include "TextUI/TextDialog.hpp"
+#include "TextUI/Query/TextQuery.hpp"
 
 #include "Helpers/Log.hpp"
@@ -37 +38 @@
 
 bool TextDialog::FileTextQuery::handle() {
-  Log() << Verbose(0) << getTitle();
+  bool badInput = false;
   std::string tempstring;
-  getline(cin,tempstring);
+  do{
+    badInput = false;
+    std::cout << getDescription() << ": ";
+    std::cin >> tempstring;
+    if(std::cin.fail()){
+      badInput = true;
+      std::cin.clear();
+      std::cin.ignore(std::numeric_limits<streamsize>::max(),'\n');
+      std::cout << "Input was not a number!" << std::endl;
+      continue;
+    }
+  } while(badInput);
+  std::cin.ignore(std::numeric_limits<streamsize>::max(),'\n');
   tmp = tempstring;
   return true;