weixin-bot/examples/nodejs/echo-bot.ts at main · epiral/weixin-bot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
#!/usr/bin/env npx tsx
/**
 * WeChat Echo Bot — 完整示例
 *
 * 功能：
 *   1. 首次运行扫码登录（终端渲染二维码），凭证自动保存
 *   2. 后续运行自动加载已保存凭证，跳过扫码
 *   3. 收到消息后显示"正在输入"，然后回复 "Echo: ..."
 *   4. Session 过期时自动重新扫码
 *
 * 用法：
 *   npx tsx examples/nodejs/echo-bot.ts
 *   npx tsx examples/nodejs/echo-bot.ts --force-login   # 强制重新扫码
 *
 * 依赖（example 自带，不影响 SDK）：
 *   qrcode-terminal — 终端二维码渲染
 */

import { WeixinBot } from '@pinixai/weixin-bot'
import qrterm from 'qrcode-terminal'

// ── 配置 ─────────────────────────────────────────────────────────────────

const forceLogin = process.argv.includes('--force-login')

// ── 日志工具 ──────────────────────────────────────────────────────────────

function log(level: string, msg: string) {
  const ts = new Date().toISOString()
  console.log(`${ts} [${level}] ${msg}`)
}

// ── 拦截 SDK 登录 URL，渲染二维码 ─────────────────────────────────────────

const origStderrWrite = process.stderr.write.bind(process.stderr)
process.stderr.write = ((chunk: any, ...args: any[]) => {
  const str = typeof chunk === 'string' ? chunk : chunk.toString()
  // 检测到登录 URL，渲染二维码
  if (str.startsWith('https://') && str.includes('qrcode=')) {
    const url = str.trim()
    qrterm.generate(url, { small: true }, (qr: string) => {
      origStderrWrite(qr + '\n')
    })
  }
  return origStderrWrite(chunk, ...args)
}) as typeof process.stderr.write

// ── 启动 ──────────────────────────────────────────────────────────────────

const bot = new WeixinBot({
  onError: (err) => {
    log('ERROR', err instanceof Error ? err.stack ?? err.message : String(err))
  },
})

log('INFO', forceLogin ? '强制重新扫码登录...' : '正在登录（已有凭证则自动跳过扫码）...')
const creds = await bot.login({ force: forceLogin })
log('INFO', `登录成功 — Bot ID: ${creds.accountId}`)
log('INFO', `关联用户: ${creds.userId}`)
log('INFO', `API 地址: ${creds.baseUrl}`)

// 统计
let messageCount = 0
const startTime = Date.now()

// 注册消息处理
bot.onMessage(async (msg) => {
  messageCount++
  const elapsed = ((Date.now() - startTime) / 1000).toFixed(0)

  log('RECV', `#${messageCount} | 类型: ${msg.type} | 用户: ${msg.userId}`)
  log('RECV', `内容: ${msg.text}`)

  try {
    await bot.sendTyping(msg.userId)
  } catch { /* typing 失败不影响回复 */ }

  await new Promise((resolve) => setTimeout(resolve, 1000))

  const reply = `Echo: ${msg.text}`

  try {
    await bot.reply(msg, reply)
    log('SEND', `回复成功 (${reply.length} 字符) | 运行 ${elapsed}s | 累计 ${messageCount} 条`)
  } catch (err) {
    log('ERROR', `回复失败: ${err instanceof Error ? err.message : String(err)}`)
  }
})

process.on('SIGINT', () => {
  log('INFO', `收到 SIGINT，正在停止... (共处理 ${messageCount} 条消息)`)
  bot.stop()
})

log('INFO', '开始接收微信消息 (Ctrl+C 停止)')
log('INFO', '────────────────────────────────────')
await bot.run()
log('INFO', `Bot 已停止，共处理 ${messageCount} 条消息`)