193 lines
6.9 KiB
Python
193 lines
6.9 KiB
Python
import json
|
||
from typing import Union, Any, List
|
||
|
||
from fastapi import APIRouter, BackgroundTasks, Depends
|
||
from fastapi import Request
|
||
from pywebpush import WebPushException, webpush
|
||
from sqlalchemy.orm import Session
|
||
from starlette.responses import PlainTextResponse
|
||
|
||
from app import schemas
|
||
from app.chain.message import MessageChain
|
||
from app.core.config import settings, global_vars
|
||
from app.core.security import verify_token
|
||
from app.db import get_db
|
||
from app.db.models import User
|
||
from app.db.models.message import Message
|
||
from app.db.systemconfig_oper import SystemConfigOper
|
||
from app.db.userauth import get_current_active_superuser
|
||
from app.log import logger
|
||
from app.modules.wechat.WXBizMsgCrypt3 import WXBizMsgCrypt
|
||
from app.schemas import NotificationSwitch
|
||
from app.schemas.types import SystemConfigKey, NotificationType, MessageChannel
|
||
|
||
router = APIRouter()
|
||
|
||
|
||
def start_message_chain(body: Any, form: Any, args: Any):
|
||
"""
|
||
启动链式任务
|
||
"""
|
||
MessageChain().process(body=body, form=form, args=args)
|
||
|
||
|
||
@router.post("/", summary="接收用户消息", response_model=schemas.Response)
|
||
async def user_message(background_tasks: BackgroundTasks, request: Request):
|
||
"""
|
||
用户消息响应
|
||
"""
|
||
body = await request.body()
|
||
form = await request.form()
|
||
args = request.query_params
|
||
background_tasks.add_task(start_message_chain, body, form, args)
|
||
return schemas.Response(success=True)
|
||
|
||
|
||
@router.post("/web", summary="接收WEB消息", response_model=schemas.Response)
|
||
def web_message(text: str, current_user: User = Depends(get_current_active_superuser)):
|
||
"""
|
||
WEB消息响应
|
||
"""
|
||
MessageChain().handle_message(
|
||
channel=MessageChannel.Web,
|
||
userid=current_user.name,
|
||
username=current_user.name,
|
||
text=text
|
||
)
|
||
return schemas.Response(success=True)
|
||
|
||
|
||
@router.get("/web", summary="获取WEB消息", response_model=List[dict])
|
||
def get_web_message(_: schemas.TokenPayload = Depends(verify_token),
|
||
db: Session = Depends(get_db),
|
||
page: int = 1,
|
||
count: int = 20):
|
||
"""
|
||
获取WEB消息列表
|
||
"""
|
||
ret_messages = []
|
||
messages = Message.list_by_page(db, page=page, count=count)
|
||
for message in messages:
|
||
try:
|
||
ret_messages.append(message.to_dict())
|
||
except Exception as e:
|
||
logger.error(f"获取WEB消息列表失败: {str(e)}")
|
||
continue
|
||
return ret_messages
|
||
|
||
|
||
def wechat_verify(echostr: str, msg_signature: str,
|
||
timestamp: Union[str, int], nonce: str) -> Any:
|
||
"""
|
||
微信验证响应
|
||
"""
|
||
try:
|
||
wxcpt = WXBizMsgCrypt(sToken=settings.WECHAT_TOKEN,
|
||
sEncodingAESKey=settings.WECHAT_ENCODING_AESKEY,
|
||
sReceiveId=settings.WECHAT_CORPID)
|
||
except Exception as err:
|
||
logger.error(f"微信请求验证失败: {str(err)}")
|
||
return str(err)
|
||
ret, sEchoStr = wxcpt.VerifyURL(sMsgSignature=msg_signature,
|
||
sTimeStamp=timestamp,
|
||
sNonce=nonce,
|
||
sEchoStr=echostr)
|
||
if ret != 0:
|
||
logger.error("微信请求验证失败 VerifyURL ret: %s" % str(ret))
|
||
# 验证URL成功，将sEchoStr返回给企业号
|
||
return PlainTextResponse(sEchoStr)
|
||
|
||
|
||
def vocechat_verify(token: str) -> Any:
|
||
"""
|
||
VoceChat验证响应
|
||
"""
|
||
if token == settings.API_TOKEN:
|
||
return {"status": "OK"}
|
||
return {"status": "ERROR"}
|
||
|
||
|
||
@router.get("/", summary="回调请求验证")
|
||
def incoming_verify(token: str = None, echostr: str = None, msg_signature: str = None,
|
||
timestamp: Union[str, int] = None, nonce: str = None) -> Any:
|
||
"""
|
||
微信/VoceChat等验证响应
|
||
"""
|
||
logger.info(f"收到验证请求: token={token}, echostr={echostr}, "
|
||
f"msg_signature={msg_signature}, timestamp={timestamp}, nonce={nonce}")
|
||
if echostr and msg_signature and timestamp and nonce:
|
||
return wechat_verify(echostr, msg_signature, timestamp, nonce)
|
||
return vocechat_verify(token)
|
||
|
||
|
||
@router.get("/switchs", summary="查询通知消息渠道开关", response_model=List[NotificationSwitch])
|
||
def read_switchs(_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
||
"""
|
||
查询通知消息渠道开关
|
||
"""
|
||
return_list = []
|
||
# 读取数据库
|
||
switchs = SystemConfigOper().get(SystemConfigKey.NotificationChannels)
|
||
if not switchs:
|
||
for noti in NotificationType:
|
||
return_list.append(NotificationSwitch(mtype=noti.value, wechat=True,
|
||
telegram=True, slack=True,
|
||
synologychat=True, vocechat=True))
|
||
else:
|
||
for switch in switchs:
|
||
return_list.append(NotificationSwitch(**switch))
|
||
for noti in NotificationType:
|
||
if not any([x.mtype == noti.value for x in return_list]):
|
||
return_list.append(NotificationSwitch(mtype=noti.value, wechat=True,
|
||
telegram=True, slack=True,
|
||
synologychat=True, vocechat=True))
|
||
return return_list
|
||
|
||
|
||
@router.post("/switchs", summary="设置通知消息渠道开关", response_model=schemas.Response)
|
||
def set_switchs(switchs: List[NotificationSwitch],
|
||
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
||
"""
|
||
设置通知消息渠道开关
|
||
"""
|
||
switch_list = []
|
||
for switch in switchs:
|
||
switch_list.append(switch.dict())
|
||
# 存入数据库
|
||
SystemConfigOper().set(SystemConfigKey.NotificationChannels, switch_list)
|
||
|
||
return schemas.Response(success=True)
|
||
|
||
|
||
@router.post("/webpush/subscribe", summary="客户端webpush通知订阅", response_model=schemas.Response)
|
||
def subscribe(subscription: schemas.Subscription, _: schemas.TokenPayload = Depends(verify_token)):
|
||
"""
|
||
客户端webpush通知订阅
|
||
"""
|
||
subinfo = subscription.dict()
|
||
if subinfo not in global_vars.get_subscriptions():
|
||
global_vars.push_subscription(subinfo)
|
||
logger.debug(f"通知订阅成功: {subinfo}")
|
||
return schemas.Response(success=True)
|
||
|
||
|
||
@router.post("/webpush/send", summary="发送webpush通知", response_model=schemas.Response)
|
||
def send_notification(payload: schemas.SubscriptionMessage, _: schemas.TokenPayload = Depends(verify_token)):
|
||
"""
|
||
发送webpush通知
|
||
"""
|
||
for sub in global_vars.get_subscriptions():
|
||
try:
|
||
webpush(
|
||
subscription_info=sub,
|
||
data=json.dumps(payload.dict()),
|
||
vapid_private_key=settings.VAPID.get("privateKey"),
|
||
vapid_claims={
|
||
"sub": settings.VAPID.get("subject")
|
||
},
|
||
)
|
||
except WebPushException as err:
|
||
logger.error(f"WebPush发送失败: {str(err)}")
|
||
continue
|
||
return schemas.Response(success=True)
|