com.github.insanusmokrassar.AutoPostTelegramBot.plugins.RatingTimerAutoDisablePlugin.kt Maven / Gradle / Ivy

Go to download
package com.github.insanusmokrassar.AutoPostTelegramBot.plugins

import com.github.insanusmokrassar.AutoPostTelegramBot.base.models.FinalConfig
import com.github.insanusmokrassar.AutoPostTelegramBot.base.plugins.*
import com.github.insanusmokrassar.AutoPostTelegramBot.plugins.rating.RatingPlugin
import com.github.insanusmokrassar.AutoPostTelegramBot.plugins.rating.disableLikesForPost
import com.github.insanusmokrassar.AutoPostTelegramBot.plugins.scheduler.SchedulerPlugin
import com.github.insanusmokrassar.AutoPostTelegramBot.utils.extensions.subscribe
import com.github.insanusmokrassar.AutoPostTelegramBot.utils.extensions.subscribeChecking
import com.pengrad.telegrambot.TelegramBot
import java.lang.ref.WeakReference

class RatingTimerAutoDisablePlugin : Plugin {
    override fun onInit(bot: TelegramBot, baseConfig: FinalConfig, pluginManager: PluginManager) {
        val ratingPlugin: RatingPlugin = pluginManager.plugins.firstOrNull {
            it is RatingPlugin
        } as? RatingPlugin ?:let {
            commonLogger.warning("Plugin $name was not load for the reason that rating plugin was not found")
            return
        }

        val schedulerPlugin: SchedulerPlugin = (pluginManager.plugins.firstOrNull {
            it is SchedulerPlugin
        } as? SchedulerPlugin) ?:let {
            commonLogger.warning("Plugin $name was not load for the reason that scheduler plugin was not found")
            return
        }

        val botWR = WeakReference(bot)
        val sourceChatId = baseConfig.sourceChatId

        schedulerPlugin.timerSchedulesTable.postTimeRegisteredChannel.subscribeChecking(
            {
                commonLogger.throwing(
                    name,
                    "register post time scheduler registered",
                    it
                )
                true
            }
        ) {
            botWR.get() ?.let {
                bot ->

                disableLikesForPost(
                    it.first,
                    bot,
                    sourceChatId,
                    ratingPlugin.postsLikesMessagesTable
                )

                true
            } ?: false
        }

        ratingPlugin.postsLikesMessagesTable.ratingMessageRegisteredChannel.subscribe(
            {
                commonLogger.throwing(
                    name,
                    "register post rating enabled",
                    it
                )
                true
            }
        ) {
            schedulerPlugin.timerSchedulesTable.unregisterPost(it.first)
        }
    }
}