"""
Phase 4: 个人微信 (iLink 协议 / 微信 ClawBot 插件) —— 前台用户模块
im_bot_user.py
为【平台注册用户】提供"各自绑定自己的个人微信"能力:
- 每个用户独立一条 im_bot_wechat_account 记录(按 user_id 隔离)
- 扫码登录免 AppID;消息收发走 Bearer bot_token + X-WECHAT-UIN
- 绑定后由七虎AI 大脑自动应答(复用 channels 的真实 LLM 调用)
与后台 im_bot_admin 的区别:
- 后台:平台级 IM Bot 接入中枢(飞书/企微/Telegram 等多平台、路由、统计),其个人微信
端点仅限 user_id IS NULL 的"全局 Bot",与用户绑定互不干扰。
- 前台:每个登录用户管理自己的个人微信,数据严格按 user_id 隔离。
"""
import base64
import json
import os
import random
import threading
import time
from datetime import datetime, timedelta, timezone
from typing import Optional
from urllib.parse import quote as _quote
import requests
from fastapi import APIRouter, Depends, HTTPException, Query
from sqlalchemy import text
from sqlalchemy.orm import Session
from ..auth import get_current_user
from ..database import SessionLocal
from ..models import User
router = APIRouter()
# 北京时间(UTC+8)。容器系统时区为 UTC,所有对外展示/持久化的时间戳必须按北京时间存储,
# 否则前台微信 Bot 对话窗会显示成 UTC(比系统时间早 8 小时)。
TZ_BEIJING = timezone(timedelta(hours=8))
def _beijing_now() -> datetime:
"""返回北京时间的 naive datetime(去掉 tzinfo 以便直接存入 MySQL DATETIME)。"""
return datetime.now(TZ_BEIJING).replace(tzinfo=None)
# ============================================================
# iLink 协议常量(腾讯官方 iLink Bot HTTP API,直连免 AppID)
# ============================================================
ILINK_BASE = "https://ilinkai.weixin.qq.com"
# 每个已连接账号一个后台接收线程,key = 账号表主键 id
_wechat_receiver_threads: dict[int, threading.Thread] = {}
# ============================================================
# 通用 DB 工具
# ============================================================
def _execute_raw(db: Session, sql_text, params=None):
result = db.execute(sql_text, params or {})
if result.returns_rows:
return result.fetchall()
db.commit()
return result
def _row_to_dict(row) -> dict:
if row is None:
return None
d = {}
for col in row._mapping.keys():
val = row._mapping[col]
if isinstance(val, datetime):
val = val.isoformat()
d[col] = val
return d
# ============================================================
# iLink 协议请求工具
# ============================================================
def _wechat_headers(bot_token: str) -> dict:
"""iLink 协议请求头:Bearer bot_token + 随机 X-WECHAT-UIN 防重放"""
uin = base64.b64encode(str(random.randint(0, 2 ** 32 - 1)).encode("utf-8")).decode("utf-8")
return {
"Content-Type": "application/json",
"Authorization": f"Bearer {bot_token}",
"AuthorizationType": "ilink_bot_token",
"X-WECHAT-UIN": uin,
}
def _wechat_row_for_user(db: Session, user_id: str):
rows = _execute_raw(
db,
text("SELECT * FROM im_bot_wechat_account WHERE user_id = :uid ORDER BY id DESC LIMIT 1"),
{"uid": user_id},
)
return rows[0] if rows else None
def _wechat_mask(row) -> dict:
d = _row_to_dict(row)
if d.get("bot_token"):
bt = d["bot_token"]
d["bot_token"] = ("***" + bt[-6:]) if len(bt) > 6 else "***"
d.pop("appid", None)
d.pop("secret_key", None)
return d
# ============================================================
# 七虎AI 大脑(复用 channels 的真实 LLM 调用)
# ============================================================
def _recent_history(user_id: str, limit: int = 6) -> str:
"""拼接最近若干轮对话,作为多轮上下文。"""
db = None
try:
db = SessionLocal()
rows = _execute_raw(db, text(
"SELECT role, content FROM im_bot_chat_log "
"WHERE user_id=:uid ORDER BY id DESC LIMIT :lim"
), {"uid": user_id, "lim": limit})
pairs = []
for r in reversed(rows):
dd = _row_to_dict(r)
role = "用户" if dd["role"] == "user" else "七虎AI"
pairs.append(f"{role}:{dd['content']}")
return "\n".join(pairs)
except Exception:
return ""
finally:
if db:
db.close()
def _qihua_reply(text: str, user_id: str = None) -> str:
"""调用七虎AI 生成回复。自行创建 DB 会话,避免上层线程无可用 session。"""
prompt = text
if user_id:
hist = _recent_history(user_id)
if hist:
prompt = (
"以下是你与该用户最近的对话历史,请结合上下文作答:\n"
f"{hist}\n\n"
f"用户当前问题:{text}"
)
db = None
try:
from app.api.im_llm import _call_real_llm
db = SessionLocal()
return _call_real_llm(db, None, prompt, "个人微信", user_id=user_id)
except Exception as e:
return f"(AI 回复生成失败:{e})"
finally:
if db:
db.close()
# ============================================================
# iLink 消息收发
# ============================================================
def _wechat_get_updates(baseurl: str, bot_token: str, cursor: str):
"""长轮询收消息,返回 (msgs, new_cursor)。最多 hold 35s。"""
try:
resp = requests.post(
baseurl.rstrip("/") + "/ilink/bot/getupdates",
headers=_wechat_headers(bot_token),
json={"get_updates_buf": cursor, "base_info": {"channel_version": "1.0.2"}},
timeout=40,
)
if not resp.ok:
return [], cursor
j = resp.json() if resp.content else {}
return j.get("msgs", []) or [], j.get("get_updates_buf", cursor)
except Exception:
return [], cursor
def _wechat_post_msg(baseurl: str, bot_token: str, to_user_id: str, context_token: str, item_list: list):
"""向 iLink 发送一条消息(item_list 可含文本/图片/视频等)。返回 (ok, ret, body)。
ret=-14=会话过期需重扫;ret=-2=token 失效。"""
import uuid as _uuid
payload = {
"msg": {
"from_user_id": "",
"to_user_id": to_user_id,
"client_id": "py-" + str(_uuid.uuid4()),
"message_type": 2,
"message_state": 2,
"context_token": context_token,
"item_list": item_list,
},
"base_info": {"channel_version": "1.0.2"},
}
last_body = {}
for _ in range(2):
try:
resp = requests.post(
baseurl.rstrip("/") + "/ilink/bot/sendmessage",
headers=_wechat_headers(bot_token),
json=payload,
timeout=20,
)
try:
body = resp.json() if resp.content else {}
except Exception:
body = {}
last_body = body
ret = body.get("ret") if isinstance(body, dict) else None
try:
with open("/tmp/wechat_send.log", "a", encoding="utf-8") as _f:
_f.write(_beijing_now().isoformat() + " SEND_HTTP=%s ret=%s body=%s\n"
% (resp.status_code, ret, json.dumps(body, ensure_ascii=False)[:300]))
except Exception:
pass
if resp.ok and (ret is None or ret == 0 or not body):
return True, ret, body
if ret in (-2, -14):
return False, ret, body # 鉴权/会话失效,立即停止重试
except Exception:
pass
return False, (last_body.get("ret") if isinstance(last_body, dict) else None), last_body
def _wechat_send_message(baseurl: str, bot_token: str, to_user_id: str, context_token: str, text: str):
"""发送纯文本回复。"""
return _wechat_post_msg(baseurl, bot_token, to_user_id, context_token,
[{"type": 1, "text_item": {"text": text}}])
def _platform_relay_url(raw_url: str) -> str:
"""把 AGNES 等境外媒体 URL 改写为平台中转地址,使微信/iLink 可内联渲染。
iLink 服务器(中国大陆)直连境外 AGNES 常被墙/超时,微信里只能看到"链接"。
改写为平台中转地址(平台服务器经香港代理可达 AGNES),微信即可内联显示图片/视频。
"""
access = os.environ.get("APP_ACCESS_URL", "http://210.16.161.218:12888").rstrip("/")
return f"{access}/app/api/chat/media/relay?url={_quote(raw_url, safe='')}"
def _wechat_send_media(baseurl: str, bot_token: str, to_user_id: str, context_token: str,
media_type: str, url: str):
"""发送媒体(图片/视频)并在微信内联渲染。
问题:AGNES 媒体 URL 位于境外,iLink(中国大陆服务器)直连常被墙/超时,
微信里只能看到"链接"而无法内联播放。
解决:改写为平台中转地址(经香港代理可达 AGNES),微信即可内联显示图片/视频。
"""
relay_url = _platform_relay_url(url)
if media_type == "image":
media_item = {"type": 2, "image_item": {"url": relay_url}}
caption = "🖼️ 图片已生成(若未直接显示,请点此查看):" + relay_url
else:
media_item = {"type": 5, "video_item": {"url": relay_url}}
caption = "🎬 视频已生成(若未直接播放,请点此查看):" + relay_url
return _wechat_post_msg(baseurl, bot_token, to_user_id, context_token,
[media_item, {"type": 1, "text_item": {"text": caption}}])
def _wechat_extract_text(msg: dict) -> str:
parts = []
for it in (msg.get("item_list") or []):
if it.get("type") == 1 and it.get("text_item", {}).get("text"):
parts.append(it["text_item"]["text"])
if not parts:
for k in ("text", "content", "message", "msg"):
v = msg.get(k)
if isinstance(v, str) and v.strip():
parts.append(v)
return "\n".join(parts)
def _wechat_sender_id(m: dict) -> str:
"""从入站消息中提取发送者 id(兼容多套字段名)。"""
for k in ("from_user_id", "from_user_name", "from_username", "user_id",
"sender_id", "wxid", "from_wxid", "username"):
v = m.get(k)
if v:
return str(v)
return ""
def _wechat_context_token(m: dict) -> str:
"""提取 context_token(回复必须回带,否则不关联到对话窗口)。"""
for k in ("context_token", "ctx_token", "conversation_id", "conv_id",
"msg_id", "context_id", "session_id"):
v = m.get(k)
if v:
return str(v)
return ""
def _log_chat(user_id, platform, from_user, context_token, role, content):
"""持久化 IM Bot 对话历史,便于平台内查阅。"""
db = None
try:
db = SessionLocal()
_execute_raw(db, text(
"INSERT INTO im_bot_chat_log "
"(user_id, platform, from_user, context_token, role, content, created_at) "
"VALUES (:uid, :platform, :fu, :ctx, :role, :content, :created)"
), {
"uid": user_id, "platform": platform, "fu": from_user,
"ctx": context_token, "role": role, "content": content,
"created": _beijing_now(),
})
except Exception:
pass
finally:
if db:
db.close()
# ============================================================
# 镜像到 AI 聊天(标准 conversations/messages)
# 个人微信对话作为一个独立会话窗(默认名 "WeXin Bot")出现在 AI 聊天中,
# 从而复用 AI 聊天模块的全部功能(滚动翻阅、重命名、置顶、删除等)。
# ============================================================
WECHAT_CONV_TITLE = "WeXin Bot"
def _get_or_create_wechat_conv(db: Session, user_id: str) -> str:
"""按 user_id 找/建 source='wechat_bot' 的会话,返回会话 id。"""
rows = _execute_raw(db, text(
"SELECT id FROM conversations WHERE user_id=:uid AND source='wechat_bot' "
"AND status<>'archived' ORDER BY updated_at DESC LIMIT 1"
), {"uid": user_id})
if rows:
return _row_to_dict(rows[0])["id"]
import uuid as _uuid
cid = str(_uuid.uuid4())
now = _beijing_now()
_execute_raw(db, text(
"INSERT INTO conversations "
"(id, user_id, title, status, pinned, knowledge_graph_count, source, created_at, updated_at) "
"VALUES (:id, :uid, :title, 'active', 0, 0, 'wechat_bot', :c, :u)"
), {"id": cid, "uid": user_id, "title": WECHAT_CONV_TITLE, "c": now, "u": now})
return cid
def _append_ai_message(db: Session, conv_id: str, role: str, content: str, created=None):
import uuid as _uuid
_execute_raw(db, text(
"INSERT INTO messages (id, conversation_id, role, content, token_used, created_at) "
"VALUES (:id, :cid, :role, :content, 0, :c)"
), {
"id": str(_uuid.uuid4()), "cid": conv_id, "role": role,
"content": content, "c": created or _beijing_now(),
})
def _mirror_to_ai_chat(user_id: str, user_content: str, assistant_content: str):
"""把一轮微信问答写入用户的 WeXin Bot AI 聊天会话。"""
if not user_id:
return
db = None
try:
db = SessionLocal()
cid = _get_or_create_wechat_conv(db, user_id)
now = _beijing_now()
_append_ai_message(db, cid, "user", user_content, created=now)
# assistant 比 user 晚 50ms,保证排序严格递增,避免同秒时 ORDER BY 乱序
_append_ai_message(db, cid, "assistant", assistant_content,
created=now + timedelta(milliseconds=50))
_execute_raw(db, text("UPDATE conversations SET updated_at=:u WHERE id=:id"),
{"u": _beijing_now(), "id": cid})
except Exception:
pass
finally:
if db:
db.close()
# ============================================================
# 后台接收线程(每用户独立)
# ============================================================
def _mark_wechat_expired(account_id: int):
"""发送会话失效(ret=-14/-2):标记账号需重扫,并清除失效 token。"""
db = None
try:
db = SessionLocal()
_execute_raw(db, text(
"UPDATE im_bot_wechat_account SET status='expired', bot_token=NULL, "
"updated_at=:ts WHERE id=:aid"
), {"aid": account_id, "ts": _beijing_now()})
except Exception:
pass
finally:
if db:
db.close()
def _wechat_receiver_loop(account_id: int):
"""长轮询个人微信消息 → 七虎AI 回复 → 发送。"""
cursor = ""
while True:
db = SessionLocal()
try:
row = _execute_raw(
db,
text("SELECT * FROM im_bot_wechat_account WHERE id = :aid"),
{"aid": account_id},
)
if not row:
return
d = _row_to_dict(row[0])
if d.get("status") != "connected" or not d.get("bot_token"):
return
bot_token = d["bot_token"]
baseurl = d.get("baseurl") or ILINK_BASE
msgs, cursor = _wechat_get_updates(baseurl, bot_token, cursor)
owner_uid = d.get("user_id")
for m in msgs:
if m.get("message_type") == 2:
continue # 仅跳过 bot 自己发出的回声,其余(含后续追问)均尝试处理
content = _wechat_extract_text(m)
if not content:
continue
sender = _wechat_sender_id(m)
ctx = _wechat_context_token(m)
# 诊断:记录原始入站消息,定位字段名 / 第二轮不回复
try:
with open("/tmp/wechat_inbound.log", "a", encoding="utf-8") as _f:
_f.write(_beijing_now().isoformat() + " sender=" + repr(sender)
+ " ctx=" + repr(ctx) + " raw=" + json.dumps(m, ensure_ascii=False) + "\n")
except Exception:
pass
# 先生成回复(此时历史尚未包含本条,避免重复)
reply = _qihua_reply(content, owner_uid)
# 媒体结构化返回:_call_real_llm 生成图片/视频时返回
# {"__media__": True, "media_type": "image"/"video", "url": "..."}
if isinstance(reply, dict) and reply.get("__media__"):
media_type = reply.get("media_type") or "image"
media_url = reply.get("url") or ""
ok, ret, _ = _wechat_send_media(baseurl, bot_token, sender, ctx, media_type, media_url)
# 对话历史里存可渲染的媒体标签(经平台中转,AI 聊天 WeXin Bot 窗与历史页均可内联显示)
relay_url = _platform_relay_url(media_url)
reply_for_log = (
f'
'
if media_type == "image"
else f''
)
else:
reply_text = reply if isinstance(reply, str) else str(reply)
ok, ret, _ = _wechat_send_message(baseurl, bot_token, sender, ctx, reply_text)
reply_for_log = reply_text
if ok:
# 仅发送成功才写对话历史(避免“发了但没收到”的误导记录)
if owner_uid:
_log_chat(owner_uid, "personal_wechat", sender, ctx, "user", content)
_log_chat(owner_uid, "personal_wechat", sender, ctx, "assistant", reply_for_log)
# 同步镜像到 AI 聊天的 WeXin Bot 会话窗
_mirror_to_ai_chat(owner_uid, content, reply_for_log)
else:
try:
with open("/tmp/wechat_send.log", "a", encoding="utf-8") as _f:
_f.write(_beijing_now().isoformat() + " SEND_FAIL ret=%s sender=%s\n"
% (ret, repr(sender)))
except Exception:
pass
if ret in (-2, -14):
# 会话/鉴权失效:标记需重扫但继续循环,让下一轮 poll 自然检测过期
_mark_wechat_expired(d["id"])
time.sleep(1)
except Exception:
time.sleep(5)
finally:
db.close()
def _start_wechat_receiver(account_id: int):
if account_id in _wechat_receiver_threads and _wechat_receiver_threads[account_id].is_alive():
return
t = threading.Thread(target=_wechat_receiver_loop, args=(account_id,), daemon=True)
t.start()
_wechat_receiver_threads[account_id] = t
# ============================================================
# 前台用户接口(按 user_id 隔离)
# ============================================================
@router.get("/status")
async def wechat_status(user: User = Depends(get_current_user)):
"""当前登录用户的个人微信连接状态"""
db = SessionLocal()
try:
row = _wechat_row_for_user(db, user.id)
if not row:
return {"success": True, "data": {"connected": False, "configured": False}}
d = _row_to_dict(row)
return {"success": True, "data": {
"connected": d.get("status") == "connected",
"status": d.get("status"),
"configured": True,
"baseurl": d.get("baseurl"),
"ilink_bot_id": d.get("ilink_bot_id"),
"username": d.get("username"),
"connected_at": str(d.get("connected_at")) if d.get("connected_at") else None,
}}
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
finally:
db.close()
@router.post("/qrcode")
async def wechat_qrcode(user: User = Depends(get_current_user)):
"""获取个人微信扫码登录二维码(iLink 协议,免 AppID)"""
try:
resp = requests.get(
ILINK_BASE + "/ilink/bot/get_bot_qrcode",
params={"bot_type": 3},
timeout=20,
)
if not resp.ok:
return {"success": False, "error": f"获取二维码失败 HTTP {resp.status_code}"}
j = resp.json() if resp.content else {}
qr_url = j.get("qrcode_img_content") or ""
qr_img = None
try:
import io as _io
import qrcode as _qrcode
_buf = _io.BytesIO()
_q = _qrcode.QRCode(box_size=10, border=2)
_q.add_data(qr_url)
_q.make(fit=True)
_img = _q.make_image(fill_color="black", back_color="white")
_img.save(_buf, format="PNG")
qr_img = "data:image/png;base64," + base64.b64encode(_buf.getvalue()).decode()
except Exception:
qr_img = None
return {"success": True, "data": {
"qrcode": j.get("qrcode"),
"qrcode_img_content": qr_url,
"qrcode_img": qr_img,
}}
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@router.post("/qrcode/status")
async def wechat_qrcode_status(
qrcode: str = Query(...),
user: User = Depends(get_current_user),
):
"""轮询扫码状态;confirmed 时保存该用户的 bot_token 并启动接收线程"""
try:
resp = requests.get(
ILINK_BASE + "/ilink/bot/get_qrcode_status",
params={"qrcode": qrcode},
timeout=40,
)
if not resp.ok:
return {"success": False, "error": f"查询状态失败 HTTP {resp.status_code}"}
j = resp.json() if resp.content else {}
data = j.get("data", j) if isinstance(j, dict) else j
status = data.get("status")
if status == "confirmed":
creds = data.get("credentials", {}) or {}
bot_token = data.get("bot_token") or creds.get("bot_token")
baseurl = data.get("baseurl") or creds.get("baseurl") or ILINK_BASE
ilink_bot_id = data.get("ilink_bot_id") or creds.get("ilink_bot_id")
db = SessionLocal()
try:
row = _wechat_row_for_user(db, user.id)
if not row:
_execute_raw(db, text("""INSERT INTO im_bot_wechat_account
(user_id, username, bot_token, ilink_bot_id, baseurl, status,
connected_at, created_at, updated_at)
VALUES (:user_id, :username, :bot_token, :ilink_bot_id, :baseurl,
'connected', :connected_at, :created_at, :updated_at)"""),
{
"user_id": user.id, "username": user.username,
"bot_token": bot_token, "ilink_bot_id": ilink_bot_id,
"baseurl": baseurl,
# 统一按北京时间存储,避免容器 UTC 导致显示偏差
"connected_at": _beijing_now(),
"created_at": _beijing_now(), "updated_at": _beijing_now(),
})
else:
_execute_raw(db, text("""UPDATE im_bot_wechat_account SET
user_id=:user_id, username=:username, bot_token=:bot_token,
ilink_bot_id=:ilink_bot_id, baseurl=:baseurl,
status='connected', connected_at=:connected_at, updated_at=:updated_at
WHERE id=:id"""),
{
"user_id": user.id, "username": user.username,
"bot_token": bot_token, "ilink_bot_id": ilink_bot_id,
"baseurl": baseurl,
"connected_at": _beijing_now(),
"updated_at": _beijing_now(), "id": _row_to_dict(row)["id"],
})
row = _wechat_row_for_user(db, user.id)
aid = _row_to_dict(row)["id"] if row else 1
_start_wechat_receiver(aid)
finally:
db.close()
return {"success": True, "data": {"status": "confirmed", "connected": True}}
return {"success": True, "data": {"status": status, "connected": False}}
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@router.post("/disconnect")
async def wechat_disconnect(user: User = Depends(get_current_user)):
"""断开当前用户的个人微信连接(清除 bot_token)"""
db = SessionLocal()
try:
row = _wechat_row_for_user(db, user.id)
if row:
_execute_raw(db, text(
"UPDATE im_bot_wechat_account SET bot_token=NULL, status='disconnected', "
"updated_at=:updated_at WHERE id=:id"
), {"updated_at": _beijing_now(), "id": _row_to_dict(row)["id"]})
return {"success": True, "message": "已断开个人微信连接"}
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
finally:
db.close()
# ============================================================
# 启动恢复:为所有已连接用户账号拉起接收线程
# ============================================================
@router.get("/history")
async def wechat_history(
user: User = Depends(get_current_user),
limit: int = Query(200),
):
"""查询当前用户通过个人微信与七虎AI 的对话历史"""
db = SessionLocal()
try:
rows = _execute_raw(
db,
text("SELECT id, role, content, platform, from_user, created_at "
"FROM im_bot_chat_log WHERE user_id=:uid ORDER BY id DESC LIMIT :lim"),
{"uid": user.id, "lim": min(int(limit), 500)},
)
items = [_row_to_dict(r) for r in rows]
items.reverse()
return {"success": True, "data": {"messages": items, "total": len(items)}}
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
finally:
db.close()
def start_all_wechat_receivers():
"""在应用启动时调用,恢复此前已连接用户的个人微信接收线程。"""
db = SessionLocal()
try:
rows = _execute_raw(
db,
text("SELECT id FROM im_bot_wechat_account WHERE status='connected' AND user_id IS NOT NULL"),
)
for r in rows:
_start_wechat_receiver(r._mapping["id"])
if rows:
import logging
logging.getLogger("aiflow").info(f"[IM个人微信] 已恢复 {len(rows)} 个用户微信接收线程")
except Exception as e:
import logging
logging.getLogger("aiflow").warning(f"[IM个人微信] 启动恢复失败(不影响服务): {e}")
finally:
db.close()