ee5c1ebadc
不再用图片文件大小、文件名等回退逻辑去“猜参数”。 发送接口不再“先返回发送中”，而是直接调用 SendEmoji 并等待结果： 成功才返回 表情发送成功 失败直接返回具体错误（不会再假成功）
300 lines
12 KiB
Python
300 lines
12 KiB
Python
import requests
|
|
import html
|
|
import re
|
|
from flask import Blueprint, render_template, jsonify, request, current_app, Response, stream_with_context
|
|
from .auth import login_required
|
|
from loguru import logger
|
|
import xml.etree.ElementTree as ET
|
|
from datetime import datetime
|
|
from utils.message_formatter import format_quote_message, parse_quote_message
|
|
|
|
# 创建消息管理蓝图
|
|
messages_bp = Blueprint('messages', __name__)
|
|
logger = logger
|
|
|
|
|
|
def _is_emoji_message(msg: dict) -> bool:
|
|
message_type = str(msg.get('message_type', ''))
|
|
content = msg.get('content') or ''
|
|
xml_markers = ('<emoji', '<emoticonmd5>', '<msg><emoji', '<gameext')
|
|
return message_type in {'47', '1048625', '1090519089'} or any(marker in content for marker in xml_markers)
|
|
|
|
|
|
def _is_usable_local_media_path(value: str) -> bool:
|
|
if not value:
|
|
return False
|
|
value = str(value).strip()
|
|
if value.startswith(('http://', 'https://')):
|
|
return True
|
|
if 'static/images' in value or 'static\\images' in value:
|
|
return True
|
|
if '/' in value or '\\' in value:
|
|
return True
|
|
return False
|
|
|
|
|
|
def _extract_emoji_preview_url(xml_text: str) -> str:
|
|
if not xml_text:
|
|
return ''
|
|
patterns = (
|
|
r'cdnurl\s*=\s*["\']([^"\']+)["\']',
|
|
r'encrypturl\s*=\s*["\']([^"\']+)["\']',
|
|
r'externurl\s*=\s*["\']([^"\']+)["\']'
|
|
)
|
|
for pattern in patterns:
|
|
match = re.search(pattern, xml_text)
|
|
if match:
|
|
return html.unescape(match.group(1))
|
|
return ''
|
|
|
|
|
|
def _parse_sysmsg_payload(content: str) -> dict:
|
|
payload = {
|
|
"sysmsg_type": "",
|
|
"summary": "",
|
|
"replace_msg": "",
|
|
"session": "",
|
|
"msgid": "",
|
|
"newmsgid": "",
|
|
}
|
|
text = str(content or "").strip()
|
|
if not text.startswith("<sysmsg"):
|
|
return payload
|
|
|
|
try:
|
|
root = ET.fromstring(text)
|
|
except Exception:
|
|
payload["summary"] = text
|
|
return payload
|
|
|
|
payload["sysmsg_type"] = str(root.attrib.get("type", "")).strip()
|
|
if payload["sysmsg_type"] == "revokemsg":
|
|
revoke_node = root.find("revokemsg")
|
|
if revoke_node is not None:
|
|
payload["session"] = str(revoke_node.findtext("session", "") or "").strip()
|
|
payload["msgid"] = str(revoke_node.findtext("msgid", "") or "").strip()
|
|
payload["newmsgid"] = str(revoke_node.findtext("newmsgid", "") or "").strip()
|
|
payload["replace_msg"] = str(revoke_node.findtext("replacemsg", "") or "").strip()
|
|
payload["summary"] = payload["replace_msg"] or "撤回了一条消息"
|
|
return payload
|
|
|
|
payload["summary"] = str(root.findtext(".//content", "") or "").strip() or text
|
|
return payload
|
|
|
|
|
|
def _proxy_remote_media(target_url: str) -> Response:
|
|
if not target_url:
|
|
return Response("missing url", status=400)
|
|
|
|
headers = {
|
|
"User-Agent": (
|
|
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
|
|
"AppleWebKit/537.36 (KHTML, like Gecko) "
|
|
"Chrome/123.0.0.0 Safari/537.36"
|
|
),
|
|
"Referer": "http://weixin.qq.com/"
|
|
}
|
|
range_header = request.headers.get("Range")
|
|
if range_header:
|
|
headers["Range"] = range_header
|
|
|
|
upstream = requests.get(target_url, headers=headers, stream=True, timeout=30)
|
|
|
|
response_headers = {}
|
|
for key in ("Content-Type", "Content-Length", "Content-Range", "Accept-Ranges", "Cache-Control", "ETag", "Last-Modified"):
|
|
value = upstream.headers.get(key)
|
|
if value:
|
|
response_headers[key] = value
|
|
|
|
if "Accept-Ranges" not in response_headers:
|
|
response_headers["Accept-Ranges"] = "bytes"
|
|
|
|
return Response(
|
|
stream_with_context(upstream.iter_content(chunk_size=64 * 1024)),
|
|
status=upstream.status_code,
|
|
headers=response_headers,
|
|
direct_passthrough=True
|
|
)
|
|
|
|
|
|
# 消息列表页面
|
|
@messages_bp.route('/messages')
|
|
@login_required
|
|
def message_list_page():
|
|
"""消息列表页面"""
|
|
return render_template('message_list.html')
|
|
|
|
|
|
# API路由
|
|
@messages_bp.route('/api/messages', methods=['GET'])
|
|
@login_required
|
|
def get_messages():
|
|
"""获取消息列表API"""
|
|
try:
|
|
server = current_app.dashboard_server
|
|
# 获取查询参数
|
|
group_id = request.args.get('group_id')
|
|
start_date = request.args.get('start_date', datetime.now().strftime('%Y-%m-%d'))
|
|
end_date = request.args.get('end_date', datetime.now().strftime('%Y-%m-%d'))
|
|
search_text = request.args.get('search_text')
|
|
page = int(request.args.get('page', 1))
|
|
page_size = int(request.args.get('page_size', 20))
|
|
|
|
# 调用数据库方法获取消息
|
|
result = server.message_storage.get_messages_by_filter(
|
|
group_id=group_id,
|
|
start_date=start_date,
|
|
end_date=end_date,
|
|
search_text=search_text,
|
|
page=page,
|
|
page_size=page_size
|
|
)
|
|
|
|
# 处理消息数据，添加群组名称和发送者昵称，并格式化引用消息
|
|
for msg in result['messages']:
|
|
raw_content = str(msg.get('content') or '')
|
|
# 获取群组名称
|
|
msg['group_name'] = server.contact_manager.get_nickname(msg['group_id']) or msg['group_id']
|
|
|
|
# 获取发送者昵称
|
|
msg['sender_name'] = server.contact_manager.get_group_name(msg['group_id'], msg['sender']) or msg['sender']
|
|
|
|
if raw_content.startswith("<sysmsg"):
|
|
sysmsg_payload = _parse_sysmsg_payload(raw_content)
|
|
msg['content'] = sysmsg_payload.get('summary') or '[系统消息]'
|
|
msg['message_xml'] = raw_content
|
|
msg['sysmsg_type'] = sysmsg_payload.get('sysmsg_type', '')
|
|
msg['sysmsg_summary'] = sysmsg_payload.get('summary', '')
|
|
msg['sysmsg_replace_msg'] = sysmsg_payload.get('replace_msg', '')
|
|
msg['sysmsg_session'] = sysmsg_payload.get('session', '')
|
|
msg['sysmsg_msgid'] = sysmsg_payload.get('msgid', '')
|
|
msg['sysmsg_newmsgid'] = sysmsg_payload.get('newmsgid', '')
|
|
continue
|
|
|
|
if _is_emoji_message(msg):
|
|
msg['content'] = '[表情]'
|
|
msg['emoji_preview_url'] = msg.get('image_path') or _extract_emoji_preview_url(msg.get('attachment_url', ''))
|
|
continue
|
|
|
|
# 处理消息内容，格式化引用消息
|
|
if str(msg.get('message_type')) == "49" and msg.get('content'): # 应用消息类型
|
|
try:
|
|
# 检查是否为引用消息
|
|
if '<refermsg>' in msg['content']:
|
|
# 使用格式化工具处理引用消息
|
|
quote_data = parse_quote_message(msg['content'])
|
|
msg['content'] = quote_data.get('formatted_message') or format_quote_message(msg['content'])
|
|
msg['quoted_type'] = quote_data.get('reference_type', '')
|
|
msg['quoted_preview_image'] = quote_data.get('preview_image', '')
|
|
msg['quoted_preview_video_thumb'] = quote_data.get('preview_video_thumb', '')
|
|
msg['quoted_reference_svrid'] = quote_data.get('reference_svrid', '')
|
|
msg['quoted_reference_md5'] = quote_data.get('reference_md5', '')
|
|
|
|
# 优先使用原始被引用消息自己已落库的图片/缩略图地址
|
|
reference_svrid = quote_data.get('reference_svrid')
|
|
reference_md5 = quote_data.get('reference_md5')
|
|
referenced_msg = None
|
|
if reference_svrid:
|
|
referenced_msg = server.message_storage.get_message_by_message_id(reference_svrid)
|
|
if not referenced_msg and reference_md5 and msg['quoted_type'] == 'image':
|
|
referenced_msg = server.message_storage.get_image_message_by_md5(reference_md5)
|
|
|
|
if referenced_msg:
|
|
if msg['quoted_type'] == 'image':
|
|
image_path = referenced_msg.get('image_path') or ''
|
|
msg['quoted_preview_image'] = image_path if _is_usable_local_media_path(image_path) else ''
|
|
elif msg['quoted_type'] == 'video':
|
|
video_thumb = referenced_msg.get('message_thumb') or referenced_msg.get('image_path') or ''
|
|
msg['quoted_preview_video_thumb'] = video_thumb if _is_usable_local_media_path(video_thumb) else ''
|
|
else:
|
|
# 其他类型的应用消息，解析 XML 提取标题
|
|
root = ET.fromstring(msg['content'])
|
|
title_elem = root.find('.//title')
|
|
if title_elem is not None:
|
|
msg['content'] = title_elem.text
|
|
except Exception as e:
|
|
logger.error(f"解析消息类型49出错: {e}")
|
|
try:
|
|
msg['content'] = format_quote_message(msg['content'])
|
|
except Exception:
|
|
pass
|
|
|
|
return jsonify(result)
|
|
except Exception as e:
|
|
logger.error(f"获取消息列表失败: {e}")
|
|
return jsonify({'error': str(e)}), 500
|
|
|
|
|
|
@messages_bp.route('/api/groups', methods=['GET'])
|
|
@login_required
|
|
def get_groups():
|
|
"""获取群组列表API"""
|
|
try:
|
|
server = current_app.dashboard_server
|
|
# 获取机器人管理的群组列表
|
|
groups = []
|
|
for group_id in server.contact_manager.get_contacts():
|
|
if '@chatroom' in group_id:
|
|
groups.append({
|
|
'group_id': group_id,
|
|
'group_name': server.contact_manager.get_nickname(group_id) or group_id
|
|
})
|
|
|
|
return jsonify({'groups': groups})
|
|
except Exception as e:
|
|
logger.error(f"获取群组列表失败: {e}")
|
|
return jsonify({'error': str(e)}), 500
|
|
|
|
|
|
@messages_bp.route('/api/hourly_message_trend', methods=['GET'])
|
|
@login_required
|
|
def get_hourly_message_trend():
|
|
"""获取按小时聊天趋势数据API"""
|
|
try:
|
|
server = current_app.dashboard_server
|
|
# 获取查询参数
|
|
group_id = request.args.get('group_id')
|
|
days = int(request.args.get('days', 1))
|
|
|
|
# 调用数据库方法获取按小时消息趋势数据
|
|
trend_data = server.message_storage.get_hourly_message_trend(group_id=group_id, days=days)
|
|
|
|
# 格式化数据为前端需要的格式
|
|
hours = []
|
|
counts = []
|
|
for item in trend_data:
|
|
hours.append(item['hour_slot'])
|
|
counts.append(item['message_count'])
|
|
|
|
# 获取群组名称
|
|
group_name = server.contact_manager.get_nickname(group_id) if group_id else "所有群组"
|
|
|
|
return jsonify({
|
|
'success': True,
|
|
'data': {
|
|
'hours': hours,
|
|
'counts': counts,
|
|
'group_name': group_name
|
|
}
|
|
})
|
|
except Exception as e:
|
|
logger.error(f"获取按小时聊天趋势数据失败: {e}")
|
|
return jsonify({'success': False, 'error': str(e)}), 500
|
|
|
|
|
|
@messages_bp.route('/api/messages/media_proxy', methods=['GET'])
|
|
@login_required
|
|
def message_media_proxy():
|
|
target_url = request.args.get('url', '').strip()
|
|
if not target_url:
|
|
return Response("missing url", status=400)
|
|
if target_url.startswith("<"):
|
|
return Response("invalid media url", status=400)
|
|
if not re.match(r"^https?://", target_url, flags=re.IGNORECASE):
|
|
return Response("unsupported media url", status=400)
|
|
try:
|
|
return _proxy_remote_media(target_url)
|
|
except Exception as e:
|
|
logger.error(f"消息媒体代理失败: {e}")
|
|
return Response(f"proxy failed: {e}", status=502)
|