com.github.insanusmokrassar.AutoPostTelegramBot.utils.BotLogger.kt Maven / Gradle / Ivy

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

import com.github.insanusmokrassar.AutoPostTelegramBot.base.plugins.commonLogger
import com.github.insanusmokrassar.AutoPostTelegramBot.utils.extensions.splitForMessageWithAdditionalStep
import com.github.insanusmokrassar.TelegramBotAPI.bot.RequestException
import com.github.insanusmokrassar.TelegramBotAPI.bot.RequestsExecutor
import com.github.insanusmokrassar.TelegramBotAPI.requests.send.SendMessage
import com.github.insanusmokrassar.TelegramBotAPI.types.ChatIdentifier
import com.github.insanusmokrassar.TelegramBotAPI.types.ParseMode.MarkdownParseMode
import kotlinx.coroutines.*
import kotlinx.coroutines.channels.Channel
import java.lang.ref.WeakReference
import java.util.logging.*

private class LoggerHandler(
    loggerToHandle: Logger,
    executor: RequestsExecutor,
    private val logsChatId: ChatIdentifier
) : Handler() {
    private val botWR = WeakReference(executor)

    private val logsChannel = Channel(Channel.UNLIMITED)
    private val recordsSendingJob = GlobalScope.launch {
        coroutineScope {
            launch {
                for (msg in logsChannel) {
                    val bot = botWR.get() ?: break
                    formatter.format(msg).splitForMessageWithAdditionalStep(6).forEach { record ->
                        try {
                            bot.execute(
                                SendMessage(
                                    logsChatId,
                                    record,
                                    MarkdownParseMode
                                )
                            )
                        } catch (e: RequestException) {
                            bot.execute(
                                SendMessage(
                                    logsChatId,
                                    record
                                )
                            )
                        }
                    }
                }
            }
        }
    }

    init {
        loggerToHandle.addHandler(this)

        formatter = SimpleFormatter()
    }

    override fun publish(record: LogRecord?) {
        record ?: return
        GlobalScope.launch {
            logsChannel.send(record)
        }
    }

    override fun flush() {}

    override fun close() {
        botWR.clear()
        logsChannel.cancel()
        recordsSendingJob.cancel()
    }
}

@Synchronized
fun initHandler(executor: RequestsExecutor, logsChatId: ChatIdentifier) {
    commonLogger.handlers.firstOrNull { it is LoggerHandler } ?: run {
        LoggerHandler(commonLogger, executor, logsChatId)
    }
}