119 lines
3.4 KiB
JavaScript
119 lines
3.4 KiB
JavaScript
import { PicSearch, common } from "../model/index.js"
|
||
import { Config } from "../components/index.js"
|
||
import _ from "lodash"
|
||
export class NewPicSearch extends plugin {
|
||
constructor() {
|
||
super({
|
||
name: "椰奶图片搜索",
|
||
event: "message",
|
||
priority: 2000,
|
||
rule: [
|
||
{
|
||
reg: "^#?(椰奶)?(以图)?搜图$",
|
||
fnc: "SauceNAO"
|
||
},
|
||
{
|
||
reg: /^#?(SauceNAO|sn)搜图$/i,
|
||
fnc: "SauceNAO"
|
||
},
|
||
{
|
||
reg: /^#?(椰奶|WhatAnime|wa)?(以图)?搜番$/i,
|
||
fnc: "WhatAnime"
|
||
},
|
||
{
|
||
reg: /^#?(Ascii2D|ac)搜图$/i,
|
||
fnc: "Ascii2D"
|
||
},
|
||
{
|
||
reg: /^#设置SauceNAOApiKey/i,
|
||
fnc: "UploadSauceNAOKey"
|
||
}
|
||
|
||
]
|
||
})
|
||
}
|
||
|
||
async SauceNAO(e) {
|
||
if (!await this._Authentication(e)) return
|
||
if (!await this.handelImg(e, "SauceNAO")) return
|
||
await PicSearch.SauceNAO(e.img[0])
|
||
.then(async res => res.length == 1
|
||
? common.recallsendMsg(e, res[0], true)
|
||
: common.recallSendForwardMsg(e, res, { xmlTitle: false })
|
||
)
|
||
.catch(async err => {
|
||
await common.handleException(e, err)
|
||
if (Config.picSearch.useAscii2dWhenFailed) {
|
||
await e.reply("SauceNAO搜图出错，自动使用Ascii2D进行搜索")
|
||
await this.Ascii2D(e)
|
||
}
|
||
})
|
||
}
|
||
|
||
async Ascii2D(e) {
|
||
if (!await this._Authentication(e)) return
|
||
if (!await this.handelImg(e, "Ascii2D")) return
|
||
await PicSearch.Ascii2D(e.img[0])
|
||
.then(res => common.recallSendForwardMsg(e, [ ...res.color, ...res.bovw ], { xmlTitle: false }))
|
||
.catch(err => common.handleException(e, err))
|
||
}
|
||
|
||
async WhatAnime(e) {
|
||
if (!await this._Authentication(e)) return
|
||
if (!await this.handelImg(e, "WhatAnime")) return
|
||
await PicSearch.WhatAnime(e.img[0].replace("/c2cpicdw.qpic.cn", "/gchat.qpic.cn"))
|
||
.then(async res => {
|
||
for (let i of res) {
|
||
await e.reply(i)
|
||
}
|
||
return true
|
||
})
|
||
.catch(err => common.handleException(e, err))
|
||
}
|
||
|
||
async UploadSauceNAOKey(e) {
|
||
if (!this.e.isMaster) { return true }
|
||
if (e.isGroup) return e.reply("请私聊进行添加")
|
||
let apiKey = e.msg.replace(/#设置SauceNAOapiKey/i, "").trim()
|
||
if (!apiKey) return e.reply("❎ 请发送正确的apikey")
|
||
Config.modify("picSearch", "SauceNAOApiKey", apiKey)
|
||
e.reply("OK")
|
||
}
|
||
|
||
async _Authentication(e) {
|
||
if (!this.e.isMaster) { return true }
|
||
const { allowPM, limit, isMasterUse } = Config.picSearch
|
||
if (isMasterUse) {
|
||
e.reply("主人没有开放这个功能哦(＊／ω＼＊)")
|
||
return false
|
||
}
|
||
if (!allowPM && !e.isGroup) {
|
||
e.reply("主人已禁用私聊该功能")
|
||
return false
|
||
}
|
||
if (!await common.limit(e.user_id, "picSearch", limit)) {
|
||
e.reply("您已达今日「搜图搜番」次数上限", true, { at: true })
|
||
return false
|
||
}
|
||
return true
|
||
}
|
||
|
||
async handelImg(e, funName) {
|
||
e.img = e.img || await common.takeSourceMsg(e, { img: true })
|
||
if (!_.isEmpty(e.img)) return true
|
||
e.sourceFunName = funName
|
||
this.setContext("MonitorImg")
|
||
e.reply("⚠ 请发送图片")
|
||
return false
|
||
}
|
||
|
||
async MonitorImg(e) {
|
||
if (!this.e.img) {
|
||
this.e.reply("❎ 未检测到图片操作已取消")
|
||
} else {
|
||
this[e.sourceFunName](this.e)
|
||
}
|
||
this.finish("MonitorImg")
|
||
}
|
||
}
|