Miao-Yunzai/plugins/genshin/apps/abbrSet.js

import plugin from '../../../lib/plugins/plugin.js'
import common from '../../../lib/common/common.js'
import fs from 'node:fs'
import gsCfg from '../model/gsCfg.js'
import YAML from 'yaml'
import lodash from 'lodash'

export class abbrSet extends plugin {
  constructor(e) {
    super({
      name: '别名设置',
      dsc: '角色别名设置',
      event: 'message',
      priority: 600,
      rule: [
        {
          reg: '^#(星铁)?(设置|配置)(.*)(别名|昵称)$',
          fnc: 'abbr'
        },
        {
          reg: '^#(星铁)?删除(别名|昵称)(.*)$',
          fnc: 'delAbbr'
        },
        {
          reg: '^#(星铁)?(.*)(别名|昵称)$',
          fnc: 'abbrList'
        }
      ]
    })
    this.isSr = false
    this.file = './plugins/genshin/config/role.name.yaml'
  }

  async init() {
    if (!fs.existsSync(this.file)) {
      fs.writeFileSync(this.file, `神里绫华:
  - 龟龟
  - 小乌龟`)
    }
  }

  async abbr() {
    if (!await this.checkAuth()) return
    let role = gsCfg.getRole(this.e.msg, '#|星铁|设置|配置|别名|昵称', this.e.isSr)
    if (!role) return false
    this.e.role = role
    this.isSr = this.e.isSr
    this.setContext('setAbbr')

    await this.reply(`请发送${role.alias}别名，多个用空格隔开`)
  }

  async checkAuth() {
    if (!this.e.isGroup && !this.e.isMaster) {
      await this.reply('禁止私聊设置角色别名')
      return false
    }

    let abbrSetAuth = gsCfg.getConfig('mys', 'set').abbrSetAuth
    /** 所有人都允许 */
    if (abbrSetAuth === 0) return true
    /** 主人默认允许 */
    if (this.e.isMaster) return true
    /** 管理员 */
    if (abbrSetAuth == 1) {
      if (!this.e.bot.gml.has(this.e.group_id)) {
        return false
      }
      if (!this.e.bot.gml.get(this.e.group_id).get(this.e.user_id)) {
        return false
      }
      if (!this.e.member.is_admin) {
        this.e.reply('暂无权限，只有管理员才能操作')
        return false
      }
    }

    return true
  }

  async setAbbr() {
    if (!this.e.msg || this.e.at || this.e.img) {
      await this.reply('设置错误：请发送正确内容')
      return
    }

    let { setAbbr = {} } = this.getContext()
    this.finish('setAbbr')

    let role = setAbbr.role
    let setName = this.e.msg.split(' ')

    let nameArr = gsCfg.getConfig('role', 'name')

    if (!nameArr[role.name]) {
      nameArr[role.name] = []
    }

    let ret = []
    for (let name of setName) {
      name = name.replace(/#|星铁|设置|配置|别名|昵称/g, '')
      if (!name) continue
      /** 重复添加 */
      if (nameArr[role.name].includes(name) || gsCfg.roleNameToID(name, this.isSr)) {
        continue
      }

      nameArr[role.name].push(name)
      ret.push(name)
    }
    if (ret.length <= 0) {
      await this.reply('设置失败：别名错误或已存在')
      return
    }

    this.save(nameArr)

    gsCfg[this.isSr ? 'sr_nameID' : 'nameID'] = false

    await this.reply(`设置别名成功：${ret.join('、')}`)
  }

  save(data) {
    data = YAML.stringify(data)
    fs.writeFileSync(this.file, data)
  }

  async delAbbr() {
    let role = gsCfg.getRole(this.e.msg, '#|删除|别名|昵称')
    if (!role) return false

    let nameArr = gsCfg.getConfig('role', 'name')

    if (!nameArr[role.name]) {
      await this.reply('默认别名设置，不能删除！')
      return true
    }

    nameArr[role.name] = nameArr[role.name].filter((v) => {
      if (v == role.alias) return false
      return v
    })

    this.save(nameArr)

    await this.reply(`设置${role.name}别名成功：${role.alias}`)
  }

  async abbrList() {
    let role = gsCfg.getRole(this.e.msg, '#|星铁|别名|昵称', this.e.isSr)

    if (!role) return false

    let name = gsCfg.getdefSet('role', this.e.isSr ? 'sr_name' : 'name')[role.roleId]
    let nameUser = gsCfg.getConfig('role', 'name')[role.name] ?? []

    let list = lodash.uniq([...name, ...nameUser])

    let msg = []
    for (let i in list) {
      let num = Number(i) + 1
      msg.push(`${num}.${list[i]}\n`)
    }

    let title = `${role.name}别名，${list.length}个`

    msg = await common.makeForwardMsg(this.e, msg, title)

    await this.e.reply(msg)
  }

  async makeForwardMsg(qq, title, msg) {
    let nickname = this.e.bot.nickname
    if (this.e.isGroup) {
      let info = await this.e.bot.getGroupMemberInfo(this.e.group_id, qq)
      nickname = info.card ?? info.nickname
    }
    let userInfo = {
      user_id: this.e.bot.uin,
      nickname
    }

    let forwardMsg = [
      {
        ...userInfo,
        message: title
      },
      {
        ...userInfo,
        message: msg
      }
    ]

    /** 制作转发内容 */
    if (this.e.isGroup) {
      forwardMsg = await this.e.group.makeForwardMsg(forwardMsg)
    } else {
      forwardMsg = await this.e.friend.makeForwardMsg(forwardMsg)
    }

    /** 处理描述 */
    forwardMsg.data = forwardMsg.data
      .replace(/\n/g, '')
      .replace(/<title color="#777777" size="26">(.+?)<\/title>/g, '___')
      .replace(/___+/, `<title color="#777777" size="26">${title}</title>`)

    return forwardMsg
  }
}