feat: AddOrReplace collection method

percula · Oct 28, 2018 · d1fa42d · d1fa42d
1 parent cb9ed66
commit d1fa42d
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/library/src/main/java/com/perculacreative/ktx/Collection.kt b/library/src/main/java/com/perculacreative/ktx/Collection.kt
@@ -28,6 +28,19 @@ fun <T> MutableCollection<T>.replaceWith(collection: Collection<T>) {
     addAll(collection)
 }
 
+/**
+ * Replaces the first item within the list that matches the given predicate or adds the item to
+ * the list if none match.
+ */
+fun <T> MutableList<T>.addOrReplace(item: T, predicate: (T) -> Boolean): Boolean {
+    return this.indexOfFirst { predicate.invoke(it) }
+            .takeIf { it >= 0 }
+            ?.let { this[it] = item }
+            ?.let { true }
+    ?: this.add(item)
+            .let { false }
+}
+
 fun <S: MutableList<T>, T> S.addAnd(index: Int, item: T): S {
     add(index, item)
     return this