io.github.dehuckakpyt.telegrambot.source.callback.DatabaseCallbackContentSource.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of telegram-bot-source-jpa Show documentation

Kotlin Telegram Bot library

There is a newer version: 0.11.2

Show newest version

package io.github.dehuckakpyt.telegrambot.source.callback

import io.github.dehuckakpyt.telegrambot.constant.Default.MAX_CALLBACK_CONTENTS_PER_USER
import io.github.dehuckakpyt.telegrambot.exception.chat.ChatException
import io.github.dehuckakpyt.telegrambot.model.callback.DatabaseCallbackContent
import io.github.dehuckakpyt.telegrambot.model.source.CallbackContent
import io.github.dehuckakpyt.telegrambot.repository.OffsetLimitPageable
import io.github.dehuckakpyt.telegrambot.repository.callback.DatabaseCallbackContentRepository
import org.springframework.transaction.annotation.Transactional
import java.time.LocalDateTime
import java.util.*

open class DatabaseCallbackContentSource(
    private val repository: DatabaseCallbackContentRepository,
    private val maxCallbackContentsPerUser: Long = MAX_CALLBACK_CONTENTS_PER_USER,
) : CallbackContentSource {

    private val pageable = OffsetLimitPageable(maxCallbackContentsPerUser - 1, 1)

    @Transactional
    override suspend fun save(chatId: Long, fromId: Long, content: String): CallbackContent {
        val callbackContent = findLast(chatId, fromId)?.apply {
            this.callbackId = UUID.randomUUID()
            this.content = content
            this.updateDate = LocalDateTime.now()
        } ?: DatabaseCallbackContent(
            chatId = chatId,
            fromId = fromId,
            callbackId = UUID.randomUUID(),
            content = content,
            updateDate = LocalDateTime.now(),
        )

        return repository.save(callbackContent)
    }

    @Transactional(readOnly = true)
    override suspend fun get(callbackId: UUID): CallbackContent {
        return repository.findFirstByCallbackId(callbackId)
            ?: throw ChatException("Содержание для callback'а не найдено :(")
    }

    private fun findLast(chatId: Long, fromId: Long): DatabaseCallbackContent? {
        if (maxCallbackContentsPerUser < 1) return null

        return repository.findByChatIdAndFromIdOrderByUpdateDateDesc(chatId, fromId, pageable).firstOrNull()
    }
}