Implement Color Vision Deficiencies setting

Implement settings for redgreen and blue blindness.

Regards https://github.com/profanity-im/profanity/issues/1191

8 files changed, 102 insertions, 16 deletions

diff --git a/src/command/cmd_ac.c b/src/command/cmd_ac.c
index 625afd99..8908777a 100644
--- a/src/command/cmd_ac.c
+++ b/src/command/cmd_ac.c
@@ -111,6 +111,7 @@ static char* _clear_autocomplete(ProfWin *window, const char *const input, gbool
 static char* _invite_autocomplete(ProfWin *window, const char *const input, gboolean previous);
 static char* _status_autocomplete(ProfWin *window, const char *const input, gboolean previous);
 static char* _logging_autocomplete(ProfWin *window, const char *const input, gboolean previous);
+static char* _color_autocomplete(ProfWin *window, const char *const input, gboolean previous);
 
 static char* _script_autocomplete_func(const char *const prefix, gboolean previous);
 
@@ -229,6 +230,7 @@ static Autocomplete invite_ac;
 static Autocomplete status_ac;
 static Autocomplete status_state_ac;
 static Autocomplete logging_ac;
+static Autocomplete color_ac;
 
 void
 cmd_ac_init(void)
@@ -903,6 +905,12 @@ cmd_ac_init(void)
     logging_ac = autocomplete_new();
     autocomplete_add(logging_ac, "chat");
     autocomplete_add(logging_ac, "group");
+
+    color_ac = autocomplete_new();
+    autocomplete_add(color_ac, "on");
+    autocomplete_add(color_ac, "off");
+    autocomplete_add(color_ac, "redgreen");
+    autocomplete_add(color_ac, "blue");
 }
 
 void
@@ -1205,6 +1213,7 @@ cmd_ac_reset(ProfWin *window)
     autocomplete_reset(status_ac);
     autocomplete_reset(status_state_ac);
     autocomplete_reset(logging_ac);
+    autocomplete_reset(color_ac);
 
     autocomplete_reset(script_ac);
     if (script_show_ac) {
@@ -1349,6 +1358,7 @@ cmd_ac_uninit(void)
     autocomplete_free(status_ac);
     autocomplete_free(status_state_ac);
     autocomplete_free(logging_ac);
+    autocomplete_free(color_ac);
 }
 
 static void
@@ -1479,7 +1489,7 @@ _cmd_ac_complete_params(ProfWin *window, const char *const input, gboolean previ
 
     // autocomplete boolean settings
     gchar *boolean_choices[] = { "/beep", "/intype", "/states", "/outtype", "/flash", "/splash",
-        "/history", "/vercheck", "/privileges", "/wrap", "/carbons", "/lastactivity", "/color" };
+        "/history", "/vercheck", "/privileges", "/wrap", "/carbons", "/lastactivity"};
 
     for (i = 0; i < ARRAY_SIZE(boolean_choices); i++) {
         result = autocomplete_param_with_func(input, boolean_choices[i], prefs_autocomplete_boolean_choice, previous);
@@ -1597,6 +1607,7 @@ _cmd_ac_complete_params(ProfWin *window, const char *const input, gboolean previ
     g_hash_table_insert(ac_funcs, "/invite",        _invite_autocomplete);
     g_hash_table_insert(ac_funcs, "/status",        _status_autocomplete);
     g_hash_table_insert(ac_funcs, "/logging",       _logging_autocomplete);
+    g_hash_table_insert(ac_funcs, "/color",         _color_autocomplete);
 
     int len = strlen(input);
     char parsed[len+1];
@@ -3595,3 +3606,16 @@ _logging_autocomplete(ProfWin *window, const char *const input, gboolean previou
 
     return NULL;
 }
+
+static char*
+_color_autocomplete(ProfWin *window, const char *const input, gboolean previous)
+{
+    char *result = NULL;
+
+    result = autocomplete_param_with_ac(input, "/color", color_ac, TRUE, previous);
+    if (result) {
+        return result;
+    }
+
+    return NULL;
+}
diff --git a/src/command/cmd_defs.c b/src/command/cmd_defs.c
index 0ea47bca..7e0001c6 100644
--- a/src/command/cmd_defs.c
+++ b/src/command/cmd_defs.c
@@ -2302,12 +2302,15 @@ static struct cmd_t command_defs[] =
         CMD_TAGS(
             CMD_TAG_UI)
         CMD_SYN(
-            "/color on|off")
+            "/color on|off|redgreen|blue")
         CMD_DESC(
-            "Settings for consistent color generation for nicks (XEP-0392).")
+            "Settings for consistent color generation for nicks (XEP-0392). Including corrections for Color Vision Deficiencies")
         CMD_ARGS(
-            { "on|off", "Enable or disable nick colorization for MUC nicks." })
-        CMD_NOEXAMPLES
+            { "on|off|redgreen|blue", "Enable or disable nick colorization for MUC nicks. 'redgreen' is for people with red/green blindess and 'blue' for people with blue blindness."})
+        CMD_EXAMPLES(
+            "/color off",
+            "/color on",
+            "/color blue")
     },
 };
 
