初始化模版工程
This commit is contained in:
Cloud Bot
354
remote-control/bots/lark/handlers.ts
Normal file
354
remote-control/bots/lark/handlers.ts
Normal file
@@ -0,0 +1,354 @@
|
||||
import type * as lark from '@larksuiteoapi/node-sdk'
|
||||
import { HTTPClient } from '@/http'
|
||||
import { getDefaultAgentId } from '@/app/api/nova-config'
|
||||
import { getConversationId, setConversationId } from './user-store'
|
||||
import { sendMessageAndWaitResponse } from '@/remote-control/shared/nova-bridge'
|
||||
import { uploadBotAttachments, type BotAttachment } from '@/remote-control/shared/file-uploader'
|
||||
import { BotLogger } from '@/remote-control/shared/logger'
|
||||
|
||||
const oapiClient = new HTTPClient({
|
||||
baseURL: process.env.NOVA_BASE_URL,
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'Tenant-Id': process.env.NOVA_TENANT_ID!,
|
||||
'Authorization': process.env.NOVA_ACCESS_KEY!,
|
||||
},
|
||||
})
|
||||
|
||||
const userQueues = new Map<string, Promise<void>>()
|
||||
const processedMessages = new Set<string>()
|
||||
|
||||
function extractTextContent(messageType: string, content: string): string {
|
||||
try {
|
||||
const parsed = JSON.parse(content)
|
||||
|
||||
if (messageType === 'text') {
|
||||
return (parsed.text || '').trim()
|
||||
}
|
||||
|
||||
if (messageType === 'post') {
|
||||
// 富文本：遍历 content 数组提取 text 类型的文本
|
||||
const contentArray = parsed.content || []
|
||||
const texts: string[] = []
|
||||
for (const paragraph of contentArray) {
|
||||
if (Array.isArray(paragraph)) {
|
||||
for (const item of paragraph) {
|
||||
if (item.tag === 'text' && item.text) {
|
||||
texts.push(item.text)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return texts.join(' ').trim()
|
||||
}
|
||||
|
||||
return ''
|
||||
} catch {
|
||||
return ''
|
||||
}
|
||||
}
|
||||
|
||||
function cleanMentions(text: string): string {
|
||||
// 清除 @_user_N 占位符
|
||||
return text.replace(/@_user_\d+/g, '').trim()
|
||||
}
|
||||
|
||||
async function extractAttachments(
|
||||
client: lark.Client,
|
||||
messageType: string,
|
||||
content: string,
|
||||
messageId: string,
|
||||
): Promise<BotAttachment[]> {
|
||||
const attachments: BotAttachment[] = []
|
||||
|
||||
try {
|
||||
const parsed = JSON.parse(content)
|
||||
|
||||
// 处理 image 类型
|
||||
if (messageType === 'image' && parsed.image_key) {
|
||||
const resp = await client.im.messageResource.get({
|
||||
path: { message_id: messageId, file_key: parsed.image_key },
|
||||
params: { type: 'image' },
|
||||
})
|
||||
const stream = resp.getReadableStream()
|
||||
const chunks: Buffer[] = []
|
||||
for await (const chunk of stream) {
|
||||
chunks.push(chunk)
|
||||
}
|
||||
const buffer = Buffer.concat(chunks)
|
||||
|
||||
// 将 buffer 转为临时 URL（通过 data URI）
|
||||
const dataUri = `data:image/png;base64,${buffer.toString('base64')}`
|
||||
attachments.push({
|
||||
url: dataUri,
|
||||
fileName: `lark_image_${Date.now()}.png`,
|
||||
})
|
||||
}
|
||||
|
||||
// 处理 file 类型
|
||||
if (messageType === 'file' && parsed.file_key) {
|
||||
const resp = await client.im.messageResource.get({
|
||||
path: { message_id: messageId, file_key: parsed.file_key },
|
||||
params: { type: 'file' },
|
||||
})
|
||||
const stream = resp.getReadableStream()
|
||||
const chunks: Buffer[] = []
|
||||
for await (const chunk of stream) {
|
||||
chunks.push(chunk)
|
||||
}
|
||||
const buffer = Buffer.concat(chunks)
|
||||
|
||||
const dataUri = `data:application/octet-stream;base64,${buffer.toString('base64')}`
|
||||
attachments.push({
|
||||
url: dataUri,
|
||||
fileName: parsed.file_name || `lark_file_${Date.now()}`,
|
||||
})
|
||||
}
|
||||
|
||||
// 处理 post 富文本中的图片
|
||||
if (messageType === 'post' && parsed.content) {
|
||||
for (const paragraph of parsed.content) {
|
||||
if (Array.isArray(paragraph)) {
|
||||
for (const item of paragraph) {
|
||||
if (item.tag === 'img' && item.image_key) {
|
||||
const resp = await client.im.messageResource.get({
|
||||
path: { message_id: messageId, file_key: item.image_key },
|
||||
params: { type: 'image' },
|
||||
})
|
||||
const stream = resp.getReadableStream()
|
||||
const chunks: Buffer[] = []
|
||||
for await (const chunk of stream) {
|
||||
chunks.push(chunk)
|
||||
}
|
||||
const buffer = Buffer.concat(chunks)
|
||||
|
||||
const dataUri = `data:image/png;base64,${buffer.toString('base64')}`
|
||||
attachments.push({
|
||||
url: dataUri,
|
||||
fileName: `lark_post_image_${Date.now()}.png`,
|
||||
})
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
} catch (err) {
|
||||
console.error('[Lark Bot] 提取附件失败:', err)
|
||||
}
|
||||
|
||||
return attachments
|
||||
}
|
||||
|
||||
async function getOrCreateConversation(openId: string, name: string): Promise<string> {
|
||||
const existing = getConversationId(openId)
|
||||
if (existing) return existing
|
||||
|
||||
const agentId = getDefaultAgentId()
|
||||
|
||||
const res = await oapiClient.post<{ conversation_id: string }>(
|
||||
'/v1/oapi/super_agent/chat/create_conversation',
|
||||
{
|
||||
agent_id: agentId,
|
||||
title: `Lark User ${name || openId}`,
|
||||
},
|
||||
)
|
||||
|
||||
const conversationId = res.conversation_id
|
||||
setConversationId(openId, conversationId, name)
|
||||
return conversationId
|
||||
}
|
||||
|
||||
async function replyMessage(
|
||||
client: lark.Client,
|
||||
chatId: string,
|
||||
text: string,
|
||||
): Promise<void> {
|
||||
// 检查文本中是否包含 Markdown 图片语法
|
||||
const imageRegex = /!\[([^\]]*)\]\(([^)]+)\)/g
|
||||
const images: Array<{ alt: string; url: string }> = []
|
||||
let match: RegExpExecArray | null
|
||||
|
||||
while ((match = imageRegex.exec(text)) !== null) {
|
||||
images.push({ alt: match[1], url: match[2] })
|
||||
}
|
||||
|
||||
// 如果包含图片，使用富文本消息
|
||||
if (images.length > 0) {
|
||||
// 移除 Markdown 图片语法，保留其他文本
|
||||
const textWithoutImages = text.replace(imageRegex, '').trim()
|
||||
|
||||
// 构建富文本内容
|
||||
const content: any[][] = []
|
||||
|
||||
// 添加文本段落
|
||||
if (textWithoutImages) {
|
||||
const lines = textWithoutImages.split('\n').filter(line => line.trim())
|
||||
for (const line of lines) {
|
||||
content.push([{ tag: 'text', text: line }])
|
||||
}
|
||||
}
|
||||
|
||||
// 上传图片并添加到富文本中
|
||||
for (const image of images) {
|
||||
try {
|
||||
// 下载图片
|
||||
const response = await fetch(image.url)
|
||||
const buffer = Buffer.from(await response.arrayBuffer())
|
||||
|
||||
// 上传到飞书
|
||||
const uploadResp = await client.im.image.create({
|
||||
data: {
|
||||
image_type: 'message',
|
||||
image: buffer,
|
||||
},
|
||||
})
|
||||
|
||||
const imageKey = uploadResp.image_key || uploadResp.data?.image_key
|
||||
if (imageKey) {
|
||||
// 添加图片到富文本
|
||||
content.push([{ tag: 'img', image_key: imageKey }])
|
||||
} else {
|
||||
// 如果没有 image_key，添加链接
|
||||
content.push([{ tag: 'a', text: image.alt || '查看图片', href: image.url }])
|
||||
}
|
||||
} catch (err) {
|
||||
console.error('[Lark Bot] 上传图片失败:', err)
|
||||
// 上传失败，添加图片链接
|
||||
content.push([{ tag: 'a', text: image.alt || '查看图片', href: image.url }])
|
||||
}
|
||||
}
|
||||
|
||||
// 飞书富文本格式需要包含语言标识
|
||||
await client.im.message.create({
|
||||
params: { receive_id_type: 'chat_id' },
|
||||
data: {
|
||||
receive_id: chatId,
|
||||
content: JSON.stringify({
|
||||
zh_cn: {
|
||||
title: '',
|
||||
content: content
|
||||
}
|
||||
}),
|
||||
msg_type: 'post',
|
||||
},
|
||||
})
|
||||
} else {
|
||||
// 纯文本消息
|
||||
await client.im.message.create({
|
||||
params: { receive_id_type: 'chat_id' },
|
||||
data: {
|
||||
receive_id: chatId,
|
||||
content: JSON.stringify({ text }),
|
||||
msg_type: 'text',
|
||||
},
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
async function processMessage(
|
||||
client: lark.Client,
|
||||
data: any,
|
||||
onMessage?: () => void,
|
||||
): Promise<void> {
|
||||
const message = data.message
|
||||
const sender = data.sender
|
||||
const messageId = message.message_id
|
||||
const openId = sender.sender_id.open_id
|
||||
const name = sender.sender_id.union_id || openId
|
||||
const chatId = message.chat_id
|
||||
const messageType = message.message_type
|
||||
const content = message.content
|
||||
|
||||
// 消息去重
|
||||
if (processedMessages.has(messageId)) {
|
||||
return
|
||||
}
|
||||
processedMessages.add(messageId)
|
||||
// 限制 Set 大小，防止内存泄漏
|
||||
if (processedMessages.size > 10000) {
|
||||
const firstItem = processedMessages.values().next().value
|
||||
if (firstItem) {
|
||||
processedMessages.delete(firstItem)
|
||||
}
|
||||
}
|
||||
|
||||
// 提取文本和附件
|
||||
let text = extractTextContent(messageType, content)
|
||||
text = cleanMentions(text)
|
||||
|
||||
const attachments = await extractAttachments(client, messageType, content, messageId)
|
||||
|
||||
if (!text && attachments.length === 0) {
|
||||
return
|
||||
}
|
||||
|
||||
BotLogger.log({
|
||||
platform: 'lark',
|
||||
eventType: 'message_received',
|
||||
severity: 'info',
|
||||
message: `收到来自 ${name} 的消息 (${messageType}, ${attachments.length} 附件)`,
|
||||
details: { openId, chatId },
|
||||
})
|
||||
|
||||
try {
|
||||
// 上传附件
|
||||
let uploadFileIds: string[] | undefined
|
||||
if (attachments.length > 0) {
|
||||
uploadFileIds = await uploadBotAttachments(attachments)
|
||||
if (uploadFileIds.length === 0) uploadFileIds = undefined
|
||||
}
|
||||
|
||||
const conversationId = await getOrCreateConversation(openId, name)
|
||||
const response = await sendMessageAndWaitResponse(
|
||||
`lark:${openId}`,
|
||||
conversationId,
|
||||
text || '',
|
||||
uploadFileIds,
|
||||
)
|
||||
|
||||
if (!response) {
|
||||
await replyMessage(client, chatId, '没有收到回复，请稍后重试。')
|
||||
} else {
|
||||
await replyMessage(client, chatId, response)
|
||||
}
|
||||
|
||||
if (onMessage) onMessage()
|
||||
|
||||
BotLogger.log({
|
||||
platform: 'lark',
|
||||
eventType: 'message_sent',
|
||||
severity: 'info',
|
||||
message: `已回复用户 ${name}`,
|
||||
})
|
||||
} catch (error) {
|
||||
const errMsg = error instanceof Error ? error.message : '未知错误'
|
||||
console.error('[Lark Bot] 处理消息失败:', errMsg)
|
||||
await replyMessage(client, chatId, `处理消息时出错: ${errMsg}`).catch(() => {})
|
||||
BotLogger.log({
|
||||
platform: 'lark',
|
||||
eventType: 'error',
|
||||
severity: 'error',
|
||||
message: `处理消息失败: ${errMsg}`,
|
||||
details: { openId },
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
export function createEventHandler(client: lark.Client, onMessage?: () => void) {
|
||||
return async (data: any) => {
|
||||
const openId = data.sender?.sender_id?.open_id
|
||||
if (!openId) return
|
||||
|
||||
// 使用队列确保单用户消息顺序处理
|
||||
const currentQueue = userQueues.get(openId) ?? Promise.resolve()
|
||||
const newQueue = currentQueue.then(() => processMessage(client, data, onMessage))
|
||||
userQueues.set(openId, newQueue)
|
||||
|
||||
newQueue.finally(() => {
|
||||
if (userQueues.get(openId) === newQueue) {
|
||||
userQueues.delete(openId)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
146
remote-control/bots/lark/index.ts
Normal file
146
remote-control/bots/lark/index.ts
Normal file
@@ -0,0 +1,146 @@
|
||||
import * as lark from '@larksuiteoapi/node-sdk'
|
||||
import { createEventHandler } from './handlers'
|
||||
import { closeConnectionsForPlatform } from '@/remote-control/shared/nova-bridge'
|
||||
import { BotLogger } from '@/remote-control/shared/logger'
|
||||
import { ConfigManager, type ConfigChangeEvent } from '@/remote-control/config/manager'
|
||||
|
||||
let client: lark.Client | null = null
|
||||
let wsClient: lark.WSClient | null = null
|
||||
let currentAppId: string = ''
|
||||
let currentAppSecret: string = ''
|
||||
let startTime: number = 0
|
||||
let messagesProcessed: number = 0
|
||||
let lastError: Error | null = null
|
||||
|
||||
export interface BotStatus {
|
||||
platform: string
|
||||
status: 'connected' | 'disconnected' | 'connecting'
|
||||
uptime?: number
|
||||
messagesProcessed: number
|
||||
error?: string
|
||||
}
|
||||
|
||||
export async function startBot(appId: string, appSecret: string): Promise<void> {
|
||||
if (!appId || !appSecret) {
|
||||
const msg = '飞书 App ID 或 App Secret 为空，跳过启动'
|
||||
console.warn(`[Lark Bot] ${msg}`)
|
||||
BotLogger.log({ platform: 'lark', eventType: 'error', severity: 'warning', message: msg })
|
||||
return
|
||||
}
|
||||
|
||||
// 幂等检查：如果已有活跃连接且凭据未变，直接跳过
|
||||
if (wsClient && currentAppId === appId && currentAppSecret === appSecret) {
|
||||
console.log('[Lark Bot] 已有活跃连接，跳过重复启动')
|
||||
return
|
||||
}
|
||||
|
||||
// 如果已有旧连接（凭据变更），先停止
|
||||
if (wsClient) {
|
||||
console.log('[Lark Bot] 凭据已变更，先停止旧连接')
|
||||
await stopBot()
|
||||
}
|
||||
|
||||
currentAppId = appId
|
||||
currentAppSecret = appSecret
|
||||
|
||||
try {
|
||||
client = new lark.Client({ appId, appSecret })
|
||||
|
||||
wsClient = new lark.WSClient({
|
||||
appId,
|
||||
appSecret,
|
||||
loggerLevel: lark.LoggerLevel.info,
|
||||
})
|
||||
|
||||
const eventHandler = createEventHandler(client, () => { messagesProcessed++ })
|
||||
|
||||
wsClient.start({
|
||||
eventDispatcher: new lark.EventDispatcher({}).register({
|
||||
'im.message.receive_v1': eventHandler,
|
||||
}),
|
||||
})
|
||||
|
||||
startTime = Date.now()
|
||||
lastError = null
|
||||
console.log('[Lark Bot] 已启动，等待消息...')
|
||||
BotLogger.log({
|
||||
platform: 'lark',
|
||||
eventType: 'connection',
|
||||
severity: 'info',
|
||||
message: '飞书 Bot 已启动',
|
||||
})
|
||||
} catch (err) {
|
||||
const error = err instanceof Error ? err : new Error(String(err))
|
||||
lastError = error
|
||||
client = null
|
||||
wsClient = null
|
||||
currentAppId = ''
|
||||
currentAppSecret = ''
|
||||
console.error('[Lark Bot] 启动失败:', error.message)
|
||||
BotLogger.log({
|
||||
platform: 'lark',
|
||||
eventType: 'error',
|
||||
severity: 'error',
|
||||
message: `飞书 Bot 启动失败: ${error.message}`,
|
||||
details: { hint: '请检查 App ID 和 App Secret 是否正确' },
|
||||
})
|
||||
throw error
|
||||
}
|
||||
}
|
||||
|
||||
export async function stopBot(): Promise<void> {
|
||||
closeConnectionsForPlatform('lark')
|
||||
if (wsClient) {
|
||||
wsClient.close()
|
||||
wsClient = null
|
||||
client = null
|
||||
currentAppId = ''
|
||||
currentAppSecret = ''
|
||||
startTime = 0
|
||||
console.log('[Lark Bot] 已停止')
|
||||
BotLogger.log({
|
||||
platform: 'lark',
|
||||
eventType: 'disconnection',
|
||||
severity: 'info',
|
||||
message: '飞书 Bot 已停止',
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
export function getStatus(): BotStatus {
|
||||
return {
|
||||
platform: 'lark',
|
||||
status: wsClient ? 'connected' : 'disconnected',
|
||||
uptime: wsClient && startTime ? Math.floor((Date.now() - startTime) / 1000) : undefined,
|
||||
messagesProcessed,
|
||||
error: lastError?.message,
|
||||
}
|
||||
}
|
||||
|
||||
const configManager = ConfigManager.getInstance()
|
||||
configManager.on('config-changed', async (event: ConfigChangeEvent) => {
|
||||
const { newConfig } = event
|
||||
const platformConfig = newConfig.lark
|
||||
|
||||
try {
|
||||
if (platformConfig.enabled && !wsClient) {
|
||||
await startBot(platformConfig.appId, platformConfig.appSecret)
|
||||
} else if (!platformConfig.enabled && wsClient) {
|
||||
await stopBot()
|
||||
} else if (platformConfig.enabled && wsClient) {
|
||||
if (platformConfig.appId !== currentAppId || platformConfig.appSecret !== currentAppSecret) {
|
||||
await stopBot()
|
||||
await startBot(platformConfig.appId, platformConfig.appSecret)
|
||||
}
|
||||
}
|
||||
lastError = null
|
||||
} catch (error) {
|
||||
lastError = error instanceof Error ? error : new Error(String(error))
|
||||
BotLogger.log({
|
||||
platform: 'lark',
|
||||
eventType: 'error',
|
||||
severity: 'error',
|
||||
message: `配置变更处理失败: ${lastError.message}`,
|
||||
})
|
||||
}
|
||||
})
|
||||
47
remote-control/bots/lark/user-store.ts
Normal file
47
remote-control/bots/lark/user-store.ts
Normal file
@@ -0,0 +1,47 @@
|
||||
import fs from 'node:fs'
|
||||
import path from 'node:path'
|
||||
|
||||
interface UserMapping {
|
||||
conversationId: string
|
||||
name: string
|
||||
createdAt: string
|
||||
}
|
||||
|
||||
type UserStore = Record<string, UserMapping>
|
||||
|
||||
const STORE_PATH = path.join(process.cwd(), 'remote-control', 'data', 'lark-users.json')
|
||||
|
||||
function readStore(): UserStore {
|
||||
try {
|
||||
if (!fs.existsSync(STORE_PATH)) {
|
||||
return {}
|
||||
}
|
||||
const data = fs.readFileSync(STORE_PATH, 'utf-8')
|
||||
return JSON.parse(data)
|
||||
} catch {
|
||||
return {}
|
||||
}
|
||||
}
|
||||
|
||||
function writeStore(store: UserStore): void {
|
||||
const dir = path.dirname(STORE_PATH)
|
||||
if (!fs.existsSync(dir)) {
|
||||
fs.mkdirSync(dir, { recursive: true })
|
||||
}
|
||||
fs.writeFileSync(STORE_PATH, JSON.stringify(store, null, 2), 'utf-8')
|
||||
}
|
||||
|
||||
export function getConversationId(openId: string): string | null {
|
||||
const store = readStore()
|
||||
return store[openId]?.conversationId ?? null
|
||||
}
|
||||
|
||||
export function setConversationId(openId: string, conversationId: string, name: string): void {
|
||||
const store = readStore()
|
||||
store[openId] = {
|
||||
conversationId,
|
||||
name,
|
||||
createdAt: new Date().toISOString(),
|
||||
}
|
||||
writeStore(store)
|
||||
}
|
||||
Reference in New Issue
Block a user