From c2d6afccc2f004e18da64e98eef5037968054355 Mon Sep 17 00:00:00 2001
From: InsanusMokrassar <ovsyannikov.alexey95@gmail.com>
Date: Sun, 22 Sep 2024 18:12:46 +0600
Subject: [PATCH] add koin extensions for plugins

---
 .../kotlin/dev/inmo/plagubot/HelloPlugin.kt   |  6 +--
 .../dev/inmo/plagubot/KoinExtensions.kt       | 48 +++++++++++++++++++
 2 files changed, 50 insertions(+), 4 deletions(-)

diff --git a/plugin/src/main/kotlin/dev/inmo/plagubot/HelloPlugin.kt b/plugin/src/main/kotlin/dev/inmo/plagubot/HelloPlugin.kt
index ecb53f1..c6a388b 100644
--- a/plugin/src/main/kotlin/dev/inmo/plagubot/HelloPlugin.kt
+++ b/plugin/src/main/kotlin/dev/inmo/plagubot/HelloPlugin.kt
@@ -26,9 +26,7 @@ object HelloPlugin : Plugin {
     )
 
     override fun Module.setupDI(config: JsonObject) {
-        single {
-            get<Json>().decodeFromJsonElement(HelloPluginConfig.serializer(), config["helloPlugin"] ?: return@single null)
-        }
+        registerConfig<HelloPluginConfig>("helloPlugin")
     }
 
     private sealed interface InternalFSMState : State {
@@ -43,7 +41,7 @@ object HelloPlugin : Plugin {
     }
 
     override suspend fun BehaviourContextWithFSM<State>.setupBotPlugin(koin: Koin) {
-        val toPrint = koin.getOrNull<HelloPluginConfig>() ?.print ?: "Hello :)"
+        val toPrint = koin.configOrNull<HelloPluginConfig>() ?.print ?: "Hello :)"
         logger.d { toPrint }
         logger.dS { getMe().toString() }
         onCommand("hello_world") {
diff --git a/plugin/src/main/kotlin/dev/inmo/plagubot/KoinExtensions.kt b/plugin/src/main/kotlin/dev/inmo/plagubot/KoinExtensions.kt
index d3ed49d..c230e65 100644
--- a/plugin/src/main/kotlin/dev/inmo/plagubot/KoinExtensions.kt
+++ b/plugin/src/main/kotlin/dev/inmo/plagubot/KoinExtensions.kt
@@ -1,11 +1,59 @@
 package dev.inmo.plagubot
 
+import kotlinx.serialization.InternalSerializationApi
+import kotlinx.serialization.KSerializer
+import kotlinx.serialization.json.Json
+import kotlinx.serialization.json.JsonObject
+import kotlinx.serialization.serializer
 import org.jetbrains.exposed.sql.Database
 import org.koin.core.Koin
+import org.koin.core.module.Module
 import org.koin.core.scope.Scope
+import kotlin.reflect.KClass
 
 val Scope.database: Database
     get() = get()
 
 val Koin.database: Database
     get() = get()
+
+/**
+ * Using [single] to register `T` with serializer [configSerializer]
+ *
+ * @param optional If passed, absence of [field] in config will lead to null config value in koin. If passed as false
+ * (default), absence of [field] in config will lead to an error
+ */
+inline fun <reified T> Module.registerConfig(configSerializer: KSerializer<T>, field: String?, optional: Boolean = false) {
+    single {
+        val fieldValue = get<JsonObject>().let {
+            if (field == null) {
+                it
+            } else {
+                it[field] ?: if (optional) return@single null else error("Unable to take field $field from config")
+            }
+        }
+        get<Json>().decodeFromJsonElement(configSerializer, fieldValue)
+    }
+}
+
+/**
+ * Using [single] to register config with getting of [serializer] from [kClass]
+ *
+ * @param optional If passed, absence of [field] in config will lead to null config value in koin. If passed as false
+ * (default), absence of [field] in config will lead to an error
+ */
+@OptIn(InternalSerializationApi::class)
+inline fun <reified T : Any> Module.registerConfig(kClass: KClass<T>, field: String?, optional: Boolean = false) = registerConfig(kClass.serializer(), field, optional)
+
+/**
+ * Using [single] to register config with getting of [serializer] from [kClass]
+ *
+ * @param optional If passed, absence of [field] in config will lead to null config value in koin. If passed as false
+ * (default), absence of [field] in config will lead to an error
+ */
+inline fun <reified T : Any> Module.registerConfig(field: String?, optional: Boolean = false) = registerConfig(T::class, field, optional)
+
+inline fun <reified T : Any> Scope.config() = get<T>()
+inline fun <reified T : Any> Koin.config() = get<T>()
+inline fun <reified T : Any> Scope.configOrNull() = getOrNull<T>()
+inline fun <reified T : Any> Koin.configOrNull() = getOrNull<T>()