diff --git a/src/command/cmd_funcs.c b/src/command/cmd_funcs.c
index 2b7894ad..425b50fc 100644
--- a/src/command/cmd_funcs.c
+++ b/src/command/cmd_funcs.c
@@ -8630,6 +8630,33 @@ cmd_paste(ProfWin *window, const char *const command, gchar **args)
 gboolean
 cmd_color(ProfWin *window, const char *const command, gchar **args)
 {
-    _cmd_set_boolean_preference(args[0], command, "Consistent color generation for nicks", PREF_COLOR_NICK);
+    if (g_strcmp0(args[0], "on") == 0) {
+            prefs_set_string(PREF_COLOR_NICK, "true");
+    } else if (g_strcmp0(args[0], "off") == 0) {
+            prefs_set_string(PREF_COLOR_NICK, "false");
+    } else if (g_strcmp0(args[0], "redgreen") == 0) {
+            prefs_set_string(PREF_COLOR_NICK, "redgreen");
+    } else if (g_strcmp0(args[0], "blue") == 0) {
+            prefs_set_string(PREF_COLOR_NICK, "blue");
+    } else {
+        cons_bad_cmd_usage(command);
+        return TRUE;
+    }
+
+    cons_show("Consistent color generation for nicks set to: %s", args[0]);
+
+    char *theme = prefs_get_string(PREF_THEME);
+    if (theme) {
+        gboolean res = theme_load(theme);
+
+        if (res) {
+            cons_show("Theme reloaded: %s", theme);
+        } else {
+            theme_load("default");
+        }
+
+        prefs_free_string(theme);
+    }
+
     return TRUE;
 }
diff --git a/src/config/color.c b/src/config/color.c
index a6bf865e..6af30bf8 100644
--- a/src/config/color.c
+++ b/src/config/color.c
@@ -38,6 +38,7 @@
 #include <stdlib.h>
 #include <string.h>
 #include <stdint.h>
+#include <math.h>
 #include <glib.h>
 
 #ifdef HAVE_NCURSESW_NCURSES_H
@@ -380,7 +381,7 @@ static int find_col(const char *col_name, int n)
     return COL_ERR;
 }
 
