refactor(yogurt): move scripting logic into single file

[skip ci]

Author: Wesley-Young

yogurt/src/commonMain/kotlin/org/ntqqrev/yogurt/YogurtApp.kt

@@ -1,6 +1,5 @@
 package org.ntqqrev.yogurt
 
-import com.dokar.quickjs.QuickJs
 import com.github.ajalt.mordant.rendering.TextColors
 import com.github.ajalt.mordant.terminal.Terminal
 import io.ktor.http.*
@@ -11,7 +10,6 @@ import io.ktor.server.cio.*
 import io.ktor.server.engine.*
 import io.ktor.server.plugins.contentnegotiation.*
 import io.ktor.server.plugins.cors.routing.*
-import io.ktor.server.plugins.di.*
 import io.ktor.server.routing.*
 import io.ktor.server.sse.*
 import io.ktor.server.websocket.*
@@ -25,8 +23,7 @@ import org.ntqqrev.milky.milkyVersion
 import org.ntqqrev.yogurt.config.loadConfigAndUpdate
 import org.ntqqrev.yogurt.debug.configureDebugFaceDetailsApi
 import org.ntqqrev.yogurt.fs.withFs
-import org.ntqqrev.yogurt.script.createScriptEnvironment
-import org.ntqqrev.yogurt.script.loadScripts
+import org.ntqqrev.yogurt.scripting.configureScripting
 import org.ntqqrev.yogurt.util.*
 import kotlin.time.Duration.Companion.milliseconds
 
@@ -147,12 +144,7 @@ object YogurtApp {
                 }
             }
 
-            val qjs = createScriptEnvironment()
-
-            dependencies {
-                provide<MilkyContext> { ctx }
-                provide<QuickJs> { qjs } cleanup { it.close() }
-            }
+            configureScripting()
         }
 
         monitor.subscribe(ApplicationStarted) {
@@ -162,7 +154,6 @@ object YogurtApp {
 
             launch {
                 botLogin()
-                loadScripts()
             }
         }
     }

yogurt/src/commonMain/kotlin/org/ntqqrev/yogurt/script/Handles.kt

@@ -1,29 +1,41 @@
-package org.ntqqrev.yogurt.script
+package org.ntqqrev.yogurt.scripting
 
 import com.dokar.quickjs.QuickJs
 import com.dokar.quickjs.binding.ObjectBindingScope
 import com.dokar.quickjs.binding.define
 import io.ktor.server.application.*
 import io.ktor.server.plugins.di.*
-import kotlinx.coroutines.Dispatchers
-import kotlinx.coroutines.runBlocking
-import kotlinx.serialization.ExperimentalSerializationApi
+import kotlinx.coroutines.*
 import org.ntqqrev.acidify.AbstractBot
 import org.ntqqrev.acidify.milky.MilkyContext
 import org.ntqqrev.acidify.milky.api.MilkyApiHandler
 import org.ntqqrev.acidify.milky.api.handler.*
 import org.ntqqrev.acidify.milky.mediaSourceScoped
 import org.ntqqrev.milky.milkyJsonModule
-import org.ntqqrev.yogurt.script.stdlib.defineConsole
-import org.ntqqrev.yogurt.script.stdlib.defineHttp
+import org.ntqqrev.yogurt.fs.withFs
+import org.ntqqrev.yogurt.scripting.stdlib.defineStdlibConsole
+import org.ntqqrev.yogurt.scripting.stdlib.defineStdlibHttp
+import org.ntqqrev.yogurt.scriptsPath
 import kotlin.time.DurationUnit
 import kotlin.time.measureTime
 
+const val rootHandle = "yogurt"
+const val apiHandle = "api"
+const val eventHandle = "event"
+
+const val internalApiHandle = "__yogurtInternal_api"
+const val internalEventMapHandle = "__yogurtInternal_eventMap"
+const val internalEmitHandle = "__yogurtInternal_emit"
+
+class Script(
+    val name: String,
+    val content: String,
+)
+
 context(ctx: MilkyContext)
