From 349ae02d9201dffcca98cdba270091d6dfe2d114 Mon Sep 17 00:00:00 2001 From: Vito Caputo Date: Tue, 13 Dec 2016 07:26:39 -0800 Subject: util: introduce ask_(string,num) helpers quick and dirty stdio dialog helpers --- util.c | 38 ++++++++++++++++++++++++++++++++++++++ util.h | 2 ++ 2 files changed, 40 insertions(+) diff --git a/util.c b/util.c index c07589e..0e5825a 100644 --- a/util.c +++ b/util.c @@ -1,5 +1,6 @@ #include #include +#include #include #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 */ -- cgit v1.2.3