feat: add support for delegation

ShindouMihou · Oct 16, 2023 · 279895c · 279895c
1 parent b66c37e
commit 279895c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/main/java/pw/mihou/nexus/features/command/react/React.kt b/src/main/java/pw/mihou/nexus/features/command/react/React.kt
@@ -25,11 +25,13 @@ import java.io.InputStream
 import java.time.Instant
 import java.util.concurrent.atomic.AtomicReference
 import java.util.concurrent.locks.ReentrantLock
+import kotlin.properties.Delegates
+import kotlin.reflect.KProperty
 
 typealias Subscription<T> = (oldValue: T, newValue: T) -> Unit
 typealias Unsubscribe = () -> Unit
 
-class React(private val api: DiscordApi) {
+class React internal constructor(private val api: DiscordApi) {
     private var message: NexusMessage = NexusMessage()
     private var unsubscribe: Unsubscribe = {}
     private var component: (Component.() -> Unit)? = null
@@ -91,6 +93,12 @@ class React(private val api: DiscordApi) {
     class Writable<T>(value: T) {
         private val subscribers: MutableList<Subscription<T>> = mutableListOf()
         private val _value: AtomicReference<T> = AtomicReference(value)
+        operator fun getValue(thisRef: Any?, property: KProperty<*>): T {
+            return _value.get()
+        }
+        operator fun setValue(thisRef: Any?, property: KProperty<*>, value: T) {
+            set(value)
+        }
         fun set(value: T) {
             val oldValue = _value.get()
             _value.set(value)

diff --git a/src/test/java/commands/ReactiveTest.kt b/src/test/java/commands/ReactiveTest.kt
@@ -8,7 +8,7 @@ class ReactiveTest: NexusHandler {
     val description = "A test regarding React"
     override fun onEvent(event: NexusCommandEvent) {
         event.R {
-            val clicks = writable(0)
+            var clicks by writable(0)
             render {
                 Embed {
                     Title("R.Embeds")
@@ -21,7 +21,7 @@ class ReactiveTest: NexusHandler {
                 }
                 Button(label = "Click to be DM'd") { event ->
                     event.interaction.user.sendMessage("Hi")
-                    clicks.getAndUpdate { it + 1 }
+                    clicks += 1
                 }
             }
         }