82 lines
2.2 KiB
JavaScript
82 lines
2.2 KiB
JavaScript
import { analyzeWechatMessages } from '../../analysis/wechatAnalyzer.js'
|
||
import { getWechatRuntimeConfig } from '../../config/env.js'
|
||
import { runOpenCli } from '../../adapters/opencli.js'
|
||
|
||
function stripMention(content, botName) {
|
||
return content.replace(botName, '').trim()
|
||
}
|
||
|
||
function parseTarget(tokens) {
|
||
const type = tokens[1]
|
||
const value = tokens.slice(2).join(' ').trim()
|
||
|
||
if (['群', '群聊', 'room', 'group'].includes(type)) {
|
||
return { room: value }
|
||
}
|
||
|
||
if (['好友', 'friend', 'contact'].includes(type)) {
|
||
return { friend: value }
|
||
}
|
||
|
||
return {}
|
||
}
|
||
|
||
export async function handleWechatCommand(content, context = {}) {
|
||
const config = getWechatRuntimeConfig()
|
||
const normalized = stripMention(content, config.botName)
|
||
|
||
if (!normalized.startsWith(config.commandPrefix)) {
|
||
return { handled: false }
|
||
}
|
||
|
||
const commandLine = normalized.slice(config.commandPrefix.length).trim()
|
||
const tokens = commandLine.split(/\s+/).filter(Boolean)
|
||
const command = tokens[0]
|
||
|
||
if (['分析', 'analyze', '统计', 'stats'].includes(command)) {
|
||
const statsOnly = ['统计', 'stats'].includes(command)
|
||
const target = parseTarget(tokens)
|
||
const result = await analyzeWechatMessages({
|
||
...target,
|
||
serviceType: context.serviceType,
|
||
dataDir: config.dataDir,
|
||
statsOnly,
|
||
})
|
||
|
||
if (statsOnly || !result.analysis) {
|
||
return {
|
||
handled: true,
|
||
reply: [
|
||
`${result.target}`,
|
||
`消息数：${result.stats.totalMessages}`,
|
||
`文本消息：${result.stats.textMessages}`,
|
||
`平均长度：${result.stats.averageTextLength}`,
|
||
`高频发言：${result.stats.topSpeakers.map((item) => `${item.name}(${item.count})`).join('，') || '无'}`,
|
||
].join('\n'),
|
||
}
|
||
}
|
||
|
||
return {
|
||
handled: true,
|
||
reply: result.analysis,
|
||
}
|
||
}
|
||
|
||
if (command === 'opencli') {
|
||
if (!config.enableRemoteOpenCli) {
|
||
return {
|
||
handled: true,
|
||
reply: '远程 OpenCLI 执行未开启。需要在 .env 中显式设置 ENABLE_REMOTE_OPENCLI=true。',
|
||
}
|
||
}
|
||
|
||
await runOpenCli(tokens.slice(1))
|
||
return {
|
||
handled: true,
|
||
reply: 'OpenCLI 命令已执行，结果请看本机控制台。',
|
||
}
|
||
}
|
||
|
||
return { handled: false }
|
||
}
|