Add cancel / triggerChange extension methods to watcher functions

orx-file-watcher/src/main/kotlin/FileWatcher.kt

@@ -11,8 +11,6 @@ import java.nio.file.WatchKey
 import kotlin.concurrent.thread
 
 class FileWatcher(private val program: Program, val file: File, private val onChange: (File) -> Unit) {
-    init {
-        watchThread
     val path = file.absoluteFile.toPath()
     val parent = path.parent
     val key = pathKeys.getOrPut(parent) {
@@ -21,12 +19,18 @@ class FileWatcher(private val program: Program, val file: File, private val onCh
                 SensitivityWatchEventModifier.HIGH
         )
     }
+    init {
+        watchThread
         watching.getOrPut(path) {
             mutableListOf()
         }.add(this)
         keyPaths.getOrPut(key) { parent }
     }
 
+    fun stop() {
+        watching[path]?.remove(this)
+    }
+
     internal fun triggerChange() {
         program.launch {
             onChange(file)
@@ -34,20 +38,45 @@ class FileWatcher(private val program: Program, val file: File, private val onCh
     }
 }
 
+private val watchers = mutableMapOf< ()->Any, FileWatcher>()
+
 fun <T> watchFile(program: Program, file: File, transducer: (File) -> T): () -> T {
     var result = transducer(file)
-    FileWatcher(program, file) {
+    val watcher = FileWatcher(program, file) {
         try {
             result = transducer(file)
         } catch (e: Throwable) {
             e.printStackTrace()
         }
     }
-    return {
+
+    val function = {
         result
     }
+
+    @Suppress("UNCHECKED_CAST")
+    watchers[function as ()->Any] = watcher
+    return function
 }
 
+/**
+ * Stops the watcher
+ */
+fun <T> (()->T).stop() {
+    @Suppress("UNCHECKED_CAST")
+    watchers[this as ()->Any]?.stop()
+
+}
+
+/**
+ * Triggers reload
+ */
+fun <T> (()->T).triggerChange() {
+    @Suppress("UNCHECKED_CAST")
+    watchers[this as ()->Any]?.triggerChange()
+}
+
+
 @JvmName("programWatchFile")
 fun <T> Program.watchFile(file: File, transducer: (File) -> T): () -> T = watchFile(this, file, transducer)
 
@@ -67,9 +96,8 @@ private val watchThread by lazy {
             key.pollEvents().forEach {
                 val contextPath = it.context() as Path
                 val fullPath = path?.resolve(contextPath)
-                watching[fullPath]?.forEach {
+                watching[fullPath]?.forEach { w ->
-
+                    w.triggerChange()
-                    it.triggerChange()
                 }
             }
             key.reset()
@@ -82,12 +110,12 @@ fun main() {
         it.readText()
     }
 
-
+    a.stop()
+    a.triggerChange()
 
     while (true) {
         println(a())
         Thread.sleep(2000)
-
     }
 }