mirror of
https://github.com/InsanusMokrassar/PlaguBot.git
synced 2025-09-16 13:59:26 +00:00
improve overloads for start of PlaguBot and fill logs
This commit is contained in:
@@ -2,12 +2,12 @@ package dev.inmo.plagubot
|
|||||||
|
|
||||||
import dev.inmo.kslog.common.*
|
import dev.inmo.kslog.common.*
|
||||||
import dev.inmo.micro_utils.common.Warning
|
import dev.inmo.micro_utils.common.Warning
|
||||||
|
import dev.inmo.micro_utils.coroutines.runCatchingLogging
|
||||||
import dev.inmo.micro_utils.coroutines.runCatchingSafely
|
import dev.inmo.micro_utils.coroutines.runCatchingSafely
|
||||||
import dev.inmo.micro_utils.fsm.common.State
|
import dev.inmo.micro_utils.fsm.common.State
|
||||||
import dev.inmo.micro_utils.fsm.common.StatesManager
|
import dev.inmo.micro_utils.fsm.common.StatesManager
|
||||||
import dev.inmo.micro_utils.fsm.common.managers.*
|
import dev.inmo.micro_utils.fsm.common.managers.*
|
||||||
import dev.inmo.micro_utils.koin.getAllDistinct
|
import dev.inmo.micro_utils.koin.getAllDistinct
|
||||||
import dev.inmo.micro_utils.pagination.utils.getAll
|
|
||||||
import dev.inmo.micro_utils.startup.launcher.StartLauncherPlugin
|
import dev.inmo.micro_utils.startup.launcher.StartLauncherPlugin
|
||||||
import dev.inmo.plagubot.config.*
|
import dev.inmo.plagubot.config.*
|
||||||
import dev.inmo.tgbotapi.bot.TelegramBot
|
import dev.inmo.tgbotapi.bot.TelegramBot
|
||||||
@@ -24,9 +24,39 @@ import org.koin.core.module.Module
|
|||||||
import org.koin.core.scope.Scope
|
import org.koin.core.scope.Scope
|
||||||
import java.io.File
|
import java.io.File
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Central plugin-object that:
|
||||||
|
* - Provides DI bindings for the bot, configuration and optional database
|
||||||
|
* - Creates and configures Telegram bot client
|
||||||
|
* - Builds BehaviourBuilder with FSM support and launches long polling
|
||||||
|
* - Loads and initializes other plugins
|
||||||
|
* - Exposes several start(...) overloads to bootstrap the application from JSON or strongly-typed configs
|
||||||
|
*
|
||||||
|
* Typical usage:
|
||||||
|
* - Prepare JSON or typed configs
|
||||||
|
* - Call one of start(...) methods
|
||||||
|
* - The returned Job represents the bot lifecycle; awaiting it keeps the app running
|
||||||
|
*/
|
||||||
@OptIn(Warning::class)
|
@OptIn(Warning::class)
|
||||||
@Serializable
|
@Serializable
|
||||||
object PlaguBot : Plugin {
|
object PlaguBot : Plugin {
|
||||||
|
/**
|
||||||
|
* JSON format used for encoding/decoding PlaguBot-related configuration.
|
||||||
|
*
|
||||||
|
* Note:
|
||||||
|
* - This property is not reactive. Set it only before starting the bot to affect configuration parsing/serialization.
|
||||||
|
*/
|
||||||
|
var defaultJsonFormat = dev.inmo.plagubot.config.defaultJsonFormat
|
||||||
|
@Warning("This is not reactive set. Use this only BEFORE starting of bot") set
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Allows other plugins to customize the [KtorRequestsExecutorBuilder] used by the bot client.
|
||||||
|
*
|
||||||
|
* For each plugin in DI (excluding this object), calls its own setupBotClient, passing the same [scope] and [params].
|
||||||
|
*
|
||||||
|
* @param scope Koin scope where the bot is being created
|
||||||
|
* @param params Raw JSON params of the bot part of the configuration
|
||||||
|
*/
|
||||||
override fun KtorRequestsExecutorBuilder.setupBotClient(scope: Scope, params: JsonObject) {
|
override fun KtorRequestsExecutorBuilder.setupBotClient(scope: Scope, params: JsonObject) {
|
||||||
scope.plugins.filter { it !== this@PlaguBot }.forEach {
|
scope.plugins.filter { it !== this@PlaguBot }.forEach {
|
||||||
with(it) {
|
with(it) {
|
||||||
@@ -35,6 +65,19 @@ object PlaguBot : Plugin {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Sets up application DI for the bot.
|
||||||
|
*
|
||||||
|
* Registers:
|
||||||
|
* - Decoded typed [Config] from provided [config] JSON
|
||||||
|
* - Raw [JsonObject] config itself for low-level access
|
||||||
|
* - Database-related config and database instance (if present in [Config])
|
||||||
|
* - This [PlaguBot] plugin instance
|
||||||
|
* - All plugins declared in launcher config as DI singletons
|
||||||
|
* - Configured [TelegramBot] built via [telegramBot] and customized by all plugins' setupBotClient
|
||||||
|
*
|
||||||
|
* @param config Raw PlaguBot JSON configuration
|
||||||
|
*/
|
||||||
override fun Module.setupDI(config: JsonObject) {
|
override fun Module.setupDI(config: JsonObject) {
|
||||||
single { get<Json>().decodeFromJsonElement(Config.serializer(), config) }
|
single { get<Json>().decodeFromJsonElement(Config.serializer(), config) }
|
||||||
single { config }
|
single { config }
|
||||||
@@ -55,15 +98,14 @@ object PlaguBot : Plugin {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Getting all [OnStartContextsConflictResolver], [OnUpdateContextsConflictResolver], [StatesManager] and [DefaultStatesManagerRepo]
|
* Builds and starts the bot behaviour and FSM.
|
||||||
* and pass them into [buildBehaviourWithFSM] on top of [TelegramBot] took from [koin]. In time of
|
|
||||||
* [buildBehaviourWithFSM] configuration will call [setupBotPlugin] and [deleteWebhook].
|
|
||||||
*
|
*
|
||||||
* After all preparation, the result of [buildBehaviourWithFSM] will be passed to [startGettingOfUpdatesByLongPolling]
|
* Flow:
|
||||||
* as [CoroutineScope] and [UpdatesFilter].
|
* - Collects [OnStartContextsConflictResolver], [OnUpdateContextsConflictResolver], [StatesManager] and [DefaultStatesManagerRepo] from DI (or creates defaults)
|
||||||
*
|
* - Builds BehaviourBuilder with FSM, logs errors, and calls [setupBotPlugin] for all plugins
|
||||||
* The [Job] took from [startGettingOfUpdatesByLongPolling] will be used to prevent app stopping by calling [Job.join]
|
* - Ensures webhook is removed before long polling with [deleteWebhook]
|
||||||
* on it
|
* - Starts long polling via [startGettingOfUpdatesByLongPolling] using the created behaviour context as scope and filter
|
||||||
|
* - Awaits the long-polling [Job] to keep the app alive
|
||||||
*/
|
*/
|
||||||
override suspend fun startPlugin(koin: Koin) {
|
override suspend fun startPlugin(koin: Koin) {
|
||||||
super.startPlugin(koin)
|
super.startPlugin(koin)
|
||||||
@@ -102,12 +144,18 @@ object PlaguBot : Plugin {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Initializing [Plugin]s from [koin] took by [plugins] extension. [PlaguBot] itself will be filtered out from
|
* Initializes and loads all other [Plugin]s from DI for the bot runtime.
|
||||||
* list of plugins to be inited
|
*
|
||||||
|
* Each plugin is:
|
||||||
|
* - Logged as loading started
|
||||||
|
* - Executed via plugin.setupBotPlugin(...) inside a safe block
|
||||||
|
* - Logged on failure (as warning) or success (as info)
|
||||||
|
*
|
||||||
|
* [PlaguBot] itself is excluded from this list.
|
||||||
*/
|
*/
|
||||||
override suspend fun BehaviourContextWithFSM<State>.setupBotPlugin(koin: Koin) {
|
override suspend fun BehaviourContextWithFSM<State>.setupBotPlugin(koin: Koin) {
|
||||||
koin.plugins.filter { it !== this@PlaguBot }.forEach { plugin ->
|
koin.plugins.filter { it !== this@PlaguBot }.forEach { plugin ->
|
||||||
runCatchingSafely {
|
runCatchingLogging(logger = logger) {
|
||||||
logger.i("Start loading of $plugin")
|
logger.i("Start loading of $plugin")
|
||||||
with(plugin) {
|
with(plugin) {
|
||||||
setupBotPlugin(koin)
|
setupBotPlugin(koin)
|
||||||
@@ -121,23 +169,45 @@ object PlaguBot : Plugin {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Starting plugins system using [StartLauncherPlugin.start]. In time of parsing [initialJson] [PlaguBot] may
|
* Starts the application using raw JSON and a launcher config, optionally overriding PlaguBot config.
|
||||||
* add itself in its `plugins` section in case of its absence there. So, by launching this [start] it is guaranteed
|
*
|
||||||
* that [PlaguBot] will be in list of plugins to be loaded by [StartLauncherPlugin]
|
* Behavior:
|
||||||
|
* - If [PlaguBot] is not present in [config.plugins], injects it into the plugins list
|
||||||
|
* - Optionally merges [plaguBotConfig] into [json]
|
||||||
|
* - Delegates to [StartLauncherPlugin.start] and returns the lifecycle [Job]
|
||||||
|
*
|
||||||
|
* @param json Raw JSON that may contain both launcher and PlaguBot configs
|
||||||
|
* @param config Parsed launcher configuration
|
||||||
|
* @param plaguBotConfig Optional typed PlaguBot config to merge/add
|
||||||
|
* @return Job representing the bot lifecycle
|
||||||
*/
|
*/
|
||||||
suspend fun start(initialJson: JsonObject): Job {
|
suspend fun start(
|
||||||
val initialConfig = defaultJsonFormat.decodeFromJsonElement(dev.inmo.micro_utils.startup.launcher.Config.serializer(), initialJson)
|
json: JsonObject,
|
||||||
|
config: dev.inmo.micro_utils.startup.launcher.Config,
|
||||||
|
plaguBotConfig: Config? = null
|
||||||
|
): Job {
|
||||||
KSLog.i("Config has been read")
|
KSLog.i("Config has been read")
|
||||||
|
|
||||||
// Adding of PlaguBot when it absent in config
|
// Adding of PlaguBot when it absent in config
|
||||||
val (resultJson, resultConfig) = if (PlaguBot in initialConfig.plugins) {
|
val (resultJson, resultConfig) = if (PlaguBot in config.plugins) {
|
||||||
KSLog.i("Initial config contains PlaguBot, pass config as is to StartLauncherPlugin")
|
KSLog.i("Initial config contains PlaguBot, pass config as is to StartLauncherPlugin")
|
||||||
initialJson to initialConfig
|
json to config
|
||||||
} else {
|
} else {
|
||||||
KSLog.i("Start fixing of PlaguBot absence. If PlaguBot has been skipped by some reason, use dev.inmo.micro_utils.startup.launcher.main as startup point or StartLauncherPlugin directly")
|
KSLog.i("Start fixing of PlaguBot absence. If PlaguBot has been skipped by some reason, use dev.inmo.micro_utils.startup.launcher.main as startup point or StartLauncherPlugin directly")
|
||||||
|
|
||||||
|
val encodedPlaguBotConfig = plaguBotConfig ?.let {
|
||||||
|
defaultJsonFormat.encodeToJsonElement(Config.serializer(), it).jsonObject
|
||||||
|
} ?: JsonObject(emptyMap())
|
||||||
|
|
||||||
val resultJson = JsonObject(
|
val resultJson = JsonObject(
|
||||||
initialJson + Pair("plugins", JsonArray(initialJson["plugins"]!!.jsonArray + JsonPrimitive(PlaguBot::class.qualifiedName!!)))
|
encodedPlaguBotConfig + JsonObject(
|
||||||
|
json + Pair(
|
||||||
|
"plugins",
|
||||||
|
JsonArray(
|
||||||
|
(json["plugins"] as? JsonArray ?: JsonArray(emptyList())) + JsonPrimitive(PlaguBot::class.qualifiedName!!)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
)
|
||||||
)
|
)
|
||||||
val resultConfig = defaultJsonFormat.decodeFromJsonElement(dev.inmo.micro_utils.startup.launcher.Config.serializer(), resultJson)
|
val resultConfig = defaultJsonFormat.decodeFromJsonElement(dev.inmo.micro_utils.startup.launcher.Config.serializer(), resultJson)
|
||||||
resultJson to resultConfig
|
resultJson to resultConfig
|
||||||
@@ -151,9 +221,91 @@ object PlaguBot : Plugin {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Accepts single argument in [args] which will be interpreted as [File] path with [StartLauncherPlugin]
|
* Starts the application using typed launcher config and typed PlaguBot config.
|
||||||
* configuration content. After reading of that file as [JsonObject] will pass it in [start] with [JsonObject] as
|
*
|
||||||
* argument
|
* Behavior:
|
||||||
|
* - Merges both configs into a single JSON
|
||||||
|
* - Ensures [PlaguBot] is present in plugins list
|
||||||
|
* - Delegates to [start] that accepts JSON and launcher config
|
||||||
|
*
|
||||||
|
* @param pluginsConfig Launcher configuration
|
||||||
|
* @param plaguBotConfig Typed PlaguBot configuration
|
||||||
|
* @return Job representing the bot lifecycle
|
||||||
|
*/
|
||||||
|
suspend fun start(
|
||||||
|
pluginsConfig: dev.inmo.micro_utils.startup.launcher.Config,
|
||||||
|
plaguBotConfig: Config
|
||||||
|
): Job {
|
||||||
|
val json = JsonObject(
|
||||||
|
defaultJsonFormat.encodeToJsonElement(
|
||||||
|
dev.inmo.micro_utils.startup.launcher.Config.serializer(),
|
||||||
|
pluginsConfig
|
||||||
|
).jsonObject + defaultJsonFormat.encodeToJsonElement(
|
||||||
|
Config.serializer(),
|
||||||
|
plaguBotConfig
|
||||||
|
).jsonObject
|
||||||
|
)
|
||||||
|
val pluginsConfig = if (pluginsConfig.plugins.contains(PlaguBot)) {
|
||||||
|
pluginsConfig
|
||||||
|
} else {
|
||||||
|
pluginsConfig.copy(
|
||||||
|
plugins = pluginsConfig.plugins + PlaguBot,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
return start(json, pluginsConfig)
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Starts the application using raw JSON and a typed PlaguBot config.
|
||||||
|
*
|
||||||
|
* Behavior:
|
||||||
|
* - Encodes [plaguBotConfig] and merges it into [initialJson]
|
||||||
|
* - Delegates to [start] that takes a JSON only
|
||||||
|
*
|
||||||
|
* @param initialJson Initial JSON to start from
|
||||||
|
* @param plaguBotConfig Typed PlaguBot configuration that will be encoded and merged
|
||||||
|
* @return Job representing the bot lifecycle
|
||||||
|
*/
|
||||||
|
suspend fun start(
|
||||||
|
initialJson: JsonObject,
|
||||||
|
plaguBotConfig: Config
|
||||||
|
): Job {
|
||||||
|
val encodedPlaguBotConfig = defaultJsonFormat.encodeToJsonElement(Config.serializer(), plaguBotConfig).jsonObject
|
||||||
|
|
||||||
|
return start(
|
||||||
|
JsonObject(initialJson + encodedPlaguBotConfig)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Starts the plugins system using [StartLauncherPlugin.start].
|
||||||
|
*
|
||||||
|
* Parsing notes:
|
||||||
|
* - [initialJson] is decoded into launcher config
|
||||||
|
* - If [PlaguBot] is missing from the plugins list, it will be added automatically
|
||||||
|
*
|
||||||
|
* By using this method it is guaranteed that [PlaguBot] will be included into the set of plugins to launch.
|
||||||
|
*
|
||||||
|
* @param initialJson Raw JSON that includes launcher configuration (and optionally PlaguBot config)
|
||||||
|
* @return Job representing the bot lifecycle
|
||||||
|
*/
|
||||||
|
suspend fun start(initialJson: JsonObject): Job {
|
||||||
|
val initialConfig = defaultJsonFormat.decodeFromJsonElement(dev.inmo.micro_utils.startup.launcher.Config.serializer(), initialJson)
|
||||||
|
|
||||||
|
return start(initialJson, initialConfig)
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Starts the application from CLI arguments.
|
||||||
|
*
|
||||||
|
* Expects:
|
||||||
|
* - Exactly one argument: path to a file with JSON configuration
|
||||||
|
*
|
||||||
|
* The file is read, parsed as [JsonObject], and passed to [start(JsonObject)].
|
||||||
|
*
|
||||||
|
* @param args Command-line arguments; first element is a path to the config file
|
||||||
|
* @return Job representing the bot lifecycle
|
||||||
*/
|
*/
|
||||||
suspend fun start(args: Array<String>): Job {
|
suspend fun start(args: Array<String>): Job {
|
||||||
KSLog.default = KSLog("PlaguBot")
|
KSLog.default = KSLog("PlaguBot")
|
||||||
|
Reference in New Issue
Block a user