mirror of
https://github.com/claude-code-best/claude-code.git
synced 2026-06-15 12:55:51 +00:00
* feat: 接入 weixin 服务层与命令入口 Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai> * feat: 注册内建 weixin channel 插件 Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai> * fix: 修正 channel permission relay 路由与能力判定 Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai> * fix: 修复 builtin channel 的 ChannelsNotice 误报 Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai> * docs: 补充内建 weixin channel 使用说明 Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai> * docs: 更新微信 channel 接入计划状态 Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai> * fix: 延迟加载 weixin 登录二维码依赖 Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai> * fix: 改用 qrcode 生成 weixin 登录二维码 Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai> * fix: 修正 vite 构建的 Windows 路径解析 Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai> * chore: 删除临时规划文档 wx_channel.md 并还原 package.json 排序 wx_channel.md 内容已整合到 docs/features/channels.md,不再需要。 package.json 中 @ant/model-provider 位置从原始位置被无意移动,还原。 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * refactor: 将 weixin 模块从 src/ 迁移至 packages/weixin 工作区包 将 src/services/weixin/ 中的纯业务逻辑迁入 @claude-code-best/weixin workspace 包,降低 src/ 耦合度。仅保留 server.ts 作为薄适配层。 - 迁移 7 个无修改的纯模块 (types/api/accounts/login/pairing/media/send) - monitor.ts 内联 PERMISSION_REPLY_RE 正则,解除对 src/ 的依赖 - permissions.ts 本地定义 ChannelPermissionRequestParams 接口 - cli.ts 拆分:serve 子命令通过回调注入,login/access 保留在包内 - server.ts 重写为从 @claude-code-best/weixin 导入 - 新增 cli-serve.ts 作为 serve 入口薄壳 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * fix: 修正 weixin barrel export 中 interface 的导出方式 ChannelPermissionRequestParams 是纯类型,必须用 export type 导出, 否则 Bun 运行时会报 "export not found" 错误。 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * refactor: 将 server.ts 迁入 packages/weixin,彻底移除 src/services/weixin/ 通过依赖注入(WeixinServerDeps)解耦 src/ 依赖(analytics、config、 MCP channel schema),server.ts 完全移入包内。cli.tsx 入口处一次性 注入所有依赖。 src/services/weixin/ 目录已完全删除。 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * fix: 修复 markdownToPlainText 中代码块正则的 ReDoS 风险 用非正则的线性扫描替代 \`\`\`[\s\S]*?\n([\s\S]*?)\`\`\` 匹配, 避免在含有大量重复 \`\`\` 序列的输入上触发多项式回溯。 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> --------- Co-authored-by: 1111 <11111@asd.c> Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
11
packages/weixin/package.json
Normal file
11
packages/weixin/package.json
Normal file
@@ -0,0 +1,11 @@
|
||||
{
|
||||
"name": "@claude-code-best/weixin",
|
||||
"version": "1.0.0",
|
||||
"private": true,
|
||||
"type": "module",
|
||||
"main": "./src/index.ts",
|
||||
"types": "./src/index.ts",
|
||||
"dependencies": {
|
||||
"qrcode": "^1.5.4"
|
||||
}
|
||||
}
|
||||
54
packages/weixin/src/__tests__/accounts.test.ts
Normal file
54
packages/weixin/src/__tests__/accounts.test.ts
Normal file
@@ -0,0 +1,54 @@
|
||||
import { afterEach, describe, expect, test } from 'bun:test'
|
||||
import { mkdtempSync, rmSync, statSync } from 'node:fs'
|
||||
import { tmpdir } from 'node:os'
|
||||
import { join } from 'node:path'
|
||||
|
||||
const testDir = mkdtempSync(join(tmpdir(), 'weixin-test-accounts-'))
|
||||
process.env.WEIXIN_STATE_DIR = testDir
|
||||
|
||||
import { clearAccount, loadAccount, saveAccount } from '../accounts.js'
|
||||
|
||||
afterEach(() => {
|
||||
rmSync(testDir, { recursive: true, force: true })
|
||||
})
|
||||
|
||||
describe('account storage', () => {
|
||||
test('loadAccount returns null when no account exists', () => {
|
||||
expect(loadAccount()).toBeNull()
|
||||
})
|
||||
|
||||
test('saveAccount and loadAccount round-trip', () => {
|
||||
const data = {
|
||||
token: 'test-token',
|
||||
baseUrl: 'https://example.com',
|
||||
userId: 'user1',
|
||||
savedAt: '2025-01-01T00:00:00.000Z',
|
||||
}
|
||||
saveAccount(data)
|
||||
expect(loadAccount()).toEqual(data)
|
||||
})
|
||||
|
||||
test('saveAccount sets file permissions to 0600', () => {
|
||||
saveAccount({
|
||||
token: 'test',
|
||||
baseUrl: 'https://example.com',
|
||||
savedAt: new Date().toISOString(),
|
||||
})
|
||||
const stats = statSync(join(testDir, 'account.json'))
|
||||
if (process.platform === 'win32') {
|
||||
expect(stats.isFile()).toBe(true)
|
||||
return
|
||||
}
|
||||
expect(stats.mode & 0o777).toBe(0o600)
|
||||
})
|
||||
|
||||
test('clearAccount removes the file', () => {
|
||||
saveAccount({
|
||||
token: 'test',
|
||||
baseUrl: 'https://example.com',
|
||||
savedAt: new Date().toISOString(),
|
||||
})
|
||||
clearAccount()
|
||||
expect(loadAccount()).toBeNull()
|
||||
})
|
||||
})
|
||||
90
packages/weixin/src/__tests__/media.test.ts
Normal file
90
packages/weixin/src/__tests__/media.test.ts
Normal file
@@ -0,0 +1,90 @@
|
||||
import { describe, expect, test } from 'bun:test'
|
||||
import { randomBytes } from 'node:crypto'
|
||||
import {
|
||||
aesEcbPaddedSize,
|
||||
buildCdnDownloadUrl,
|
||||
buildCdnUploadUrl,
|
||||
decryptAesEcb,
|
||||
encryptAesEcb,
|
||||
guessMediaType,
|
||||
parseAesKey,
|
||||
} from '../media.js'
|
||||
import { UploadMediaType } from '../types.js'
|
||||
|
||||
describe('AES-128-ECB', () => {
|
||||
test('encrypt then decrypt returns original data', () => {
|
||||
const key = randomBytes(16)
|
||||
const plaintext = Buffer.from('hello world test data!!')
|
||||
const ciphertext = encryptAesEcb(plaintext, key)
|
||||
expect(decryptAesEcb(ciphertext, key)).toEqual(plaintext)
|
||||
})
|
||||
|
||||
test('different keys produce different ciphertext', () => {
|
||||
const plaintext = Buffer.from('test data')
|
||||
expect(
|
||||
encryptAesEcb(plaintext, randomBytes(16)),
|
||||
).not.toEqual(encryptAesEcb(plaintext, randomBytes(16)))
|
||||
})
|
||||
})
|
||||
|
||||
describe('aesEcbPaddedSize', () => {
|
||||
test('pads to next 16-byte boundary', () => {
|
||||
expect(aesEcbPaddedSize(1)).toBe(16)
|
||||
expect(aesEcbPaddedSize(16)).toBe(32)
|
||||
expect(aesEcbPaddedSize(17)).toBe(32)
|
||||
expect(aesEcbPaddedSize(32)).toBe(48)
|
||||
})
|
||||
})
|
||||
|
||||
describe('parseAesKey', () => {
|
||||
test('parses 16 raw bytes from base64', () => {
|
||||
const raw = randomBytes(16)
|
||||
expect(parseAesKey(raw.toString('base64'))).toEqual(raw)
|
||||
})
|
||||
|
||||
test('parses hex-encoded key from base64', () => {
|
||||
const raw = randomBytes(16)
|
||||
const b64 = Buffer.from(raw.toString('hex'), 'ascii').toString('base64')
|
||||
expect(parseAesKey(b64)).toEqual(raw)
|
||||
})
|
||||
|
||||
test('throws on invalid key length', () => {
|
||||
expect(() => parseAesKey(Buffer.from('short').toString('base64'))).toThrow(
|
||||
'Invalid aes_key',
|
||||
)
|
||||
})
|
||||
})
|
||||
|
||||
describe('CDN URL builders', () => {
|
||||
test('buildCdnDownloadUrl encodes param', () => {
|
||||
expect(buildCdnDownloadUrl('abc=123', 'https://cdn.example.com')).toBe(
|
||||
'https://cdn.example.com/download?encrypted_query_param=abc%3D123',
|
||||
)
|
||||
})
|
||||
|
||||
test('buildCdnUploadUrl encodes params', () => {
|
||||
expect(
|
||||
buildCdnUploadUrl('https://cdn.example.com', 'param1', 'key1'),
|
||||
).toBe(
|
||||
'https://cdn.example.com/upload?encrypted_query_param=param1&filekey=key1',
|
||||
)
|
||||
})
|
||||
})
|
||||
|
||||
describe('guessMediaType', () => {
|
||||
test('detects image extensions', () => {
|
||||
expect(guessMediaType('photo.jpg')).toBe(UploadMediaType.IMAGE)
|
||||
expect(guessMediaType('photo.png')).toBe(UploadMediaType.IMAGE)
|
||||
expect(guessMediaType('photo.webp')).toBe(UploadMediaType.IMAGE)
|
||||
})
|
||||
|
||||
test('detects video extensions', () => {
|
||||
expect(guessMediaType('video.mp4')).toBe(UploadMediaType.VIDEO)
|
||||
expect(guessMediaType('video.mov')).toBe(UploadMediaType.VIDEO)
|
||||
})
|
||||
|
||||
test('defaults to FILE for unknown extensions', () => {
|
||||
expect(guessMediaType('doc.pdf')).toBe(UploadMediaType.FILE)
|
||||
expect(guessMediaType('archive.zip')).toBe(UploadMediaType.FILE)
|
||||
})
|
||||
})
|
||||
22
packages/weixin/src/__tests__/monitor.test.ts
Normal file
22
packages/weixin/src/__tests__/monitor.test.ts
Normal file
@@ -0,0 +1,22 @@
|
||||
import { describe, expect, test } from 'bun:test'
|
||||
import { extractPermissionReply } from '../monitor.js'
|
||||
|
||||
describe('extractPermissionReply', () => {
|
||||
test('parses allow replies', () => {
|
||||
expect(extractPermissionReply('yes abcde')).toEqual({
|
||||
requestId: 'abcde',
|
||||
behavior: 'allow',
|
||||
})
|
||||
})
|
||||
|
||||
test('parses deny replies', () => {
|
||||
expect(extractPermissionReply('No abcde')).toEqual({
|
||||
requestId: 'abcde',
|
||||
behavior: 'deny',
|
||||
})
|
||||
})
|
||||
|
||||
test('ignores unrelated text', () => {
|
||||
expect(extractPermissionReply('yes please do it')).toBeNull()
|
||||
})
|
||||
})
|
||||
78
packages/weixin/src/__tests__/pairing.test.ts
Normal file
78
packages/weixin/src/__tests__/pairing.test.ts
Normal file
@@ -0,0 +1,78 @@
|
||||
import { afterEach, describe, expect, test } from 'bun:test'
|
||||
import { mkdtempSync, rmSync } from 'node:fs'
|
||||
import { tmpdir } from 'node:os'
|
||||
import { join } from 'node:path'
|
||||
|
||||
const testDir = mkdtempSync(join(tmpdir(), 'weixin-test-pairing-'))
|
||||
process.env.WEIXIN_STATE_DIR = testDir
|
||||
|
||||
import {
|
||||
addPendingPairing,
|
||||
confirmPairing,
|
||||
isAllowed,
|
||||
loadAccessConfig,
|
||||
saveAccessConfig,
|
||||
} from '../pairing.js'
|
||||
|
||||
afterEach(() => {
|
||||
rmSync(testDir, { recursive: true, force: true })
|
||||
})
|
||||
|
||||
describe('loadAccessConfig', () => {
|
||||
test('returns default config when no file exists', () => {
|
||||
const config = loadAccessConfig()
|
||||
expect(config.policy).toBe('pairing')
|
||||
expect(config.allowFrom).toEqual([])
|
||||
})
|
||||
|
||||
test('round-trips saved config', () => {
|
||||
saveAccessConfig({ policy: 'allowlist', allowFrom: ['user1'] })
|
||||
const config = loadAccessConfig()
|
||||
expect(config.policy).toBe('allowlist')
|
||||
expect(config.allowFrom).toEqual(['user1'])
|
||||
})
|
||||
})
|
||||
|
||||
describe('isAllowed', () => {
|
||||
test('returns false for unknown user under pairing policy', () => {
|
||||
expect(isAllowed('unknown')).toBe(false)
|
||||
})
|
||||
|
||||
test('returns true for allowed user', () => {
|
||||
saveAccessConfig({ policy: 'pairing', allowFrom: ['user1'] })
|
||||
expect(isAllowed('user1')).toBe(true)
|
||||
})
|
||||
|
||||
test('returns true for any user under disabled policy', () => {
|
||||
saveAccessConfig({ policy: 'disabled', allowFrom: [] })
|
||||
expect(isAllowed('anyone')).toBe(true)
|
||||
})
|
||||
})
|
||||
|
||||
describe('pairing flow', () => {
|
||||
test('generates 6-digit code', () => {
|
||||
expect(addPendingPairing('user1')).toMatch(/^\d{6}$/)
|
||||
})
|
||||
|
||||
test('returns same code for same user', () => {
|
||||
const code1 = addPendingPairing('user1')
|
||||
const code2 = addPendingPairing('user1')
|
||||
expect(code1).toBe(code2)
|
||||
})
|
||||
|
||||
test('confirm adds user to allowlist', () => {
|
||||
const code = addPendingPairing('user1')
|
||||
expect(confirmPairing(code)).toBe('user1')
|
||||
expect(isAllowed('user1')).toBe(true)
|
||||
})
|
||||
|
||||
test('confirm returns null for invalid code', () => {
|
||||
expect(confirmPairing('000000')).toBeNull()
|
||||
})
|
||||
|
||||
test('code cannot be reused after confirmation', () => {
|
||||
const code = addPendingPairing('user1')
|
||||
confirmPairing(code)
|
||||
expect(confirmPairing(code)).toBeNull()
|
||||
})
|
||||
})
|
||||
43
packages/weixin/src/__tests__/permissions.test.ts
Normal file
43
packages/weixin/src/__tests__/permissions.test.ts
Normal file
@@ -0,0 +1,43 @@
|
||||
import { afterEach, describe, expect, test } from 'bun:test'
|
||||
import {
|
||||
clearPermissionStateForTests,
|
||||
consumePendingPermission,
|
||||
getActivePermissionChat,
|
||||
savePendingPermission,
|
||||
setActivePermissionChat,
|
||||
} from '../permissions.js'
|
||||
|
||||
afterEach(() => {
|
||||
clearPermissionStateForTests()
|
||||
})
|
||||
|
||||
describe('permission state', () => {
|
||||
test('tracks active permission chat', () => {
|
||||
setActivePermissionChat('user-1', 'ctx-1')
|
||||
expect(getActivePermissionChat()).toEqual({
|
||||
chatId: 'user-1',
|
||||
contextToken: 'ctx-1',
|
||||
updatedAt: expect.any(Number),
|
||||
})
|
||||
})
|
||||
|
||||
test('consumes pending permission only for matching user', () => {
|
||||
savePendingPermission(
|
||||
{
|
||||
request_id: 'abcde',
|
||||
tool_name: 'Bash',
|
||||
description: 'Run a command',
|
||||
input_preview: '{"command":"pwd"}',
|
||||
},
|
||||
'user-1',
|
||||
'ctx-1',
|
||||
)
|
||||
|
||||
expect(consumePendingPermission('abcde', 'user-2')).toBeNull()
|
||||
expect(consumePendingPermission('ABCDE', 'user-1')).toMatchObject({
|
||||
request_id: 'abcde',
|
||||
chatId: 'user-1',
|
||||
})
|
||||
expect(consumePendingPermission('abcde', 'user-1')).toBeNull()
|
||||
})
|
||||
})
|
||||
32
packages/weixin/src/__tests__/send.test.ts
Normal file
32
packages/weixin/src/__tests__/send.test.ts
Normal file
@@ -0,0 +1,32 @@
|
||||
import { describe, expect, test } from 'bun:test'
|
||||
import { markdownToPlainText } from '../send.js'
|
||||
|
||||
describe('markdownToPlainText', () => {
|
||||
test('removes bold markers', () => {
|
||||
expect(markdownToPlainText('**bold**')).toBe('bold')
|
||||
})
|
||||
|
||||
test('removes italic markers', () => {
|
||||
expect(markdownToPlainText('*italic*')).toBe('italic')
|
||||
})
|
||||
|
||||
test('removes inline code backticks', () => {
|
||||
expect(markdownToPlainText('`code`')).toBe('code')
|
||||
})
|
||||
|
||||
test('removes code block fences', () => {
|
||||
expect(markdownToPlainText("```js\nconsole.log('hi');\n```"))
|
||||
.toBe("console.log('hi');")
|
||||
})
|
||||
|
||||
test('converts links to text with URL', () => {
|
||||
expect(markdownToPlainText('[click](https://example.com)')).toBe(
|
||||
'click (https://example.com)',
|
||||
)
|
||||
})
|
||||
|
||||
test('handles mixed markdown', () => {
|
||||
expect(markdownToPlainText('# Hello\n\n**bold** and *italic* with `code`'))
|
||||
.toBe('Hello\n\nbold and italic with code')
|
||||
})
|
||||
})
|
||||
57
packages/weixin/src/accounts.ts
Normal file
57
packages/weixin/src/accounts.ts
Normal file
@@ -0,0 +1,57 @@
|
||||
import {
|
||||
chmodSync,
|
||||
existsSync,
|
||||
mkdirSync,
|
||||
readFileSync,
|
||||
unlinkSync,
|
||||
writeFileSync,
|
||||
} from 'node:fs'
|
||||
import { homedir } from 'node:os'
|
||||
import { join } from 'node:path'
|
||||
|
||||
export const DEFAULT_BASE_URL = 'https://ilinkai.weixin.qq.com'
|
||||
export const CDN_BASE_URL = 'https://novac2c.cdn.weixin.qq.com/c2c'
|
||||
|
||||
export interface AccountData {
|
||||
token: string
|
||||
baseUrl: string
|
||||
userId?: string
|
||||
savedAt: string
|
||||
}
|
||||
|
||||
export function getStateDir(): string {
|
||||
const dir =
|
||||
process.env.WEIXIN_STATE_DIR ||
|
||||
join(homedir(), '.claude', 'channels', 'weixin')
|
||||
if (!existsSync(dir)) {
|
||||
mkdirSync(dir, { recursive: true })
|
||||
}
|
||||
return dir
|
||||
}
|
||||
|
||||
function accountPath(): string {
|
||||
return join(getStateDir(), 'account.json')
|
||||
}
|
||||
|
||||
export function loadAccount(): AccountData | null {
|
||||
const path = accountPath()
|
||||
if (!existsSync(path)) return null
|
||||
try {
|
||||
return JSON.parse(readFileSync(path, 'utf-8')) as AccountData
|
||||
} catch {
|
||||
return null
|
||||
}
|
||||
}
|
||||
|
||||
export function saveAccount(data: AccountData): void {
|
||||
const path = accountPath()
|
||||
writeFileSync(path, JSON.stringify(data, null, 2), 'utf-8')
|
||||
chmodSync(path, 0o600)
|
||||
}
|
||||
|
||||
export function clearAccount(): void {
|
||||
const path = accountPath()
|
||||
if (existsSync(path)) {
|
||||
unlinkSync(path)
|
||||
}
|
||||
}
|
||||
148
packages/weixin/src/api.ts
Normal file
148
packages/weixin/src/api.ts
Normal file
@@ -0,0 +1,148 @@
|
||||
import { randomBytes } from 'node:crypto'
|
||||
import type {
|
||||
BaseInfo,
|
||||
GetConfigResp,
|
||||
GetUpdatesReq,
|
||||
GetUpdatesResp,
|
||||
GetUploadUrlReq,
|
||||
GetUploadUrlResp,
|
||||
SendMessageReq,
|
||||
SendTypingReq,
|
||||
SendTypingResp,
|
||||
} from './types.js'
|
||||
|
||||
const CHANNEL_VERSION = '0.1.0'
|
||||
|
||||
function baseInfo(): BaseInfo {
|
||||
return { channel_version: CHANNEL_VERSION }
|
||||
}
|
||||
|
||||
function randomUin(): string {
|
||||
return randomBytes(4).toString('base64')
|
||||
}
|
||||
|
||||
function buildHeaders(token?: string): Record<string, string> {
|
||||
const headers: Record<string, string> = {
|
||||
'Content-Type': 'application/json',
|
||||
'X-WECHAT-UIN': randomUin(),
|
||||
}
|
||||
if (token) {
|
||||
headers.AuthorizationType = 'ilink_bot_token'
|
||||
headers.Authorization = `Bearer ${token}`
|
||||
}
|
||||
return headers
|
||||
}
|
||||
|
||||
async function post<T>(
|
||||
baseUrl: string,
|
||||
path: string,
|
||||
body: unknown,
|
||||
token?: string,
|
||||
timeoutMs = 40_000,
|
||||
signal?: AbortSignal,
|
||||
): Promise<T> {
|
||||
const controller = new AbortController()
|
||||
const timeout = setTimeout(() => controller.abort(), timeoutMs)
|
||||
|
||||
if (signal) {
|
||||
signal.addEventListener('abort', () => controller.abort(), { once: true })
|
||||
}
|
||||
|
||||
try {
|
||||
const response = await fetch(`${baseUrl}${path}`, {
|
||||
method: 'POST',
|
||||
headers: buildHeaders(token),
|
||||
body: JSON.stringify(body),
|
||||
signal: controller.signal,
|
||||
})
|
||||
|
||||
if (!response.ok) {
|
||||
throw new Error(`HTTP ${response.status}: ${response.statusText}`)
|
||||
}
|
||||
|
||||
return (await response.json()) as T
|
||||
} finally {
|
||||
clearTimeout(timeout)
|
||||
}
|
||||
}
|
||||
|
||||
export async function getUpdates(
|
||||
baseUrl: string,
|
||||
token: string,
|
||||
getUpdatesBuf: string,
|
||||
signal?: AbortSignal,
|
||||
): Promise<GetUpdatesResp> {
|
||||
const body: GetUpdatesReq = {
|
||||
get_updates_buf: getUpdatesBuf,
|
||||
base_info: baseInfo(),
|
||||
}
|
||||
|
||||
try {
|
||||
return await post<GetUpdatesResp>(
|
||||
baseUrl,
|
||||
'/ilink/bot/getupdates',
|
||||
body,
|
||||
token,
|
||||
40_000,
|
||||
signal,
|
||||
)
|
||||
} catch (error) {
|
||||
if (error instanceof Error && error.name === 'AbortError') {
|
||||
return { ret: 0, msgs: [], get_updates_buf: getUpdatesBuf }
|
||||
}
|
||||
throw error
|
||||
}
|
||||
}
|
||||
|
||||
export async function sendMessage(
|
||||
baseUrl: string,
|
||||
token: string,
|
||||
msg: SendMessageReq['msg'],
|
||||
): Promise<void> {
|
||||
const body: SendMessageReq = { msg, base_info: baseInfo() }
|
||||
await post(baseUrl, '/ilink/bot/sendmessage', body, token)
|
||||
}
|
||||
|
||||
export async function getUploadUrl(
|
||||
baseUrl: string,
|
||||
token: string,
|
||||
params: Omit<GetUploadUrlReq, 'base_info'>,
|
||||
): Promise<GetUploadUrlResp> {
|
||||
return post<GetUploadUrlResp>(
|
||||
baseUrl,
|
||||
'/ilink/bot/getuploadurl',
|
||||
{ ...params, base_info: baseInfo() },
|
||||
token,
|
||||
)
|
||||
}
|
||||
|
||||
export async function getConfig(
|
||||
baseUrl: string,
|
||||
token: string,
|
||||
userId: string,
|
||||
contextToken?: string,
|
||||
): Promise<GetConfigResp> {
|
||||
return post<GetConfigResp>(
|
||||
baseUrl,
|
||||
'/ilink/bot/getconfig',
|
||||
{
|
||||
ilink_user_id: userId,
|
||||
context_token: contextToken,
|
||||
base_info: baseInfo(),
|
||||
},
|
||||
token,
|
||||
)
|
||||
}
|
||||
|
||||
export async function sendTyping(
|
||||
baseUrl: string,
|
||||
token: string,
|
||||
req: Omit<SendTypingReq, 'base_info'>,
|
||||
): Promise<SendTypingResp> {
|
||||
return post<SendTypingResp>(
|
||||
baseUrl,
|
||||
'/ilink/bot/sendtyping',
|
||||
{ ...req, base_info: baseInfo() },
|
||||
token,
|
||||
)
|
||||
}
|
||||
119
packages/weixin/src/cli.ts
Normal file
119
packages/weixin/src/cli.ts
Normal file
@@ -0,0 +1,119 @@
|
||||
import { clearAccount, DEFAULT_BASE_URL, loadAccount, saveAccount } from './accounts.js'
|
||||
import { startLogin, waitForLogin } from './login.js'
|
||||
import { confirmPairing } from './pairing.js'
|
||||
import { runWeixinMcpServer } from './server.js'
|
||||
import type { WeixinServerDeps } from './server.js'
|
||||
|
||||
function printUsage(): void {
|
||||
process.stdout.write(
|
||||
[
|
||||
'Usage:',
|
||||
' ccb weixin serve',
|
||||
' ccb weixin login',
|
||||
' ccb weixin login clear',
|
||||
' ccb weixin access pair <code>',
|
||||
'',
|
||||
'Session enablement:',
|
||||
' ccb --channels plugin:weixin@builtin',
|
||||
].join('\n') + '\n',
|
||||
)
|
||||
}
|
||||
|
||||
async function runLogin(clear = false): Promise<void> {
|
||||
if (clear) {
|
||||
clearAccount()
|
||||
process.stdout.write('WeChat account cleared.\n')
|
||||
return
|
||||
}
|
||||
|
||||
const existing = loadAccount()
|
||||
if (existing) {
|
||||
process.stdout.write(
|
||||
[
|
||||
'Already connected:',
|
||||
` User ID: ${existing.userId || 'unknown'}`,
|
||||
` Connected since: ${existing.savedAt}`,
|
||||
'',
|
||||
'Run `ccb weixin login clear` to disconnect.',
|
||||
'Restart Claude Code with:',
|
||||
' ccb --channels plugin:weixin@builtin',
|
||||
].join('\n') + '\n',
|
||||
)
|
||||
return
|
||||
}
|
||||
|
||||
process.stdout.write('Starting WeChat QR login...\n\n')
|
||||
const qr = await startLogin(DEFAULT_BASE_URL)
|
||||
process.stdout.write(
|
||||
`\nScan the QR code above with WeChat, or open this URL:\n${qr.qrcodeUrl || ''}\n\n`,
|
||||
)
|
||||
|
||||
const result = await waitForLogin({
|
||||
qrcodeId: qr.qrcodeId,
|
||||
apiBaseUrl: DEFAULT_BASE_URL,
|
||||
})
|
||||
|
||||
if (!result.connected || !result.token) {
|
||||
process.stderr.write(`Login failed: ${result.message}\n`)
|
||||
process.exit(1)
|
||||
}
|
||||
|
||||
saveAccount({
|
||||
token: result.token,
|
||||
baseUrl: result.baseUrl || DEFAULT_BASE_URL,
|
||||
userId: result.userId,
|
||||
savedAt: new Date().toISOString(),
|
||||
})
|
||||
|
||||
process.stdout.write(
|
||||
[
|
||||
'Connected successfully!',
|
||||
` User ID: ${result.userId || 'unknown'}`,
|
||||
` Base URL: ${result.baseUrl || DEFAULT_BASE_URL}`,
|
||||
'',
|
||||
'Restart Claude Code with:',
|
||||
' ccb --channels plugin:weixin@builtin',
|
||||
].join('\n') + '\n',
|
||||
)
|
||||
}
|
||||
|
||||
function runAccess(args: string[]): void {
|
||||
if (args[0] !== 'pair' || !args[1]) {
|
||||
printUsage()
|
||||
process.exit(1)
|
||||
}
|
||||
|
||||
const userId = confirmPairing(args[1])
|
||||
if (!userId) {
|
||||
process.stderr.write('Invalid or expired pairing code.\n')
|
||||
process.exit(1)
|
||||
}
|
||||
|
||||
process.stdout.write(`Paired successfully: ${userId}\n`)
|
||||
}
|
||||
|
||||
export async function handleWeixinCli(
|
||||
args: string[],
|
||||
serverDeps?: WeixinServerDeps,
|
||||
version?: string,
|
||||
): Promise<void> {
|
||||
const [subcommand, ...rest] = args
|
||||
|
||||
switch (subcommand) {
|
||||
case 'serve':
|
||||
if (!serverDeps) {
|
||||
process.stderr.write('[weixin] serve handler not available in this context.\n')
|
||||
process.exit(1)
|
||||
}
|
||||
await runWeixinMcpServer(version ?? '0.0.0', serverDeps)
|
||||
return
|
||||
case 'login':
|
||||
await runLogin(rest[0] === 'clear')
|
||||
return
|
||||
case 'access':
|
||||
runAccess(rest)
|
||||
return
|
||||
default:
|
||||
printUsage()
|
||||
}
|
||||
}
|
||||
115
packages/weixin/src/index.ts
Normal file
115
packages/weixin/src/index.ts
Normal file
@@ -0,0 +1,115 @@
|
||||
// @claude-code-best/weixin — WeChat channel integration
|
||||
|
||||
// Types
|
||||
export {
|
||||
MessageType,
|
||||
MessageItemType,
|
||||
MessageState,
|
||||
UploadMediaType,
|
||||
TypingStatus,
|
||||
} from './types.js'
|
||||
export type {
|
||||
BaseInfo,
|
||||
CDNMedia,
|
||||
TextItem,
|
||||
ImageItem,
|
||||
VoiceItem,
|
||||
FileItem,
|
||||
VideoItem,
|
||||
RefMessage,
|
||||
MessageItem,
|
||||
WeixinMessage,
|
||||
GetUpdatesReq,
|
||||
GetUpdatesResp,
|
||||
SendMessageReq,
|
||||
GetUploadUrlReq,
|
||||
GetUploadUrlResp,
|
||||
GetConfigResp,
|
||||
SendTypingReq,
|
||||
SendTypingResp,
|
||||
} from './types.js'
|
||||
|
||||
// API client
|
||||
export {
|
||||
getUpdates,
|
||||
sendMessage,
|
||||
getUploadUrl,
|
||||
getConfig,
|
||||
sendTyping,
|
||||
} from './api.js'
|
||||
|
||||
// Account management
|
||||
export {
|
||||
DEFAULT_BASE_URL,
|
||||
CDN_BASE_URL,
|
||||
getStateDir,
|
||||
loadAccount,
|
||||
saveAccount,
|
||||
clearAccount,
|
||||
} from './accounts.js'
|
||||
export type { AccountData } from './accounts.js'
|
||||
|
||||
// Login
|
||||
export { startLogin, waitForLogin } from './login.js'
|
||||
export type { QRCodeResult, LoginResult } from './login.js'
|
||||
|
||||
// Pairing / access control
|
||||
export {
|
||||
loadAccessConfig,
|
||||
saveAccessConfig,
|
||||
isAllowed,
|
||||
addPendingPairing,
|
||||
confirmPairing,
|
||||
} from './pairing.js'
|
||||
export type { AccessConfig } from './pairing.js'
|
||||
|
||||
// Media encryption / upload
|
||||
export {
|
||||
encryptAesEcb,
|
||||
decryptAesEcb,
|
||||
aesEcbPaddedSize,
|
||||
buildCdnDownloadUrl,
|
||||
buildCdnUploadUrl,
|
||||
parseAesKey,
|
||||
downloadAndDecrypt,
|
||||
uploadFile,
|
||||
guessMediaType,
|
||||
downloadRemoteToTemp,
|
||||
} from './media.js'
|
||||
export type { UploadedFileInfo } from './media.js'
|
||||
|
||||
// Message sending
|
||||
export { markdownToPlainText, sendText, sendMediaFile } from './send.js'
|
||||
|
||||
// Monitor (message polling)
|
||||
export {
|
||||
getContextToken,
|
||||
extractPermissionReply,
|
||||
startPollLoop,
|
||||
} from './monitor.js'
|
||||
export type {
|
||||
ParsedMessage,
|
||||
OnMessageCallback,
|
||||
PermissionResponse,
|
||||
OnPermissionResponseCallback,
|
||||
} from './monitor.js'
|
||||
|
||||
// Permission state
|
||||
export {
|
||||
setActivePermissionChat,
|
||||
getActivePermissionChat,
|
||||
savePendingPermission,
|
||||
consumePendingPermission,
|
||||
} from './permissions.js'
|
||||
export type {
|
||||
ChannelPermissionRequestParams,
|
||||
PendingPermissionRequest,
|
||||
ActivePermissionChat,
|
||||
} from './permissions.js'
|
||||
|
||||
// Server (MCP)
|
||||
export { createWeixinMcpServer, runWeixinMcpServer } from './server.js'
|
||||
export type { WeixinServerDeps } from './server.js'
|
||||
|
||||
// CLI
|
||||
export { handleWeixinCli } from './cli.js'
|
||||
134
packages/weixin/src/login.ts
Normal file
134
packages/weixin/src/login.ts
Normal file
@@ -0,0 +1,134 @@
|
||||
import { toString as qrToString } from 'qrcode'
|
||||
|
||||
export interface QRCodeResult {
|
||||
qrcodeUrl?: string
|
||||
qrcodeId: string
|
||||
message: string
|
||||
}
|
||||
|
||||
export interface LoginResult {
|
||||
connected: boolean
|
||||
token?: string
|
||||
accountId?: string
|
||||
baseUrl?: string
|
||||
userId?: string
|
||||
message: string
|
||||
}
|
||||
|
||||
async function renderQrCodeToTerminal(qrcodeUrl: string): Promise<void> {
|
||||
const output = await qrToString(qrcodeUrl, {
|
||||
type: 'terminal',
|
||||
errorCorrectionLevel: 'L',
|
||||
small: true,
|
||||
})
|
||||
process.stderr.write(`${output}\n`)
|
||||
}
|
||||
|
||||
export async function startLogin(apiBaseUrl: string): Promise<QRCodeResult> {
|
||||
const response = await fetch(`${apiBaseUrl}/ilink/bot/get_bot_qrcode?bot_type=3`)
|
||||
if (!response.ok) {
|
||||
throw new Error(`Failed to get QR code: HTTP ${response.status}`)
|
||||
}
|
||||
|
||||
const data = (await response.json()) as {
|
||||
qrcode?: string
|
||||
qrcode_img_content?: string
|
||||
}
|
||||
|
||||
if (!data.qrcode) {
|
||||
throw new Error('No qrcode in response')
|
||||
}
|
||||
|
||||
const qrcodeUrl = data.qrcode_img_content || ''
|
||||
if (qrcodeUrl) {
|
||||
await renderQrCodeToTerminal(qrcodeUrl)
|
||||
}
|
||||
|
||||
return {
|
||||
qrcodeUrl,
|
||||
qrcodeId: data.qrcode,
|
||||
message: 'Scan the QR code with WeChat to connect.',
|
||||
}
|
||||
}
|
||||
|
||||
export async function waitForLogin(params: {
|
||||
qrcodeId: string
|
||||
apiBaseUrl: string
|
||||
timeoutMs?: number
|
||||
maxRetries?: number
|
||||
}): Promise<LoginResult> {
|
||||
const { qrcodeId, apiBaseUrl, timeoutMs = 480_000, maxRetries = 3 } = params
|
||||
const deadline = Date.now() + timeoutMs
|
||||
let currentQrcodeId = qrcodeId
|
||||
let retryCount = 0
|
||||
|
||||
while (Date.now() < deadline) {
|
||||
try {
|
||||
const controller = new AbortController()
|
||||
const timeout = setTimeout(() => controller.abort(), 60_000)
|
||||
|
||||
const response = await fetch(
|
||||
`${apiBaseUrl}/ilink/bot/get_qrcode_status?qrcode=${encodeURIComponent(currentQrcodeId)}`,
|
||||
{
|
||||
headers: { 'iLink-App-ClientVersion': '1' },
|
||||
signal: controller.signal,
|
||||
},
|
||||
)
|
||||
clearTimeout(timeout)
|
||||
|
||||
if (!response.ok) {
|
||||
throw new Error(`HTTP ${response.status}`)
|
||||
}
|
||||
|
||||
const data = (await response.json()) as {
|
||||
status?: string
|
||||
bot_token?: string
|
||||
ilink_bot_id?: string
|
||||
baseurl?: string
|
||||
ilink_user_id?: string
|
||||
}
|
||||
|
||||
switch (data.status) {
|
||||
case 'confirmed':
|
||||
return {
|
||||
connected: true,
|
||||
token: data.bot_token,
|
||||
accountId: data.ilink_bot_id,
|
||||
baseUrl: data.baseurl,
|
||||
userId: data.ilink_user_id,
|
||||
message: 'Connected to WeChat successfully!',
|
||||
}
|
||||
case 'scaned':
|
||||
process.stderr.write(
|
||||
'QR code scanned, waiting for confirmation...\n',
|
||||
)
|
||||
break
|
||||
case 'expired': {
|
||||
retryCount += 1
|
||||
if (retryCount >= maxRetries) {
|
||||
return {
|
||||
connected: false,
|
||||
message: 'QR code expired after maximum retries.',
|
||||
}
|
||||
}
|
||||
process.stderr.write('QR code expired, refreshing...\n')
|
||||
const refreshed = await startLogin(apiBaseUrl)
|
||||
currentQrcodeId = refreshed.qrcodeId
|
||||
break
|
||||
}
|
||||
case 'wait':
|
||||
default:
|
||||
break
|
||||
}
|
||||
} catch (error) {
|
||||
if (error instanceof Error && error.name === 'AbortError') {
|
||||
continue
|
||||
}
|
||||
throw error
|
||||
}
|
||||
|
||||
await new Promise(resolve => setTimeout(resolve, 1000))
|
||||
}
|
||||
|
||||
return { connected: false, message: 'Login timed out.' }
|
||||
}
|
||||
163
packages/weixin/src/media.ts
Normal file
163
packages/weixin/src/media.ts
Normal file
@@ -0,0 +1,163 @@
|
||||
import {
|
||||
createCipheriv,
|
||||
createDecipheriv,
|
||||
createHash,
|
||||
randomBytes,
|
||||
} from 'node:crypto'
|
||||
import {
|
||||
existsSync,
|
||||
mkdirSync,
|
||||
readFileSync,
|
||||
writeFileSync,
|
||||
} from 'node:fs'
|
||||
import { tmpdir } from 'node:os'
|
||||
import { basename, extname, join } from 'node:path'
|
||||
import { getUploadUrl } from './api.js'
|
||||
import { UploadMediaType } from './types.js'
|
||||
|
||||
export function encryptAesEcb(plaintext: Buffer, key: Buffer): Buffer {
|
||||
const cipher = createCipheriv('aes-128-ecb', key, null)
|
||||
return Buffer.concat([cipher.update(plaintext), cipher.final()])
|
||||
}
|
||||
|
||||
export function decryptAesEcb(ciphertext: Buffer, key: Buffer): Buffer {
|
||||
const decipher = createDecipheriv('aes-128-ecb', key, null)
|
||||
return Buffer.concat([decipher.update(ciphertext), decipher.final()])
|
||||
}
|
||||
|
||||
export function aesEcbPaddedSize(size: number): number {
|
||||
return size + (16 - (size % 16))
|
||||
}
|
||||
|
||||
export function buildCdnDownloadUrl(
|
||||
encryptedQueryParam: string,
|
||||
cdnBaseUrl: string,
|
||||
): string {
|
||||
return `${cdnBaseUrl}/download?encrypted_query_param=${encodeURIComponent(encryptedQueryParam)}`
|
||||
}
|
||||
|
||||
export function buildCdnUploadUrl(
|
||||
cdnBaseUrl: string,
|
||||
uploadParam: string,
|
||||
filekey: string,
|
||||
): string {
|
||||
return `${cdnBaseUrl}/upload?encrypted_query_param=${encodeURIComponent(uploadParam)}&filekey=${encodeURIComponent(filekey)}`
|
||||
}
|
||||
|
||||
export function parseAesKey(aesKeyBase64: string): Buffer {
|
||||
const decoded = Buffer.from(aesKeyBase64, 'base64')
|
||||
if (decoded.length === 16) {
|
||||
return decoded
|
||||
}
|
||||
if (decoded.length === 32 && /^[0-9a-fA-F]{32}$/.test(decoded.toString('ascii'))) {
|
||||
return Buffer.from(decoded.toString('ascii'), 'hex')
|
||||
}
|
||||
throw new Error(
|
||||
`Invalid aes_key: expected 16 raw bytes or 32 hex chars, got ${decoded.length} bytes`,
|
||||
)
|
||||
}
|
||||
|
||||
export async function downloadAndDecrypt(params: {
|
||||
encryptQueryParam: string
|
||||
aesKey: string
|
||||
cdnBaseUrl: string
|
||||
}): Promise<Buffer> {
|
||||
const url = buildCdnDownloadUrl(params.encryptQueryParam, params.cdnBaseUrl)
|
||||
const response = await fetch(url)
|
||||
if (!response.ok) {
|
||||
throw new Error(`CDN download failed: HTTP ${response.status}`)
|
||||
}
|
||||
const ciphertext = Buffer.from(await response.arrayBuffer())
|
||||
return decryptAesEcb(ciphertext, parseAesKey(params.aesKey))
|
||||
}
|
||||
|
||||
export interface UploadedFileInfo {
|
||||
encryptQueryParam: string
|
||||
aesKey: string
|
||||
fileSize: number
|
||||
rawSize: number
|
||||
fileName: string
|
||||
}
|
||||
|
||||
export async function uploadFile(params: {
|
||||
filePath: string
|
||||
toUserId: string
|
||||
mediaType: number
|
||||
apiBaseUrl: string
|
||||
token: string
|
||||
cdnBaseUrl: string
|
||||
}): Promise<UploadedFileInfo> {
|
||||
const plaintext = readFileSync(params.filePath)
|
||||
const rawSize = plaintext.length
|
||||
const rawMd5 = createHash('md5').update(plaintext).digest('hex')
|
||||
const aesKey = randomBytes(16)
|
||||
const filekey = randomBytes(16).toString('hex')
|
||||
const ciphertext = encryptAesEcb(plaintext, aesKey)
|
||||
const fileSize = ciphertext.length
|
||||
|
||||
const uploadResp = await getUploadUrl(params.apiBaseUrl, params.token, {
|
||||
filekey,
|
||||
media_type: params.mediaType,
|
||||
to_user_id: params.toUserId,
|
||||
rawsize: rawSize,
|
||||
rawfilemd5: rawMd5,
|
||||
filesize: fileSize,
|
||||
no_need_thumb: true,
|
||||
aeskey: aesKey.toString('hex'),
|
||||
})
|
||||
|
||||
if (!uploadResp.upload_param) {
|
||||
throw new Error('No upload_param in response')
|
||||
}
|
||||
|
||||
const uploadUrl = buildCdnUploadUrl(
|
||||
params.cdnBaseUrl,
|
||||
uploadResp.upload_param,
|
||||
filekey,
|
||||
)
|
||||
const uploadResult = await fetch(uploadUrl, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/octet-stream' },
|
||||
body: new Uint8Array(ciphertext),
|
||||
})
|
||||
|
||||
if (!uploadResult.ok) {
|
||||
throw new Error(`CDN upload failed: HTTP ${uploadResult.status}`)
|
||||
}
|
||||
|
||||
return {
|
||||
encryptQueryParam: uploadResult.headers.get('x-encrypted-param') || '',
|
||||
aesKey: Buffer.from(aesKey.toString('hex')).toString('base64'),
|
||||
fileSize,
|
||||
rawSize,
|
||||
fileName: basename(params.filePath),
|
||||
}
|
||||
}
|
||||
|
||||
export function guessMediaType(filePath: string): number {
|
||||
const ext = extname(filePath).toLowerCase()
|
||||
const imageExts = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.webp', '.heic']
|
||||
const videoExts = ['.mp4', '.mov', '.avi', '.mkv', '.webm']
|
||||
|
||||
if (imageExts.includes(ext)) return UploadMediaType.IMAGE
|
||||
if (videoExts.includes(ext)) return UploadMediaType.VIDEO
|
||||
return UploadMediaType.FILE
|
||||
}
|
||||
|
||||
export async function downloadRemoteToTemp(
|
||||
url: string,
|
||||
destDir?: string,
|
||||
): Promise<string> {
|
||||
const dir = destDir || join(tmpdir(), 'weixin-downloads')
|
||||
if (!existsSync(dir)) mkdirSync(dir, { recursive: true })
|
||||
|
||||
const response = await fetch(url)
|
||||
if (!response.ok) throw new Error(`Download failed: HTTP ${response.status}`)
|
||||
|
||||
const buffer = Buffer.from(await response.arrayBuffer())
|
||||
const urlPath = new URL(url).pathname
|
||||
const name = basename(urlPath) || `file_${Date.now()}`
|
||||
const dest = join(dir, name)
|
||||
writeFileSync(dest, buffer)
|
||||
return dest
|
||||
}
|
||||
303
packages/weixin/src/monitor.ts
Normal file
303
packages/weixin/src/monitor.ts
Normal file
@@ -0,0 +1,303 @@
|
||||
import {
|
||||
existsSync,
|
||||
mkdirSync,
|
||||
readFileSync,
|
||||
writeFileSync,
|
||||
} from 'node:fs'
|
||||
import { tmpdir } from 'node:os'
|
||||
import { basename, join } from 'node:path'
|
||||
// Matches the canonical definition in src/services/mcp/channelPermissions.ts
|
||||
const PERMISSION_REPLY_RE = /^\s*(y|yes|n|no)\s+([a-km-z]{5})\s*$/i
|
||||
import { getUpdates } from './api.js'
|
||||
import { getStateDir } from './accounts.js'
|
||||
import { downloadAndDecrypt } from './media.js'
|
||||
import { addPendingPairing, isAllowed } from './pairing.js'
|
||||
import { consumePendingPermission, setActivePermissionChat } from './permissions.js'
|
||||
import { sendText } from './send.js'
|
||||
import { MessageItemType, MessageType, type MessageItem, type WeixinMessage } from './types.js'
|
||||
|
||||
const contextTokens = new Map<string, string>()
|
||||
|
||||
export function getContextToken(userId: string): string | undefined {
|
||||
return contextTokens.get(userId)
|
||||
}
|
||||
|
||||
function cursorPath(): string {
|
||||
return join(getStateDir(), 'cursor.txt')
|
||||
}
|
||||
|
||||
function loadCursor(): string {
|
||||
const path = cursorPath()
|
||||
if (existsSync(path)) return readFileSync(path, 'utf-8').trim()
|
||||
return ''
|
||||
}
|
||||
|
||||
function saveCursor(cursor: string): void {
|
||||
writeFileSync(cursorPath(), cursor, 'utf-8')
|
||||
}
|
||||
|
||||
async function downloadMedia(
|
||||
item: MessageItem,
|
||||
cdnBaseUrl: string,
|
||||
): Promise<{ path: string; type: string } | null> {
|
||||
let encryptQueryParam: string | undefined
|
||||
let aesKey: string | undefined
|
||||
let ext = ''
|
||||
let mediaType = ''
|
||||
|
||||
switch (item.type) {
|
||||
case MessageItemType.IMAGE:
|
||||
encryptQueryParam = item.image_item?.media?.encrypt_query_param
|
||||
aesKey = item.image_item?.aeskey
|
||||
? Buffer.from(item.image_item.aeskey, 'hex').toString('base64')
|
||||
: item.image_item?.media?.aes_key
|
||||
ext = '.jpg'
|
||||
mediaType = 'image'
|
||||
break
|
||||
case MessageItemType.VOICE:
|
||||
encryptQueryParam = item.voice_item?.media?.encrypt_query_param
|
||||
aesKey = item.voice_item?.media?.aes_key
|
||||
ext = '.silk'
|
||||
mediaType = 'voice'
|
||||
break
|
||||
case MessageItemType.FILE:
|
||||
encryptQueryParam = item.file_item?.media?.encrypt_query_param
|
||||
aesKey = item.file_item?.media?.aes_key
|
||||
ext = item.file_item?.file_name
|
||||
? `.${item.file_item.file_name.split('.').pop()}`
|
||||
: ''
|
||||
mediaType = 'file'
|
||||
break
|
||||
case MessageItemType.VIDEO:
|
||||
encryptQueryParam = item.video_item?.media?.encrypt_query_param
|
||||
aesKey = item.video_item?.media?.aes_key
|
||||
ext = '.mp4'
|
||||
mediaType = 'video'
|
||||
break
|
||||
default:
|
||||
return null
|
||||
}
|
||||
|
||||
if (!encryptQueryParam || !aesKey) return null
|
||||
|
||||
try {
|
||||
const data = await downloadAndDecrypt({
|
||||
encryptQueryParam,
|
||||
aesKey,
|
||||
cdnBaseUrl,
|
||||
})
|
||||
const dir = join(tmpdir(), 'weixin-media')
|
||||
if (!existsSync(dir)) mkdirSync(dir, { recursive: true })
|
||||
const rawFileName = item.file_item?.file_name || `${Date.now()}${ext}`
|
||||
const fileName = basename(rawFileName)
|
||||
const filePath = join(dir, fileName)
|
||||
writeFileSync(filePath, data)
|
||||
return { path: filePath, type: mediaType }
|
||||
} catch (error) {
|
||||
process.stderr.write(`[weixin] Failed to download media: ${error}\n`)
|
||||
return null
|
||||
}
|
||||
}
|
||||
|
||||
export interface ParsedMessage {
|
||||
fromUserId: string
|
||||
messageId: string
|
||||
text: string
|
||||
attachmentPath?: string
|
||||
attachmentType?: string
|
||||
}
|
||||
|
||||
export type OnMessageCallback = (msg: ParsedMessage) => Promise<void>
|
||||
|
||||
export type PermissionResponse = {
|
||||
requestId: string
|
||||
behavior: 'allow' | 'deny'
|
||||
fromUserId: string
|
||||
}
|
||||
|
||||
export type OnPermissionResponseCallback = (
|
||||
response: PermissionResponse,
|
||||
) => Promise<void>
|
||||
|
||||
export function extractPermissionReply(
|
||||
text: string,
|
||||
): { requestId: string; behavior: 'allow' | 'deny' } | null {
|
||||
const match = text.match(PERMISSION_REPLY_RE)
|
||||
if (!match) return null
|
||||
const behavior =
|
||||
match[1]?.toLowerCase().startsWith('y') ? 'allow' : 'deny'
|
||||
const requestId = match[2]?.toLowerCase()
|
||||
if (!requestId) return null
|
||||
return { requestId, behavior }
|
||||
}
|
||||
|
||||
export async function startPollLoop(params: {
|
||||
baseUrl: string
|
||||
cdnBaseUrl: string
|
||||
token: string
|
||||
onMessage: OnMessageCallback
|
||||
onPermissionResponse?: OnPermissionResponseCallback
|
||||
abortSignal: AbortSignal
|
||||
}): Promise<void> {
|
||||
const {
|
||||
baseUrl,
|
||||
cdnBaseUrl,
|
||||
token,
|
||||
onMessage,
|
||||
onPermissionResponse,
|
||||
abortSignal,
|
||||
} = params
|
||||
let cursor = loadCursor()
|
||||
let consecutiveErrors = 0
|
||||
|
||||
process.stderr.write('[weixin] Starting message poll loop...\n')
|
||||
|
||||
while (!abortSignal.aborted) {
|
||||
try {
|
||||
const response = await getUpdates(baseUrl, token, cursor, abortSignal)
|
||||
|
||||
if (response.errcode === -14) {
|
||||
process.stderr.write(
|
||||
'[weixin] Session expired (errcode -14). Pausing for 30s...\n',
|
||||
)
|
||||
await new Promise(resolve => setTimeout(resolve, 30_000))
|
||||
continue
|
||||
}
|
||||
|
||||
if (response.ret !== 0 && response.ret !== undefined) {
|
||||
throw new Error(
|
||||
`getUpdates error: ret=${response.ret} errcode=${response.errcode} ${response.errmsg}`,
|
||||
)
|
||||
}
|
||||
|
||||
consecutiveErrors = 0
|
||||
|
||||
if (response.get_updates_buf) {
|
||||
cursor = response.get_updates_buf
|
||||
saveCursor(cursor)
|
||||
}
|
||||
|
||||
if (response.msgs && response.msgs.length > 0) {
|
||||
for (const msg of response.msgs) {
|
||||
await processMessage(msg, {
|
||||
baseUrl,
|
||||
cdnBaseUrl,
|
||||
token,
|
||||
onMessage,
|
||||
onPermissionResponse,
|
||||
})
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
if (abortSignal.aborted) break
|
||||
|
||||
consecutiveErrors += 1
|
||||
process.stderr.write(
|
||||
`[weixin] Poll error (${consecutiveErrors}): ${error instanceof Error ? error.message : String(error)}\n`,
|
||||
)
|
||||
|
||||
if (consecutiveErrors >= 3) {
|
||||
process.stderr.write(
|
||||
'[weixin] Too many consecutive errors, backing off 30s...\n',
|
||||
)
|
||||
await new Promise(resolve => setTimeout(resolve, 30_000))
|
||||
consecutiveErrors = 0
|
||||
} else {
|
||||
await new Promise(resolve => setTimeout(resolve, 2000))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
process.stderr.write('[weixin] Poll loop stopped.\n')
|
||||
}
|
||||
|
||||
async function processMessage(
|
||||
msg: WeixinMessage,
|
||||
ctx: {
|
||||
baseUrl: string
|
||||
cdnBaseUrl: string
|
||||
token: string
|
||||
onMessage: OnMessageCallback
|
||||
onPermissionResponse?: OnPermissionResponseCallback
|
||||
},
|
||||
): Promise<void> {
|
||||
if (msg.message_type !== MessageType.USER) return
|
||||
const fromUserId = msg.from_user_id
|
||||
if (!fromUserId) return
|
||||
|
||||
if (msg.context_token) {
|
||||
contextTokens.set(fromUserId, msg.context_token)
|
||||
}
|
||||
|
||||
if (!isAllowed(fromUserId)) {
|
||||
const code = addPendingPairing(fromUserId)
|
||||
try {
|
||||
await sendText({
|
||||
to: fromUserId,
|
||||
text: `Your pairing code is: ${code}\n\nAsk the operator to confirm:\nccb weixin access pair ${code}`,
|
||||
baseUrl: ctx.baseUrl,
|
||||
token: ctx.token,
|
||||
contextToken: msg.context_token || '',
|
||||
})
|
||||
} catch (error) {
|
||||
process.stderr.write(`[weixin] Failed to send pairing code: ${error}\n`)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
setActivePermissionChat(fromUserId, msg.context_token)
|
||||
|
||||
let textContent = ''
|
||||
let mediaPath: string | undefined
|
||||
let mediaType: string | undefined
|
||||
|
||||
if (msg.item_list) {
|
||||
for (const item of msg.item_list) {
|
||||
if (item.type === MessageItemType.TEXT && item.text_item?.text) {
|
||||
textContent += `${textContent ? '\n' : ''}${item.text_item.text}`
|
||||
} else if (
|
||||
item.type === MessageItemType.IMAGE ||
|
||||
item.type === MessageItemType.VOICE ||
|
||||
item.type === MessageItemType.FILE ||
|
||||
item.type === MessageItemType.VIDEO
|
||||
) {
|
||||
const downloaded = await downloadMedia(item, ctx.cdnBaseUrl)
|
||||
if (downloaded) {
|
||||
mediaPath = downloaded.path
|
||||
mediaType = downloaded.type
|
||||
}
|
||||
if (item.type === MessageItemType.VOICE && item.voice_item?.text) {
|
||||
textContent += `${textContent ? '\n' : ''}[Voice transcription]: ${item.voice_item.text}`
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!textContent && !mediaPath) return
|
||||
|
||||
if (textContent && ctx.onPermissionResponse) {
|
||||
const permissionReply = extractPermissionReply(textContent)
|
||||
if (permissionReply) {
|
||||
const pending = consumePendingPermission(
|
||||
permissionReply.requestId,
|
||||
fromUserId,
|
||||
)
|
||||
if (pending) {
|
||||
await ctx.onPermissionResponse({
|
||||
requestId: pending.request_id,
|
||||
behavior: permissionReply.behavior,
|
||||
fromUserId,
|
||||
})
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
await ctx.onMessage({
|
||||
fromUserId,
|
||||
messageId: String(msg.message_id || ''),
|
||||
text: textContent || '(media attachment)',
|
||||
attachmentPath: mediaPath,
|
||||
attachmentType: mediaType,
|
||||
})
|
||||
}
|
||||
101
packages/weixin/src/pairing.ts
Normal file
101
packages/weixin/src/pairing.ts
Normal file
@@ -0,0 +1,101 @@
|
||||
import { existsSync, readFileSync, writeFileSync } from 'node:fs'
|
||||
import { join } from 'node:path'
|
||||
import { getStateDir } from './accounts.js'
|
||||
|
||||
export interface AccessConfig {
|
||||
policy: 'pairing' | 'allowlist' | 'disabled'
|
||||
allowFrom: string[]
|
||||
}
|
||||
|
||||
interface PendingEntry {
|
||||
userId: string
|
||||
expiresAt: number
|
||||
}
|
||||
|
||||
function configPath(): string {
|
||||
return join(getStateDir(), 'access.json')
|
||||
}
|
||||
|
||||
function pendingPath(): string {
|
||||
return join(getStateDir(), 'pending-pairings.json')
|
||||
}
|
||||
|
||||
function loadPending(): Record<string, PendingEntry> {
|
||||
const path = pendingPath()
|
||||
if (!existsSync(path)) return {}
|
||||
try {
|
||||
return JSON.parse(readFileSync(path, 'utf-8')) as Record<string, PendingEntry>
|
||||
} catch {
|
||||
return {}
|
||||
}
|
||||
}
|
||||
|
||||
function savePending(data: Record<string, PendingEntry>): void {
|
||||
writeFileSync(pendingPath(), JSON.stringify(data, null, 2), 'utf-8')
|
||||
}
|
||||
|
||||
export function loadAccessConfig(): AccessConfig {
|
||||
const path = configPath()
|
||||
if (!existsSync(path)) {
|
||||
return { policy: 'pairing', allowFrom: [] }
|
||||
}
|
||||
try {
|
||||
return JSON.parse(readFileSync(path, 'utf-8')) as AccessConfig
|
||||
} catch {
|
||||
return { policy: 'pairing', allowFrom: [] }
|
||||
}
|
||||
}
|
||||
|
||||
export function saveAccessConfig(config: AccessConfig): void {
|
||||
writeFileSync(configPath(), JSON.stringify(config, null, 2), 'utf-8')
|
||||
}
|
||||
|
||||
export function isAllowed(userId: string): boolean {
|
||||
const config = loadAccessConfig()
|
||||
if (config.policy === 'disabled') return true
|
||||
return config.allowFrom.includes(userId)
|
||||
}
|
||||
|
||||
export function addPendingPairing(userId: string): string {
|
||||
const pending = loadPending()
|
||||
const now = Date.now()
|
||||
|
||||
for (const code of Object.keys(pending)) {
|
||||
if (pending[code]!.expiresAt < now) {
|
||||
delete pending[code]
|
||||
}
|
||||
}
|
||||
|
||||
for (const [code, entry] of Object.entries(pending)) {
|
||||
if (entry.userId === userId) {
|
||||
savePending(pending)
|
||||
return code
|
||||
}
|
||||
}
|
||||
|
||||
const code = String(Math.floor(100000 + Math.random() * 900000))
|
||||
pending[code] = { userId, expiresAt: now + 10 * 60 * 1000 }
|
||||
savePending(pending)
|
||||
return code
|
||||
}
|
||||
|
||||
export function confirmPairing(code: string): string | null {
|
||||
const pending = loadPending()
|
||||
const entry = pending[code]
|
||||
if (!entry || entry.expiresAt < Date.now()) {
|
||||
delete pending[code]
|
||||
savePending(pending)
|
||||
return null
|
||||
}
|
||||
|
||||
delete pending[code]
|
||||
savePending(pending)
|
||||
|
||||
const config = loadAccessConfig()
|
||||
if (!config.allowFrom.includes(entry.userId)) {
|
||||
config.allowFrom.push(entry.userId)
|
||||
saveAccessConfig(config)
|
||||
}
|
||||
|
||||
return entry.userId
|
||||
}
|
||||
83
packages/weixin/src/permissions.ts
Normal file
83
packages/weixin/src/permissions.ts
Normal file
@@ -0,0 +1,83 @@
|
||||
/** Mirrors ChannelPermissionRequestParams from src/services/mcp/channelNotification.ts */
|
||||
export interface ChannelPermissionRequestParams {
|
||||
request_id: string
|
||||
tool_name: string
|
||||
description: string
|
||||
input_preview: string
|
||||
channel_context?: {
|
||||
source_server?: string
|
||||
chat_id?: string
|
||||
}
|
||||
}
|
||||
|
||||
export type PendingPermissionRequest = ChannelPermissionRequestParams & {
|
||||
chatId: string
|
||||
contextToken?: string
|
||||
createdAt: number
|
||||
expiresAt: number
|
||||
}
|
||||
|
||||
export type ActivePermissionChat = {
|
||||
chatId: string
|
||||
contextToken?: string
|
||||
updatedAt: number
|
||||
}
|
||||
|
||||
const PENDING_PERMISSION_TTL_MS = 15 * 60 * 1000
|
||||
|
||||
const pendingPermissions = new Map<string, PendingPermissionRequest>()
|
||||
let activePermissionChat: ActivePermissionChat | null = null
|
||||
|
||||
function pruneExpiredPendingPermissions(now = Date.now()): void {
|
||||
for (const [requestId, entry] of pendingPermissions.entries()) {
|
||||
if (entry.expiresAt <= now) {
|
||||
pendingPermissions.delete(requestId)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
export function setActivePermissionChat(
|
||||
chatId: string,
|
||||
contextToken?: string,
|
||||
): void {
|
||||
activePermissionChat = { chatId, contextToken, updatedAt: Date.now() }
|
||||
}
|
||||
|
||||
export function getActivePermissionChat(): ActivePermissionChat | null {
|
||||
return activePermissionChat
|
||||
}
|
||||
|
||||
export function savePendingPermission(
|
||||
request: ChannelPermissionRequestParams,
|
||||
chatId: string,
|
||||
contextToken?: string,
|
||||
): PendingPermissionRequest {
|
||||
pruneExpiredPendingPermissions()
|
||||
const entry: PendingPermissionRequest = {
|
||||
...request,
|
||||
chatId,
|
||||
contextToken,
|
||||
createdAt: Date.now(),
|
||||
expiresAt: Date.now() + PENDING_PERMISSION_TTL_MS,
|
||||
}
|
||||
pendingPermissions.set(request.request_id.toLowerCase(), entry)
|
||||
return entry
|
||||
}
|
||||
|
||||
export function consumePendingPermission(
|
||||
requestId: string,
|
||||
fromUserId: string,
|
||||
): PendingPermissionRequest | null {
|
||||
pruneExpiredPendingPermissions()
|
||||
const key = requestId.toLowerCase()
|
||||
const entry = pendingPermissions.get(key)
|
||||
if (!entry) return null
|
||||
if (entry.chatId !== fromUserId) return null
|
||||
pendingPermissions.delete(key)
|
||||
return entry
|
||||
}
|
||||
|
||||
export function clearPermissionStateForTests(): void {
|
||||
pendingPermissions.clear()
|
||||
activePermissionChat = null
|
||||
}
|
||||
180
packages/weixin/src/send.ts
Normal file
180
packages/weixin/src/send.ts
Normal file
@@ -0,0 +1,180 @@
|
||||
import { randomUUID } from 'node:crypto'
|
||||
import type { CDNMedia, MessageItem } from './types.js'
|
||||
import { sendMessage } from './api.js'
|
||||
import { guessMediaType, uploadFile } from './media.js'
|
||||
import { MessageItemType, MessageState, MessageType } from './types.js'
|
||||
|
||||
function stripCodeBlocks(text: string): string {
|
||||
// Non-regex approach to avoid ReDoS on inputs with many ``` sequences.
|
||||
let result = ''
|
||||
let i = 0
|
||||
while (i < text.length) {
|
||||
if (text.startsWith('```', i)) {
|
||||
// Skip the opening fence (including optional language tag on same line)
|
||||
let j = i + 3
|
||||
// skip to end of first line (the fence line itself)
|
||||
while (j < text.length && text[j] !== '\n') j++
|
||||
if (j < text.length) j++ // skip the \n
|
||||
// Collect content until closing ```
|
||||
const contentStart = j
|
||||
while (j < text.length) {
|
||||
if (text.startsWith('```', j)) {
|
||||
result += text.slice(contentStart, j)
|
||||
// skip closing fence and its trailing newline
|
||||
j += 3
|
||||
while (j < text.length && text[j] !== '\n') j++
|
||||
if (j < text.length) j++ // skip \n
|
||||
break
|
||||
}
|
||||
j++
|
||||
}
|
||||
// If no closing fence found, include rest as-is
|
||||
if (j >= text.length && !text.startsWith('```', j - 3)) {
|
||||
result += text.slice(i)
|
||||
}
|
||||
i = j
|
||||
} else {
|
||||
result += text[i]
|
||||
i++
|
||||
}
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
export function markdownToPlainText(text: string): string {
|
||||
return stripCodeBlocks(text)
|
||||
.replace(/`([^`]+)`/g, '$1')
|
||||
.replace(/\*\*\*(.+?)\*\*\*/g, '$1')
|
||||
.replace(/\*\*(.+?)\*\*/g, '$1')
|
||||
.replace(/\*(.+?)\*/g, '$1')
|
||||
.replace(/___(.+?)___/g, '$1')
|
||||
.replace(/__(.+?)__/g, '$1')
|
||||
.replace(/_(.+?)_/g, '$1')
|
||||
.replace(/~~(.+?)~~/g, '$1')
|
||||
.replace(/^#{1,6}\s+/gm, '')
|
||||
.replace(/\[([^\]]+)\]\(([^)]+)\)/g, '$1 ($2)')
|
||||
.replace(/!\[([^\]]*)\]\([^)]+\)/g, '[$1]')
|
||||
.replace(/^>\s+/gm, '')
|
||||
.replace(/^[-*_]{3,}$/gm, '---')
|
||||
.replace(/^[\s]*[-*+]\s+/gm, '- ')
|
||||
.replace(/^[\s]*(\d+)\.\s+/gm, '$1. ')
|
||||
.replace(/\n{3,}/g, '\n\n')
|
||||
.trim()
|
||||
}
|
||||
|
||||
export async function sendText(params: {
|
||||
to: string
|
||||
text: string
|
||||
baseUrl: string
|
||||
token: string
|
||||
contextToken: string
|
||||
}): Promise<{ messageId: string }> {
|
||||
const clientId = randomUUID()
|
||||
await sendMessage(params.baseUrl, params.token, {
|
||||
to_user_id: params.to,
|
||||
from_user_id: '',
|
||||
client_id: clientId,
|
||||
message_type: MessageType.BOT,
|
||||
message_state: MessageState.FINISH,
|
||||
context_token: params.contextToken,
|
||||
item_list: [
|
||||
{
|
||||
type: MessageItemType.TEXT,
|
||||
text_item: { text: markdownToPlainText(params.text) },
|
||||
},
|
||||
],
|
||||
})
|
||||
|
||||
return { messageId: clientId }
|
||||
}
|
||||
|
||||
async function sendItems(params: {
|
||||
items: MessageItem[]
|
||||
to: string
|
||||
baseUrl: string
|
||||
token: string
|
||||
contextToken: string
|
||||
}): Promise<string> {
|
||||
let lastClientId = ''
|
||||
for (const item of params.items) {
|
||||
lastClientId = randomUUID()
|
||||
await sendMessage(params.baseUrl, params.token, {
|
||||
to_user_id: params.to,
|
||||
from_user_id: '',
|
||||
client_id: lastClientId,
|
||||
message_type: MessageType.BOT,
|
||||
message_state: MessageState.FINISH,
|
||||
context_token: params.contextToken,
|
||||
item_list: [item],
|
||||
})
|
||||
}
|
||||
return lastClientId
|
||||
}
|
||||
|
||||
export async function sendMediaFile(params: {
|
||||
filePath: string
|
||||
to: string
|
||||
text: string
|
||||
baseUrl: string
|
||||
token: string
|
||||
contextToken: string
|
||||
cdnBaseUrl: string
|
||||
}): Promise<{ messageId: string }> {
|
||||
const mediaType = guessMediaType(params.filePath)
|
||||
const uploaded = await uploadFile({
|
||||
filePath: params.filePath,
|
||||
toUserId: params.to,
|
||||
mediaType,
|
||||
apiBaseUrl: params.baseUrl,
|
||||
token: params.token,
|
||||
cdnBaseUrl: params.cdnBaseUrl,
|
||||
})
|
||||
|
||||
const cdnMedia: CDNMedia = {
|
||||
encrypt_query_param: uploaded.encryptQueryParam,
|
||||
aes_key: uploaded.aesKey,
|
||||
encrypt_type: 1,
|
||||
}
|
||||
|
||||
const items: MessageItem[] = []
|
||||
if (params.text) {
|
||||
items.push({
|
||||
type: MessageItemType.TEXT,
|
||||
text_item: { text: markdownToPlainText(params.text) },
|
||||
})
|
||||
}
|
||||
|
||||
switch (mediaType) {
|
||||
case 1:
|
||||
items.push({
|
||||
type: MessageItemType.IMAGE,
|
||||
image_item: { media: cdnMedia, mid_size: uploaded.fileSize },
|
||||
})
|
||||
break
|
||||
case 2:
|
||||
items.push({
|
||||
type: MessageItemType.VIDEO,
|
||||
video_item: { media: cdnMedia, video_size: uploaded.fileSize },
|
||||
})
|
||||
break
|
||||
default:
|
||||
items.push({
|
||||
type: MessageItemType.FILE,
|
||||
file_item: {
|
||||
media: cdnMedia,
|
||||
file_name: uploaded.fileName,
|
||||
len: String(uploaded.rawSize),
|
||||
},
|
||||
})
|
||||
break
|
||||
}
|
||||
|
||||
const messageId = await sendItems({
|
||||
items,
|
||||
to: params.to,
|
||||
baseUrl: params.baseUrl,
|
||||
token: params.token,
|
||||
contextToken: params.contextToken,
|
||||
})
|
||||
return { messageId }
|
||||
}
|
||||
353
packages/weixin/src/server.ts
Normal file
353
packages/weixin/src/server.ts
Normal file
@@ -0,0 +1,353 @@
|
||||
import { existsSync } from 'node:fs'
|
||||
import { Server } from '@modelcontextprotocol/sdk/server/index.js'
|
||||
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
|
||||
import {
|
||||
CallToolRequestSchema,
|
||||
ListToolsRequestSchema,
|
||||
} from '@modelcontextprotocol/sdk/types.js'
|
||||
import {
|
||||
CDN_BASE_URL,
|
||||
DEFAULT_BASE_URL,
|
||||
loadAccount,
|
||||
getConfig,
|
||||
sendTyping,
|
||||
getContextToken,
|
||||
startPollLoop,
|
||||
getActivePermissionChat,
|
||||
savePendingPermission,
|
||||
sendMediaFile,
|
||||
sendText,
|
||||
TypingStatus,
|
||||
} from './index.js'
|
||||
import type { ParsedMessage } from './monitor.js'
|
||||
import type { ChannelPermissionRequestParams } from './permissions.js'
|
||||
|
||||
export interface WeixinServerDeps {
|
||||
enableConfigs(): void
|
||||
initializeAnalyticsSink(): void
|
||||
shutdownDatadog(): Promise<void>
|
||||
shutdown1PEventLogging(): Promise<void>
|
||||
logForDebugging(message: string): void
|
||||
registerPermissionHandler(
|
||||
server: Server,
|
||||
handler: (request: ChannelPermissionRequestParams) => Promise<void>,
|
||||
): void
|
||||
}
|
||||
|
||||
function formatPermissionRequestMessage(
|
||||
request: ChannelPermissionRequestParams,
|
||||
): string {
|
||||
return [
|
||||
'Claude Code needs your approval.',
|
||||
'',
|
||||
`Tool: ${request.tool_name}`,
|
||||
`Reason: ${request.description}`,
|
||||
`Input: ${request.input_preview}`,
|
||||
'',
|
||||
`Reply with: yes ${request.request_id}`,
|
||||
`Or deny with: no ${request.request_id}`,
|
||||
].join('\n')
|
||||
}
|
||||
|
||||
export function createWeixinMcpServer(version: string): Server {
|
||||
const server = new Server(
|
||||
{ name: 'weixin', version },
|
||||
{
|
||||
capabilities: {
|
||||
experimental: {
|
||||
'claude/channel': {},
|
||||
'claude/channel/permission': {},
|
||||
},
|
||||
tools: {},
|
||||
},
|
||||
instructions:
|
||||
'Messages from WeChat arrive as <channel source="plugin:weixin:weixin" chat_id="..." sender_id="...">. Reply using the reply tool with the chat_id from the channel tag. Use absolute paths for file attachments.',
|
||||
},
|
||||
)
|
||||
|
||||
server.setRequestHandler(ListToolsRequestSchema, async () => ({
|
||||
tools: [
|
||||
{
|
||||
name: 'reply',
|
||||
description:
|
||||
'Reply to a WeChat message. Pass the chat_id from the channel tag.',
|
||||
inputSchema: {
|
||||
type: 'object' as const,
|
||||
properties: {
|
||||
chat_id: {
|
||||
type: 'string',
|
||||
description: 'The chat_id from the channel notification',
|
||||
},
|
||||
text: { type: 'string', description: 'The reply text' },
|
||||
files: {
|
||||
type: 'array',
|
||||
items: { type: 'string' },
|
||||
description: 'Optional absolute file paths to attach',
|
||||
},
|
||||
},
|
||||
required: ['chat_id', 'text'],
|
||||
},
|
||||
},
|
||||
{
|
||||
name: 'send_typing',
|
||||
description: 'Send a typing indicator to a WeChat user.',
|
||||
inputSchema: {
|
||||
type: 'object' as const,
|
||||
properties: {
|
||||
chat_id: { type: 'string', description: 'The chat_id (user ID)' },
|
||||
},
|
||||
required: ['chat_id'],
|
||||
},
|
||||
},
|
||||
],
|
||||
}))
|
||||
|
||||
server.setRequestHandler(CallToolRequestSchema, async request => {
|
||||
const { name, arguments: args } = request.params
|
||||
const account = loadAccount()
|
||||
if (!account) {
|
||||
return {
|
||||
content: [
|
||||
{
|
||||
type: 'text',
|
||||
text: 'WeChat not connected. Run `ccb weixin login` first.',
|
||||
},
|
||||
],
|
||||
isError: true,
|
||||
}
|
||||
}
|
||||
|
||||
const baseUrl = account.baseUrl || DEFAULT_BASE_URL
|
||||
const cdnBaseUrl = CDN_BASE_URL
|
||||
|
||||
switch (name) {
|
||||
case 'reply': {
|
||||
const chatId = typeof args?.chat_id === 'string' ? args.chat_id : ''
|
||||
const text = typeof args?.text === 'string' ? args.text : ''
|
||||
const files = Array.isArray(args?.files)
|
||||
? args.files.filter((value): value is string => typeof value === 'string')
|
||||
: undefined
|
||||
|
||||
if (!chatId || !text) {
|
||||
return {
|
||||
content: [
|
||||
{ type: 'text', text: 'Missing chat_id or text parameter.' },
|
||||
],
|
||||
isError: true,
|
||||
}
|
||||
}
|
||||
|
||||
const contextToken = getContextToken(chatId) || ''
|
||||
|
||||
try {
|
||||
if (files && files.length > 0) {
|
||||
for (const [index, filePath] of files.entries()) {
|
||||
if (!existsSync(filePath)) {
|
||||
return {
|
||||
content: [
|
||||
{ type: 'text', text: `File not found: ${filePath}` },
|
||||
],
|
||||
isError: true,
|
||||
}
|
||||
}
|
||||
await sendMediaFile({
|
||||
filePath,
|
||||
to: chatId,
|
||||
text: index === 0 ? text : '',
|
||||
baseUrl,
|
||||
token: account.token,
|
||||
contextToken,
|
||||
cdnBaseUrl,
|
||||
})
|
||||
}
|
||||
|
||||
return {
|
||||
content: [{ type: 'text', text: 'Message sent with attachments.' }],
|
||||
}
|
||||
}
|
||||
|
||||
await sendText({
|
||||
to: chatId,
|
||||
text,
|
||||
baseUrl,
|
||||
token: account.token,
|
||||
contextToken,
|
||||
})
|
||||
return { content: [{ type: 'text', text: 'Message sent.' }] }
|
||||
} catch (error) {
|
||||
return {
|
||||
content: [{ type: 'text', text: `Failed to send: ${error}` }],
|
||||
isError: true,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
case 'send_typing': {
|
||||
const chatId = typeof args?.chat_id === 'string' ? args.chat_id : ''
|
||||
if (!chatId) {
|
||||
return {
|
||||
content: [{ type: 'text', text: 'Missing chat_id parameter.' }],
|
||||
isError: true,
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
const contextToken = getContextToken(chatId)
|
||||
const config = await getConfig(
|
||||
baseUrl,
|
||||
account.token,
|
||||
chatId,
|
||||
contextToken,
|
||||
)
|
||||
if (config.typing_ticket) {
|
||||
await sendTyping(baseUrl, account.token, {
|
||||
ilink_user_id: chatId,
|
||||
typing_ticket: config.typing_ticket,
|
||||
status: TypingStatus.TYPING,
|
||||
})
|
||||
}
|
||||
return {
|
||||
content: [{ type: 'text', text: 'Typing indicator sent.' }],
|
||||
}
|
||||
} catch (error) {
|
||||
return {
|
||||
content: [{ type: 'text', text: `Failed to send typing: ${error}` }],
|
||||
isError: true,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
default:
|
||||
return {
|
||||
content: [{ type: 'text', text: `Unknown tool: ${name}` }],
|
||||
isError: true,
|
||||
}
|
||||
}
|
||||
})
|
||||
|
||||
return server
|
||||
}
|
||||
|
||||
export async function runWeixinMcpServer(
|
||||
version: string,
|
||||
deps: WeixinServerDeps,
|
||||
): Promise<void> {
|
||||
deps.enableConfigs()
|
||||
deps.initializeAnalyticsSink()
|
||||
|
||||
const account = loadAccount()
|
||||
if (!account) {
|
||||
process.stderr.write(
|
||||
'[weixin] No account configured. Run `ccb weixin login` to connect your WeChat account.\n',
|
||||
)
|
||||
await Promise.all([deps.shutdown1PEventLogging(), deps.shutdownDatadog()])
|
||||
process.exit(1)
|
||||
}
|
||||
|
||||
const server = createWeixinMcpServer(version)
|
||||
const transport = new StdioServerTransport()
|
||||
|
||||
deps.registerPermissionHandler(server, async request => {
|
||||
const targetChatId = request.channel_context?.chat_id
|
||||
const targetChat = targetChatId
|
||||
? {
|
||||
chatId: targetChatId,
|
||||
contextToken: getContextToken(targetChatId),
|
||||
}
|
||||
: getActivePermissionChat()
|
||||
|
||||
if (!targetChat) {
|
||||
deps.logForDebugging(
|
||||
`[Weixin MCP] No active chat available for permission request ${request.request_id}`,
|
||||
)
|
||||
return
|
||||
}
|
||||
|
||||
try {
|
||||
savePendingPermission(
|
||||
request,
|
||||
targetChat.chatId,
|
||||
targetChat.contextToken,
|
||||
)
|
||||
await sendText({
|
||||
to: targetChat.chatId,
|
||||
text: formatPermissionRequestMessage(request),
|
||||
baseUrl,
|
||||
token: account.token,
|
||||
contextToken: targetChat.contextToken || '',
|
||||
})
|
||||
} catch (error) {
|
||||
process.stderr.write(
|
||||
`[weixin] Failed to relay permission request ${request.request_id}: ${error}\n`,
|
||||
)
|
||||
}
|
||||
})
|
||||
|
||||
await server.connect(transport)
|
||||
|
||||
const baseUrl = account.baseUrl || DEFAULT_BASE_URL
|
||||
const controller = new AbortController()
|
||||
|
||||
let exiting = false
|
||||
const shutdownAndExit = async (): Promise<void> => {
|
||||
if (exiting) return
|
||||
exiting = true
|
||||
if (!controller.signal.aborted) {
|
||||
controller.abort()
|
||||
}
|
||||
await Promise.all([deps.shutdown1PEventLogging(), deps.shutdownDatadog()])
|
||||
process.exit(0)
|
||||
}
|
||||
|
||||
process.stdin.on('end', () => void shutdownAndExit())
|
||||
process.stdin.on('error', () => void shutdownAndExit())
|
||||
process.on('SIGINT', () => void shutdownAndExit())
|
||||
process.on('SIGTERM', () => void shutdownAndExit())
|
||||
process.on('SIGHUP', () => void shutdownAndExit())
|
||||
|
||||
const ppid = process.ppid
|
||||
const parentCheck = setInterval(() => {
|
||||
try {
|
||||
process.kill(ppid, 0)
|
||||
} catch {
|
||||
process.stderr.write('[weixin] Parent process exited, shutting down...\n')
|
||||
clearInterval(parentCheck)
|
||||
void shutdownAndExit()
|
||||
}
|
||||
}, 5000)
|
||||
|
||||
deps.logForDebugging('[Weixin MCP] Starting poll loop')
|
||||
await startPollLoop({
|
||||
baseUrl,
|
||||
cdnBaseUrl: CDN_BASE_URL,
|
||||
token: account.token,
|
||||
onMessage: async (msg: ParsedMessage) => {
|
||||
await server.notification({
|
||||
method: 'notifications/claude/channel',
|
||||
params: {
|
||||
content: msg.text,
|
||||
meta: {
|
||||
chat_id: msg.fromUserId,
|
||||
sender_id: msg.fromUserId,
|
||||
message_id: msg.messageId,
|
||||
...(msg.attachmentPath && { attachment_path: msg.attachmentPath }),
|
||||
...(msg.attachmentType && { attachment_type: msg.attachmentType }),
|
||||
},
|
||||
},
|
||||
})
|
||||
},
|
||||
onPermissionResponse: async response => {
|
||||
await server.notification({
|
||||
method: 'notifications/claude/channel/permission',
|
||||
params: {
|
||||
request_id: response.requestId,
|
||||
behavior: response.behavior,
|
||||
},
|
||||
})
|
||||
},
|
||||
abortSignal: controller.signal,
|
||||
})
|
||||
|
||||
clearInterval(parentCheck)
|
||||
await shutdownAndExit()
|
||||
}
|
||||
178
packages/weixin/src/types.ts
Normal file
178
packages/weixin/src/types.ts
Normal file
@@ -0,0 +1,178 @@
|
||||
export const MessageType = {
|
||||
NONE: 0,
|
||||
USER: 1,
|
||||
BOT: 2,
|
||||
} as const
|
||||
|
||||
export const MessageItemType = {
|
||||
NONE: 0,
|
||||
TEXT: 1,
|
||||
IMAGE: 2,
|
||||
VOICE: 3,
|
||||
FILE: 4,
|
||||
VIDEO: 5,
|
||||
} as const
|
||||
|
||||
export const MessageState = {
|
||||
NEW: 0,
|
||||
GENERATING: 1,
|
||||
FINISH: 2,
|
||||
} as const
|
||||
|
||||
export const UploadMediaType = {
|
||||
IMAGE: 1,
|
||||
VIDEO: 2,
|
||||
FILE: 3,
|
||||
VOICE: 4,
|
||||
} as const
|
||||
|
||||
export const TypingStatus = {
|
||||
TYPING: 1,
|
||||
CANCEL: 2,
|
||||
} as const
|
||||
|
||||
export interface BaseInfo {
|
||||
channel_version?: string
|
||||
}
|
||||
|
||||
export interface CDNMedia {
|
||||
encrypt_query_param?: string
|
||||
aes_key?: string
|
||||
encrypt_type?: number
|
||||
}
|
||||
|
||||
export interface TextItem {
|
||||
text?: string
|
||||
}
|
||||
|
||||
export interface ImageItem {
|
||||
media?: CDNMedia
|
||||
thumb_media?: CDNMedia
|
||||
aeskey?: string
|
||||
url?: string
|
||||
mid_size?: number
|
||||
thumb_size?: number
|
||||
thumb_height?: number
|
||||
thumb_width?: number
|
||||
hd_size?: number
|
||||
}
|
||||
|
||||
export interface VoiceItem {
|
||||
media?: CDNMedia
|
||||
encode_type?: number
|
||||
bits_per_sample?: number
|
||||
sample_rate?: number
|
||||
playtime?: number
|
||||
text?: string
|
||||
}
|
||||
|
||||
export interface FileItem {
|
||||
media?: CDNMedia
|
||||
file_name?: string
|
||||
md5?: string
|
||||
len?: string
|
||||
}
|
||||
|
||||
export interface VideoItem {
|
||||
media?: CDNMedia
|
||||
video_size?: number
|
||||
play_length?: number
|
||||
video_md5?: string
|
||||
thumb_media?: CDNMedia
|
||||
thumb_size?: number
|
||||
thumb_height?: number
|
||||
thumb_width?: number
|
||||
}
|
||||
|
||||
export interface RefMessage {
|
||||
message_item?: MessageItem
|
||||
title?: string
|
||||
}
|
||||
|
||||
export interface MessageItem {
|
||||
type?: number
|
||||
create_time_ms?: number
|
||||
update_time_ms?: number
|
||||
is_completed?: boolean
|
||||
msg_id?: string
|
||||
ref_msg?: RefMessage
|
||||
text_item?: TextItem
|
||||
image_item?: ImageItem
|
||||
voice_item?: VoiceItem
|
||||
file_item?: FileItem
|
||||
video_item?: VideoItem
|
||||
}
|
||||
|
||||
export interface WeixinMessage {
|
||||
seq?: number
|
||||
message_id?: number
|
||||
from_user_id?: string
|
||||
to_user_id?: string
|
||||
client_id?: string
|
||||
create_time_ms?: number
|
||||
update_time_ms?: number
|
||||
delete_time_ms?: number
|
||||
session_id?: string
|
||||
group_id?: string
|
||||
message_type?: number
|
||||
message_state?: number
|
||||
item_list?: MessageItem[]
|
||||
context_token?: string
|
||||
}
|
||||
|
||||
export interface GetUpdatesReq {
|
||||
get_updates_buf?: string
|
||||
base_info?: BaseInfo
|
||||
}
|
||||
|
||||
export interface GetUpdatesResp {
|
||||
ret?: number
|
||||
errcode?: number
|
||||
errmsg?: string
|
||||
msgs?: WeixinMessage[]
|
||||
get_updates_buf?: string
|
||||
longpolling_timeout_ms?: number
|
||||
}
|
||||
|
||||
export interface SendMessageReq {
|
||||
msg?: WeixinMessage
|
||||
base_info?: BaseInfo
|
||||
}
|
||||
|
||||
export interface GetUploadUrlReq {
|
||||
filekey?: string
|
||||
media_type?: number
|
||||
to_user_id?: string
|
||||
rawsize?: number
|
||||
rawfilemd5?: string
|
||||
filesize?: number
|
||||
thumb_rawsize?: number
|
||||
thumb_rawfilemd5?: string
|
||||
thumb_filesize?: number
|
||||
no_need_thumb?: boolean
|
||||
aeskey?: string
|
||||
base_info?: BaseInfo
|
||||
}
|
||||
|
||||
export interface GetUploadUrlResp {
|
||||
upload_param?: string
|
||||
thumb_upload_param?: string
|
||||
}
|
||||
|
||||
export interface GetConfigResp {
|
||||
ret?: number
|
||||
errmsg?: string
|
||||
typing_ticket?: string
|
||||
}
|
||||
|
||||
export interface SendTypingReq {
|
||||
ilink_user_id?: string
|
||||
typing_ticket?: string
|
||||
status?: number
|
||||
base_info?: BaseInfo
|
||||
}
|
||||
|
||||
export interface SendTypingResp {
|
||||
ret?: number
|
||||
errmsg?: string
|
||||
}
|
||||
5
packages/weixin/tsconfig.json
Normal file
5
packages/weixin/tsconfig.json
Normal file
@@ -0,0 +1,5 @@
|
||||
{
|
||||
"extends": "../../tsconfig.base.json",
|
||||
"include": ["src/**/*.ts"],
|
||||
"exclude": ["node_modules", "dist"]
|
||||
}
|
||||
Reference in New Issue
Block a user