cc-wechat 0.1.0

package/src/server.ts

@@ -0,0 +1,445 @@
+#!/usr/bin/env node
+/**
+ * cc-wechat MCP Server 主入口
+ * Claude Code Channel 插件 — 微信消息桥接
+ */
+
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { ListToolsRequestSchema, CallToolRequestSchema } from '@modelcontextprotocol/sdk/types.js';
+import fs from 'node:fs';
+
+import { getActiveAccount, saveAccount, loadSyncBuf, saveSyncBuf } from './store.js';
+import { getUpdates, sendMessage, sendTyping, getConfig } from './ilink-api.js';
+import { loginBrowser } from './auth.js';
+import { uploadMedia, downloadMedia } from './cdn.js';
+import { stripMarkdown, chunkText } from './text-utils.js';
+import type { WeixinMessage, AccountData } from './types.js';
+import { MessageItemType } from './types.js';
+
+// ─── 状态变量 ─────────────────────────────────────────
+
+let pollingActive = false;
+let pollingAbort: AbortController | null = null;
+const typingTicketCache = new Map<string, string>();
+
+// ─── Session 过期处理常量 ─────────────────────────────
+
+const SESSION_EXPIRED_ERRCODE = -14;
+const MAX_SESSION_RETRIES = 3;
+const MAX_CONSECUTIVE_FAILURES = 5;
+const INITIAL_RETRY_DELAY_MS = 2_000;
+const MAX_RETRY_DELAY_MS = 30_000;
+const SESSION_PAUSE_MS = 5 * 60_000;
+
+// ─── 辅助函数 ─────────────────────────────────────────
+
+/** 可中断的 sleep */
+function sleep(ms: number, signal?: AbortSignal): Promise<void> {
+  return new Promise((resolve, reject) => {
+    const t = setTimeout(resolve, ms);
+    signal?.addEventListener('abort', () => { clearTimeout(t); reject(new Error('aborted')); }, { once: true });
+  });
+}
+
+/** 从消息提取可读文本（异步，支持媒体下载） */
+async function extractText(msg: WeixinMessage): Promise<string> {
+  const parts: string[] = [];
+  for (const item of msg.item_list ?? []) {
+    const t = item.type ?? 0;
+
+    if (t === MessageItemType.TEXT) {
+      if (item.text_item?.text) parts.push(item.text_item.text);
+    } else if (t === MessageItemType.IMAGE) {
+      let desc = '[图片]';
+      if (item.image_item?.media?.encrypt_query_param && item.image_item?.media?.aes_key) {
+        try {
+          const filePath = await downloadMedia({
+            encryptQueryParam: item.image_item.media.encrypt_query_param,
+            aesKeyBase64: item.image_item.media.aes_key,
+          });
+          desc += `\n[附件: ${filePath}]`;
+        } catch {
+          // 下载失败不阻塞消息处理
+        }
+      }
+      parts.push(desc);
+    } else if (t === MessageItemType.VOICE) {
+      parts.push(`[语音] ${item.voice_item?.text ?? ''}`);
+    } else if (t === MessageItemType.FILE) {
+      let desc = `[文件: ${item.file_item?.file_name ?? 'unknown'}]`;
+      if (item.file_item?.media?.encrypt_query_param && item.file_item?.media?.aes_key) {
+        try {
+          const filePath = await downloadMedia({
+            encryptQueryParam: item.file_item.media.encrypt_query_param,
+            aesKeyBase64: item.file_item.media.aes_key,
+            fileName: item.file_item.file_name,
+          });
+          desc += `\n[附件: ${filePath}]`;
+        } catch {
+          // 下载失败不阻塞消息处理
+        }
+      }
+      parts.push(desc);
+    } else if (t === MessageItemType.VIDEO) {
+      parts.push('[视频]');
+    }
+  }
+  return parts.join('\n') || '[空消息]';
+}
+
+// ─── MCP Server 创建 ─────────────────────────────────
+
+const server = new Server(
+  { name: 'wechat-channel', version: '0.1.0' },
+  {
+    capabilities: {
+      experimental: { 'claude/channel': {} },
+      tools: {},
+    },
+    instructions: `Messages arrive as <channel source="wechat-channel" user_id="..." context_token="...">.
+Reply using the reply tool. Pass user_id and context_token from the channel tag.
+For media: set media to an absolute local file path to send image/video/file.
+IMPORTANT: Always use the reply tool to respond to WeChat messages. Do not just output text.`,
+  },
+);
+
+// ─── Tools — ListToolsRequestSchema ──────────────────
+
+server.setRequestHandler(ListToolsRequestSchema, async () => ({
+  tools: [
+    {
+      name: 'login',
+      description: '扫码登录微信。首次使用或 session 过期后运行。',
+      inputSchema: {
+        type: 'object' as const,
+        properties: {},
+      },
+    },
+    {
+      name: 'reply',
+      description: '回复微信消息',
+      inputSchema: {
+        type: 'object' as const,
+        properties: {
+          user_id: { type: 'string', description: '微信用户 ID（来自消息 meta 的 user_id）' },
+          context_token: { type: 'string', description: '会话上下文令牌（来自消息 meta 的 context_token）' },
+          content: { type: 'string', description: '回复文本内容' },
+          media: { type: 'string', description: '可选：本地文件绝对路径，发送图片/视频/文件' },
+        },
+        required: ['user_id', 'context_token', 'content'],
+      },
+    },
+  ],
+}));
+
+// ─── Tools — CallToolRequestSchema ───────────────────
+
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+  const { name, arguments: args } = request.params;
+
+  // ── login tool ──
+  if (name === 'login') {
+    try {
+      const result = await loginBrowser();
+      saveAccount({
+        token: result.token,
+        baseUrl: result.baseUrl ?? '',
+        botId: result.accountId,
+        savedAt: new Date().toISOString(),
+      });
+      startPolling();
+      return {
+        content: [{ type: 'text' as const, text: `登录成功！账号 ID: ${result.accountId}` }],
+      };
+    } catch (err) {
+      return {
+        content: [{ type: 'text' as const, text: `登录失败: ${String(err)}` }],
+        isError: true,
+      };
+    }
+  }
+
+  // ── reply tool ──
+  if (name === 'reply') {
+    const userId = args?.user_id as string | undefined;
+    const contextToken = args?.context_token as string | undefined;
+    const content = args?.content as string | undefined;
+    const media = args?.media as string | undefined;
+
+    // 验证必填参数
+    if (!userId || !contextToken || !content) {
+      return {
+        content: [{ type: 'text' as const, text: '缺少必填参数: user_id, context_token, content' }],
+        isError: true,
+      };
+    }
+
+    // 验证账号存在
+    const account = getActiveAccount();
+    if (!account) {
+      return {
+        content: [{ type: 'text' as const, text: '未登录，请先使用 login 工具扫码登录' }],
+        isError: true,
+      };
+    }
+
+    // 检查媒体文件是否存在
+    if (media && !fs.existsSync(media)) {
+      return {
+        content: [{ type: 'text' as const, text: `媒体文件不存在: ${media}` }],
+        isError: true,
+      };
+    }
+
+    try {
+      // 发送 typing 状态（best-effort）
+      try {
+        let ticket = typingTicketCache.get(userId);
+        if (!ticket) {
+          const config = await getConfig(account.token, userId, contextToken, account.baseUrl);
+          ticket = config.typing_ticket ?? '';
+          if (ticket) typingTicketCache.set(userId, ticket);
+        }
+        if (ticket) {
+          await sendTyping(account.token, userId, ticket, 1, account.baseUrl);
+        }
+      } catch {
+        // typing 失败不阻塞
+      }
+
+      // 清理 Markdown 并分段发送
+      const plainText = stripMarkdown(content);
+      const chunks = chunkText(plainText, 3900);
+      for (const chunk of chunks) {
+        await sendMessage(account.token, userId, chunk, contextToken, account.baseUrl);
+      }
+
+      // 发送媒体文件（如有）
+      let mediaError = '';
+      if (media) {
+        try {
+          await uploadMedia({
+            token: account.token,
+            toUser: userId,
+            contextToken,
+            filePath: media,
+            baseUrl: account.baseUrl,
+          });
+        } catch (err) {
+          mediaError = `（媒体发送失败: ${String(err)}）`;
+        }
+      }
+
+      // 停止 typing 状态（best-effort）
+      try {
+        const ticket = typingTicketCache.get(userId);
+        if (ticket) {
+          await sendTyping(account.token, userId, ticket, 2, account.baseUrl);
+        }
+      } catch {
+        // typing 失败不阻塞
+      }
+
+      return {
+        content: [{
+          type: 'text' as const,
+          text: `已发送 ${chunks.length} 段文本${media ? ' + 1 个媒体文件' : ''}${mediaError}`,
+        }],
+      };
+    } catch (err) {
+      return {
+        content: [{ type: 'text' as const, text: `发送失败: ${String(err)}` }],
+        isError: true,
+      };
+    }
+  }
+
+  return {
+    content: [{ type: 'text' as const, text: `未知工具: ${name}` }],
+    isError: true,
+  };
+});
+
+// ─── 轮询循环 ─────────────────────────────────────────
+
+/** 消息长轮询循环 */
+async function pollLoop(account: AccountData): Promise<void> {
+  let buf = loadSyncBuf();
+  let consecutiveFailures = 0;
+  let sessionRetries = 0;
+  let retryDelay = INITIAL_RETRY_DELAY_MS;
+  let nextTimeoutMs: number | undefined;
+
+  while (pollingActive && !pollingAbort?.signal.aborted) {
+    try {
+      const resp = await getUpdates(account.token, buf, account.baseUrl, nextTimeoutMs);
+
+      // 更新长轮询超时
+      if (resp.longpolling_timeout_ms) {
+        nextTimeoutMs = resp.longpolling_timeout_ms;
+      }
+
+      // 检查 API 错误
+      if ((resp.ret !== undefined && resp.ret !== 0) || (resp.errcode !== undefined && resp.errcode !== 0)) {
+        const errcode = resp.errcode ?? resp.ret ?? 0;
+
+        if (errcode === SESSION_EXPIRED_ERRCODE) {
+          sessionRetries++;
+          process.stderr.write(`[wechat-channel] Session 过期 (${sessionRetries}/${MAX_SESSION_RETRIES})\n`);
+
+          if (sessionRetries >= MAX_SESSION_RETRIES) {
+            pollingActive = false;
+            // 通知 Claude session 过期
+            server.notification({
+              method: 'notifications/message',
+              params: {
+                level: 'error',
+                data: 'WeChat session expired, please use login tool to reconnect',
+              },
+            });
+            return;
+          }
+
+          await sleep(SESSION_PAUSE_MS, pollingAbort?.signal);
+          continue;
+        }
+
+        // 其他错误
+        consecutiveFailures++;
+        process.stderr.write(
+          `[wechat-channel] API 错误 errcode=${errcode} errmsg=${resp.errmsg ?? ''} ` +
+          `(${consecutiveFailures}/${MAX_CONSECUTIVE_FAILURES})\n`,
+        );
+
+        if (consecutiveFailures >= MAX_CONSECUTIVE_FAILURES) {
+          process.stderr.write(`[wechat-channel] 连续失败过多，暂停 ${SESSION_PAUSE_MS / 1000}s\n`);
+          await sleep(SESSION_PAUSE_MS, pollingAbort?.signal);
+          consecutiveFailures = 0;
+        } else {
+          await sleep(retryDelay, pollingAbort?.signal);
+          retryDelay = Math.min(retryDelay * 2, MAX_RETRY_DELAY_MS);
+        }
+        continue;
+      }
+
+      // 成功 → 重置计数器
+      consecutiveFailures = 0;
+      retryDelay = INITIAL_RETRY_DELAY_MS;
+
+      // 保存 sync buf
+      if (resp.get_updates_buf) {
+        buf = resp.get_updates_buf;
+        saveSyncBuf(buf);
+      }
+
+      // 处理消息（仅用户消息 message_type === 1）
+      for (const msg of resp.msgs ?? []) {
+        if (msg.message_type !== 1) continue;
+
+        const fromUser = msg.from_user_id ?? '';
+        const contextToken = msg.context_token ?? '';
+
+        // 提取文本
+        const text = await extractText(msg);
+
+        // 缓存 typing ticket（best-effort）
+        try {
+          const config = await getConfig(account.token, fromUser, contextToken, account.baseUrl);
+          if (config.typing_ticket) {
+            typingTicketCache.set(fromUser, config.typing_ticket);
+          }
+        } catch {
+          // 忽略
+        }
+
+        // 发送 typing 状态（best-effort）
+        try {
+          const ticket = typingTicketCache.get(fromUser);
+          if (ticket) {
+            await sendTyping(account.token, fromUser, ticket, 1, account.baseUrl);
+          }
+        } catch {
+          // 忽略
+        }
+
+        // 通知 Claude 有新消息
+        server.notification({
+          method: 'notifications/claude/channel',
+          params: {
+            content: text,
+            meta: {
+              source: 'wechat',
+              user_id: fromUser,
+              context_token: contextToken,
+              message_id: String(msg.message_id ?? ''),
+              session_id: msg.session_id ?? '',
+            },
+          },
+        });
+      }
+    } catch (err) {
+      if (pollingAbort?.signal.aborted) return;
+
+      // 网络错误 → 指数退避
+      consecutiveFailures++;
+      process.stderr.write(
+        `[wechat-channel] 网络错误: ${String(err)} ` +
+        `(${consecutiveFailures}/${MAX_CONSECUTIVE_FAILURES})\n`,
+      );
+
+      if (consecutiveFailures >= MAX_CONSECUTIVE_FAILURES) {
+        process.stderr.write(`[wechat-channel] 连续失败过多，暂停 ${SESSION_PAUSE_MS / 1000}s\n`);
+        await sleep(SESSION_PAUSE_MS, pollingAbort?.signal);
+        consecutiveFailures = 0;
+      } else {
+        await sleep(retryDelay, pollingAbort?.signal);
+        retryDelay = Math.min(retryDelay * 2, MAX_RETRY_DELAY_MS);
+      }
+    }
+  }
+}
+
+// ─── 轮询控制 ─────────────────────────────────────────
+
+/** 启动消息轮询 */
+function startPolling(): void {
+  const account = getActiveAccount();
+  if (!account || pollingActive) return;
+  pollingActive = true;
+  pollingAbort = new AbortController();
+  pollLoop(account).catch((err) => {
+    if (!pollingAbort?.signal.aborted) {
+      process.stderr.write(`[wechat-channel] Poll loop crashed: ${String(err)}\n`);
+    }
+    pollingActive = false;
+  });
+}
+
+/** 停止消息轮询 */
+function stopPolling(): void {
+  pollingActive = false;
+  pollingAbort?.abort();
+  pollingAbort = null;
+}
+
+// ─── 主入口 ───────────────────────────────────────────
+
+async function main(): Promise<void> {
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+  process.stderr.write('[wechat-channel] MCP server started\n');
+
+  const account = getActiveAccount();
+  if (account) {
+    process.stderr.write(`[wechat-channel] Found saved account: ${account.botId}\n`);
+    startPolling();
+  } else {
+    process.stderr.write('[wechat-channel] No saved account. Use the login tool to connect.\n');
+  }
+}
+
+main().catch((err) => {
+  process.stderr.write(`[wechat-channel] Fatal: ${String(err)}\n`);
+  process.exit(1);
+});

