import type { Bot } from 'el-bot'
import type { EventType, MessageType } from 'mirai-ts'
import { check } from 'mirai-ts'
import { block, displayList, initBlacklist, unBlock } from './utils'

export default async function (ctx: Bot) {
  if (!ctx.db)
    return
  const { mirai, cli } = ctx

  const blacklist = await initBlacklist()

  mirai.beforeListener.push(
    (msg: MessageType.ChatMessage | EventType.Event) => {
      const isFriendBlocked
        = check.isChatMessage(msg) && blacklist.friends.has(msg.sender.id)
      const isGroupBlocked
        = msg.type === 'GroupMessage'
        && blacklist.groups.has(msg.sender.group.id)

      if (isFriendBlocked || isGroupBlocked)
        mirai.active = false
      else
        mirai.active = true
    },
  )

  // register command
  // 显示当前已有的黑名单
  cli
    .command('blacklist')
    .description('黑名单')
    .option('-l, --list <type>', '当前列表', 'all')
    .action(async (options) => {
      if (!ctx.user.isAllowed(undefined, true))
        return
      const listType = options.list
      if (listType) {
        if (['friend', 'user', 'all'].includes(listType))
          ctx.reply(`当前用户黑名单：${displayList(blacklist.friends)}`)

        if (['group', 'all'].includes(listType))
          ctx.reply(`当前群聊黑名单：${displayList(blacklist.groups)}`)
      }
    })

  cli
    .command('block <type> [id]')
    .description('封禁')
    .action(async (type, id) => {
      if (!ctx.user.isAllowed(undefined, true))
        return
      const msg = ctx.mirai.curMsg
      if (await block(type, Number.parseInt(id))) {
        const info = `[blacklist] 封禁 ${type} ${id}`
        msg!.reply!(info)
      }
    })

  cli
    .command('unblock <type> [id]')
    .description('解封')
    .action(async (type, id) => {
      if (!ctx.user.isAllowed(undefined, true))
        return
      const msg = ctx.mirai.curMsg
      if (await unBlock(type, Number.parseInt(id))) {
        const info = `[blacklist] 解封 ${type} ${id}`
        msg!.reply!(info)
      }
    })
}