diff --git a/main.cpp b/main.cpp index d623a60..42ff231 100644 --- a/main.cpp +++ b/main.cpp @@ -134,6 +134,14 @@ count_type get_number(string prompt) { print_input(prompt+": " + s); ch = getch(); } + + // Check if input is a number + for(int i = 0; i < s.size(); i++) + if(s[i] < '0' || s[i] > '9') + return file_offset + cur.r; + if(s.size() == 0) + return file_offset + cur.r; + return stoi(s); } // Taking user input - string