Integrate reset functionality by @paweljaneczek

Author: rubengees

kotterknife/src/androidTest/kotlin/kotterknife/ViewTest.kt

@@ -157,6 +157,20 @@ class ViewTest {
         assertEquals(2, example.name.count())
     }
 
+    @Test
+    fun testReset() {
+        class Example(context: Context) : FrameLayout(context) {
+            val name: View? by bindOptionalView(1)
+        }
+
+        val example = Example(InstrumentationRegistry.getContext())
+        example.addView(viewWithId(1))
+        assertNotNull(example.name)
+        example.removeAllViews()
+        KotterKnife.reset(example)
+        assertNull(example.name)
+    }
+
     private fun viewWithId(id: Int): View {
         val view = View(InstrumentationRegistry.getContext())
         view.id = id

kotterknife/src/main/kotlin/kotterknife/KotterKnife.kt

@@ -8,11 +8,16 @@ import android.app.DialogFragment
 import android.app.Fragment
 import android.support.v7.widget.RecyclerView.ViewHolder
 import android.view.View
+import java.util.*
 import kotlin.properties.ReadOnlyProperty
 import kotlin.reflect.KProperty
 import android.support.v4.app.DialogFragment as SupportDialogFragment
 import android.support.v4.app.Fragment as SupportFragment
 
+object KotterKnife {
+    fun reset(target: Any) = LazyRegistry.reset(target)
+}
+
 fun <V : View> View.bindView(id: Int)
         : ReadOnlyProperty<View, V> = required(id, viewFinder)
 
@@ -155,10 +160,29 @@ private class Lazy<in T, out V>(private val initializer: (T, KProperty<*>) -> V)
     private var value: Any? = EMPTY
 
     override fun getValue(thisRef: T, property: KProperty<*>): V {
+        LazyRegistry.register(thisRef!!, this)
+
         if (value == EMPTY) {
             value = initializer(thisRef, property)
         }
+
         @Suppress("UNCHECKED_CAST")
         return value as V
     }
+
+    fun reset() {
+        value = EMPTY
+    }
+}
+
+private object LazyRegistry {
+    private val lazyMap = WeakHashMap<Any, MutableCollection<Lazy<*, *>>>()
+
+    fun register(target: Any, lazy: Lazy<*, *>) {
+        lazyMap.getOrPut(target, { Collections.newSetFromMap(WeakHashMap()) }).add(lazy)
+    }
+
+    fun reset(target: Any) {
+        lazyMap[target]?.forEach { it.reset() }
+    }
 }