Allow diffcallback customization

1 files changed, 11 insertions, 2 deletions

diff --git a/fastadapter/src/main/kotlin/ca/allanwang/kau/adapters/SingleFastAdapter.kt b/fastadapter/src/main/kotlin/ca/allanwang/kau/adapters/SingleFastAdapter.kt
index e183bde..ebb8048 100644
--- a/fastadapter/src/main/kotlin/ca/allanwang/kau/adapters/SingleFastAdapter.kt
+++ b/fastadapter/src/main/kotlin/ca/allanwang/kau/adapters/SingleFastAdapter.kt
@@ -20,6 +20,7 @@ import com.mikepenz.fastadapter.GenericItem
 import com.mikepenz.fastadapter.IAdapter
 import com.mikepenz.fastadapter.IItemAdapter
 import com.mikepenz.fastadapter.adapters.ItemAdapter
+import com.mikepenz.fastadapter.diff.DiffCallback
 import com.mikepenz.fastadapter.diff.FastAdapterDiffUtil
 
 /**
@@ -51,7 +52,15 @@ class SingleFastAdapter private constructor(val adapter: ItemAdapter<GenericItem
         throw IllegalStateException("SingleFastAdapter only allows one adapter.")
     }
 
-    fun setWithDiff(items: List<GenericItem>, detectMoves: Boolean = true) {
-        FastAdapterDiffUtil.set(adapter, items, detectMoves)
+    fun setWithDiff(
+        items: List<GenericItem>,
+        detectMoves: Boolean = true,
+        callback: DiffCallback<GenericItem>? = null
+    ) {
+        if (callback == null) {
+            FastAdapterDiffUtil.set(adapter, items, detectMoves)
+        } else {
+            FastAdapterDiffUtil.set(adapter, items, callback, detectMoves)
+        }
     }
 }