feat: iPet → iAs 完整迁移,Rust 版微信 AI 助手

Phase 1  CLI 框架 + 配置系统
  - clap 子命令: login / listen / send / whoami / usage
  - config.json + env var 替换
  - tracing 日志系统
  - state 持久化(auth/runtime 文件存 + PostgreSQL)

Phase 2  微信通道
  - wechat::client — 完整 iLink Bot HTTP API 实现
  - 扫码登录(终端二维码 + 轮询状态)
  - 长轮询 getupdates / 消息收发 / 监听注册

Phase 3  AI 对话(纯文本 + function calling)
  - LlmProvider trait: DeepSeek + LM Studio 实现
  - SSE 流式解析(text / reasoning / tool_calls delta / usage)
  - Conversation: 消息历史 + chat / chat_with_tools 工具循环

Phase 4  PostgreSQL 集成
  - app_state(认证 KV 存储)
  - chat_records(消息收发记录)
  - llm_usage(Token 用量统计缓存命中率)
  - user_memories(长期记忆持久化)
  - pending_approvals(审批确认码)
  - scheduled_tasks(定时任务表)

Phase 5  一切皆 Skill(工具系统)
  - SkillRegistry: 系统 + 用户 skills 双目录合并
  - SKILL.md 解析器 + 子进程执行器(stdin JSON → stdout)
  - 9 个系统 Skills: datetime / weather / search / email /
    shell / schedule / memos / read_memories / read_summaries
  - ApprovalManager: High 风险技能 → 确认码审批(透明模式)
  - High 风险技能:确认码审批(透明模式)

Phase 6  定时任务调度器

上下文管理
  - ChatSession: checkpoint + token budget (28K) + summaries
  - Token 估算器(中英文自适应)
  - 12h 空闲 → trigger_idle_summary(不入会话)
  - Budget 溢出 → trigger_overflow_summary(入会话 + drain 旧消息)
  - Summarizer: LLM 生成自然语言摘要(fallback 简单截断)
  - 长期记忆 / 摘要 通过 read_memories / read_summaries 工具按需读取

工具调用日志 + Token 统计
  - INFO: 工具名 + 参数 + 结果摘要
  - DEBUG: 子进程 exit/stdout/stderr
  - ias usage --since --until --model 查看用量和缓存命中率
