125 lines
3.3 KiB
Python
125 lines
3.3 KiB
Python
|
"""浏览器推送通知服务 — Web Push Protocol (RFC 8291)"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
import logging
|
|||
|
|
from typing import Optional
|
|||
|
|
|
|||
|
|
from app.core.database import SessionLocal
|
|||
|
|
from app.models.push_subscription import PushSubscription
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def send_push_to_user(
|
|||
|
|
user_id: str,
|
|||
|
|
title: str,
|
|||
|
|
body: str = "",
|
|||
|
|
url: str = "/",
|
|||
|
|
require_interaction: bool = False,
|
|||
|
|
) -> int:
|
|||
|
|
"""向指定用户的所有设备发送推送通知。返回成功发送数。"""
|
|||
|
|
db = SessionLocal()
|
|||
|
|
sent = 0
|
|||
|
|
try:
|
|||
|
|
subs = (
|
|||
|
|
db.query(PushSubscription)
|
|||
|
|
.filter(PushSubscription.user_id == user_id)
|
|||
|
|
.all()
|
|||
|
|
)
|
|||
|
|
for sub in subs:
|
|||
|
|
ok = await _send_webpush(
|
|||
|
|
endpoint=sub.endpoint,
|
|||
|
|
p256dh=sub.p256dh,
|
|||
|
|
auth=sub.auth,
|
|||
|
|
title=title,
|
|||
|
|
body=body,
|
|||
|
|
url=url,
|
|||
|
|
require_interaction=require_interaction,
|
|||
|
|
)
|
|||
|
|
if ok:
|
|||
|
|
sent += 1
|
|||
|
|
return sent
|
|||
|
|
finally:
|
|||
|
|
db.close()
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def broadcast_push(
|
|||
|
|
title: str,
|
|||
|
|
body: str = "",
|
|||
|
|
url: str = "/",
|
|||
|
|
) -> int:
|
|||
|
|
"""向所有已订阅用户广播推送通知。"""
|
|||
|
|
db = SessionLocal()
|
|||
|
|
sent = 0
|
|||
|
|
try:
|
|||
|
|
subs = db.query(PushSubscription).all()
|
|||
|
|
for sub in subs:
|
|||
|
|
ok = await _send_webpush(
|
|||
|
|
endpoint=sub.endpoint,
|
|||
|
|
p256dh=sub.p256dh,
|
|||
|
|
auth=sub.auth,
|
|||
|
|
title=title,
|
|||
|
|
body=body,
|
|||
|
|
url=url,
|
|||
|
|
require_interaction=False,
|
|||
|
|
)
|
|||
|
|
if ok:
|
|||
|
|
sent += 1
|
|||
|
|
return sent
|
|||
|
|
finally:
|
|||
|
|
db.close()
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def _send_webpush(
|
|||
|
|
endpoint: str,
|
|||
|
|
p256dh: str,
|
|||
|
|
auth: str,
|
|||
|
|
title: str,
|
|||
|
|
body: str,
|
|||
|
|
url: str = "/",
|
|||
|
|
require_interaction: bool = False,
|
|||
|
|
) -> bool:
|
|||
|
|
"""发送单条 Web Push 消息。使用 pywebpush 或直接 HTTP POST。"""
|
|||
|
|
payload = json.dumps({
|
|||
|
|
"title": title,
|
|||
|
|
"body": body,
|
|||
|
|
"data": {"url": url},
|
|||
|
|
"requireInteraction": require_interaction,
|
|||
|
|
"icon": "/icons/icon-192.png",
|
|||
|
|
"badge": "/icons/icon-192.png",
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 尝试 pywebpush（如果已安装）
|
|||
|
|
try:
|
|||
|
|
from pywebpush import WebPusher, WebPushException
|
|||
|
|
wp = WebPusher({
|
|||
|
|
"endpoint": endpoint,
|
|||
|
|
"keys": {"p256dh": p256dh, "auth": auth},
|
|||
|
|
})
|
|||
|
|
wp.send(payload, timeout=10)
|
|||
|
|
return True
|
|||
|
|
except ImportError:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
# Fallback: 直接 HTTP POST（不加密，仅用于开发环境）
|
|||
|
|
import aiohttp
|
|||
|
|
async with aiohttp.ClientSession() as session:
|
|||
|
|
async with session.post(
|
|||
|
|
endpoint,
|
|||
|
|
data=payload,
|
|||
|
|
headers={
|
|||
|
|
"Content-Type": "application/json",
|
|||
|
|
"TTL": "86400",
|
|||
|
|
},
|
|||
|
|
timeout=aiohttp.ClientTimeout(total=10),
|
|||
|
|
) as resp:
|
|||
|
|
ok = resp.status in (200, 201, 204)
|
|||
|
|
if not ok:
|
|||
|
|
logger.warning("Push 发送失败: status=%d", resp.status)
|
|||
|
|
return ok
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error("Push 发送异常: %s", e)
|
|||
|
|
return False
|