main.top.mrxiaom.overflow.spi.MediaURLService.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of overflow-core-api Show documentation

One of the Overflow project modules

There is a newer version: 1.0.2

@file:Suppress("INVISIBLE_MEMBER", "INVISIBLE_REFERENCE")
package top.mrxiaom.overflow.spi

import net.mamoe.mirai.message.data.Audio
import net.mamoe.mirai.message.data.Image
import net.mamoe.mirai.message.data.ShortVideo
import net.mamoe.mirai.spi.BaseService
import net.mamoe.mirai.spi.SpiServiceLoader
import top.mrxiaom.overflow.contact.RemoteBot

/**
 * 媒体消息下载链接获取服务
 */
interface MediaURLService : BaseService {

    /**
     * 获取图片链接
     *
     * @return 图片链接，返回 null 代表不进行处理，交给其它服务或 Overflow 自带服务处理
     */
    suspend fun queryImageUrl(bot: RemoteBot, image: Image): String?

    /**
     * 获取语音链接
     *
     * @return 语音链接，返回 null 代表不进行处理，交给其它服务或 Overflow 自带服务处理
     */
    fun queryAudioUrl(audio: Audio): String?

    /**
     * 获取视频链接
     *
     * @return 视频链接，返回 null 代表不进行处理，交给其它服务或 Overflow 自带服务处理
     */
    fun queryVideoUrl(video: ShortVideo): String?

    companion object {
        private val loader = SpiServiceLoader(MediaURLService::class)

        internal val instances: List
            get() = loader.allServices.sortedBy { it.priority }

        suspend fun List.queryImageUrl(bot: RemoteBot, image: Image): String? {
            for (ext in this) {
                val url = ext.queryImageUrl(bot, image) ?: continue
                return url
            }
            return null
        }
        fun List.queryAudioUrl(audio: Audio): String? {
            for (ext in this) {
                val url = ext.queryAudioUrl(audio) ?: continue
                return url
            }
            return null
        }
        fun List.queryVideoUrl(video: ShortVideo): String? {
            for (ext in this) {
                val url = ext.queryVideoUrl(video) ?: continue
                return url
            }
            return null
        }
    }
}