feat: 第五轮开发 - 14项未来路线图功能完整实现
W1-W14 全部完成: - W1: 消息搜索 (ILIKE全文检索 + SearchModal) - W2: 对话导出 (JSON/Markdown/TXT三格式) - W3: 记忆时间线 DevTools 可视化 - W4: 通知推送系统 (WebSocket + Browser Notification API) - W5: 定时提醒 (30s轮询 + 重复提醒 + WebSocket推送) - W6: 每日简报 (08:00自动生成: 天气+新闻+提醒+AI摘要) - W7: IoT场景自动化 (规则引擎 10s轮询 + 条件评估 + 场景执行) - W8: 语音输入 (浏览器 Speech Recognition API) - W9: STT服务 (voice-service + whisper.cpp) - W10: TTS服务 (浏览器 Speech Synthesis + edge-tts三档回退) - W11: 文件管理 (上传/下载/缩略图/纯Go bilinear缩放) - W12: 知识库RAG (PostgreSQL tsvector + 文档分块 + 检索) - W13: 多模态 (图片上传+分析: Vision API + 本地Go分析回退) - W14: PWA (Service Worker + 离线页 + install prompt) 总计: 6个Go微服务 + 10+前端组件 + 10+ PostgreSQL表 + 4个后台调度器
This commit is contained in:
@@ -0,0 +1,162 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"log"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
|
||||
"github.com/yourname/cyrene-ai/gateway/internal/config"
|
||||
"github.com/yourname/cyrene-ai/gateway/internal/ws"
|
||||
)
|
||||
|
||||
// NotificationHandler 通知推送处理器
|
||||
type NotificationHandler struct {
|
||||
cfg *config.Config
|
||||
hub *ws.Hub
|
||||
}
|
||||
|
||||
// NewNotificationHandler 创建通知处理器
|
||||
func NewNotificationHandler(cfg *config.Config, hub *ws.Hub) *NotificationHandler {
|
||||
return &NotificationHandler{cfg: cfg, hub: hub}
|
||||
}
|
||||
|
||||
// PushNotificationRequest 推送通知请求体
|
||||
type PushNotificationRequest struct {
|
||||
UserID string `json:"user_id" binding:"required"`
|
||||
Type string `json:"type" binding:"required,oneof=info warning success thinking reminder"`
|
||||
Title string `json:"title" binding:"required"`
|
||||
Body string `json:"body" binding:"required"`
|
||||
Data map[string]interface{} `json:"data,omitempty"`
|
||||
}
|
||||
|
||||
// Push 推送通知到指定用户 (需要 JWT 认证)
|
||||
// POST /api/v1/notifications/push
|
||||
func (h *NotificationHandler) Push(c *gin.Context) {
|
||||
var req PushNotificationRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "请求参数无效: " + err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
// 生成通知
|
||||
notif := h.buildNotification(req)
|
||||
|
||||
// 序列化 WS 消息
|
||||
msg := ws.ServerMessage{
|
||||
Type: "notification",
|
||||
MessageID: "notif_" + generateID(),
|
||||
Timestamp: time.Now().UnixMilli(),
|
||||
Notification: notif,
|
||||
}
|
||||
|
||||
data, err := json.Marshal(msg)
|
||||
if err != nil {
|
||||
log.Printf("[notification] 序列化通知失败: %v", err)
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "内部错误"})
|
||||
return
|
||||
}
|
||||
|
||||
// 通过 Hub 推送给指定用户
|
||||
h.hub.SendToUser(req.UserID, data)
|
||||
|
||||
log.Printf("[notification] 通知已推送: user=%s type=%s title=%s", req.UserID, req.Type, req.Title)
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"success": true,
|
||||
"notification": gin.H{
|
||||
"id": notif.ID,
|
||||
"type": notif.Type,
|
||||
"title": notif.Title,
|
||||
"user_id": req.UserID,
|
||||
"timestamp": notif.Timestamp,
|
||||
"delivered": h.hub.UserClientCount(req.UserID) > 0,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
// InternalNotify 内部服务推送通知 (使用内部 service token)
|
||||
// POST /api/v1/internal/notify
|
||||
func (h *NotificationHandler) InternalNotify(c *gin.Context) {
|
||||
var req PushNotificationRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "请求参数无效: " + err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
// 生成通知
|
||||
notif := h.buildNotification(req)
|
||||
|
||||
// 序列化 WS 消息
|
||||
msg := ws.ServerMessage{
|
||||
Type: "notification",
|
||||
MessageID: "notif_" + generateID(),
|
||||
Timestamp: time.Now().UnixMilli(),
|
||||
Notification: notif,
|
||||
}
|
||||
|
||||
data, err := json.Marshal(msg)
|
||||
if err != nil {
|
||||
log.Printf("[notification] 序列化通知失败: %v", err)
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "内部错误"})
|
||||
return
|
||||
}
|
||||
|
||||
// 通过 Hub 推送给指定用户
|
||||
h.hub.SendToUser(req.UserID, data)
|
||||
|
||||
log.Printf("[notification] 内部通知已推送: user=%s type=%s title=%s", req.UserID, req.Type, req.Title)
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"success": true,
|
||||
"notification": gin.H{
|
||||
"id": notif.ID,
|
||||
"type": notif.Type,
|
||||
"title": notif.Title,
|
||||
"user_id": req.UserID,
|
||||
"timestamp": notif.Timestamp,
|
||||
"delivered": h.hub.UserClientCount(req.UserID) > 0,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
// InternalNotifyAuth 内部服务认证中间件
|
||||
func (h *NotificationHandler) InternalNotifyAuth() gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
token := c.GetHeader("X-Internal-Token")
|
||||
if token == "" {
|
||||
token = c.GetHeader("Authorization")
|
||||
if len(token) > 7 && token[:7] == "Bearer " {
|
||||
token = token[7:]
|
||||
}
|
||||
}
|
||||
|
||||
if token != h.cfg.InternalServiceToken {
|
||||
c.JSON(http.StatusUnauthorized, gin.H{"error": "内部认证失败"})
|
||||
c.Abort()
|
||||
return
|
||||
}
|
||||
c.Next()
|
||||
}
|
||||
}
|
||||
|
||||
// buildNotification 构建 NotificationInfo
|
||||
func (h *NotificationHandler) buildNotification(req PushNotificationRequest) *ws.NotificationInfo {
|
||||
notifID := "notif_" + generateID()
|
||||
now := time.Now().UTC().Format(time.RFC3339)
|
||||
|
||||
if req.Data == nil {
|
||||
req.Data = make(map[string]interface{})
|
||||
}
|
||||
|
||||
return &ws.NotificationInfo{
|
||||
ID: notifID,
|
||||
Type: req.Type,
|
||||
Title: req.Title,
|
||||
Body: req.Body,
|
||||
Timestamp: now,
|
||||
Data: req.Data,
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user