394 lines
15 KiB
Python
394 lines
15 KiB
Python
from fastapi import APIRouter, Request, Response, Query
|
|
from app.core.wecombot import WecomBot
|
|
from app.models.database import get_db
|
|
from app.models.community import CommunityDB
|
|
from app.core.config import settings
|
|
import xml.etree.ElementTree as ET
|
|
import logging
|
|
import hashlib
|
|
import base64
|
|
from Crypto.Cipher import AES
|
|
import typing
|
|
import struct
|
|
from app.core.wecomclient import wecom_client
|
|
from app.models.user import UserDB
|
|
from app.api.deps import get_current_user
|
|
from fastapi import Depends
|
|
from sqlalchemy.orm import Session
|
|
from app.core.response import error_response, success_response, ResponseModel
|
|
from pydantic import BaseModel
|
|
from app.models.wecom_external_chat import WecomExternalChatDB, WecomExternalChatInfo, WecomExternalChatMemberDB, WecomExternalChatMemberInfo
|
|
|
|
router = APIRouter()
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
|
|
|
|
def decrypt_msg(msg_encrypt: str, signature: str, timestamp: str, nonce: str) -> typing.Optional[str]:
|
|
"""解密企业微信消息"""
|
|
try:
|
|
token = settings.WECHAT_CORP_TOKEN
|
|
encoding_aes_key = settings.WECHAT_CORP_ENCODING_AES_KEY
|
|
corpid = settings.WECHAT_CORP_ID
|
|
|
|
# 1. 验证签名
|
|
sort_list = [token, timestamp, nonce, msg_encrypt]
|
|
sort_list.sort()
|
|
sha1 = hashlib.sha1()
|
|
sha1.update("".join(sort_list).encode())
|
|
calc_signature = sha1.hexdigest()
|
|
if calc_signature != signature:
|
|
return None
|
|
|
|
# 2. 解密消息
|
|
aes_key = base64.b64decode(encoding_aes_key + "=")
|
|
aes = AES.new(aes_key, AES.MODE_CBC, aes_key[:16])
|
|
|
|
# 解密
|
|
text = base64.b64decode(msg_encrypt)
|
|
decrypted_text = aes.decrypt(text)
|
|
|
|
# 去除补位
|
|
pad = decrypted_text[-1]
|
|
if not isinstance(pad, int):
|
|
pad = ord(pad)
|
|
content = decrypted_text[:-pad]
|
|
|
|
# 验证corpid
|
|
xml_len = struct.unpack('!I', content[16:20])[0]
|
|
xml_content = content[20 : 20 + xml_len]
|
|
from_corpid = content[20 + xml_len:]
|
|
|
|
if from_corpid.decode() != corpid:
|
|
return None
|
|
|
|
return xml_content.decode()
|
|
|
|
except Exception as e:
|
|
logging.exception("解密企业微信消息失败")
|
|
return None
|
|
|
|
@router.get("")
|
|
async def verify_callback(
|
|
msg_signature: str = Query(..., description="签名"),
|
|
timestamp: str = Query(..., description="时间戳"),
|
|
nonce: str = Query(..., description="随机数"),
|
|
echostr: str = Query(..., description="随机字符串")
|
|
):
|
|
"""验证回调配置"""
|
|
try:
|
|
# 解密echostr
|
|
decrypted_str = decrypt_msg(echostr, msg_signature, timestamp, nonce)
|
|
if not decrypted_str:
|
|
return Response(status_code=403)
|
|
|
|
return Response(content=decrypted_str, media_type="text/plain")
|
|
|
|
except Exception as e:
|
|
logging.exception("验证回调配置失败")
|
|
return Response(status_code=403)
|
|
|
|
@router.post("")
|
|
async def wechat_corp_callback(
|
|
request: Request,
|
|
msg_signature: str = Query(..., description="签名"),
|
|
timestamp: str = Query(..., description="时间戳"),
|
|
nonce: str = Query(..., description="随机数")
|
|
):
|
|
"""处理企业微信回调消息"""
|
|
try:
|
|
# 读取原始XML数据
|
|
body = await request.body()
|
|
body_str = body.decode()
|
|
|
|
# 解析XML获取加密消息
|
|
root = ET.fromstring(body_str)
|
|
encrypt_msg = root.find('Encrypt').text
|
|
|
|
# 解密消息
|
|
decrypted_msg = decrypt_msg(encrypt_msg, msg_signature, timestamp, nonce)
|
|
if not decrypted_msg:
|
|
return Response(content="success", media_type="text/plain")
|
|
|
|
# 解析解密后的XML
|
|
msg_root = ET.fromstring(decrypted_msg)
|
|
|
|
logging.info(f"企业微信回调消息：{decrypted_msg}")
|
|
|
|
# 解析基本信息
|
|
msg_type = msg_root.find('MsgType').text
|
|
|
|
logging.info(f"msg_type: {msg_type}")
|
|
|
|
# 处理事件消息
|
|
if msg_type == 'event':
|
|
event = msg_root.find('Event').text
|
|
|
|
logging.info(f"event: {event}")
|
|
|
|
# 处理外部群聊变更事件
|
|
if event == 'change_external_chat':
|
|
chat_id = msg_root.find('ChatId').text
|
|
change_type = msg_root.find('ChangeType').text
|
|
update_detail = msg_root.find('UpdateDetail').text
|
|
|
|
join_user_id = None
|
|
if update_detail == 'add_member' and msg_root.find('JoinScene') is not None:
|
|
logging.info(f"有新成员加入群聊")
|
|
# 获取加入的成员ID
|
|
join_user_id_elem = msg_root.find('JoinUserID')
|
|
if join_user_id_elem is not None:
|
|
join_user_id = join_user_id_elem.text
|
|
|
|
logging.info(f"chat_id: {chat_id}, change_type: {change_type}, update_detail: {update_detail}, join_user_id: {join_user_id}")
|
|
|
|
# 处理群聊变更事件
|
|
await wecom_client.handle_chat_change_event(
|
|
chat_id=chat_id,
|
|
change_type=change_type,
|
|
update_detail=update_detail,
|
|
join_user_id=join_user_id
|
|
)
|
|
|
|
if update_detail == 'add_member' and join_user_id:
|
|
logging.info(f"发送欢迎消息到群聊：{chat_id}")
|
|
await wecom_client.send_welcome_message(chat_id)
|
|
|
|
return Response(content="success", media_type="text/plain")
|
|
|
|
except Exception as e:
|
|
logging.exception("处理企业微信回调消息异常")
|
|
return Response(content="success", media_type="text/plain")
|
|
|
|
class UnionidToExternalUseridRequest(BaseModel):
|
|
unionid: str
|
|
openid: str
|
|
|
|
@router.post("/unionid_to_external_userid")
|
|
async def unionid_to_external_userid(
|
|
request: UnionidToExternalUseridRequest
|
|
):
|
|
"""根据unionid获取external_userid"""
|
|
result = await wecom_client.unionid_to_external_userid(request.unionid, request.openid)
|
|
print(f"根据unionid获取external_userid结果: {result}")
|
|
return success_response(message="获取external_userid成功", data=result)
|
|
|
|
@router.get("/external-chats", response_model=ResponseModel)
|
|
async def get_external_chats(
|
|
db: Session = Depends(get_db),
|
|
current_user: UserDB = Depends(get_current_user)
|
|
):
|
|
"""获取企业微信外部群聊列表"""
|
|
try:
|
|
# 检查是否为管理员
|
|
if current_user.userid != settings.PLATFORM_USER_ID:
|
|
return error_response(code=403, message="权限不足")
|
|
|
|
# 获取群聊列表
|
|
chats = db.query(WecomExternalChatDB).filter(
|
|
WecomExternalChatDB.is_active == True
|
|
).order_by(WecomExternalChatDB.update_time.desc()).all()
|
|
|
|
# 转换为Pydantic模型
|
|
chat_list = [WecomExternalChatInfo.model_validate(chat) for chat in chats]
|
|
|
|
return success_response(message="获取群聊列表成功", data=chat_list)
|
|
except Exception as e:
|
|
logging.exception("获取群聊列表异常")
|
|
return error_response(code=500, message=f"获取群聊列表失败: {str(e)}")
|
|
|
|
@router.get("/external-chats/{chat_id}/members", response_model=ResponseModel)
|
|
async def get_external_chat_members(
|
|
chat_id: str,
|
|
db: Session = Depends(get_db),
|
|
current_user: UserDB = Depends(get_current_user)
|
|
):
|
|
"""获取企业微信外部群聊成员列表"""
|
|
try:
|
|
# 检查是否为管理员
|
|
if current_user.userid != settings.PLATFORM_USER_ID:
|
|
return error_response(code=403, message="权限不足")
|
|
|
|
# 检查群聊是否存在
|
|
chat = db.query(WecomExternalChatDB).filter(
|
|
WecomExternalChatDB.chat_id == chat_id
|
|
).first()
|
|
|
|
if not chat:
|
|
return error_response(code=404, message="群聊不存在")
|
|
|
|
# 获取成员列表
|
|
members = db.query(WecomExternalChatMemberDB).filter(
|
|
WecomExternalChatMemberDB.chat_id == chat_id
|
|
).order_by(WecomExternalChatMemberDB.join_time.desc()).all()
|
|
|
|
# 转换为Pydantic模型
|
|
member_list = [WecomExternalChatMemberInfo.model_validate(member) for member in members]
|
|
|
|
return success_response(message="获取群聊成员列表成功", data=member_list)
|
|
except Exception as e:
|
|
logging.exception("获取群聊成员列表异常")
|
|
return error_response(code=500, message=f"获取群聊成员列表失败: {str(e)}")
|
|
|
|
@router.post("/sync-chat/{chat_id}", response_model=ResponseModel)
|
|
async def sync_external_chat(
|
|
chat_id: str,
|
|
db: Session = Depends(get_db),
|
|
current_user: UserDB = Depends(get_current_user)
|
|
):
|
|
"""同步企业微信外部群聊信息"""
|
|
try:
|
|
# 检查是否为管理员
|
|
if current_user.userid != settings.PLATFORM_USER_ID:
|
|
return error_response(code=403, message="权限不足")
|
|
|
|
# 获取群聊信息
|
|
result = await wecom_client.handle_chat_change_event(
|
|
chat_id=chat_id,
|
|
change_type="create",
|
|
update_detail=""
|
|
)
|
|
|
|
if not result:
|
|
return error_response(code=500, message="同步群聊信息失败")
|
|
|
|
return success_response(message="同步群聊信息成功")
|
|
except Exception as e:
|
|
logging.exception("同步群聊信息异常")
|
|
return error_response(code=500, message=f"同步群聊信息失败: {str(e)}")
|
|
|
|
@router.get("/chat-dashboard")
|
|
async def chat_dashboard(
|
|
current_user: UserDB = Depends(get_current_user),
|
|
db: Session = Depends(get_db)
|
|
):
|
|
"""显示企业微信外部群聊信息的HTML页面"""
|
|
# 检查是否为管理员
|
|
if current_user.userid != settings.PLATFORM_USER_ID:
|
|
return Response(content="权限不足", media_type="text/html")
|
|
|
|
# 获取群聊列表
|
|
chats = db.query(WecomExternalChatDB).filter(
|
|
WecomExternalChatDB.is_active == True
|
|
).order_by(WecomExternalChatDB.update_time.desc()).all()
|
|
|
|
# 生成HTML
|
|
html = """
|
|
<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<title>企业微信外部群聊信息</title>
|
|
<meta charset="utf-8">
|
|
<meta name="viewport" content="width=device-width, initial-scale=1">
|
|
<style>
|
|
body { font-family: Arial, sans-serif; margin: 0; padding: 20px; }
|
|
h1, h2 { color: #333; }
|
|
table { border-collapse: collapse; width: 100%; margin-bottom: 20px; }
|
|
th, td { border: 1px solid #ddd; padding: 8px; text-align: left; }
|
|
th { background-color: #f2f2f2; }
|
|
tr:nth-child(even) { background-color: #f9f9f9; }
|
|
.container { max-width: 1200px; margin: 0 auto; }
|
|
.btn {
|
|
display: inline-block;
|
|
padding: 6px 12px;
|
|
margin-bottom: 0;
|
|
font-size: 14px;
|
|
font-weight: 400;
|
|
line-height: 1.42857143;
|
|
text-align: center;
|
|
white-space: nowrap;
|
|
vertical-align: middle;
|
|
cursor: pointer;
|
|
background-image: none;
|
|
border: 1px solid transparent;
|
|
border-radius: 4px;
|
|
color: #fff;
|
|
background-color: #337ab7;
|
|
text-decoration: none;
|
|
}
|
|
.btn:hover { background-color: #286090; }
|
|
</style>
|
|
<script>
|
|
function loadMembers(chatId) {
|
|
fetch(`/api/wecom/external-chats/${chatId}/members`)
|
|
.then(response => response.json())
|
|
.then(data => {
|
|
if (data.code === 0) {
|
|
const members = data.data;
|
|
let html = '<table>';
|
|
html += '<tr><th>ID</th><th>用户ID</th><th>类型</th><th>姓名</th><th>加入时间</th><th>是否已发送欢迎</th></tr>';
|
|
|
|
members.forEach(member => {
|
|
html += `<tr>
|
|
<td>${member.id}</td>
|
|
<td>${member.user_id}</td>
|
|
<td>${member.type}</td>
|
|
<td>${member.name || '未知'}</td>
|
|
<td>${new Date(member.join_time).toLocaleString()}</td>
|
|
<td>${member.welcome_sent ? '是' : '否'}</td>
|
|
</tr>`;
|
|
});
|
|
|
|
html += '</table>';
|
|
document.getElementById('members-' + chatId).innerHTML = html;
|
|
} else {
|
|
alert('获取成员列表失败: ' + data.message);
|
|
}
|
|
})
|
|
.catch(error => {
|
|
console.error('Error:', error);
|
|
alert('获取成员列表失败');
|
|
});
|
|
}
|
|
|
|
function syncChat(chatId) {
|
|
fetch(`/api/wecom/sync-chat/${chatId}`, {
|
|
method: 'POST'
|
|
})
|
|
.then(response => response.json())
|
|
.then(data => {
|
|
if (data.code === 0) {
|
|
alert('同步成功');
|
|
location.reload();
|
|
} else {
|
|
alert('同步失败: ' + data.message);
|
|
}
|
|
})
|
|
.catch(error => {
|
|
console.error('Error:', error);
|
|
alert('同步失败');
|
|
});
|
|
}
|
|
</script>
|
|
</head>
|
|
<body>
|
|
<div class="container">
|
|
<h1>企业微信外部群聊信息</h1>
|
|
"""
|
|
|
|
if not chats:
|
|
html += "<p>暂无群聊信息</p>"
|
|
else:
|
|
for chat in chats:
|
|
html += f"""
|
|
<div style="margin-bottom: 30px; border: 1px solid #ddd; padding: 15px; border-radius: 5px;">
|
|
<h2>{chat.name or '未命名群聊'} ({chat.chat_id})</h2>
|
|
<p>
|
|
<strong>创建时间:</strong> {chat.create_time.strftime('%Y-%m-%d %H:%M:%S')}<br>
|
|
<strong>更新时间:</strong> {chat.update_time.strftime('%Y-%m-%d %H:%M:%S') if chat.update_time else '无'}<br>
|
|
<strong>成员数量:</strong> {chat.member_count}<br>
|
|
<strong>群主:</strong> {chat.owner or '未知'}<br>
|
|
<strong>公告:</strong> {chat.notice or '无'}<br>
|
|
</p>
|
|
<button class="btn" onclick="loadMembers('{chat.chat_id}')">查看成员</button>
|
|
<button class="btn" onclick="syncChat('{chat.chat_id}')">同步群信息</button>
|
|
<div id="members-{chat.chat_id}" style="margin-top: 15px;"></div>
|
|
</div>
|
|
"""
|
|
|
|
html += """
|
|
</div>
|
|
</body>
|
|
</html>
|
|
"""
|
|
|
|
return Response(content=html, media_type="text/html") |