import cloud from '@lafjs/cloud'
import { decrypt, getSignature } from '@wecom/crypto'
import xml2js from 'xml2js'
function genConversationKey(userName) {
return `${.WXWORK_AGENTID}:${userName}`
}
function genWxAppAccessTokenKey() {
return `${.WXWORK_AGENTID}:access-token`
}
async function getToken() {
('[getToken] called')
const cache = .get(genWxAppAccessTokenKey())
if (cache && >= Date.now()) return
const res = await ({
url: '/cgi-bin/gettoken',
method: 'GET',
params: {
corpid: .WXWORK_CORPID,
corpsecret: .WXWORK_CORPSECRET,
}
})
const token = res.data.access_token
.set(genWxAppAccessTokenKey(), { token, expires: Date.now() + res.data.expires_in * 1000 })
return token
}
async function sendWxMessage(message, user) {
('[sendWxMessage] called', user, message)
const res = await ({
url: '/cgi-bin/message/send',
method: 'POST',
params: {
access_token: await getToken()
},
data: {
"touser": user,
"msgtype": "text",
"agentid": .WXWORK_AGENTID,
"text": {
"content": message
},
"safe": 0,
"enable_id_trans": 0,
"enable_duplicate_check": 0,
"duplicate_check_interval": 1800
},
})
('[sendWxMessage] received', res.data)
}
async function sendDifyMessage(message, userName, onMessage) {
('[sendDifyMessage] called', message, userName)
const conversationId = .get(genConversationKey(userName)) || null
let newConversationId = ''
let responseText = ''
try {
const response = await ({
url: '/v1/chat-messages',
method: 'POST',
headers: {
'Authorization': `Bearer ${.DIFY_APPTOKEN}`
},
data: {
inputs: {},
response_mode: "streaming",
query: message,
user: userName,
conversation_id: conversationId
},
responseType: "stream"
})
let firstHalfMessage = ''
response.data.on('data', (data) => {
let message = data.toString()
try {
if (firstHalfMessage) {
message += firstHalfMessage
firstHalfMessage = ''
}
// 检查是不是sse协议
if (!('data: ')) return
const parsedChunk: Record<string, any> = ((6))
if (!newConversationId) {
newConversationId = parsedChunk.conversation_id
.set(genConversationKey(userName), newConversationId)
}
const { answer } = parsedChunk
responseText += answer
// 伪流式响应
if (('\n\n') || (responseText.length > 120 && /[?。；！]$/.test(responseText))) {
onMessage(responseText.replace('\n\n', ''))
('[sendDifyMessage] received', responseText, newConversationId)
responseText = ''
}
} catch (e) {
firstHalfMessage = message
console.error('[sendDifyMessage] error', message)
}
})
// stream结束时把剩下的消息全部发出去
response.data.on('end', () => {
onMessage(responseText.replace('\n\n', ''))
})
} catch (e) {
console.error("[sendDifyMessage] error", e)
}
}
async function asyncSendMessage(xml) {
('[asyncSendMessage] called', xml)
if ([0] !== 'text') return
const message = xml.Content[0]
const userName = [0]
if (message === '/new') {
// 重置conversation id
.set(genConversationKey(userName), null)
sendWxMessage('新建成功，开始新的对话吧~~', userName)
return
}
sendWxMessage('AI思考中, 请耐心等待~~', userName)
try {
sendDifyMessage(message, userName, (message) => {
sendWxMessage(message, userName)
})
}
catch (e) {
console.error('[sendDifyMessage] error', e)
sendWxMessage('接口请求失败，请联系管理员查看错误信息', userName)
}
}
export default async function (ctx: FunctionContext) {
const { query } = ctx
const { msg_signature, timestamp, nonce, echostr } = query
const token = .WXWORK_TOKEN
const key = .WXWORK_AESKEY
('[main] called', ctx.method, )
// 签名验证专用
if (ctx.method === 'GET') {
const signature = getSignature(token, timestamp, nonce, echostr)
if (signature !== msg_signature) {
return { message: '签名验证失败', code: 401 }
}
const { message } = decrypt(key, echostr)
return message
}
const payload =
const encrypt = [0]
const signature = getSignature(token, timestamp, nonce, encrypt)
if (signature !== msg_signature) {
return { message: '签名验证失败', code: 401 }
}
const { message } = decrypt(key, encrypt)
const {
xml
} = await xml2(message)
// 由于GPT API耗时较久，这里提前返回，防止企业微信超时重试，后续再手动调用发消息接口
(200)
await asyncSendMessage(xml)
return { message: true, code: 0 }
}