add storage and update context

alice-ktx/build.gradle.kts

@@ -25,6 +25,9 @@ dependencies {
     implementation("ch.qos.logback:logback-classic:1.5.6")
 
     implementation("io.ktor:ktor-serialization-kotlinx-json-jvm")
+
+    testImplementation("junit:junit:4.13.2")
+    testImplementation("org.jetbrains.kotlinx:kotlinx-coroutines-test:1.9.0")
 }
 
 kotlin {

alice-ktx/src/main/kotlin/com/github/alice/ktx/Dispatch.kt

@@ -25,27 +25,28 @@ fun Skill.Builder.dispatch(body: Dispatcher.() -> Unit) {
  * @param fsmStrategy Стратегия конечного автомата состояний (FSM), используемая для управления состояниями.
  */
 class Dispatcher internal constructor(
-    val fsmStrategy: FSMStrategy,
+    internal val fsmStrategy: FSMStrategy,
     val dialogApi: DialogApi? = null,
-    internal val fsmContext: (message: MessageRequest) -> FSMContext
+    internal val fsmContext: (message: MessageRequest) -> FSMContext,
+    internal val enableApiStorage: Boolean = false
 ) {
     internal val commandHandlers = linkedSetOf<Handler>()
     internal val networkErrorHandlers = linkedSetOf<NetworkErrorHandler>()
-    internal val middlewares = linkedMapOf<MiddlewareType, MutableList<Middleware>>()
+    internal val middlewares = mutableMapOf<MiddlewareType, LinkedHashSet<Middleware>>()
 
     init {
-        MiddlewareType.entries.forEach { middlewares[it] = mutableListOf() }
+        MiddlewareType.entries.forEach { middlewares[it] = linkedSetOf() }
     }
 
-    internal fun addHandler(handler: Handler) {
+    fun addHandler(handler: Handler) {
         commandHandlers.add(handler)
     }
 
-    internal fun addMiddleware(middleware: Middleware, type: MiddlewareType) {
+    fun addMiddleware(middleware: Middleware, type: MiddlewareType) {
         middlewares[type]?.add(middleware)
     }
 
-    internal fun addNetworkErrorHandler(handler: NetworkErrorHandler) {
+    fun addNetworkErrorHandler(handler: NetworkErrorHandler) {
         networkErrorHandlers.add(handler)
     }
 }

alice-ktx/src/main/kotlin/com/github/alice/ktx/Skill.kt

@@ -8,7 +8,9 @@ import com.github.alice.ktx.models.response.MessageResponse
 import com.github.alice.ktx.server.WebServer
 import com.github.alice.ktx.server.WebServerResponseListener
 import com.github.alice.ktx.state.FSMContext
-import com.github.alice.ktx.state.impl.KotlinxSerializationFSMContext
+import com.github.alice.ktx.state.impl.BaseFSMContext
+import com.github.alice.ktx.storage.apiStorage.EnableApiStorage
+import com.github.alice.ktx.storage.impl.memoryStorage
 import kotlinx.serialization.json.Json
 
 /**
@@ -43,22 +45,23 @@ class Skill internal constructor(
         var fsmStrategy: FSMStrategy = FSMStrategy.USER
         internal var dispatcherConfiguration: Dispatcher.() -> Unit = { }
 
+        var storage = memoryStorage { }
+
         var fsmContext: (message: MessageRequest) -> FSMContext = { message ->
-            KotlinxSerializationFSMContext(message, fsmStrategy, json)
+            BaseFSMContext(storage, id, fsmStrategy, message)
         }
 
         internal fun build(body: Builder.() -> Unit): Skill {
             body()
 
-            val dispatcher = Dispatcher(
-                fsmStrategy = fsmStrategy,
-                dialogApi = dialogApi,
-                fsmContext = fsmContext
-            ).apply(dispatcherConfiguration)
-
             return Skill(
                 webServer = webServer,
-                dispatcher = dispatcher
+                dispatcher = Dispatcher(
+                    fsmStrategy = fsmStrategy,
+                    dialogApi = dialogApi,
+                    fsmContext = fsmContext,
+                    enableApiStorage = storage.javaClass.isAnnotationPresent(EnableApiStorage::class.java)
+                ).apply(dispatcherConfiguration)
             )
         }
     }
@@ -80,7 +83,7 @@ class Skill internal constructor(
         override suspend fun messageHandle(model: MessageRequest): MessageResponse? {
             runMiddlewares(model, MiddlewareType.OUTER)?.let { return it }
             dispatcher.commandHandlers.forEach { handler ->
-                if(handler.event(model)) {
+                if (handler.event(model)) {
                     runMiddlewares(model, MiddlewareType.INNER)?.let { return it }
                     return handler.handle(model)
                 }

alice-ktx/src/main/kotlin/com/github/alice/ktx/handlers/MessageHandler.kt

@@ -30,7 +30,7 @@ fun Dispatcher.message(
     val messageHandler = MessageHandler(
         eventBlock = { message ->
             val eventMessage = EventMessage(
-                context = request(message).state,
+                context = request(message).context,
                 message = message
             )
             event(eventMessage)

alice-ktx/src/main/kotlin/com/github/alice/ktx/handlers/impl/ButtonPressedHandler.kt

@@ -23,7 +23,7 @@ fun Dispatcher.buttonPressed(
             event(
                 EventButtonPressed(
                     payload = message.request.payload!!,
-                    context = request(message).state
+                    context = request(message).context
                 )
             )
         },

alice-ktx/src/main/kotlin/com/github/alice/ktx/models/Request.kt

@@ -3,15 +3,24 @@ package com.github.alice.ktx.models
 import com.github.alice.ktx.Dispatcher
 import com.github.alice.ktx.models.request.MessageRequest
 import com.github.alice.ktx.state.FSMContext
+import com.github.alice.ktx.state.MutableFSMContext
+import com.github.alice.ktx.storage.apiStorage.ApiStorageDetails
+
+internal suspend fun Dispatcher.request(message: MessageRequest): Request {
+    val context = fsmContext(message)
+    context.init()
 
-internal fun Dispatcher.request(message: MessageRequest): Request {
     return Request(
         message = message,
-        state = fsmContext(message)
+        context = context,
+        fsmStrategy = fsmStrategy,
+        enableApiStorage = enableApiStorage
     )
 }
 
 data class Request(
     val message: MessageRequest,
-    val state: FSMContext
+    val context: MutableFSMContext,
+    internal val fsmStrategy: FSMStrategy,
+    internal val enableApiStorage: Boolean
 )

alice-ktx/src/main/kotlin/com/github/alice/ktx/models/response/MessageResponse.kt

@@ -7,13 +7,15 @@ import com.github.alice.ktx.models.button.Button
 import com.github.alice.ktx.models.card.Card
 import com.github.alice.ktx.models.request.AccountLinking
 import com.github.alice.ktx.models.response.analytics.Analytics
-import com.github.alice.ktx.models.response.analytics.AnalyticsEvent
-import com.github.alice.ktx.state.FSMContext
+import com.github.alice.ktx.state.ReadOnlyFSMContext
 import kotlinx.serialization.SerialName
 import kotlinx.serialization.Serializable
 
-fun Request.response(body: MessageResponse.Builder.() -> Unit): MessageResponse {
-    return MessageResponse.Builder(this).build(body)
+suspend fun Request.response(body: suspend MessageResponse.Builder.() -> Unit): MessageResponse {
+    return MessageResponse.Builder(this)
+        .setFSMStrategy(fsmStrategy)
+        .setEnableApiStorage(enableApiStorage)
+        .build(body)
 }
 
 /**
@@ -23,10 +25,10 @@ suspend fun Request.authorization(
     onAlreadyAuthenticated: (suspend () -> MessageResponse)? = null,
     onAuthorizationFailed: (suspend () -> MessageResponse)? = null
 ): MessageResponse {
-    if(message.session.user?.accessToken != null && onAlreadyAuthenticated != null)
+    if (message.session.user?.accessToken != null && onAlreadyAuthenticated != null)
         return onAlreadyAuthenticated()
 
-    if(message.meta.interfaces.accountLinking == null && onAuthorizationFailed != null)
+    if (message.meta.interfaces.accountLinking == null && onAuthorizationFailed != null)
         return onAuthorizationFailed()
 
     return MessageResponse.AuthorizationBuilder(request = this).build()
@@ -59,12 +61,24 @@ data class MessageResponse internal constructor(
         internal var audioPlayer: AudioPlayer? = null
         private val buttons = mutableListOf<Button>()
         var analytics: Analytics? = null
+        private var fSMStrategy = FSMStrategy.USER
+        private var enableApiStorage = false
 
         internal fun addButton(button: Button) {
             buttons.add(button)
         }
 
-        internal fun build(body: Builder.() -> Unit): MessageResponse {
+        internal fun setFSMStrategy(strategy: FSMStrategy): Builder {
+            fSMStrategy = strategy
+            return this
+        }
+
+        internal fun setEnableApiStorage(enable: Boolean): Builder {
+            this.enableApiStorage = enable
+            return this
+        }
+
+        internal suspend fun build(body: suspend Builder.() -> Unit): MessageResponse {
             body()
 
             val response = MessageResponse(
@@ -83,7 +97,7 @@ data class MessageResponse internal constructor(
                 analytics = analytics
             )
 
-            response.setState(request.state)
+            if (enableApiStorage) response.setState(fSMStrategy, request.context)
 
             return response
         }
@@ -101,9 +115,9 @@ data class MessageResponse internal constructor(
         }
     }
 
-    private fun setState(state: FSMContext) {
+    private suspend fun setState(fSMStrategy: FSMStrategy, state: ReadOnlyFSMContext) {
         val stateResponse = getStateResponse(state.getState(), state.getData())
-        when (state.getStrategy()) {
+        when (fSMStrategy) {
             FSMStrategy.USER -> userState = stateResponse
             FSMStrategy.SESSION -> sessionState = stateResponse
             FSMStrategy.APPLICATION -> applicationState = stateResponse

alice-ktx/src/main/kotlin/com/github/alice/ktx/state/FSMContext.kt

@@ -1,52 +1,10 @@
 package com.github.alice.ktx.state
 
-import kotlin.reflect.KClass
-
 /**
  * Интерфейс для доступа к информации состояния конкретного пользователя.
  * Создаётся и передаётся в обработчики при каждом событии.
  * */
-interface FSMContext: ReadOnlyFSMContext {
-
-    /**
-     * Установить состояние по ключу.
-     * */
-    fun setState(key: String)
-
-    /**
-     * Записать данные (перезапись) с произвольным типом.
-     * */
-    fun <V: Any> setTypedData(vararg pairs: Pair<String, V>, clazz: KClass<V>)
-
-    /**
-     * Записать данные (перезапись).
-     * */
-    fun setData(vararg pairs: Pair<String, String>)
-
-    /**
-     * Обновление данных в хранилище по ключу с произвольным типом.
-     * @return Обновленные данные.
-     * */
-    fun <V: Any> updateTypedData(vararg pairs: Pair<String, V>, clazz: KClass<V>): Map<String, String>
-
-    /**
-     * Обновление данные в хранилище по ключу.
-     * @return Обновленные данные.
-     * */
-    fun updateData(vararg pairs: Pair<String, String>): Map<String, String>
-
-    /**
-     * Удалить данные по ключу.
-     * */
-    fun removeData(key: String): String?
-
-    /**
-     * Удалить данные по ключу с произвольным типом.
-     * */
-    fun <V: Any> removeTypedData(key: String, clazz: KClass<V>): V?
+interface FSMContext: MutableFSMContext {
 
-    /**
-     * Очистить состояние и данные.
-     * */
-    fun clear()
+    suspend fun init()
 }

alice-ktx/src/main/kotlin/com/github/alice/ktx/state/MutableFSMContext.kt

@@ -0,0 +1,48 @@
+package com.github.alice.ktx.state
+
+import kotlin.reflect.KClass
+
+interface MutableFSMContext : ReadOnlyFSMContext {
+
+    /**
+     * Установить состояние по ключу.
+     * */
+    suspend fun setState(key: String)
+
+    /**
+     * Записать данные (перезапись) с произвольным типом.
+     * */
+    suspend fun <V: Any> setTypedData(vararg pairs: Pair<String, V>, clazz: KClass<V>)
+
+    /**
+     * Записать данные (перезапись).
+     * */
+    suspend fun setData(vararg pairs: Pair<String, String>)
+
+    /**
+     * Обновление данных в хранилище по ключу с произвольным типом.
+     * @return Обновленные данные.
+     * */
+    suspend fun <V: Any> updateTypedData(vararg pairs: Pair<String, V>, clazz: KClass<V>): Map<String, String>
+
+    /**
+     * Обновление данные в хранилище по ключу.
+     * @return Обновленные данные.
+     * */
+    suspend fun updateData(vararg pairs: Pair<String, String>): Map<String, String>
+
+    /**
+     * Удалить данные по ключу.
+     * */
+    suspend fun removeData(key: String): String?
+
+    /**
+     * Удалить данные по ключу с произвольным типом.
+     * */
+    suspend fun <V: Any> removeTypedData(key: String, clazz: KClass<V>): V?
+
+    /**
+     * Очистить состояние и данные.
+     * */
+    suspend fun clear()
+}

alice-ktx/src/main/kotlin/com/github/alice/ktx/state/ReadOnlyFSMContext.kt

@@ -1,32 +1,26 @@
 package com.github.alice.ktx.state
 
-import com.github.alice.ktx.models.FSMStrategy
 import kotlin.reflect.KClass
 
 interface ReadOnlyFSMContext {
 
-    /**
-     * Получить стратегию FSM.
-     * */
-    fun getStrategy(): FSMStrategy
-
     /**
      * Получить текущее состояние.
      * */
-    fun getState(): String?
+    suspend fun getState(): String?
 
     /**
      * Получить все данные.
      * */
-    fun getData(): Map<String, String>
+    suspend fun getData(): Map<String, String>
 
     /**
      * Получить данные по ключу с произвольным типом.
      * */
-    fun <V: Any> getTypedData(key: String, clazz: KClass<V>): V?
+    suspend fun <V: Any> getTypedData(key: String, clazz: KClass<V>): V?
 
     /**
      * Получить данные по ключу.
      * */
-    fun getData(key: String): String?
+    suspend fun getData(key: String): String?
 }