util: introduce ask_(string,num) helpers

quick and dirty stdio dialog helpers

2 files changed, 40 insertions, 0 deletions

diff --git a/util.c b/util.c
index c07589e..0e5825a 100644
--- a/util.c
+++ b/util.c
@@ -1,5 +1,6 @@
 #include <limits.h>
 #include <stdio.h>
+#include <string.h>
 #include <unistd.h>
 
 #include "util.h"
@@ -20,3 +21,40 @@ unsigned get_ncpus(void)
 
 	return n == 0 ? 1 : n;
 }
+
+
+static void query(const char *prompt, const char *def, char *buf, int len)
+{
+	buf[0] = '\0';
+
+	printf("%s [%s]: ", prompt, def);
+	fflush(stdout);
+
+	fgets(buf, len, stdin);
+	if (buf[0] == '\0' || buf[0] == '\n') {
+		snprintf(buf, len, "%s", def);
+	} else if(strchr(buf, '\n')) {
+		*strchr(buf, '\n') = '\0';
+	}
+}
+
+
+void ask_string(char *buf, int len, const char *prompt, const char *def)
+{
+	query(prompt, def, buf, len);
+}
+
+
+void ask_num(int *res, int max, const char *prompt, int def)
+{
+	char	buf[21], buf2[256];
+	int	num;
+
+	snprintf(buf, sizeof(buf), "%i", def);
+	do {
+		query(prompt, buf, buf2, sizeof(buf2));
+		num = atoi(buf2); /* TODO: errors (strtol)*/
+	} while (num > max);
+
+	*res = num;
+}
diff --git a/util.h b/util.h
index c82b623..549b529 100644
--- a/util.h
+++ b/util.h
@@ -22,5 +22,7 @@
 	(sizeof(_str) - 1)
 
 unsigned get_ncpus(void);
+void ask_string(char *buf, int len, const char *prompt, const char *def);
+void ask_num(int *res, int max, const char *prompt, int def);
 
 #endif /* _UTIL_H */