This commit is contained in:
2026-06-01 17:21:43 +08:00
parent 3a2d2769b5
commit b9de3665d9
53 changed files with 9874 additions and 2 deletions
+151
View File
@@ -0,0 +1,151 @@
# iPet → Rust (iAs) 迁移计划
## 项目定位
将 iPetBun/JS 微信 AI 聊天机器人)迁移到 Rust 实现的 `iAs` 项目。
## 技术栈
| 模块 | Rust 方案 |
|------|-----------|
| 运行时 | `tokio`full |
| HTTP 客户端 | `reqwest`json + stream |
| 序列化 | `serde` / `serde_json` |
| CLI | `clap`derive |
| 数据库 | `sqlx`postgres + uuid + chrono |
| 邮箱 | `async-imap` + `mailparse` |
| JWT | `jsonwebtoken` |
| 定时任务 | `tokio-cron-scheduler` |
| 流式 | `tokio-stream` + `futures` |
| 加密 | `sha2` + `hex` + `uuid` |
## 分阶段计划
### Phase 1 ✅ CLI 框架 + 配置系统
- [x] clap 子命令:login, listen, send, whoami
- [x] 配置文件读取(config.json + env var 替换)
- [x] 日志系统(tracing + EnvFilter
- [x] 状态管理(auth.json / runtime.json 持久化)
### Phase 2 🔜 微信通道(当前阶段)
- [x] API 类型定义
- [ ] 扫码登录(get_bot_qrcode + get_qrcode_status
- [ ] 长轮询收消息(getupdates
- [ ] 发送消息(sendmessage
- [ ] 注册/注销监听器(notifystart/notifystop
- [ ] 身份和 token 持久化
### Phase 3 ✅ AI 对话(纯对话,无工具)
- [x] DeepSeek 流式 APISSE + thinking 模式)
- [x] LM Studio 兼容(OpenAI 接口)
- [x] Provider 抽象(LlmProvider trait
- [x] 对话管理器(Conversation
- 消息历史管理
- 系统提示词
- 自动修剪历史
- 流式消费 + 自动记录
- [x] listen 命令集成 LLM 回复
### Phase 4 ✅ PostgreSQL 集成(认证+聊天记录入库)
- [x] sqlx 迁移框架(`migrations/` 目录)
- [x] `app_state` 表:认证信息 Key-Value 存储
- [x] `chat_records` 表:消息收发记录
- [x] 数据库连接池管理(`Database` 结构体)
- [x] 文件存储 → 数据库的自动迁移
- [x] 文件存储降级(无 DB 时保持可用)
- [x] 入库时机:
- login → auth 入库
- 收到消息 → inbound 记录
- echo/LLM 回复 → outbound 记录
- send 命令 → outbound 记录
### Phase 5 🎯 一切皆 Skill(工具系统)
**设计文档:** `src/tools/DESIGN.md`
**核心思想:** 没有"内置工具"和"外部技能"的区别——所有工具都是 SKILL.md。
- 系统 skills — `resources/skills/`,随 iAs 分发
- 用户 skills — `skills/`,用户自行添加
- 加载时合并,同名用户覆盖系统
**安全模型(仅两级):**
| 等级 | 行为 |
|------|------|
| Low | 自动执行 |
| High | 微信确认码审批(6位码,5分钟过期)|
**系统 Skills7个):**
| Skill | 风险 | 说明 |
|-------|------|------|
| `get_current_datetime` | Low | 获取时间 |
| `query_weather` | Low | 和风天气 |
| `search_web` | Low | Tavily 搜索 |
| `email` | High | 收发邮件 |
| `execute_shell` | High | 执行 shell |
| `list_scheduled_tasks` | Low | 列定时任务 |
| `manage_scheduled_task` | High | 管理定时任务 |
**去掉的模块:**
- ~~MCP 桥接~~(由 Skills 替代)
- ~~Shell 能力~~(由 Skills 替代)
- ~~Medium 风险层~~(仅保留 Low/High
**实施步骤:**
- [x] 设计文档(DESIGN.md
- [x] Step 1: SkillSpec/Skill/SkillRegistry 基础设施
- [x] Step 2: SKILL.md 解析器 + 目录扫描(system + user 合并)
- [x] Step 3: 技能执行器(子进程 + stdin 参数 + 结果解析)
- [x] Step 4: 系统 Skills 脚本实现(7 个)
- get_current_datetime ✅ / query_weather ✅ / search_web ✅
- email ✅ / execute_shell ✅ / list_scheduled_tasks ✅ / manage_scheduled_task ✅
- [x] Step 5: 审批系统(pending_approvals + 确认码 + oneshot 通道)
- approval.rs: ApprovalManager(内存 + 数据库双存储)
- 确认码生成/验证(SHA-256 哈希)
- 取消支持(0 / 取消)
- 3 次重试机制
- 透明模式(LLM 不感知审批过程,确认码回复被 consume 不传 LLM
- [x] Step 6: LLM function calling 集成
- ToolCall 类型 + StreamChunk::ToolCalls
- 流式解析器支持 tool_calls delta
- Conversation::chat_with_tools() 工具循环
- SkillSpec → LLM tools JSON 转换
- ToolExecutor 回调桥接 SkillRegistry
- 审批回复拦截(handle_reply 优先于 LLM
- WeChatClient 可克隆(用于 send_wechat 闭包)
### Phase 6 定时任务(PostgreSQL 调度)
- [ ] 基于 PostgreSQL 的定时任务调度
### Phase 7 上下文管理 + 信任系统
- [ ] 长短期记忆
- [ ] 滚动摘要
- [ ] 用户验证/授权
### Phase 8 优化、测试、部署
- [ ] 错误处理完善
- [ ] 日志优化
- [ ] systemd 服务
## 微信通道 API 参考
Base URL: `https://ilinkai.weixin.qq.com`
### 认证
headers:
- `Authorization: Bearer <token>`
- `AuthorizationType: ilink_bot_token`
- `Content-Type: application/json`
- `iLink-App-Id: <app_id>`
- `iLink-App-ClientVersion: <encoded_uint32>`
- `X-WECHAT-UIN: <random_uint32_b64>`
### 端点
| 方法 | 路径 | 说明 |
|------|------|------|
| GET | `/ilink/bot/get_bot_qrcode?bot_type=3` | 获取登录二维码 |
| GET | `/ilink/bot/get_qrcode_status?qrcode=<qrcode>` | 轮询扫码状态 |
| POST | `/ilink/bot/msg/notifystart` | 注册监听器 |
| POST | `/ilink/bot/msg/notifystop` | 注销监听器 |
| POST | `/ilink/bot/getupdates` | 长轮询消息 |
| POST | `/ilink/bot/sendmessage` | 发送消息 |
| POST | `/ilink/bot/getconfig` | 获取配置 |
| POST | `/ilink/bot/sendtyping` | 发送输入状态 |