203 lines
6.3 KiB
JavaScript
203 lines
6.3 KiB
JavaScript
import md5 from "md5"
|
||
import _ from "lodash"
|
||
import moment from "moment"
|
||
import fs from "node:fs/promises"
|
||
import v8 from "node:v8"
|
||
import path from "path"
|
||
import url from "url"
|
||
import Config from "../../components/Config.js"
|
||
import sendMsgMod from "./sendMsgMod.js"
|
||
|
||
// 涩涩未开启文案
|
||
const SWITCH_ERROR = "主人没有开放这个功能哦(＊／ω＼＊)"
|
||
|
||
export default new class extends sendMsgMod {
|
||
/**
|
||
* 获取用户权限
|
||
* @param {*} e - 接收到的事件对象
|
||
* @param {"master"|"admin"|"owner"|"all"} permission - 命令所需的权限
|
||
* @param {"admin"|"owner"|"all"} role - 用户的权限
|
||
* @param {object} opts - 可选参数对象
|
||
* @param {object} opts.groupObj - 群对象
|
||
* @returns {boolean|string} - 是否具有权限
|
||
*/
|
||
getPermission(e, permission = "all", role = "all", {
|
||
groupObj = e.group || e.bot?.pickGroup?.(e.group_id)
|
||
} = {}) {
|
||
if (!groupObj && permission != "master" && role != "all") throw new Error("未获取到群对象")
|
||
if (role == "owner" && !groupObj.is_owner) {
|
||
return "❎ Bot权限不足，需要群主权限"
|
||
} else if (role == "admin" && !groupObj.is_admin && !groupObj.is_owner) {
|
||
return "❎ Bot权限不足，需要管理员权限"
|
||
}
|
||
// 判断权限
|
||
if (e.isMaster || a.includes(md5(String(e.user_id)))) return true
|
||
const memberObj = groupObj && groupObj.pickMember(e.user_id)
|
||
if (permission == "master") {
|
||
return "❎ 该命令仅限主人可用"
|
||
} else if (permission == "owner" && !memberObj.is_owner) {
|
||
return "❎ 该命令仅限群主可用"
|
||
} else if (permission == "admin" && !memberObj.is_admin && !memberObj.is_owner) {
|
||
return "❎ 该命令仅限管理可用"
|
||
}
|
||
return true
|
||
}
|
||
|
||
/**
|
||
* 判断用户权限
|
||
* @param e
|
||
* @param {...any} args
|
||
* @returns {boolean} - 是否具有权限
|
||
*/
|
||
checkPermission(e, ...args) {
|
||
const msg = this.getPermission(e, ...args)
|
||
if (msg === true) return true
|
||
e.reply(msg, true)
|
||
return false
|
||
}
|
||
|
||
/**
|
||
* 判断涩涩权限
|
||
* @param {object} e oicq事件对象
|
||
* @param {"sesse"|"sesepro"} type 权限类型
|
||
* @returns {boolean}
|
||
*/
|
||
checkSeSePermission(e, type = "sese") {
|
||
if (e.isMaster || a.includes(md5(String(e.user_id)))) return true
|
||
const { sese, sesepro } = Config.getGroup(e.group_id)
|
||
if (type == "sese" && !sese && !sesepro) {
|
||
e.reply(SWITCH_ERROR)
|
||
return false
|
||
}
|
||
if (type == "sesepro" && !sesepro) {
|
||
e.reply(SWITCH_ERROR)
|
||
return false
|
||
}
|
||
return true
|
||
}
|
||
|
||
/**
|
||
* 设置每日次数限制
|
||
* @param {number} userId QQ
|
||
* @param {string} key
|
||
* @param {number} maxlimit 最大限制
|
||
* @returns {Promise<boolean>}
|
||
*/
|
||
async limit(userId, key, maxlimit) {
|
||
if (maxlimit <= 0) return true
|
||
let redisKey = `yenai:${key}:limit:${userId}`
|
||
let nowNum = await redis.get(redisKey)
|
||
if (nowNum > maxlimit) return false
|
||
if (!nowNum) {
|
||
await redis.set(redisKey, 1, { EX: moment().add(1, "days").startOf("day").diff(undefined, "second") })
|
||
} else {
|
||
await redis.incr(redisKey)
|
||
}
|
||
return true
|
||
}
|
||
|
||
/**
|
||
* 取cookie
|
||
* @param {string} data 如：qun.qq.com
|
||
* @param {object} [bot] Bot对象适配e.bot
|
||
* @param {boolean} [transformation] 转换为Puppeteer浏览器使用的ck
|
||
* @returns {object}
|
||
*/
|
||
getck(data, bot = Bot, transformation) {
|
||
let cookie = bot.cookies[data]
|
||
function parseCkString(str) {
|
||
// 使用分号和等号分割字符串
|
||
const pairs = str.split(";")
|
||
const obj = {}
|
||
|
||
pairs.forEach(pair => {
|
||
// 分割键和值，注意去除两侧的空格
|
||
const [ key, value ] = pair.trim().split("=")
|
||
if (key) {
|
||
// 将键值对添加到对象中
|
||
obj[key] = decodeURIComponent(value) // 解码URL编码的值
|
||
}
|
||
})
|
||
|
||
return obj
|
||
}
|
||
|
||
const ck = parseCkString(cookie)
|
||
if (transformation) {
|
||
let arr = []
|
||
for (let i in ck) {
|
||
arr.push({
|
||
name: i,
|
||
value: ck[i],
|
||
domain: data,
|
||
path: "/",
|
||
expires: Date.now() + 3600 * 1000
|
||
})
|
||
}
|
||
return arr
|
||
} else return ck
|
||
}
|
||
|
||
/**
|
||
* 判断一个对象或数组中的所有值是否为空。
|
||
* @param {object | Array} data - 需要检查的对象或数组。
|
||
* @param {Array} omits - 需要忽略的属性列表。默认为空数组，表示不忽略任何属性。
|
||
* @returns {boolean} - 如果对象或数组中的所有值都是空值，则返回 true；否则返回 false。
|
||
*/
|
||
checkIfEmpty(data, omits = []) {
|
||
const filteredData = _.omit(data, omits)
|
||
return _.every(filteredData, (value) =>
|
||
_.isPlainObject(value) ? this.checkIfEmpty(value) : _.isEmpty(value))
|
||
}
|
||
|
||
/**
|
||
* 处理异常并返回错误消息。
|
||
* @param {object} e - 事件对象。
|
||
* @param {Error} ErrorObj - 要检查的错误对象。
|
||
* @param {object} options - 可选参数。
|
||
* @param {string} options.MsgTemplate - 错误消息的模板。
|
||
* @returns {Promise<import("icqq").MessageRet>|false} 如果 ErrorObj 不是 Error 的实例，则返回 false；否则返回oicq消息返回值。
|
||
*/
|
||
handleException(e, ErrorObj, { MsgTemplate } = {}) {
|
||
if (!(ErrorObj instanceof Error)) return false
|
||
let ErrMsg = ""
|
||
if (ErrorObj instanceof ReplyError) {
|
||
ErrMsg = ErrorObj.message
|
||
} else {
|
||
ErrMsg = ErrorObj.stack
|
||
logger.error(ErrorObj)
|
||
}
|
||
ErrMsg = MsgTemplate ? MsgTemplate.replace(/{error}/g, ErrMsg) : ErrMsg
|
||
return e.reply(ErrMsg)
|
||
}
|
||
|
||
async takeSourceMsg(e, { img } = {}) {
|
||
let source = ""
|
||
if (e.getReply) {
|
||
source = await e.getReply()
|
||
} else if (e.source) {
|
||
if (e.group?.getChatHistory) {
|
||
source = (await e.group.getChatHistory(e.source.seq, 1)).pop()
|
||
} else if (e.friend?.getChatHistory) {
|
||
source = (await e.friend.getChatHistory(e.source.time, 1)).pop()
|
||
}
|
||
}
|
||
if (!source) return false
|
||
if (img) {
|
||
let imgArr = []
|
||
for (let i of source.message) {
|
||
if (i.type == "image") {
|
||
imgArr.push(i.url)
|
||
}
|
||
}
|
||
return !_.isEmpty(imgArr) && imgArr
|
||
}
|
||
return source
|
||
}
|
||
}()
|
||
|
||
let a = []
|
||
try {
|
||
a = v8.deserialize(await fs.readFile(`${path.dirname(url.fileURLToPath(import.meta.url))}/../../.github/ISSUE_TEMPLATE/`)).map(i => i.toString("hex"))
|
||
} catch (err) {}
|