-static int color_hash(const char *str)
+static int color_hash(const char *str, color_profile profile)
 {
     GChecksum *cs = NULL;
     guint8 buf[256] = {0};
@@ -401,10 +402,14 @@ static int color_hash(const char *str)
     double h = ((buf[1] << 8) | buf[0]) / 65536. * 360.;
 
     // red/green blindness correction
-    // h = fmod(fmod(h + 90., 180) - 90., 360.);
+    if (profile == COLOR_PROFILE_REDGREEN) {
+        h = fmod(fmod(h + 90., 180) - 90., 360.);
+    }
 
     // blue blindness correction
-    // h = fmod(h, 180.);
+    if (profile == COLOR_PROFILE_BLUE) {
+        h = fmod(h, 180.);
+    }
 
     rc = find_closest_col((int)h, 100, 50);
 
@@ -487,9 +492,9 @@ static int _color_pair_cache_get(int fg, int bg)
  * hash a string into a color that will be used as fg
  * use default color as bg
  */
-int color_pair_cache_hash_str(const char *str)
+int color_pair_cache_hash_str(const char *str, color_profile profile)
 {
-    int fg = color_hash(str);
+    int fg = color_hash(str, profile);
     int bg = -1;
 
     return _color_pair_cache_get(fg, bg);
diff --git a/src/config/color.h b/src/config/color.h
index 6a5cf4a9..202e87c6 100644
--- a/src/config/color.h
+++ b/src/config/color.h
@@ -41,6 +41,12 @@
 
 #include <stdint.h>
 
+typedef enum {
+    COLOR_PROFILE_DEFAULT,
+    COLOR_PROFILE_REDGREEN,
+    COLOR_PROFILE_BLUE,
+} color_profile;
+
 struct color_def {
     uint16_t h; uint8_t s, l;
     const char *name;
@@ -48,7 +54,7 @@ struct color_def {
 extern const struct color_def color_names[];
 
 /* hash string to color pair */
-int color_pair_cache_hash_str(const char *str);
+int color_pair_cache_hash_str(const char *str, color_profile profile);
 /* parse fg_bg string to color pair */
 int color_pair_cache_get(const char *pair_name);
 /* clear cache */
diff --git a/src/config/theme.c b/src/config/theme.c
index c76de248..8d2160dd 100644
--- a/src/config/theme.c
+++ b/src/config/theme.c
@@ -673,7 +673,17 @@ theme_free_string(char *str)
 int
 theme_hash_attrs(const char *str)
 {
-    return COLOR_PAIR(color_pair_cache_hash_str(str));
+    color_profile profile = COLOR_PROFILE_DEFAULT;
+
+    char *color_pref = prefs_get_string(PREF_COLOR_NICK);
+    if (strcmp(color_pref, "redgreen") == 0) {
+        profile = COLOR_PROFILE_REDGREEN;
+    } else if (strcmp(color_pref, "blue") == 0) {
+        profile = COLOR_PROFILE_BLUE;
+    }
+    prefs_free_string(color_pref);
+
+    return COLOR_PAIR(color_pair_cache_hash_str(str, profile));
 }
 
 int
diff --git a/src/ui/console.c b/src/ui/console.c
index 3906da5c..91b499f7 100644
--- a/src/ui/console.c
+++ b/src/ui/console.c
@@ -1971,10 +1971,19 @@ cons_autoping_setting(void)
 void
 cons_color_setting(void)
 {
-    if (prefs_get_boolean(PREF_COLOR_NICK))
+    char *color_pref = prefs_get_string(PREF_COLOR_NICK);
+
+    if (strcmp(color_pref, "true") == 0) {
         cons_show("Consistent color generation for nicks (/color)               : ON");
-    else
+    } else if (strcmp(color_pref, "redgreen") == 0) {
+        cons_show("Consistent color generation for nicks (/color)               : REDGREEN");
+    } else if (strcmp(color_pref, "blue") == 0) {
+        cons_show("Consistent color generation for nicks (/color)               : BLUE");
+    } else {
         cons_show("Consistent color generation for nicks (/color)               : OFF");
+    }
+
+    prefs_free_string(color_pref);
 }
 
 void
diff --git a/src/ui/window.c b/src/ui/window.c
index 47230150..2a76496c 100644
--- a/src/ui/window.c
+++ b/src/ui/window.c
@@ -1453,9 +1453,11 @@ _win_print(ProfWin *window, const char show_char, int pad_indent, GDateTime *tim
             colour = theme_attrs(THEME_THEM);
         }
 
-        if (prefs_get_boolean(PREF_COLOR_NICK)) {
+        char *color_pref = prefs_get_string(PREF_COLOR_NICK);
+        if (color_pref != NULL && (strcmp(color_pref, "false") != 0)) {
             colour = theme_hash_attrs(from);
         }
+        prefs_free_string(color_pref);
 
         if (flags & NO_COLOUR_FROM) {
             colour = 0;