mirror of
https://github.com/ikechan8370/chatgpt-plugin.git
synced 2025-12-17 05:47:11 +00:00
313 lines
11 KiB
JavaScript
313 lines
11 KiB
JavaScript
import { AbstractTool } from './AbstractTool.js'
|
||
import {getMasterQQ} from '../common.js'
|
||
import {Config} from '../config.js'
|
||
|
||
export class SendPictureTool extends AbstractTool {
|
||
name = 'sendPicture'
|
||
|
||
// 解析 CQ 码的方法
|
||
parseCQCode(message) {
|
||
const cqRegex = /\[CQ:([^,]+)(?:,([^\]]+))?\]/g
|
||
const result = []
|
||
let match
|
||
|
||
while ((match = cqRegex.exec(message)) !== null) {
|
||
const type = match[1]
|
||
const params = {}
|
||
|
||
if (match[2]) {
|
||
match[2].split(',').forEach(param => {
|
||
const [key, ...values] = param.split('=')
|
||
if (key && values.length) {
|
||
params[key] = values.join('=')
|
||
.replace(/&/g, '&')
|
||
.replace(/</g, '<')
|
||
.replace(/>/g, '>')
|
||
.replace(/"/g, '"')
|
||
}
|
||
})
|
||
}
|
||
|
||
if (type === 'image') {
|
||
result.push(params.url || params.file)
|
||
}
|
||
}
|
||
|
||
return result
|
||
}
|
||
|
||
parameters = {
|
||
properties: {
|
||
urlOfPicture: {
|
||
type: 'string',
|
||
description: 'the url of the pictures, not text, split with space if more than one. can be left blank.'
|
||
},
|
||
targetGroupIdOrQQNumber: {
|
||
type: 'string',
|
||
description: 'Fill in the target user\'s qq number or groupId when you need to send picture to specific user or group, otherwise leave blank'
|
||
}
|
||
},
|
||
required: ['urlOfPicture', 'targetGroupIdOrQQNumber']
|
||
}
|
||
|
||
func = async function (opt, e) {
|
||
let { urlOfPicture, targetGroupIdOrQQNumber, sender } = opt
|
||
|
||
// 处理数组格式的消息体
|
||
if (typeof urlOfPicture === 'string') {
|
||
try {
|
||
// 尝试解析为 JSON
|
||
const msgArray = JSON.parse(urlOfPicture)
|
||
if (Array.isArray(msgArray)) {
|
||
// 处理消息数组
|
||
const messages = []
|
||
for (const msg of msgArray) {
|
||
if (typeof msg === 'object' && msg.type === 'text' && msg.data?.text) {
|
||
if (msg.data.text.includes('[CQ:image')) {
|
||
// 提取 CQ 码中的参数
|
||
const cqMatch = msg.data.text.match(/\[CQ:image,([^\]]+)\]/)
|
||
if (cqMatch) {
|
||
const params = {}
|
||
cqMatch[1].split(',').forEach(param => {
|
||
const [key, ...values] = param.split('=')
|
||
if (key && values.length) {
|
||
params[key] = values.join('=')
|
||
.replace(/&/g, '&')
|
||
.replace(/</g, '<')
|
||
.replace(/>/g, '>')
|
||
.replace(/"/g, '"')
|
||
}
|
||
})
|
||
// 创建正确的图片消息格式
|
||
messages.push({
|
||
type: 'image',
|
||
data: {
|
||
file: params.url || params.file,
|
||
...params
|
||
}
|
||
})
|
||
}
|
||
} else {
|
||
// 保留普通文本消息
|
||
messages.push(msg)
|
||
}
|
||
} else {
|
||
messages.push(msg)
|
||
}
|
||
}
|
||
// 只保留图片消息
|
||
const imageMsg = messages.find(msg => msg.type === 'image')
|
||
if (imageMsg) {
|
||
urlOfPicture = imageMsg.data.file
|
||
} else {
|
||
return 'No valid image found in the message'
|
||
}
|
||
}
|
||
} catch (err) {
|
||
// 如果不是JSON格式，尝试解析 CQ 码
|
||
if (urlOfPicture.includes('[CQ:image')) {
|
||
const cqMatch = urlOfPicture.match(/\[CQ:image,([^\]]+)\]/)
|
||
if (cqMatch) {
|
||
const params = {}
|
||
cqMatch[1].split(',').forEach(param => {
|
||
const [key, ...values] = param.split('=')
|
||
if (key && values.length) {
|
||
params[key] = values.join('=')
|
||
.replace(/&/g, '&')
|
||
.replace(/</g, '<')
|
||
.replace(/>/g, '>')
|
||
.replace(/"/g, '"')
|
||
}
|
||
})
|
||
urlOfPicture = params.url || params.file
|
||
}
|
||
}
|
||
}
|
||
} else if (typeof urlOfPicture === 'object') {
|
||
if (Array.isArray(urlOfPicture)) {
|
||
// 处理消息数组
|
||
const messages = []
|
||
for (const msg of urlOfPicture) {
|
||
if (typeof msg === 'object' && msg.type === 'text' && msg.data?.text) {
|
||
if (msg.data.text.includes('[CQ:image')) {
|
||
// 提取 CQ 码中的参数
|
||
const cqMatch = msg.data.text.match(/\[CQ:image,([^\]]+)\]/)
|
||
if (cqMatch) {
|
||
const params = {}
|
||
cqMatch[1].split(',').forEach(param => {
|
||
const [key, ...values] = param.split('=')
|
||
if (key && values.length) {
|
||
params[key] = values.join('=')
|
||
.replace(/&/g, '&')
|
||
.replace(/</g, '<')
|
||
.replace(/>/g, '>')
|
||
.replace(/"/g, '"')
|
||
}
|
||
})
|
||
// 创建正确的图片消息格式
|
||
messages.push({
|
||
type: 'image',
|
||
data: {
|
||
file: params.url || params.file,
|
||
...params
|
||
}
|
||
})
|
||
}
|
||
} else {
|
||
// 保留普通文本消息
|
||
messages.push(msg)
|
||
}
|
||
} else {
|
||
messages.push(msg)
|
||
}
|
||
}
|
||
// 只保留图片消息
|
||
const imageMsg = messages.find(msg => msg.type === 'image')
|
||
if (imageMsg) {
|
||
urlOfPicture = imageMsg.data.file
|
||
} else {
|
||
return 'No valid image found in the message'
|
||
}
|
||
} else {
|
||
urlOfPicture = urlOfPicture.join(' ')
|
||
}
|
||
}
|
||
|
||
const defaultTarget = e.isGroup ? e.group_id : e.sender.user_id
|
||
const target = isNaN(targetGroupIdOrQQNumber) || !targetGroupIdOrQQNumber
|
||
? defaultTarget
|
||
: parseInt(targetGroupIdOrQQNumber) === e.bot.uin ? defaultTarget : parseInt(targetGroupIdOrQQNumber)
|
||
|
||
// 处理错误url和picture留空的情况
|
||
const urlRegex = /(?:(?:https?|ftp):\/\/)?(?:\S+(?::\S*)?@)?(?:((?:(?:[a-z0-9\u00a1-\u4dff\u9fd0-\uffff][a-z0-9\u00a1-\u4dff\u9fd0-\uffff_-]{0,62})?[a-z0-9\u00a1-\u4dff\u9fd0-\uffff]\.)+(?:[a-z\u00a1-\u4dff\u9fd0-\uffff]{2,}\.?))(?::\d{2,5})?)(?:\/[\w\u00a1-\u4dff\u9fd0-\uffff$-_.+!*'(),%]+)*(?:\?(?:[\w\u00a1-\u4dff\u9fd0-\uffff$-_.+!*(),%:@&=]|(?:[\[\]])|(?:[\u00a1-\u4dff\u9fd0-\uffff]))*)?(?:#(?:[\w\u00a1-\u4dff\u9fd0-\uffff$-_.+!*'(),;:@&=]|(?:[\[\]]))*)?\/?/i
|
||
|
||
if (!urlOfPicture) {
|
||
return 'No picture URL provided'
|
||
}
|
||
|
||
// 处理多个URL的情况
|
||
let urls = urlOfPicture.split(/\s+/).filter(url => {
|
||
return urlRegex.test(url)
|
||
})
|
||
|
||
if (urls.length === 0) {
|
||
return 'No valid picture URLs found'
|
||
}
|
||
|
||
logger.mark('pictures to send: ', urls)
|
||
let pictures = urls.map(img => {
|
||
// 检测适配器类型
|
||
const isICQQ = !!(e.bot?.adapter?.name === 'icqq')
|
||
|
||
if (img.startsWith('http')) {
|
||
if (isICQQ) {
|
||
// ICQQ适配器使用segment
|
||
return segment.image(img)
|
||
} else {
|
||
// OneBotv11适配器使用对象格式
|
||
return {
|
||
type: 'image',
|
||
data: {
|
||
file: img,
|
||
cache: false, // 禁用缓存
|
||
proxy: false, // 禁用代理
|
||
timeout: 60000, // 60秒超时
|
||
headers: {
|
||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
|
||
'Accept': 'image/avif,image/webp,image/apng,image/svg+xml,image/*,*/*;q=0.8',
|
||
'Accept-Encoding': 'gzip, deflate, br',
|
||
'Connection': 'keep-alive'
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
return segment.image(img)
|
||
})
|
||
|
||
let errs = []
|
||
try {
|
||
// 获取群列表
|
||
let groupList = await e.bot?.getGroupList?.() || e.bot.gl || []
|
||
|
||
// 发送到群
|
||
if (target.toString().length >= 6) { // 群号一般大于6位
|
||
let group = await e.bot.pickGroup(target)
|
||
if (!group) {
|
||
return `Failed to find group: ${target}`
|
||
}
|
||
|
||
for (let pic of pictures) {
|
||
let retryCount = 0
|
||
const maxRetries = 3
|
||
let lastError = null
|
||
|
||
while (retryCount < maxRetries) {
|
||
try {
|
||
// 根据适配器类型选择发送方式
|
||
const isICQQ = !!(e.bot?.adapter?.name === 'icqq')
|
||
const msgToSend = isICQQ ? pic : [pic]
|
||
|
||
// 设置超时Promise
|
||
const sendPromise = group.sendMsg(msgToSend)
|
||
const timeoutPromise = new Promise((_, reject) => {
|
||
setTimeout(() => reject(new Error('Send timeout')), 30000)
|
||
})
|
||
|
||
await Promise.race([sendPromise, timeoutPromise])
|
||
logger.mark(`图片发送成功: ${typeof pic === 'string' ? pic : pic.data?.file}`)
|
||
break
|
||
} catch (err) {
|
||
lastError = err
|
||
retryCount++
|
||
const isTimeout = err.message?.includes('ETIMEDOUT') || err.message?.includes('timeout')
|
||
logger.error(`发送图片失败 (尝试 ${retryCount}/${maxRetries}): ${isTimeout ? '连接超时' : err.message}`)
|
||
|
||
if (retryCount === maxRetries) {
|
||
errs.push(typeof pic === 'string' ? pic : (pic.data?.file || pic.url || '未知图片'))
|
||
logger.error('图片发送最终失败:', lastError.message || lastError)
|
||
} else {
|
||
// 根据错误类型调整等待时间
|
||
const waitTime = isTimeout ? 5000 * retryCount : 3000 * retryCount
|
||
await new Promise(resolve => setTimeout(resolve, waitTime))
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
if (errs.length > 0) {
|
||
return `部分图片发送失败 (${errs.length}/${pictures.length}): ${errs.join(', ')}`
|
||
}
|
||
return true
|
||
}
|
||
// 发送到私聊
|
||
else {
|
||
let masters = await getMasterQQ()
|
||
if (!Config.enableToolPrivateSend && !masters.includes(sender.toString())) {
|
||
return 'You are not allowed to send private messages'
|
||
}
|
||
|
||
let user = e.bot.pickUser(target)
|
||
for (let pic of pictures) {
|
||
try {
|
||
// 根据适配器类型选择发送方式
|
||
const isICQQ = !!(e.bot?.adapter?.name === 'icqq')
|
||
const msgToSend = isICQQ ? pic : [pic]
|
||
|
||
await user.sendMsg(msgToSend)
|
||
await new Promise(resolve => setTimeout(resolve, 1000))
|
||
} catch (err) {
|
||
logger.error('发送图片失败:', err)
|
||
errs.push(typeof pic === 'string' ? pic : (pic.data?.file || pic.url || '未知图片'))
|
||
}
|
||
}
|
||
return `Pictures have been sent to user ${target}${errs.length > 0 ? `, but ${errs.length} pictures failed to send` : ''}`
|
||
}
|
||
} catch (err) {
|
||
logger.error('发送图片过程出错:', err)
|
||
return `Failed to send pictures: ${err.message || JSON.stringify(err)}`
|
||
}
|
||
}
|
||
|
||
description = 'Useful when you want to send one or more pictures. If no extra description needed, just reply <EMPTY> at the next turn'
|
||
}
|