-suspend fun Application.createScriptEnvironment() =
-    QuickJs.create(jobDispatcher = Dispatchers.Default).apply {
-    defineConsole()
-    defineHttp()
+suspend fun Application.configureScripting() = QuickJs.create(jobDispatcher = Dispatchers.Default).apply {
+    defineStdlibConsole()
+    defineStdlibHttp()
 
     evaluate<Any?>(
         """
@@ -132,15 +144,20 @@ suspend fun Application.createScriptEnvironment() =
             }
         """.trimIndent()
     )
+
+    monitor.subscribe(ApplicationStarted) {
+        this@configureScripting.launch {
+            loadScripts()
+        }
+    }
 }
 
-@OptIn(ExperimentalSerializationApi::class)
 context(
     qjs: QuickJs,
     scope: ObjectBindingScope,
     ctx: MilkyContext,
 )
-inline fun <reified T : Any, reified R : Any> Application.defineJsApi(
+private inline fun <reified T : Any, reified R : Any> Application.defineJsApi(
     handler: MilkyApiHandler<T, R>
 ) {
     val methodName = handler.path.removePrefix("/")
@@ -185,4 +202,43 @@ inline fun <reified T : Any, reified R : Any> Application.defineJsApi(
             throw e
         }
     }
+}
+
+context(ctx: MilkyContext)
+private suspend fun QuickJs.loadScripts() = withFs {
+    val logger = ctx.application.dependencies
+        .resolve<AbstractBot>()
+        .createLogger("ScriptLoader")
+
+    if (!exists(scriptsPath)) {
+        createDirectories(scriptsPath)
+    }
+
+    val scripts = list(scriptsPath)
+        .filter { it.name.endsWith(".yogurtx.js") }
+        .map {
+            ctx.async {
+                Script(
+                    name = it.name,
+                    content = withContext(Dispatchers.IO) {
+                        it.readText()
+                    }
+                )
+            }
+        }
+        .awaitAll()
+    if (scripts.isNotEmpty()) {
+        scripts.forEach {
+            evaluate<Any?>(
+                code = it.content,
+                filename = it.name,
+                asModule = true,
+            )
+            logger.i { "脚本 ${it.name.removeSuffix(".yogurtx.js")} 加载完成" }
+        }
+        logger.i { "加载了 ${scripts.size} 个脚本" }
+        ctx.eventFlow.collect {
+            evaluate<Any?>("$internalEmitHandle(${milkyJsonModule.encodeToString(it)})")
+        }
+    }
 }

yogurt/src/commonMain/kotlin/org/ntqqrev/yogurt/script/Loader.kt

@@ -1,11 +1,11 @@
-package org.ntqqrev.yogurt.script.stdlib
+package org.ntqqrev.yogurt.scripting.stdlib
 
 import com.dokar.quickjs.QuickJs
 import com.dokar.quickjs.binding.define
 import kotlin.time.TimeMark
 import kotlin.time.TimeSource
 
-fun QuickJs.defineConsole() = define("console") {
+fun QuickJs.defineStdlibConsole() = define("console") {
     val consoleTimers = mutableMapOf<String, TimeMark>()
     val consoleCounts = mutableMapOf<String, Int>()
     var consoleGroupDepth = 0

yogurt/src/commonMain/kotlin/org/ntqqrev/yogurt/script/Script.kt

@@ -1,16 +1,11 @@
-package org.ntqqrev.yogurt.script.stdlib
+package org.ntqqrev.yogurt.scripting.stdlib
 
 import com.dokar.quickjs.QuickJs
 import com.dokar.quickjs.binding.JsObject
 import com.dokar.quickjs.binding.define
-import io.ktor.client.HttpClient
-import io.ktor.client.request.get
-import io.ktor.client.request.header
-import io.ktor.client.request.post
-import io.ktor.client.request.setBody
-import io.ktor.client.statement.HttpResponse
-import io.ktor.client.statement.bodyAsBytes
-import io.ktor.client.statement.bodyAsText
+import io.ktor.client.*
+import io.ktor.client.request.*
+import io.ktor.client.statement.*
 
 class HttpRequestOptions(
     val method: String,
@@ -46,7 +41,7 @@ class HttpRequestOptions(
 }
 
 @OptIn(ExperimentalUnsignedTypes::class)
-fun QuickJs.defineHttp() = define("http") {
+fun QuickJs.defineStdlibHttp() = define("http") {
     val httpClient = HttpClient()
 
     fun Array<Any?>.extractParams(): Pair<String, HttpRequestOptions> {