package/src/store.ts

@@ -0,0 +1,62 @@
+/**
+ * cc-wechat 凭证持久化 — account.json 原子写入 + sync buf
+ */
+
+import { mkdirSync, readFileSync, renameSync, writeFileSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+import type { AccountData } from './types.js';
+
+const ACCOUNT_FILE = 'account.json';
+const SYNC_BUF_FILE = 'sync-buf.txt';
+
+/**
+ * 获取状态目录路径，不存在则自动创建
+ */
+export function getStateDir(): string {
+  const dir = join(homedir(), '.claude', 'channels', 'wechat');
+  mkdirSync(dir, { recursive: true });
+  return dir;
+}
+
+/**
+ * 原子写入账号数据（先写 tmp 再 rename）
+ */
+export function saveAccount(data: AccountData): void {
+  const dir = getStateDir();
+  const tmpPath = join(dir, `${ACCOUNT_FILE}.tmp`);
+  const finalPath = join(dir, ACCOUNT_FILE);
+  writeFileSync(tmpPath, JSON.stringify(data, null, 2), 'utf-8');
+  renameSync(tmpPath, finalPath);
+}
+
+/**
+ * 读取当前活跃账号，文件不存在返回 null
+ */
+export function getActiveAccount(): AccountData | null {
+  try {
+    const filePath = join(getStateDir(), ACCOUNT_FILE);
+    const raw = readFileSync(filePath, 'utf-8');
+    return JSON.parse(raw) as AccountData;
+  } catch {
+    return null;
+  }
+}
+
+/**
+ * 读取 sync buf，不存在返回空字符串
+ */
+export function loadSyncBuf(): string {
+  try {
+    return readFileSync(join(getStateDir(), SYNC_BUF_FILE), 'utf-8');
+  } catch {
+    return '';
+  }
+}
+
+/**
+ * 写入 sync buf
+ */
+export function saveSyncBuf(buf: string): void {
+  writeFileSync(join(getStateDir(), SYNC_BUF_FILE), buf, 'utf-8');
+}

package/src/text-utils.ts

@@ -0,0 +1,56 @@
+/**
+ * 文本处理工具 — Markdown 清理 + 分段
+ */
+
+/** 去除 Markdown 格式，转为微信纯文本 */
+export function stripMarkdown(text: string): string {
+  let result = text;
+  // 代码围栏：去除 ``` 保留内容
+  result = result.replace(/```[^\n]*\n?([\s\S]*?)```/g, (_, code: string) => code.trim());
+  // 图片链接：移除
+  result = result.replace(/!\[[^\]]*\]\([^)]*\)/g, '');
+  // 链接：保留显示文字
+  result = result.replace(/\[([^\]]+)\]\([^)]*\)/g, '$1');
+  // 粗体
+  result = result.replace(/\*\*(.+?)\*\*/g, '$1');
+  result = result.replace(/__(.+?)__/g, '$1');
+  // 斜体
+  result = result.replace(/\*(.+?)\*/g, '$1');
+  result = result.replace(/_(.+?)_/g, '$1');
+  // 标题
+  result = result.replace(/^#{1,6}\s+/gm, '');
+  // 水平线
+  result = result.replace(/^[-*_]{3,}$/gm, '');
+  // 引用
+  result = result.replace(/^>\s?/gm, '');
+  return result.trim();
+}
+
+/** 将长文本分段（微信限制约 4000 字符） */
+export function chunkText(text: string, maxLen = 3900): string[] {
+  if (text.length <= maxLen) return [text];
+  const chunks: string[] = [];
+  let remaining = text;
+  while (remaining.length > 0) {
+    if (remaining.length <= maxLen) {
+      chunks.push(remaining);
+      break;
+    }
+    // 优先在双换行处分割
+    let breakAt = remaining.lastIndexOf('\n\n', maxLen);
+    if (breakAt < maxLen * 0.3) {
+      // 其次单换行
+      breakAt = remaining.lastIndexOf('\n', maxLen);
+    }
+    if (breakAt < maxLen * 0.3) {
+      // 其次空格（避免截断 URL）
+      breakAt = remaining.lastIndexOf(' ', maxLen);
+    }
+    if (breakAt < maxLen * 0.3) {
+      breakAt = maxLen;
+    }
+    chunks.push(remaining.slice(0, breakAt));
+    remaining = remaining.slice(breakAt).trimStart();
+  }
+  return chunks;
+}

