273 lines
7.4 KiB
JavaScript
273 lines
7.4 KiB
JavaScript
import { common, Pixiv } from './index.js'
|
||
import { Data, Plugin_Path, Config } from '../components/index.js'
|
||
import _ from 'lodash'
|
||
import { setuMsg } from '../constants/msg.js'
|
||
import request from '../lib/request/request.js'
|
||
export default new class setu {
|
||
constructor () {
|
||
this.root = `${Plugin_Path}/config/setu`
|
||
// 默认配置
|
||
this.def = Config.setu.defSet
|
||
// 存cd的变量
|
||
this.temp = {}
|
||
// 初始化
|
||
this.init()
|
||
}
|
||
|
||
async init () {
|
||
Data.createDir('config/setu')
|
||
}
|
||
|
||
/** 开始执行文案 */
|
||
get startMsg () {
|
||
return _.sample(setuMsg.start)
|
||
}
|
||
|
||
/** CD中文案 */
|
||
get CDMsg () {
|
||
return _.sample(setuMsg.cd)
|
||
}
|
||
|
||
/** 发送图片文案 */
|
||
get sendMsgs () {
|
||
return _.sample(setuMsg.send)
|
||
}
|
||
|
||
/**
|
||
* @description: 请求api
|
||
* @param {String} r18 是否r18 0或1
|
||
* @param {Number} num 数量
|
||
* @param {String} tag 关键词
|
||
* @return {Object}
|
||
*/
|
||
async setuApi (r18, num = 1, tag = []) {
|
||
let api = 'https://api.lolicon.app/setu/v2'
|
||
const { imgSize, excludeAI } = Config.setu
|
||
const size = imgSize[_.max(Object.keys(imgSize).filter(item => num > item))] || 'original'
|
||
let parans = {
|
||
r18,
|
||
num,
|
||
tag,
|
||
proxy: Pixiv.proxy,
|
||
size,
|
||
excludeAI
|
||
}
|
||
let result = await request.post(api, { data: parans }).then(res => res.json())
|
||
if (_.isEmpty(result.data)) throw Error('没有找到相关的tag')
|
||
// 消息
|
||
return await Promise.all(result.data.map(async item => {
|
||
let { pid, title, tags, author, r18, urls, aiType } = item
|
||
return [
|
||
`${this.sendMsgs}\n`,
|
||
`标题：${title}\n`,
|
||
`画师：${author}\n`,
|
||
`Pid：${pid}\n`,
|
||
`R18：${r18}\n`,
|
||
`AI：${aiType ? aiType == 1 ? '是' : '否' : '未知'}\n`,
|
||
`tag：${_.truncate(tags.join(','))}\n`,
|
||
await Pixiv._requestPixivImg(urls?.original || urls?.regular || urls?.small)
|
||
]
|
||
}))
|
||
}
|
||
|
||
/**
|
||
* @description: 发送消息和写入cd
|
||
* @param {*} e oicq
|
||
* @param {Array} img 消息数组
|
||
* @return {Boolean}
|
||
*/
|
||
async sendMsgOrSetCd (e, msg) {
|
||
// 发送消息
|
||
let res = await common.recallSendForwardMsg(e, msg, false)
|
||
if (!res) return false
|
||
// 设置CD
|
||
if (!e.isMaster) this.setCdTime(e.user_id, e.group_id)
|
||
}
|
||
|
||
/**
|
||
* @description: 设置cd
|
||
* @param {Number} userId QQ号
|
||
* @param {Number} groupId 群号不传为私聊CD
|
||
* @param {Number} cd cd时间
|
||
* @return {*}
|
||
*/
|
||
setCdTime (userId, groupId, cd = this.getCfgCd(userId, groupId)) {
|
||
let present = parseInt(Date.now() / 1000)
|
||
userId = userId - 0
|
||
groupId = groupId - 0
|
||
if (!cd) return false
|
||
if (groupId) {
|
||
this.temp[userId + groupId] = present + cd
|
||
setTimeout(() => {
|
||
delete this.temp[userId + groupId]
|
||
}, cd * 1000)
|
||
} else {
|
||
this.temp[userId] = present + cd
|
||
setTimeout(() => {
|
||
delete this.temp[userId]
|
||
}, cd * 1000)
|
||
}
|
||
return true
|
||
}
|
||
|
||
/**
|
||
* @description: 获取剩余CD时间
|
||
* @param {Number} userId QQ号
|
||
* @param {Number} groupId 群号不传则为私聊CD
|
||
* @return {String} 格式化后的时间
|
||
*/
|
||
getRemainingCd (userId, groupId) {
|
||
userId = userId - 0
|
||
groupId = groupId - 0
|
||
// 获取现在的时间并转换为秒
|
||
let present = parseInt(new Date().getTime() / 1000)
|
||
let over = 0
|
||
if (groupId) {
|
||
if (!this.temp[userId + groupId]) return false
|
||
over = (this.temp[userId + groupId] - present)
|
||
} else {
|
||
if (!this.temp[userId]) return false
|
||
over = (this.temp[userId] - present)
|
||
}
|
||
if (over <= 0) return false
|
||
return this.Secondformat(over)
|
||
}
|
||
|
||
/**
|
||
* @description: 获取配置cd
|
||
* @param {Number} userId QQ号
|
||
* @param {Number} groupId 传群号为群聊配置
|
||
* @return {*}
|
||
*/
|
||
getCfgCd (userId, groupId) {
|
||
let data = Data.readJSON(`setu${groupId ? '' : '_s'}.json`, this.root)
|
||
let CD = groupId ? data[groupId]?.cd : data[userId]
|
||
if (CD !== undefined) return CD
|
||
return this.def.cd // 默认300
|
||
}
|
||
|
||
/**
|
||
* @description: 获取r18
|
||
* @param {Number} groupID 群号不传为私聊
|
||
* @return {String} 0或1
|
||
*/
|
||
getR18 (groupID) {
|
||
let data = Data.readJSON(`setu${groupID ? '' : '_s'}.json`, this.root)
|
||
let R18 = groupID ? data[groupID]?.r18 : data.r18
|
||
if (R18 !== undefined) return R18
|
||
return this.def.r18
|
||
}
|
||
|
||
/**
|
||
* @description: 获取群的撤回时间
|
||
* @param {*} e oicq
|
||
* @return {Number}
|
||
*/
|
||
getRecallTime (groupId) {
|
||
if (!groupId) return 0
|
||
let data = Data.readJSON('setu.json', this.root)
|
||
let recalltime = data[groupId]?.recall
|
||
if (recalltime !== undefined) return recalltime
|
||
return this.def.recall // 默认120
|
||
}
|
||
|
||
/**
|
||
* @description: 设置群cd和撤回时间
|
||
* @param {Number} groupId 群号
|
||
* @param {Number} num 设置时间
|
||
* @param {Boolean} type 为true设置撤回时间反之设置CD
|
||
* @return {Boolean}
|
||
*/
|
||
setGroupRecallTimeAndCd (groupId, num, type) {
|
||
let data = Data.readJSON('setu.json', this.root)
|
||
|
||
if (!data[groupId]) data[groupId] = _.cloneDeep(this.def)
|
||
|
||
type ? data[groupId].recall = Number(num) : data[groupId].cd = Number(num)
|
||
|
||
return Data.writeJSON('setu.json', data, this.root)
|
||
}
|
||
|
||
/**
|
||
* @description: 设置CD
|
||
* @param {*} e oicq
|
||
* @param {String} qq 设置的qq
|
||
* @param {String} cd 设置的cd
|
||
*/
|
||
setUserCd (e, qq, cd) {
|
||
let data = Data.readJSON('setu_s.json', this.root)
|
||
|
||
data[qq] = Number(cd)
|
||
if (Data.writeJSON('setu_s.json', data, this.root)) {
|
||
e.reply(`✅ 设置用户${qq}的cd成功，cd时间为${cd}秒`)
|
||
delete this.temp[qq]
|
||
return true
|
||
} else {
|
||
e.reply('❎ 设置失败')
|
||
return false
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @description: 设置r18
|
||
* @param {String|Number} groupID 群聊id为假时设置私聊
|
||
* @param {Boolean} isopen 开启或关闭
|
||
*/
|
||
setR18 (groupID, isopen) {
|
||
let data = Data.readJSON(`setu${groupID ? '' : '_s'}.json`, this.root)
|
||
if (groupID) {
|
||
if (!data[groupID]) data[groupID] = _.cloneDeep(this.def)
|
||
data[groupID].r18 = isopen ? 1 : 0
|
||
} else {
|
||
data.r18 = isopen ? 1 : 0
|
||
}
|
||
if (Data.writeJSON(`setu${groupID ? '' : '_s'}.json`, data, this.root)) {
|
||
logger.mark(`[椰奶R18][${groupID ? '群聊' : '私聊'}]已${isopen ? '开启' : '关闭'}${groupID}的涩涩模式`)
|
||
return true
|
||
} else {
|
||
logger.mark(`[椰奶R18][${groupID ? '群聊' : '私聊'}]设置失败`)
|
||
return false
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @description: 获取现有设置
|
||
* @param {*} e oicq
|
||
* @return {*}
|
||
*/
|
||
getSeSeConfig (e) {
|
||
let set = _.cloneDeep(this.def)
|
||
set.cd = this.getCfgCd(e.user_id, e.group_id)
|
||
set.r18 = this.getR18(e.group_id)
|
||
set.recall = this.getRecallTime(e.group_id)
|
||
if (!e.isGroup) delete set.recall
|
||
return set
|
||
}
|
||
|
||
/**
|
||
* @description: 格式化秒
|
||
* @param {Number} value 秒
|
||
* @return {String}
|
||
*/
|
||
Secondformat (value) {
|
||
let time = common.getSecond(value)
|
||
|
||
let { second, minute, hour, day } = time
|
||
// 处理返回消息
|
||
let result = ''
|
||
if (second != 0) {
|
||
result = parseInt(second) + '秒'
|
||
}
|
||
if (minute > 0) {
|
||
result = parseInt(minute) + '分' + result
|
||
}
|
||
if (hour > 0) {
|
||
result = parseInt(hour) + '小时' + result
|
||
}
|
||
if (day > 0) {
|
||
result = parseInt(day) + '天' + result
|
||
}
|
||
return result
|
||
}
|
||
}()
|