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个后台调度器
2026-05-19 12:01:09 +08:00
parent 78e3f450c2
commit bcf4d4e621
69 changed files with 14599 additions and 150 deletions
@@ -0,0 +1,201 @@
+package handler
+
+import (
+	"encoding/json"
+	"io"
+	"log"
+	"net/http"
+	"path/filepath"
+	"strings"
+	"time"
+
+	"github.com/yourname/cyrene-ai/voice-service/internal/config"
+	"github.com/yourname/cyrene-ai/voice-service/internal/service"
+)
+
+// STTHandler HTTP API 处理器
+type STTHandler struct {
+	svc    *service.STTService
+	ttsSvc *service.TTSService
+	cfg    *config.Config
+}
+
+// NewSTTHandler 创建 STT 处理器（可选传入 TTSService 用于组合状态）
+func NewSTTHandler(svc *service.STTService, cfg *config.Config) *STTHandler {
+	return &STTHandler{svc: svc, cfg: cfg}
+}
+
+// SetTTSService 设置 TTS 服务引用，用于组合状态端点
+func (h *STTHandler) SetTTSService(ttsSvc *service.TTSService) {
+	h.ttsSvc = ttsSvc
+}
+
+// RegisterRoutes 注册所有路由到 mux
+func (h *STTHandler) RegisterRoutes(mux *http.ServeMux) {
+	mux.HandleFunc("/api/v1/transcribe", h.handleTranscribe)
+	mux.HandleFunc("/api/v1/health", h.handleHealth)
+	mux.HandleFunc("/api/v1/status", h.handleStatus)
+}
+
+// handleTranscribe POST /api/v1/transcribe
+// 接受 multipart/form-data，字段 audio (文件) 和 language (可选)
+func (h *STTHandler) handleTranscribe(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodPost {
+		writeError(w, http.StatusMethodNotAllowed, "method not allowed")
+		return
+	}
+
+	// 限制上传大小
+	r.Body = http.MaxBytesReader(w, r.Body, h.cfg.MaxAudioSize)
+
+	if err := r.ParseMultipartForm(h.cfg.MaxAudioSize); err != nil {
+		writeError(w, http.StatusBadRequest, "文件过大或解析失败，最大支持 10MB")
+		return
+	}
+
+	// 获取上传的文件
+	file, header, err := r.FormFile("audio")
+	if err != nil {
+		writeError(w, http.StatusBadRequest, "缺少 audio 文件字段")
+		return
+	}
+	defer file.Close()
+
+	// 读取文件内容
+	audioData, err := io.ReadAll(file)
+	if err != nil {
+		writeError(w, http.StatusInternalServerError, "读取音频文件失败")
+		return
+	}
+
+	if len(audioData) == 0 {
+		writeError(w, http.StatusBadRequest, "音频文件为空")
+		return
+	}
+
+	// 获取语言参数 (可选)
+	language := r.FormValue("language")
+
+	// 推断音频格式
+	format := inferFormat(header.Filename)
+	if !isSupportedFormat(format) {
+		writeError(w, http.StatusBadRequest, "不支持的音频格式: "+format+"，支持的格式: WAV, MP3, OGG, FLAC, M4A")
+		return
+	}
+
+	// 执行转录
+	startTime := time.Now()
+	text, err := h.svc.Transcribe(audioData, format, language)
+	durationMs := time.Since(startTime).Milliseconds()
+
+	if err != nil {
+		log.Printf("[stt-handler] 转录失败: %v", err)
+		writeJSON(w, http.StatusInternalServerError, map[string]interface{}{
+			"success": false,
+			"error":   err.Error(),
+		})
+		return
+	}
+
+	actualLang := language
+	if actualLang == "" {
+		actualLang = h.cfg.WhisperLanguage
+	}
+
+	writeJSON(w, http.StatusOK, map[string]interface{}{
+		"success":     true,
+		"text":        text,
+		"language":    actualLang,
+		"duration_ms": durationMs,
+	})
+}
+
+// handleHealth GET /api/v1/health
+func (h *STTHandler) handleHealth(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodGet {
+		writeError(w, http.StatusMethodNotAllowed, "method not allowed")
+		return
+	}
+
+	sttStatus := h.svc.GetStatus()
+	healthStatus := "ok"
+	if !sttStatus["available"].(bool) {
+		healthStatus = "degraded"
+	}
+
+	resp := map[string]interface{}{
+		"status":  healthStatus,
+		"service": "voice-service",
+		"stt":     sttStatus,
+	}
+
+	// 如果有 TTS 服务，也包含 TTS 状态
+	if h.ttsSvc != nil {
+		resp["tts"] = h.ttsSvc.GetEngineStatus()
+	}
+
+	writeJSON(w, http.StatusOK, resp)
+}
+
+// handleStatus GET /api/v1/status
+func (h *STTHandler) handleStatus(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodGet {
+		writeError(w, http.StatusMethodNotAllowed, "method not allowed")
+		return
+	}
+
+	resp := map[string]interface{}{
+		"service": "voice-service",
+		"stt":     h.svc.GetStatus(),
+	}
+
+	// 如果有 TTS 服务，也包含 TTS 状态
+	if h.ttsSvc != nil {
+		resp["tts"] = h.ttsSvc.GetEngineStatus()
+	}
+
+	writeJSON(w, http.StatusOK, resp)
+}
+
+// --- 辅助函数 ---
+
+func writeJSON(w http.ResponseWriter, status int, data interface{}) {
+	w.Header().Set("Content-Type", "application/json")
+	w.WriteHeader(status)
+	json.NewEncoder(w).Encode(data)
+}
+
+func writeError(w http.ResponseWriter, status int, message string) {
+	writeJSON(w, status, map[string]interface{}{
+		"error": message,
+	})
+}
+
+// inferFormat 根据文件名推断音频格式
+func inferFormat(filename string) string {
+	ext := strings.ToLower(filepath.Ext(filename))
+	switch ext {
+	case ".wav", ".wave":
+		return "wav"
+	case ".mp3", ".mpeg":
+		return "mp3"
+	case ".ogg", ".opus":
+		return "ogg"
+	case ".flac":
+		return "flac"
+	case ".m4a", ".mp4", ".aac":
+		return "m4a"
+	default:
+		return ext
+	}
+}
+
+// isSupportedFormat 检查是否支持的音频格式
+func isSupportedFormat(format string) bool {
+	switch format {
+	case "wav", "mp3", "ogg", "flac", "m4a":
+		return true
+	default:
+		return false
+	}
+}