81 lines
2.1 KiB
Kotlin
81 lines
2.1 KiB
Kotlin
package dev.inmo.tgbotapi.utils
|
|
|
|
import com.benasher44.uuid.uuid4
|
|
import io.ktor.utils.io.*
|
|
import io.ktor.utils.io.core.ByteReadPacket
|
|
import io.ktor.utils.io.core.Input
|
|
import kotlinx.serialization.Serializable
|
|
|
|
/**
|
|
* Information about file for [StorageFile]
|
|
*
|
|
* @param contentType Raw type like "application/json"
|
|
* @param fileName This filename will be used in telegram system as name of file
|
|
*/
|
|
@Serializable
|
|
data class StorageFileInfo(
|
|
val contentType: String,
|
|
val fileName: String
|
|
) {
|
|
/**
|
|
* This methods is required for random generation of name for keeping warranties about unique file name
|
|
*/
|
|
fun generateCustomName() = "${uuid4()}.${fileName.fileExtension}"
|
|
}
|
|
|
|
/**
|
|
* Contains info about file, which potentially can be sent to telegram system.
|
|
*
|
|
* @param storageFileInfo Information about this file
|
|
* @param inputSource Lambda which able to allocate [Input] for uploading/manipulating data
|
|
*
|
|
* @see StorageFileInfo
|
|
* @see asStorageFile
|
|
*/
|
|
data class StorageFile(
|
|
val storageFileInfo: StorageFileInfo,
|
|
private val inputSource: () -> Input
|
|
) {
|
|
val input: Input
|
|
get() = inputSource()
|
|
}
|
|
|
|
@Suppress("NOTHING_TO_INLINE")
|
|
inline fun StorageFile(
|
|
fileName: String,
|
|
bytes: ByteArray,
|
|
mimeType: MimeType
|
|
) = StorageFile(
|
|
StorageFileInfo(mimeType.raw, fileName)
|
|
) {
|
|
ByteReadPacket(bytes)
|
|
}
|
|
|
|
@Suppress("NOTHING_TO_INLINE")
|
|
suspend inline fun StorageFile(
|
|
fileName: String,
|
|
byteReadChannel: ByteReadChannel,
|
|
mimeType: MimeType
|
|
) = StorageFile(
|
|
StorageFileInfo(mimeType.raw, fileName),
|
|
byteReadChannel.asInput().let { { it } }
|
|
)
|
|
|
|
@Suppress("NOTHING_TO_INLINE", "unused")
|
|
inline fun ByteArray.asStorageFile(
|
|
fileName: String,
|
|
mimeType: MimeType
|
|
) = StorageFile(fileName, this, mimeType)
|
|
|
|
@Suppress("NOTHING_TO_INLINE", "unused")
|
|
suspend inline fun ByteReadChannel.asStorageFile(
|
|
fileName: String,
|
|
mimeType: MimeType
|
|
) = StorageFile(fileName, this, mimeType)
|
|
|
|
@Suppress("NOTHING_TO_INLINE", "unused")
|
|
suspend inline fun ByteReadChannelAllocator.asStorageFile(
|
|
fileName: String,
|
|
mimeType: MimeType
|
|
) = this.invoke().asStorageFile(fileName, mimeType)
|