Add array.replace() function

Author: mertyildiran

README.md

@@ -70,6 +70,19 @@ kaos> array.search(e, 'foo')
 [0, 2, 5]
 ```
 
+### list array.replace(list haystack, any needle, any replacement)
+
+Replaces all occurrences of the `needle` with the `replacement` in list `haystack`.
+
+```chaos
+kaos> list d = [1, 2, 3, 4, 5, 3, 6, 7, 3]
+kaos> list e = ['foo', 'bar', 'foo', 'baz', 'bar', 'foo']
+kaos> array.replace(d, 3, 17)
+[1, 2, 17, 4, 5, 17, 6, 7, 17]
+kaos> array.replace(e, 'foo', 'goo')
+['goo', 'bar', 'goo', 'baz', 'bar', 'goo']
+```
+
 ## Information Functions
 
 ### num array.length(list l)

array.c

@@ -212,6 +212,86 @@ int KAOS_EXPORT Kaos_search()
     return 0;
 }
 
+// list array.replace(list haystack, any needle, any replacement)
+
+char *replace_params_name[] = {
+    "haystack",
+    "needle",
+    "replacement"
+};
+unsigned replace_params_type[] = {
+    K_LIST,
+    K_ANY,
+    K_ANY
+};
+unsigned short replace_params_length = (unsigned short) sizeof(replace_params_type) / sizeof(unsigned);
+int KAOS_EXPORT Kaos_replace()
+{
+    unsigned long list_length = kaos.getListLength(replace_params_name[0]);
+    enum Type list_type = kaos.getListType(replace_params_name[0]);
+
+    kaos.startBuildingList();
+
+    for (unsigned long i = 0; i < list_length; i++) {
+        enum ValueType value_type = kaos.getListElementValueType(replace_params_name[0], i);
+
+        bool x_b, y_b;
+        long long x_i, y_i;
+        long double x_f, y_f;
+        char *x_s, *y_s;
+        char *replacement;
+
+        switch (value_type)
+        {
+            case V_BOOL:
+                x_b = kaos.getVariableBool(replace_params_name[1]);
+                y_b = kaos.getListElementBool(replace_params_name[0], i);
+                if (x_b == y_b) {
+                    kaos.createVariableBool(NULL, kaos.getVariableBool(replace_params_name[2]));
+                } else {
+                    kaos.createVariableBool(NULL, y_b);
+                }
+                break;
+            case V_INT:
+                x_i = kaos.getVariableInt(replace_params_name[1]);
+                y_i = kaos.getListElementInt(replace_params_name[0], i);
+                if (x_i == y_i) {
+                    kaos.createVariableInt(NULL, kaos.getVariableInt(replace_params_name[2]));
+                } else {
+                    kaos.createVariableInt(NULL, y_i);
+                }
+                break;
+            case V_FLOAT:
+                x_f = kaos.getVariableFloat(replace_params_name[1]);
+                y_f = kaos.getListElementFloat(replace_params_name[0], i);
+                if (x_f == y_f) {
+                    kaos.createVariableFloat(NULL, kaos.getVariableFloat(replace_params_name[2]));
+                } else {
+                    kaos.createVariableFloat(NULL, y_f);
+                }
+                break;
+            case V_STRING:
+                x_s = kaos.getVariableString(replace_params_name[1]);
+                y_s = kaos.getListElementString(replace_params_name[0], i);
+                if (strcmp(x_s, y_s) == 0) {
+                    replacement = kaos.getVariableString(replace_params_name[2]);
+                    kaos.createVariableString(NULL, replacement);
+                    free(replacement);
+                } else {
+                    kaos.createVariableString(NULL, y_s);
+                }
+                free(x_s);
+                free(y_s);
+                break;
+            default:
+                break;
+        }
+    }
+
+    kaos.returnList(list_type);
+    return 0;
+}
+
 
 // Information Functions
 
@@ -243,6 +323,7 @@ int KAOS_EXPORT KaosRegister(struct Kaos _kaos)
 
     // Searching & Replacing
     kaos.defineFunction("search", K_LIST, K_NUMBER, search_params_name, search_params_type, search_params_length);
+    kaos.defineFunction("replace", K_LIST, K_ANY, replace_params_name, replace_params_type, replace_params_length);
 
     // Information Functions
     kaos.defineFunction("length", K_NUMBER, K_ANY, length_params_name, length_params_type, length_params_length);

test.kaos

@@ -3,20 +3,22 @@ import array
 // Array Operations
 list a = [1, 2, 3]
 list b = [4, 5, 6]
+list c = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
 print array.merge(a, b)
 print array.insert(a, 5, 1)
 print array.insert(a, true, 1)
 print array.insert(a, 3.5, 1)
 print array.insert(a, 'hello', 1)
 print array.reverse(a)
-list c = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
 print array.chunk(c, 3)
 
 // Searching & Replacing
 list d = [1, 2, 3, 4, 5, 3, 6, 7, 3]
-print array.search(d, 3)
 list e = ['foo', 'bar', 'foo', 'baz', 'bar', 'foo']
+print array.search(d, 3)
 print array.search(e, 'foo')
+print array.replace(d, 3, 17)
+print array.replace(e, 'foo', 'goo')
 
 // Information Functions
 print array.length(a)

test.out

@@ -7,4 +7,6 @@
 [['a', 'b', 'c'], ['d', 'e', 'f'], ['g', 'h']]
 [2, 5, 8]
 [0, 2, 5]
+[1, 2, 17, 4, 5, 17, 6, 7, 17]
+['goo', 'bar', 'goo', 'baz', 'bar', 'goo']
 3