PlaguBot/bot/src/main/kotlin/dev/inmo/plagubot/PlaguBot.kt

80 lines
2.4 KiB
Kotlin
Raw Normal View History

2021-04-03 16:04:18 +00:00
package dev.inmo.plagubot
import dev.inmo.plagubot.config.*
2022-05-13 07:04:49 +00:00
import dev.inmo.tgbotapi.bot.ktor.telegramBot
2022-05-16 17:58:16 +00:00
import dev.inmo.tgbotapi.extensions.api.webhook.deleteWebhook
2021-11-12 08:05:42 +00:00
import dev.inmo.tgbotapi.extensions.behaviour_builder.*
2022-05-16 14:07:57 +00:00
import dev.inmo.tgbotapi.extensions.utils.updates.retrieving.startGettingOfUpdatesByLongPolling
2021-04-03 16:04:18 +00:00
import kotlinx.coroutines.*
import kotlinx.serialization.Serializable
import kotlinx.serialization.Transient
2022-05-16 14:07:57 +00:00
import kotlinx.serialization.json.JsonObject
2021-04-03 16:04:18 +00:00
import org.jetbrains.exposed.sql.Database
2022-05-16 17:58:16 +00:00
import org.koin.core.Koin
2022-05-16 14:07:57 +00:00
import org.koin.core.KoinApplication
import org.koin.core.context.GlobalContext
2022-05-16 17:58:16 +00:00
import org.koin.core.module.Module
import org.koin.core.scope.Scope
2022-05-16 14:07:57 +00:00
import org.koin.dsl.module
2021-04-03 16:04:18 +00:00
2022-05-16 17:58:16 +00:00
val Scope.plagubot: PlaguBot
2022-05-16 14:07:57 +00:00
get() = get()
2021-04-03 16:04:18 +00:00
@Serializable
data class PlaguBot(
2022-05-16 14:07:57 +00:00
private val json: JsonObject,
2021-04-03 16:04:18 +00:00
private val config: Config
) : Plugin {
@Transient
private val bot = telegramBot(config.botToken)
2022-05-16 17:58:16 +00:00
override fun Module.setupDI(database: Database, params: JsonObject) {
single { config }
single { config.plugins }
single { config.databaseConfig }
single { config.databaseConfig.database }
single { defaultJsonFormat }
single { this@PlaguBot }
includes(
config.plugins.map {
module {
with(it) {
setupDI(database, params)
}
}
}
)
}
override suspend fun BehaviourContext.setupBotPlugin(koin: Koin) {
2021-04-03 16:04:18 +00:00
config.plugins.forEach {
2022-05-16 17:58:16 +00:00
with(it) {
setupBotPlugin(koin)
}
2021-04-03 16:04:18 +00:00
}
}
/**
* This method will create an [Job] which will be the main [Job] of ran instance
*/
suspend fun start(
2022-05-16 17:58:16 +00:00
scope: CoroutineScope = CoroutineScope(Dispatchers.IO)
2022-05-16 14:07:57 +00:00
): Job {
val koinApp = KoinApplication.init()
koinApp.modules(
module {
2022-05-16 17:58:16 +00:00
setupDI(config.databaseConfig.database, json)
2022-05-16 14:07:57 +00:00
}
)
2022-05-16 17:58:16 +00:00
GlobalContext.startKoin(koinApp)
2022-05-16 14:07:57 +00:00
lateinit var behaviourContext: BehaviourContext
bot.buildBehaviour(scope = scope) {
behaviourContext = this
2022-05-16 17:58:16 +00:00
setupBotPlugin(koinApp.koin)
deleteWebhook()
2022-05-16 14:07:57 +00:00
}
return bot.startGettingOfUpdatesByLongPolling(scope = behaviourContext, updatesFilter = behaviourContext)
2021-04-03 16:04:18 +00:00
}
}