Ensure no duplicate character sets

Adds i18n_set_exists function to check the linked list for potential
duplicates before adding a new character set.

3 files changed, 33 insertions, 11 deletions

diff --git a/src/i18n_set.c b/src/i18n_set.c
index 0f6b819..a539752 100644
--- a/src/i18n_set.c
+++ b/src/i18n_set.c
@@ -52,6 +52,16 @@ struct i18n_set* i18n_set_add(struct i18n_set* set, int type) {
 }
 
 
+struct i18n_set* i18n_set_exists(struct i18n_set* set, int type) {
+  while(set) {
+    if(set->type == type)
+      break;
+    set = set->next;
+  }
+  return set;
+}
+
+
 struct i18n_set* i18n_set_rm_type(struct i18n_set* set, int type) {
   struct i18n_set* cursor = set;
   struct i18n_set* prev = NULL;
diff --git a/src/i18n_set.h b/src/i18n_set.h
index 9a79e83..ccdf2ed 100644
--- a/src/i18n_set.h
+++ b/src/i18n_set.h
@@ -20,6 +20,7 @@ struct i18n_set {
 
 struct i18n_set* i18n_set_new(int);
 struct i18n_set* i18n_set_add(struct i18n_set*, int);
+struct i18n_set* i18n_set_exists(struct i18n_set*, int);
 struct i18n_set* i18n_set_rm_type(struct i18n_set*, int);
 void i18n_set_dump(struct i18n_set*);
 void i18n_set_free(struct i18n_set*);
diff --git a/src/main.c b/src/main.c
index faa939b..50d55b9 100644
--- a/src/main.c
+++ b/src/main.c
@@ -84,31 +84,42 @@ int main(int argc, char* argv[]) {
       set = i18n_set_rm_type(set, I18N_TYPE_ASCII_UPPER);
 
     } else if(strcmp(argv[i], "-c") == 0 || strcmp(argv[i], "--capitalize") == 0) {
-      i18n_set_add(set, I18N_TYPE_ASCII_UPPER);
+      if(!i18n_set_exists(set, I18N_TYPE_ASCII_UPPER))
+        i18n_set_add(set, I18N_TYPE_ASCII_UPPER);
 
     } else if(strcmp(argv[i], "-n") == 0 || strcmp(argv[i], "--numerals") == 0) {
-      i18n_set_add(set, I18N_TYPE_ASCII_NUMERALS);
+      if(!i18n_set_exists(set, I18N_TYPE_ASCII_NUMERALS))
+        i18n_set_add(set, I18N_TYPE_ASCII_NUMERALS);
 
     } else if(strcmp(argv[i], "-y") == 0 || strcmp(argv[i], "--symbols") == 0) {
-      i18n_set_add(set, I18N_TYPE_ASCII_SYMBOLS);
+      if(!i18n_set_exists(set, I18N_TYPE_ASCII_SYMBOLS))
+        i18n_set_add(set, I18N_TYPE_ASCII_SYMBOLS);
 
     } else if(strcmp(argv[i], "-i") == 0 || strcmp(argv[i], "--i18n") == 0) {
-      i18n_set_add(set, I18N_TYPE_ONE);
-      i18n_set_add(set, I18N_TYPE_TWO);
-      i18n_set_add(set, I18N_TYPE_THREE);
-      i18n_set_add(set, I18N_TYPE_FOUR);
+      if(!i18n_set_exists(set, I18N_TYPE_ONE))
+        i18n_set_add(set, I18N_TYPE_ONE);
+      if(!i18n_set_exists(set, I18N_TYPE_TWO))
+        i18n_set_add(set, I18N_TYPE_TWO);
+      if(!i18n_set_exists(set, I18N_TYPE_THREE))
+        i18n_set_add(set, I18N_TYPE_THREE);
+      if(!i18n_set_exists(set, I18N_TYPE_FOUR))
+        i18n_set_add(set, I18N_TYPE_FOUR);
 
     } else if(strcmp(argv[i], "-1") == 0) {
-      i18n_set_add(set, I18N_TYPE_ONE);
+      if(!i18n_set_exists(set, I18N_TYPE_ONE))
+        i18n_set_add(set, I18N_TYPE_ONE);
 
     } else if(strcmp(argv[i], "-2") == 0) {
-      i18n_set_add(set, I18N_TYPE_TWO);
+      if(!i18n_set_exists(set, I18N_TYPE_TWO))
+        i18n_set_add(set, I18N_TYPE_TWO);
 
     } else if(strcmp(argv[i], "-3") == 0) {
-      i18n_set_add(set, I18N_TYPE_THREE);
+      if(!i18n_set_exists(set, I18N_TYPE_THREE))
+        i18n_set_add(set, I18N_TYPE_THREE);
 
     } else if(strcmp(argv[i], "-4") == 0) {
-      i18n_set_add(set, I18N_TYPE_FOUR);
+      if(!i18n_set_exists(set, I18N_TYPE_FOUR))
+        i18n_set_add(set, I18N_TYPE_FOUR);
 
     } else if(strcmp(argv[i], "-h") == 0 || strcmp(argv[i], "--help") == 0) {
       usage();