77 lines
2.4 KiB
Kotlin
77 lines
2.4 KiB
Kotlin
package dev.inmo.tgbotapi.CommonAbstracts
|
|
|
|
import dev.inmo.tgbotapi.types.MessageEntity.toTextParts
|
|
import dev.inmo.tgbotapi.types.captionLength
|
|
import dev.inmo.tgbotapi.types.textLength
|
|
|
|
typealias FullTextSourcesList = List<TextSource>
|
|
typealias FullTextPartsList = List<TextPart>
|
|
|
|
interface TextSource {
|
|
val asMarkdownSource: String
|
|
val asMarkdownV2Source: String
|
|
val asHtmlSource: String
|
|
val source: String
|
|
}
|
|
|
|
|
|
interface MultilevelTextSource : TextSource {
|
|
@Deprecated("Will be removed in near major release")
|
|
val textParts: List<TextPart>
|
|
get() = textParts(0)
|
|
val textSources: List<TextSource>
|
|
}
|
|
|
|
data class TextPart(
|
|
val range: IntRange,
|
|
val source: TextSource
|
|
)
|
|
|
|
fun List<TextPart>.justTextSources() = map { it.source }
|
|
fun List<TextSource>.makeString() = joinToString("") { it.source }
|
|
internal fun MultilevelTextSource.textParts(offset: Int): List<TextPart> = textSources.toTextParts(offset)
|
|
fun List<TextSource>.separateForMessage(limit: IntRange, numberOfParts: Int? = null): List<List<TextSource>> {
|
|
if (isEmpty()) {
|
|
return emptyList()
|
|
}
|
|
|
|
val resultList = mutableListOf<MutableList<TextSource>>(mutableListOf())
|
|
var currentPartLength = 0
|
|
val maxSize = limit.last + 1
|
|
|
|
for (current in this) {
|
|
if (current.source.length > maxSize) {
|
|
error("Currently unsupported parts with size more than target one-message parts (${current.source.length} > ${maxSize})")
|
|
}
|
|
|
|
if (currentPartLength + current.source.length > maxSize) {
|
|
if (numberOfParts == null || numberOfParts < resultList.size) {
|
|
resultList.add(mutableListOf())
|
|
currentPartLength = 0
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
|
|
resultList.last().add(current)
|
|
currentPartLength += current.source.length
|
|
}
|
|
|
|
return resultList
|
|
}
|
|
|
|
/**
|
|
* This method will prepare [TextSource]s list for messages. Remember, that first part will be separated with
|
|
* [captionLength] and all others with
|
|
*/
|
|
fun List<TextSource>.separateForCaption(): List<List<TextSource>> {
|
|
val captionPart = separateForMessage(captionLength, 1).first()
|
|
return listOf(captionPart) + minus(captionPart).separateForMessage(textLength)
|
|
}
|
|
|
|
/**
|
|
* This method will prepare [TextSource]s list for messages with [textLength]
|
|
*/
|
|
@Suppress("NOTHING_TO_INLINE")
|
|
inline fun List<TextSource>.separateForText(): List<List<TextSource>> = separateForMessage(textLength)
|