package/src/types.ts

@@ -0,0 +1,94 @@
+/**
+ * cc-wechat 类型定义 — iLink Bot API 请求/响应/消息结构
+ */
+
+// 消息内容类型枚举
+export const MessageItemType = {
+  NONE: 0,
+  TEXT: 1,
+  IMAGE: 2,
+  VOICE: 3,
+  FILE: 4,
+  VIDEO: 5,
+} as const;
+
+// CDN 媒体引用
+export interface CDNMedia {
+  encrypt_query_param?: string;
+  aes_key?: string;
+  encrypt_type?: number;
+}
+
+// 消息内容项
+export interface MessageItem {
+  type?: number;
+  text_item?: { text?: string };
+  image_item?: { media?: CDNMedia; url?: string; mid_size?: number };
+  voice_item?: { media?: CDNMedia; text?: string; playtime?: number };
+  file_item?: { media?: CDNMedia; file_name?: string; len?: string; md5?: string };
+  video_item?: { media?: CDNMedia; video_size?: number };
+  ref_msg?: { title?: string; message_item?: MessageItem };
+  msg_id?: string;
+}
+
+// 微信消息
+export interface WeixinMessage {
+  seq?: number;
+  message_id?: number;
+  from_user_id?: string;
+  to_user_id?: string;
+  client_id?: string;
+  create_time_ms?: number;
+  session_id?: string;
+  message_type?: number;  // 1=USER, 2=BOT
+  message_state?: number; // 0=NEW, 1=GENERATING, 2=FINISH
+  item_list?: MessageItem[];
+  context_token?: string;
+}
+
+// API 响应
+export interface GetUpdatesResp {
+  ret?: number;
+  errcode?: number;
+  errmsg?: string;
+  msgs?: WeixinMessage[];
+  get_updates_buf?: string;
+  longpolling_timeout_ms?: number;
+}
+
+export interface QRCodeResponse {
+  qrcode: string;
+  qrcode_img_content: string;
+}
+
+export interface QRStatusResponse {
+  status: 'wait' | 'scaned' | 'confirmed' | 'expired';
+  bot_token?: string;
+  ilink_bot_id?: string;
+  baseurl?: string;
+  ilink_user_id?: string;
+}
+
+export interface GetConfigResp {
+  ret?: number;
+  typing_ticket?: string;
+}
+
+export interface GetUploadUrlResp {
+  upload_param?: string;
+  thumb_upload_param?: string;
+  filekey?: string;
+}
+
+// 本地存储
+export interface AccountData {
+  token: string;
+  baseUrl: string;
+  botId: string;
+  userId?: string;
+  savedAt: string;
+}
+
+export interface BaseInfo {
+  channel_version?: string;
+}

package/tsconfig.json

@@ -0,0 +1,17 @@
+{
+  "compilerOptions": {
+    "target": "ES2024",
+    "module": "NodeNext",
+    "moduleResolution": "NodeNext",
+    "outDir": "dist",
+    "rootDir": "src",
+    "strict": true,
+    "noImplicitAny": true,
+    "declaration": true,
+    "sourceMap": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true
+  },
+  "include": ["src"],
+  "exclude": ["node_modules", "dist"]
+}