KtorPipelineStepsHolder.kt

package me.centralhardware.znatoki.telegram.statistic.telegram

import dev.inmo.tgbotapi.bot.ktor.KtorCallFactory
import dev.inmo.tgbotapi.bot.ktor.KtorPipelineStepsHolder
import dev.inmo.tgbotapi.requests.GetUpdates
import dev.inmo.tgbotapi.requests.abstracts.Request
import io.ktor.client.plugins.HttpRequestTimeoutException

class KtorPipelineStepsHolder : KtorPipelineStepsHolder {

    companion object {
        var health: Boolean = false;
    }

    override suspend fun <T : Any> onRequestException(request: Request<T>, t: Throwable): T? {
        if (t is HttpRequestTimeoutException &&
            t.message!!.startsWith("Request timeout has expired [url=https://api.telegram.org/bot")
        ) {
            health = true;
        } else {
            health = false;
        }

        return super.onRequestException(request, t)
    }

    override suspend fun <T : Any> onRequestReturnResult(
        result: Result<T>,
        request: Request<T>,
        callsFactories: List<KtorCallFactory>
    ): T {
        if (request is GetUpdates && result.isSuccess) {
            health = true
        }

        return super.onRequestReturnResult(result, request, callsFactories)
    }

}