QihuAI Design
acde481bed
One-time alignment to close drift between the running container (qihu-aiflow / image qihuaiflow:with-deps) and this Git repository. Container hotfixes previously persisted only via `docker commit` are now captured in Git so Git becomes the single source of truth. Backend: - chat_service.py: accumulated fixes incl. EigenFlux README guidance - chat_context.py / chat_media.py: URL/web-repo context injection - api/url_reader.py: NEW SSRF-safe web/README reader (Issue #725) - skills_admin.py & Skills.tsx: 14-scene capability unification - pgvector_search.py, orchestrator.py, platform_knowledge.py, market.py, admin/models.py: drift sync Frontend: - ChatPage.tsx: collapsible AI think/act block (Issue #724) - MarketPage / EvolutionDashboard / TokenDashboard styling sync Migrations: - 018_normalize_marketapp_scene.sql: fixed capability allowlist (prevents capability wipe on restart) Alignment performed 2026-07-24 07:44:19. No deletions; secrets/venv/node_modules/ caches were not touched (tracked-source-only copy).
665 lines
27 KiB
Python
665 lines
27 KiB
Python
"""
|
||
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'<img src="{relay_url}" alt="生成图片"/>'
|
||
if media_type == "image"
|
||
else f'<video src="{relay_url}" controls></video>'
|
||
)
|
||
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()
|