yenai-plugin/model/GroupBannedWords.js

import _ from 'lodash'
import moment from 'moment'
import { Data, Plugin_Path } from '../components/index.js'

export default new class {
  constructor () {
    this.root = `${Plugin_Path}/config/group`
    this.penaltyTypeMap = {
      1: '踢',
      2: '禁',
      3: '撤',
      4: '踢撤',
      5: '禁撤'
    }
    this.matchTypeMap = {
      1: '精确',
      2: '模糊',
      3: '正则'
    }
    this.dataCach = new Map()
    this.muteTimeCach = new Map()
    this.groupTitleCach = new Map()
  }

  addBannedWords (
    groupId, words, matchType = '精确', penaltyType = '禁', addedBy
  ) {
    let data = Data.readJSON(`${groupId}.json`, this.root)
    if (!data.bannedWords) data.bannedWords = {}
    if (data.bannedWords[words]) throw Error(`❎ 违禁词${words}已存在`)
    // 翻转对象
    let matchTypeMapMirr = _.invert(this.matchTypeMap)
    let penaltyTypeMapMirr = _.invert(this.penaltyTypeMap)
    data.bannedWords[words] = {
      matchType: Number(matchTypeMapMirr[matchType]),
      penaltyType: Number(penaltyTypeMapMirr[penaltyType]),
      date: moment().format('MMM Do YY'),
      addedBy
    }
    Data.writeJSON(`${groupId}.json`, data, this.root)
    this.dataCach.delete(groupId)
    return {
      words: this.keyWordTran(words),
      matchType,
      penaltyType
    }
  }

  delBannedWords (groupId, words) {
    let data = Data.readJSON(`${groupId}.json`, this.root)
    if (!data.bannedWords[words]) throw Error(`❎ 违禁词${words}不存在`)
    delete data.bannedWords[words]
    this.dataCach.delete(groupId)
    Data.writeJSON(`${groupId}.json`, data, this.root)
    return this.keyWordTran(words)
  }

  queryBannedWords (groupId, words) {
    let data = Data.readJSON(`${groupId}.json`, this.root)
    if (!data.bannedWords[words]) throw Error(`❎ 违禁词${words}不存在`)
    let { matchType, penaltyType } = data.bannedWords[words]
    return {
      ...data.bannedWords[words],
      words: this.keyWordTran(words),
      matchType: this.matchTypeMap[matchType],
      penaltyType: this.penaltyTypeMap[penaltyType]
    }
  }

  setMuteTime (groupId, time) {
    let data = Data.readJSON(`${groupId}.json`, this.root)
    data.muteTime = Number(time)
    Data.writeJSON(`${groupId}.json`, data, this.root)
    this.muteTimeCach.delete(groupId)
    return true
  }

  getMuteTime (groupId) {
    if (this.muteTimeCach.get(groupId)) return this.muteTimeCach.get(groupId)
    let data = Data.readJSON(`${groupId}.json`, this.root)
    this.muteTimeCach.set(groupId, data.muteTime ?? 300)
    return data.muteTime ?? 300
  }

  /**
   * 关键词转换成可发送消息
   * @param msg
   */
  async keyWordTran (msg) {
    /** 图片 */
    if (msg.includes('{image')) {
      let tmp = msg.split('{image')
      if (tmp.length > 2) return false

      let md5 = tmp[1].replace(/}|_|:/g, '')

      msg = segment.image(`http://gchat.qpic.cn/gchatpic_new/0/0-0-${md5}/0`)
      msg.asface = true
    } else if (msg.includes('{at:')) {
      let tmp = msg.match(/{at:(.+?)}/g)

      for (let qq of tmp) {
        qq = qq.match(/[1-9][0-9]{4,14}/g)[0]
        let member = await Bot.getGroupMemberInfo(this.group_id, Number(qq)).catch(() => { })
        let name = member?.card ?? member?.nickname
        if (!name) continue
        msg = msg.replace(`{at:${qq}}`, `@${name}`)
      }
    } else if (msg.includes('{face')) {
      let tmp = msg.match(/{face(:|_)(.+?)}/g)
      if (!tmp) return msg
      msg = []
      for (let face of tmp) {
        let id = face.match(/\d+/g)
        msg.push(segment.face(id))
      }
    }

    return msg
  }

  /**
   * 初始化已添加内容
   * @param groupId
   */
  initTextArr (groupId) {
    if (this.dataCach.get(groupId)) return this.dataCach.get(groupId)

    try {
      const data = Data.readJSON(`${groupId}.json`, this.root)?.bannedWords
      const _data = new Map()
      for (const item in data) {
        data[item].rawItem = item
        if (data[item].matchType == 2) {
          _data.set(new RegExp(item), data[item])
        } else if (data[item].matchType == 3) {
          _data.set(global.eval(item), data[item])
        } else {
          _data.set(new RegExp(`^${item}$`), data[item])
        }
      }
      this.dataCach.set(groupId, _data)
      return _data
    } catch (error) {
      logger.error(error)
      logger.error(`json格式错误：${this.root}/${groupId}.json`)
      delete this.dataCach[groupId]
      return false
    }
  }

  setTitleFilterModeChange (groupId) {
    let data = Data.readJSON(`${groupId}.json`, this.root)
    data.TitleFilterModeChange = data.TitleFilterModeChange ? 0 : 1
    Data.writeJSON(`${groupId}.json`, data, this.root)
    return data.TitleFilterModeChange
  }

  getTitleFilterModeChange (groupId) {
    let data = Data.readJSON(`${groupId}.json`, this.root)
    return data.TitleFilterModeChange ?? 0
  }

  addTitleBannedWords (groupId, arr) {
    let data = Data.readJSON(`${groupId}.json`, this.root)
    if (!data.TitleBannedWords)data.TitleBannedWords = []
    data.TitleBannedWords.push(...arr)
    Data.writeJSON(`${groupId}.json`, data, this.root)
    this.groupTitleCach.delete(groupId)
  }

  getTitleBannedWords (groupId) {
    if (this.groupTitleCach.get(groupId)) return this.groupTitleCach.get(groupId)
    let data = Data.readJSON(`${groupId}.json`, this.root).TitleBannedWords ?? []
    this.groupTitleCach.set(groupId, data)
    return data
  }

  delTitleBannedWords (groupId, arr) {
    let data = Data.readJSON(`${groupId}.json`, this.root)
    data.TitleBannedWords = _.differenceBy(data.TitleBannedWords, arr)
    Data.writeJSON(`${groupId}.json`, data, this.root)
    this.groupTitleCach.delete(groupId)
  }
}()