Add range based prompt functions

speelbarrow · Jun 20, 2023 · 634b59c · 634b59c
1 parent 123ab87
commit 634b59c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/include/amadeus/prompt.h b/include/amadeus/prompt.h
@@ -2,5 +2,8 @@
 #define PROMPT_H
 
 int prompt(const char *message);
+int prompt_min(const char *message, int min);
+int prompt_max(const char *message, int max);
+int prompt_range(const char *message, int min, int max);
 
 #endif // PROMPT_H
diff --git a/src/prompt.c b/src/prompt.c
@@ -2,14 +2,9 @@
 
 #include <stdio.h>
 #include <stdlib.h>
+#include <math.h>
 
-int prompt_counted(const char *message, int count);
-
-int prompt(const char *message) {
-    return prompt_counted(message, 0);
-}
-
-int prompt_counted(const char *message, int count) {
+int prompt_counted(const char *message, int min, int max, int count) {
     printf("%s: ", message);
 
     char *buffer = NULL;
@@ -21,9 +16,15 @@ int prompt_counted(const char *message, int count) {
         return result;
     } else if (++count < 3) {
         printf("Invalid input. Please try again.\n");
-        return prompt_counted(message, count);
+        return prompt_counted(message, min, max, count);
     } else {
         printf("Too many invalid inputs. Exiting.\n");
         exit(1);
     }
 }
+int prompt_counted(const char *message, int min, int max, int count);
+
+int prompt(const char *message) { return prompt_range(message, -INFINITY, INFINITY); }
+int prompt_min(const char *message, int min) { return prompt_range(message, min, INFINITY); }
+int prompt_max(const char *message, int max) { return prompt_range(message, -INFINITY, max); }
+int prompt_range(const char *message, int min, int max) { return prompt_counted(message, min, max, 0); }