connector.SampleRestConnectorCallback.kt Maven / Gradle / Ivy

Go to download
package ai.tock.bot.open.data.connector

import ai.tock.bot.connector.ConnectorCallbackBase
import ai.tock.bot.engine.action.Action
import ai.tock.bot.engine.action.SendSentence
import ai.tock.bot.engine.event.Event
import ai.tock.shared.jackson.mapper
import io.vertx.ext.web.RoutingContext
import mu.KotlinLogging
import java.util.Locale
import java.util.concurrent.CopyOnWriteArrayList

internal class SampleRestConnectorCallback(
    applicationId: String,
    val locale: Locale,
    private val context: RoutingContext,
    private val actions: MutableList = CopyOnWriteArrayList()
) : ConnectorCallbackBase(applicationId, sampleRestConnectorType) {

    private val logger = KotlinLogging.logger {}


    fun addAction(event: Event) {
        if (event is Action) {
            actions.add(event)
        } else {
            logger.trace { "unsupported event: $event" }
        }
    }

    fun sendResponse() {
        val messages = actions
            .filterIsInstance()
            .mapNotNull {
                if (it.stringText != null) {
                    SampleMessage(it.stringText!!)
                } else it.message(sampleRestConnectorType)?.let {
                    it as? SampleMessage
                }

            }
        context.response().end(mapper.writeValueAsString(SampleConnectorResponse(messages)))
    }
}