From ab73dc59a302651b46b07329cb2d8303e3ea36dd Mon Sep 17 00:00:00 2001
From: Allan Wang <allanwang@google.com>
Date: Wed, 22 Jan 2020 13:33:34 -0800
Subject: Allow diffcallback customization

---
 .../kotlin/ca/allanwang/kau/adapters/SingleFastAdapter.kt   | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

(limited to 'fastadapter/src/main/kotlin/ca/allanwang')

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)
+        }
     }
 }
-- 
cgit v1.2.3