xyz.cssxsh.mirai.bilibili.command.BiliDynamicCommand.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bilibili-helper Show documentation
bilibili-helper
The newest version!
package xyz.cssxsh.mirai.bilibili.command

import com.cronutils.model.*
import net.mamoe.mirai.console.command.*
import net.mamoe.mirai.console.permission.*
import net.mamoe.mirai.contact.*
import xyz.cssxsh.bilibili.api.*
import xyz.cssxsh.bilibili.data.*
import xyz.cssxsh.mirai.bilibili.*
import xyz.cssxsh.mirai.bilibili.data.*
import java.time.*

object BiliDynamicCommand : CompositeCommand(
    owner = BiliHelperPlugin,
    "bili-dynamic", "B动态",
    description = "B站动态指令",
    overrideContext = BiliCommandArgumentContext
), BiliTasker by BiliDynamicLoader {

    @SubCommand("add", "添加")
    suspend fun CommandSender.add(uid: Long, contact: Contact = subject()) = sendMessage(
        message = task(uid, contact)
    )

    @SubCommand("stop", "停止")
    suspend fun CommandSender.stop(uid: Long, contact: Contact = subject()) = sendMessage(
        message = remove(uid, contact)
    )

    @SubCommand("time", "定时")
    suspend fun CommandSender.cron(uid: Long, cron: Cron) = sendMessage(
        message = time(uid, cron)
    )

    @SubCommand("list", "列表")
    suspend fun CommandSender.detail(contact: Contact = subject()) = sendMessage(
        message = list(contact)
    )

    @SubCommand("forbid", "屏蔽")
    suspend fun CommandSender.forbid(pattern: String, add: Boolean = true) {
        val message = if (add) {
            try {
                pattern.toRegex()
                BiliTaskerConfig.dynamicForbidRegexes.add(pattern)
                "动态正则屏蔽添加成功"
            } catch (cause: Exception) {
                logger.warning(cause)
                "动态正则屏蔽添加失败，${cause.message}"
            }
        } else {
            BiliTaskerConfig.dynamicForbidRegexes.remove(pattern)
            "动态正则屏蔽取消成功"
        }

        sendMessage(message = message)
    }

    @SubCommand("filter", "过滤")
    suspend fun CommandSender.filter(type: String, add: Boolean = true) {
        val keys = when (type) {
            "回复" -> DynamicType.DYNAMIC_TYPE_FORWARD.keys
            "图片" -> DynamicType.DYNAMIC_TYPE_DRAW.keys
            "文本" -> DynamicType.DYNAMIC_TYPE_WORD.keys
            "视频" -> DynamicType.DYNAMIC_TYPE_AV.keys + DynamicType.DYNAMIC_TYPE_MEDIALIST.keys
            "专栏" -> DynamicType.DYNAMIC_TYPE_ARTICLE.keys
            "音乐" -> DynamicType.DYNAMIC_TYPE_MUSIC.keys
            "剧集" -> DynamicType.DYNAMIC_TYPE_PGC.keys
            "删除" -> DynamicType.DYNAMIC_TYPE_NONE.keys
            "番剧" -> setOf(0x0000_0200)
            "电视" -> setOf(0x0000_1003)
            "直播" -> DynamicType.DYNAMIC_TYPE_LIVE.keys + DynamicType.DYNAMIC_TYPE_LIVE_RCMD.keys
            else -> {
                sendMessage(message = "莫得这个选项 <$type>")
                return
            }
        }

        if (add) {
            BiliTaskerConfig.dynamicForbidType.addAll(keys)
            sendMessage(message = "动态类型 $keys 屏蔽添加成功")
        } else {
            BiliTaskerConfig.dynamicForbidType.removeIf { it in keys }
            sendMessage(message = "动态类型 $keys 屏蔽取消成功")
        }
    }

    @SubCommand("sleep", "睡眠")
    suspend fun CommandSender.sleep(target: PermitteeId, start: LocalTime, end: LocalTime) {
        val interval = BiliInterval(start, end)
        if (interval.isEmpty()) {
            sleep.remove(target)
            sendMessage("睡眠时间取消成功")
        } else {
            sleep[target] = BiliInterval(start, end)
            sendMessage("睡眠时间添加成功")
        }
    }

    @SubCommand("at", "艾特")
    suspend fun CommandSender.at(target: PermitteeId, start: LocalTime, end: LocalTime) {
        val interval = BiliInterval(start, end)
        if (interval.isEmpty()) {
            at.remove(target)
            sendMessage("艾特时间取消成功")
        } else {
            at[target] = BiliInterval(start, end)
            sendMessage("艾特时间添加成功")
        }
    }

    @SubCommand("test", "测试")
    suspend fun CommandSender.test(dynamicId: Long) {
        val dynamic = client.getDynamicInfo(dynamicId = dynamicId).dynamic
        val match = with(BiliDynamicLoader) {
            dynamic.check()
        }
        if (match) {
            sendMessage("没有命中")
        } else {
            sendMessage("命中过滤")
        }
    }
}