Small refactor of the forground color code

modified:   README.md
	modified:   src/main.c
	modified:   src/nms.c
	modified:   src/nms.h

README.md

@@ -70,7 +70,8 @@ eliminating the need for the user to press a key to start it.
 ls -l / | nms -a
 ```
 
-Use the `-f` option to set foreground color to either white, yellow, black, magenta, blue, green, or red - this is blue by default.
+Use the `-f` option to set foreground color to either white, yellow, black, magenta, blue, green, or
+red - this is blue by default.
 ```
 ls -l / | nms -f green
 ```
@@ -129,19 +130,18 @@ Here is how the structure is defined:
 ```
 typedef struct {
     char *src;
+    char *foreground_color;
     char *return_opts;
     int input_cursor_x;
     int input_cursor_y;
     bool show_cursor;
     bool auto_decrypt;
-    int foreground_color;
 } NmsArgs;
 ```
 * `char *src`
   * Pointer to the string of characters on which to perform the effect.
-
-Useful for displaying menus:
-
+* `char *foreground_color`
+   * Pointer to a string containing the desired foreground color: white, yellow, black, magenta, blue, green, red.
 * `char *return_opts`
   * String pointer containing only the character options that the user must choose from once the src characters are revealed. For example, if you are showing a menu with six options, this string might be "123456". The user will have to choose one of these characters before execution is handed back to the calling function. Note that the character selected is returned by `nms_exec()`;
 * `int input_cursor_x` and `int input_cursor_y`
@@ -150,8 +150,6 @@ Useful for displaying menus:
   * Set to `true` if you want the cursor to be visible during the text decryption effect. It is set to `false` by default.
 * `bool auto_decrypt`
   * Set to `true` to automatically start the decryption effect, eliminating the need for the user to press a key to start it.
- * `int foreground_color`
-   * Use this to assign a default (ncurses) color to the foreground
 
 Assign values to the structure members as needed. Then simply pass a pointer to the structure to the
 nms_exec() function:

src/main.c

@@ -16,8 +16,8 @@ int main(int argc, char *argv[]) {
 	while ((o = getopt(argc, argv, "f:av")) != -1) {
 		switch (o) {
 			case 'f':
-                args.foreground_color = getColorByName(optarg, args.foreground_color);
-                break;
+				args.foreground_color = optarg;
+				break;
 			case 'a':
                 args.auto_decrypt = true;
                 break;

src/nms.c

@@ -39,6 +39,7 @@ struct winpos {
 
 // Function prototypes (internal)
 char getMaskChar(void);
+int getColorByName(char *);
 
 /*
  * void nms_exec(NmsArgs *)
@@ -86,7 +87,7 @@ char nms_exec(NmsArgs *args) {
 	// Setting up and starting colors if terminal supports them
 	if (has_colors()) {
 		start_color();
-		init_pair(1, args->foreground_color, COLOR_BLACK);
+		init_pair(1, getColorByName(args->foreground_color), COLOR_BLACK);
 	}
 
 	// Get terminal window size
@@ -288,11 +289,19 @@ char getMaskChar(void) {
 /*
  * char getColorByName(char *string, int fallback)
  *
+ * ARGS:
+ *
+ * char *string - new color setting (white, yellow, black, magenta, blue, green, red)
+ *
  * DESCR:
  * Returns an ncurses color by its name.
  *
  */
-int getColorByName(char *string, int fallback) {
+int getColorByName(char *string) {
+
+	if(string == NULL) {
+		return COLOR_BLUE;
+	}
 
 	if(strcmp("white", string) == 0) {
 		return COLOR_WHITE;
@@ -322,5 +331,5 @@ int getColorByName(char *string, int fallback) {
 		return COLOR_RED;
 	}
 
-	return fallback;
+	return COLOR_BLUE;
 }

src/nms.h

@@ -3,25 +3,22 @@
 
 #include <stddef.h>
 #include <stdbool.h>
-#include <ncurses.h>
 
 // Default arguments for nms_exec()
-#define INIT_NMSARGS { NULL, NULL, -1, -1, false, false , COLOR_BLUE }
+#define INIT_NMSARGS { NULL, NULL, NULL, -1, -1, false, false }
 
 // Argument structure for nms_exec()
 typedef struct {
 	char *src;
+	char* foreground_color;
 	char *return_opts;
 	int input_cursor_x;
 	int input_cursor_y;
 	bool show_cursor;
 	bool auto_decrypt;
-	int foreground_color;
 } NmsArgs;
 
 // Display the characters stored in the display queue
 char nms_exec(NmsArgs *);
 
-int  getColorByName(char *string, int fallback);
-
 #endif