package http import ( "context" "embed" "encoding/csv" "encoding/json" "fmt" "html/template" "io" "io/fs" "math" "net/http" "os" "path/filepath" "regexp" "sort" "strings" "time" "github.com/gin-gonic/gin" "github.com/rs/zerolog/log" "github.com/sjzar/chatlog/internal/errors" "github.com/sjzar/chatlog/internal/model" "github.com/sjzar/chatlog/pkg/util" "github.com/sjzar/chatlog/pkg/util/dat2img" "github.com/sjzar/chatlog/pkg/util/silk" ) //go:embed static var EFS embed.FS // 统一的 HTML 预览组件片段 var previewHTMLSnippetBase = `
预览
` var previewVoiceSnippet = ` ` var previewHTMLSnippet = previewHTMLSnippetBase + previewVoiceSnippet var chatlogHTMLHeadTemplate = `%s` func writeChatlogHTMLHeader(w io.Writer, title string) { fmt.Fprintf(w, chatlogHTMLHeadTemplate, template.HTMLEscapeString(title)) } func (s *Service) initRouter() { s.initBaseRouter() s.initMediaRouter() s.initAPIRouter() s.initMCPRouter() } func (s *Service) initBaseRouter() { staticDir, _ := fs.Sub(EFS, "static") s.router.StaticFS("/static", http.FS(staticDir)) s.router.StaticFileFS("/favicon.ico", "./favicon.ico", http.FS(staticDir)) s.router.StaticFileFS("/", "./index.htm", http.FS(staticDir)) s.router.GET("/health", func(ctx *gin.Context) { ctx.JSON(http.StatusOK, gin.H{"status": "ok"}) }) s.router.NoRoute(func(c *gin.Context) { path := c.Request.URL.Path if strings.HasPrefix(path, "/api") || strings.HasPrefix(path, "/static") { c.JSON(http.StatusNotFound, gin.H{"error": "Not found"}) return } c.Header("Cache-Control", "no-cache, no-store, max-age=0, must-revalidate, value") c.Redirect(http.StatusFound, "/") }) } func (s *Service) initMediaRouter() { s.router.GET("/image/*key", func(c *gin.Context) { s.handleMedia(c, "image") }) s.router.GET("/video/*key", func(c *gin.Context) { s.handleMedia(c, "video") }) s.router.GET("/file/*key", func(c *gin.Context) { s.handleMedia(c, "file") }) s.router.GET("/voice/*key", func(c *gin.Context) { s.handleMedia(c, "voice") }) s.router.GET("/data/*path", s.handleMediaData) s.router.GET("/avatar/:username", s.handleAvatar) } func (s *Service) initAPIRouter() { api := s.router.Group("/api/v1") { api.GET("/setting", s.handleGetSetting) api.POST("/setting", s.handleUpdateSetting) actions := api.Group("/actions") actions.POST("/get-data-key", s.handleActionGetDataKey) actions.POST("/decrypt", s.handleActionDecrypt) actions.POST("/http/start", s.handleActionStartHTTP) actions.POST("/http/stop", s.handleActionStopHTTP) actions.POST("/auto-decrypt/start", s.handleActionStartAutoDecrypt) actions.POST("/auto-decrypt/stop", s.handleActionStopAutoDecrypt) dataAPI := api.Group("", s.checkDBStateMiddleware()) dataAPI.GET("/chatlog", s.handleChatlog) dataAPI.GET("/contact", s.handleContacts) dataAPI.GET("/chatroom", s.handleChatRooms) dataAPI.GET("/session", s.handleSessions) dataAPI.GET("/diary", s.handleDiary) dataAPI.GET("/dashboard", s.handleDashboard) dataAPI.GET("/search", s.handleSearch) } } func (s *Service) initMCPRouter() { s.router.Any("/mcp", func(c *gin.Context) { s.mcpStreamableServer.ServeHTTP(c.Writer, c.Request) }) s.router.Any("/sse", func(c *gin.Context) { s.mcpSSEServer.ServeHTTP(c.Writer, c.Request) }) s.router.Any("/message", func(c *gin.Context) { s.mcpSSEServer.ServeHTTP(c.Writer, c.Request) }) } // GET /api/v1/dashboard func (s *Service) handleDashboard(c *gin.Context) { // 基础聚合 gstats, err := s.db.GetDB().GlobalMessageStats() if err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": "global stats failed", "detail": err.Error()}) return } groupCounts, _ := s.db.GetDB().GroupMessageCounts() // 文件与目录大小 dataDir := s.conf.GetDataDir() workDir := dataDir if s.db != nil { if wd := s.db.GetWorkDir(); wd != "" { workDir = wd } } dirSize := safeDirSize(dataDir) dbSize := estimateDBSize(workDir) // 当前账号昵称（overview.user）：优先从 WorkDir/DataDir 路径中提取 wxid_***，再用联系人 NickName 映射；找不到则回退 wxid extractWxid := func(p string) string { p = strings.TrimSpace(p) if p == "" { return "" } // 遍历路径片段，优先返回形如 wxid_ 开头的片段 parts := strings.Split(filepath.Clean(p), string(filepath.Separator)) for _, seg := range parts { if strings.HasPrefix(strings.ToLower(seg), "wxid_") { return seg } } // 兜底返回最后一段 return filepath.Base(filepath.Clean(p)) } currentUser := "" accountID := "" // 先从 WorkDir 提取（更贴近实际解密目录结构），再从 DataDir 提取 if wd := s.db.GetWorkDir(); wd != "" && accountID == "" { accountID = extractWxid(wd) } if accountID == "" { accountID = extractWxid(dataDir) } // 若拿到候选 accountID，则尝试用联系人映射 NickName if accountID != "" && accountID != "." && accountID != string(filepath.Separator) { // Windows WeChat 4.x: v3 对应 wxid 可能带有第二段后缀，如 wxid_xxx_yyyy // 查找昵称时需要去掉第二个下划线及其后内容 lookupID := accountID low := strings.ToLower(lookupID) if strings.HasPrefix(low, "wxid_") { // 定位第二个下划线位置 rest := lookupID[len("wxid_"):] if idx := strings.Index(rest, "_"); idx >= 0 { lookupID = lookupID[:len("wxid_")+idx] } } if clist, err := s.db.GetContacts(lookupID, 0, 0); err == nil && clist != nil { for _, it := range clist.Items { if it != nil && it.UserName == lookupID { if strings.TrimSpace(it.NickName) != "" { currentUser = it.NickName } break } } if currentUser == "" && len(clist.Items) > 0 && clist.Items[0] != nil && clist.Items[0].UserName == lookupID { currentUser = clist.Items[0].NickName } } // 最终兜底：回退为 wxid/accountID if strings.TrimSpace(currentUser) == "" { currentUser = accountID } } // 使用结构体固定 JSON 输出顺序 type DBStats struct { DbSizeMB float64 `json:"db_size_mb"` DirSizeMB float64 `json:"dir_size_mb"` } type MsgStats struct { TotalMsgs int64 `json:"total_msgs"` SentMsgs int64 `json:"sent_msgs"` ReceivedMsgs int64 `json:"received_msgs"` UniqueMsgTypes int `json:"unique_msg_types"` } type OverviewGroup struct { ChatRoomName string `json:"ChatRoomName"` NickName string `json:"NickName"` MemberCount int `json:"member_count"` MessageCount int64 `json:"message_count"` } type Timeline struct { Earliest int64 `json:"earliest_msg_time"` Latest int64 `json:"latest_msg_time"` Duration int `json:"duration_days"` } type Migration struct { ID int `json:"id"` File string `json:"file"` Status string `json:"status"` CreatedAt string `json:"created_at"` } type Overview struct { User string `json:"user"` DBStats DBStats `json:"dbStats"` MsgStats MsgStats `json:"msgStats"` MsgTypes map[string]int64 `json:"msgTypes"` Groups []OverviewGroup `json:"groups"` Timeline Timeline `json:"timeline"` Migrations []Migration `json:"migrations"` } type GroupOverview struct { TotalGroups int `json:"total_groups"` ActiveGroups int `json:"active_groups"` TodayMessages int `json:"today_messages"` WeeklyAvg int `json:"weekly_avg"` MostActiveHour string `json:"most_active_hour"` } type ContentAnalysis struct { Text int64 `json:"text_messages"` Images int64 `json:"images"` Voice int64 `json:"voice_messages"` Files int64 `json:"files"` Links int64 `json:"links"` Others int64 `json:"others"` } type GroupListItem struct { Name string `json:"name"` Members int `json:"members"` Messages int64 `json:"messages"` Active bool `json:"active"` } type GroupAnalysis struct { Title string `json:"title"` Overview GroupOverview `json:"overview"` ContentAnalysis ContentAnalysis `json:"content_analysis"` GroupList []GroupListItem `json:"group_list"` } type ContentTypeStats struct { Count int64 `json:"count"` Percentage float64 `json:"percentage"` SizeMB *float64 `json:"size_mb,omitempty"` Trend *string `json:"trend,omitempty"` } type SourceChannel struct { Count int64 `json:"count"` Percentage float64 `json:"percentage"` } type ProcessingStatus struct { Processed int `json:"processed"` Processing int `json:"processing"` Pending int `json:"pending"` } type QualityMetrics struct { DataIntegrity float64 `json:"data_integrity"` ClassificationAccuracy float64 `json:"classification_accuracy"` DuplicateRate float64 `json:"duplicate_rate"` ErrorRate float64 `json:"error_rate"` } type DataTypeAnalysis struct { Title string `json:"title"` ContentTypes map[string]ContentTypeStats `json:"content_types"` SourceChannels map[string]SourceChannel `json:"source_channels"` ProcessingStatus ProcessingStatus `json:"processing_status"` QualityMetrics QualityMetrics `json:"quality_metrics"` PieGradient string `json:"pieGradient,omitempty"` } type VisualizationDefaults struct { SelectedGroupIndex int `json:"selectedGroupIndex"` } type RelationshipNode struct { Name string `json:"name"` Type string `json:"type"` Messages int64 `json:"messages"` Avatar string `json:"avatar,omitempty"` } type RelationshipNetwork struct { Nodes []RelationshipNode `json:"nodes"` } type Visualization struct { Defaults VisualizationDefaults `json:"defaults"` GroupAnalysis GroupAnalysis `json:"groupAnalysis"` DataTypeAnalysis DataTypeAnalysis `json:"dataTypeAnalysis"` RelationshipNetwork RelationshipNetwork `json:"relationshipNetwork"` } type Dashboard struct { Overview Overview `json:"overview"` Visualization Visualization `json:"visualization"` } // 群信息（合并消息计数） type groupAggregate struct { id string nickName string members int messages int64 active bool } groupAggs := make([]groupAggregate, 0) activeGroups := 0 if rooms, err := s.db.GetChatRooms("", 0, 0); err == nil { for _, r := range rooms.Items { if strings.TrimSpace(r.NickName) == "" { continue } mc := groupCounts[r.Name] active := mc > 0 if active { activeGroups++ } groupAggs = append(groupAggs, groupAggregate{ id: r.Name, nickName: r.NickName, members: len(r.Users), messages: mc, active: active, }) } } sort.Slice(groupAggs, func(i, j int) bool { if groupAggs[i].messages == groupAggs[j].messages { return groupAggs[i].nickName < groupAggs[j].nickName } return groupAggs[i].messages > groupAggs[j].messages }) overviewGroups := make([]OverviewGroup, 0, len(groupAggs)) groupList := make([]GroupListItem, 0, len(groupAggs)) for _, g := range groupAggs { overviewGroups = append(overviewGroups, OverviewGroup{ ChatRoomName: g.id, NickName: g.nickName, MemberCount: g.members, MessageCount: g.messages, }) groupList = append(groupList, GroupListItem{ Name: g.nickName, Members: g.members, Messages: g.messages, Active: g.active, }) } totalGroups := len(groupAggs) // msgTypes 依据最新文档 + 衍生细分（文件消息 / 链接消息）补齐 msgTypes := map[string]int64{ "文本消息": 0, "图片消息": 0, "语音消息": 0, "好友验证消息": 0, "好友推荐消息": 0, "聊天表情": 0, "位置消息": 0, "XML消息": 0, // 未细分的 49 类或其他 XML "文件消息": 0, "链接消息": 0, "音视频通话": 0, "手机端操作消息": 0, "系统通知": 0, "撤回消息": 0, } for k, v := range gstats.ByType { if _, ok := msgTypes[k]; ok { msgTypes[k] += v } } // 时间轴 durationDays := 0 if gstats.EarliestUnix > 0 && gstats.LatestUnix >= gstats.EarliestUnix { span := gstats.LatestUnix - gstats.EarliestUnix if span < 0 { span = 0 } durationDays = int(math.Round(float64(span) / 86400.0)) } uniqueTypes := 0 for _, v := range msgTypes { if v > 0 { uniqueTypes++ } } // 今日每小时统计用于 most_active_hour perHourTotal := make([]int64, 24) if s.db != nil && s.db.GetDB() != nil { if hours, err := s.db.GetDB().GlobalTodayHourly(); err == nil { for i := 0; i < 24; i++ { perHourTotal[i] = hours[i] } } } maxHour := 0 for h := 1; h < 24; h++ { if perHourTotal[h] > perHourTotal[maxHour] { maxHour = h } } mostActiveHour := fmt.Sprintf("%02d:00-%02d:00", maxHour, (maxHour+1)%24) // 内容占比（基于 msgTypes） totalMsgs := gstats.Total pct := func(n int64) float64 { if totalMsgs == 0 { return 0 } return math.Round((float64(n) * 10000.0 / float64(totalMsgs))) / 100.0 } // 私聊/群聊分布（用于 DataTypeAnalysis.SourceChannels） var groupTotal int64 for _, v := range groupCounts { groupTotal += v } privateTotal := totalMsgs - groupTotal // ====== 今日群聊消息数统计 ====== todayMessages := int64(0) if s.db != nil && s.db.GetDB() != nil { if todayCounts, err := s.db.GetDB().GroupTodayMessageCounts(); err == nil { for _, v := range todayCounts { todayMessages += v } } } // ====== 本周群聊平均每天消息数 ====== weeklyAvg := 0 if s.db != nil && s.db.GetDB() != nil { if weekTotal, err := s.db.GetDB().GroupWeekMessageCount(); err == nil && weekTotal > 0 { // 计算已过天数：周一=1, 周二=2 ... 周六=6, 周日=7（显示完整7天平均） now := time.Now() wday := int(now.Weekday()) // Sunday=0 passed := 0 if wday == 0 { // Sunday passed = 7 } else { passed = wday } if passed <= 0 { passed = 1 } avg := float64(weekTotal) / float64(passed) weeklyAvg = int(math.Round(avg)) } } // ===== 归一化 content_types 百分比（合计 100%）===== // 参与归一化的类别列表（与 DataTypeAnalysis.content_types 一致） ctKeys := []string{ "XML消息", "位置消息", "图片消息", "好友推荐消息", "好友验证消息", "手机端操作消息", "撤回消息", "文件消息", "文本消息", "系统通知", "聊天表情", "语音消息", "链接消息", "音视频通话", } var sumCT int64 maxKey := "" var maxCnt int64 for _, k := range ctKeys { sumCT += msgTypes[k] if msgTypes[k] > maxCnt { maxCnt = msgTypes[k] maxKey = k } } round2 := func(f float64) float64 { return math.Round(f*100) / 100 } pctCT := func(n int64) float64 { if sumCT == 0 { return 0 } return round2(float64(n) * 100.0 / float64(sumCT)) } // 先计算每类百分比与总和 ctPerc := make(map[string]float64, len(ctKeys)) sumPerc := 0.0 for _, k := range ctKeys { p := pctCT(msgTypes[k]) ctPerc[k] = p sumPerc += p } // 差额校正到 100% if diff := round2(100.0 - sumPerc); diff != 0 && maxKey != "" { ctPerc[maxKey] = round2(ctPerc[maxKey] + diff) } // ===== 关系网络（亲密度）===== relationshipNodes := make([]RelationshipNode, 0) if s.db != nil && s.db.GetDB() != nil { if ibase, err := s.db.GetDB().IntimacyBase(); err == nil && len(ibase) > 0 { skipIDs := map[string]struct{}{ "filehelper": {}, "weixin": {}, "notifymessage": {}, "fmessage": {}, } contactMap := map[string]*model.Contact{} if clist, err := s.db.GetContacts("", 0, 0); err == nil && clist != nil { for _, ct := range clist.Items { if ct != nil { contactMap[ct.UserName] = ct } } } type pair struct { k string v *model.IntimacyBase } arr := make([]pair, 0, len(ibase)) for k, v := range ibase { arr = append(arr, pair{k, v}) } sort.Slice(arr, func(i, j int) bool { ai, aj := arr[i].v, arr[j].v if ai.Last90DaysMsg != aj.Last90DaysMsg { return ai.Last90DaysMsg > aj.Last90DaysMsg } if ai.MsgCount != aj.MsgCount { return ai.MsgCount > aj.MsgCount } return ai.Past7DaysSentMsg > aj.Past7DaysSentMsg }) maxN := 24 if len(arr) < maxN { maxN = len(arr) } added := 0 for idx := 0; idx < len(arr) && added < maxN; idx++ { k := arr[idx].k v := arr[idx].v if accountID != "" && k == accountID { continue } if _, skip := skipIDs[k]; skip { continue } ct := contactMap[k] display := k if ct != nil { if strings.TrimSpace(ct.Remark) != "" { display = ct.Remark } else if strings.TrimSpace(ct.NickName) != "" { display = ct.NickName } } relationshipNodes = append(relationshipNodes, RelationshipNode{ Name: display, Type: "contact", Messages: v.MsgCount, Avatar: s.composeAvatarURL(k), }) added++ } } } others := totalMsgs - (msgTypes["文本消息"] + msgTypes["图片消息"] + msgTypes["语音消息"] + msgTypes["文件消息"] + msgTypes["链接消息"]) if others < 0 { others = 0 } defaultSelectedIndex := 0 if len(groupList) == 0 { defaultSelectedIndex = -1 } processingStatus := ProcessingStatus{} if totalMsgs > 0 { processingStatus.Processed = 100 } qualityMetrics := QualityMetrics{} floatPtr := func(v float64) *float64 { return &v } stringPtr := func(v string) *string { return &v } vis := Visualization{ Defaults: VisualizationDefaults{SelectedGroupIndex: defaultSelectedIndex}, GroupAnalysis: GroupAnalysis{ Title: "群聊分析", Overview: GroupOverview{ TotalGroups: totalGroups, ActiveGroups: activeGroups, TodayMessages: int(todayMessages), WeeklyAvg: weeklyAvg, MostActiveHour: mostActiveHour, }, ContentAnalysis: ContentAnalysis{ Text: msgTypes["文本消息"], Images: msgTypes["图片消息"], Voice: msgTypes["语音消息"], Files: msgTypes["文件消息"], Links: msgTypes["链接消息"], Others: others, }, GroupList: groupList, }, DataTypeAnalysis: DataTypeAnalysis{ Title: "数据类型统计", ContentTypes: map[string]ContentTypeStats{ "文本消息": {Count: msgTypes["文本消息"], Percentage: ctPerc["文本消息"]}, "图片消息": {Count: msgTypes["图片消息"], Percentage: ctPerc["图片消息"]}, "语音消息": {Count: msgTypes["语音消息"], Percentage: ctPerc["语音消息"]}, "文件消息": {Count: msgTypes["文件消息"], Percentage: ctPerc["文件消息"]}, "链接消息": {Count: msgTypes["链接消息"], Percentage: ctPerc["链接消息"], SizeMB: floatPtr(0), Trend: stringPtr("")}, "XML消息": {Count: msgTypes["XML消息"], Percentage: ctPerc["XML消息"]}, "好友验证消息": {Count: msgTypes["好友验证消息"], Percentage: ctPerc["好友验证消息"]}, "好友推荐消息": {Count: msgTypes["好友推荐消息"], Percentage: ctPerc["好友推荐消息"]}, "聊天表情": {Count: msgTypes["聊天表情"], Percentage: ctPerc["聊天表情"]}, "位置消息": {Count: msgTypes["位置消息"], Percentage: ctPerc["位置消息"]}, "音视频通话": {Count: msgTypes["音视频通话"], Percentage: ctPerc["音视频通话"]}, "手机端操作消息": {Count: msgTypes["手机端操作消息"], Percentage: ctPerc["手机端操作消息"]}, "系统通知": {Count: msgTypes["系统通知"], Percentage: ctPerc["系统通知"]}, "撤回消息": {Count: msgTypes["撤回消息"], Percentage: ctPerc["撤回消息"]}, }, SourceChannels: map[string]SourceChannel{ "私聊数据": {Count: privateTotal, Percentage: pct(privateTotal)}, "群聊数据": {Count: groupTotal, Percentage: pct(groupTotal)}, }, ProcessingStatus: processingStatus, QualityMetrics: qualityMetrics, PieGradient: "#3b82f6 0deg 180deg, #10b981 180deg 270deg, #f59e0b 270deg 315deg, #ef4444 315deg 360deg", }, RelationshipNetwork: RelationshipNetwork{Nodes: relationshipNodes}, } resp := Dashboard{ Overview: Overview{ User: currentUser, DBStats: DBStats{DbSizeMB: roundMB(dbSize), DirSizeMB: roundMB(dirSize)}, MsgStats: MsgStats{TotalMsgs: gstats.Total, SentMsgs: gstats.Sent, ReceivedMsgs: gstats.Received, UniqueMsgTypes: uniqueTypes}, MsgTypes: msgTypes, Groups: overviewGroups, Timeline: Timeline{Earliest: gstats.EarliestUnix, Latest: gstats.LatestUnix, Duration: durationDays}, Migrations: []Migration{}, }, Visualization: vis, } // ===== 持久化 dashboard （单一文件）===== // 仅保存一个固定文件：/dashboard.json baseDir := "" if s.db != nil { if wd := strings.TrimSpace(s.db.GetWorkDir()); wd != "" { baseDir = wd } } if baseDir == "" { if dir := strings.TrimSpace(s.conf.GetDataDir()); dir != "" { baseDir = dir } } if baseDir == "" { if cwd, err := os.Getwd(); err == nil { baseDir = cwd } } if baseDir != "" { if err := os.MkdirAll(baseDir, 0o755); err == nil { if b, err := json.Marshal(resp); err == nil { path := filepath.Join(baseDir, "dashboard.json") _ = os.WriteFile(path, b, 0o644) } } } if c.Query("download") == "1" { b, err := json.MarshalIndent(resp, "", " ") if err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": "marshal failed", "detail": err.Error()}) return } c.Header("Content-Type", "application/json") c.Header("Content-Disposition", "attachment; filename=dashboard.json") c.Data(http.StatusOK, "application/json", b) return } c.JSON(http.StatusOK, resp) } func roundMB(bytes int64) float64 { if bytes <= 0 { return 0 } // 1 MB = 1024*1024 mb := float64(bytes) / (1024.0 * 1024.0) // round to 2 decimals return float64(int(mb*100+0.5)) / 100.0 } // safeDirSize walks a directory and sums file sizes; returns 0 on error. func safeDirSize(path string) int64 { var total int64 if path == "" { return 0 } _ = filepath.Walk(path, func(p string, info os.FileInfo, err error) error { if err != nil { return nil } if info == nil || info.IsDir() { return nil } total += info.Size() return nil }) return total } // estimateDBSize sums sizes of common DB files under workDir func estimateDBSize(workDir string) int64 { if workDir == "" { return 0 } var total int64 _ = filepath.Walk(workDir, func(p string, info os.FileInfo, err error) error { if err != nil || info == nil || info.IsDir() { return nil } name := strings.ToLower(info.Name()) if strings.HasSuffix(name, ".db") || strings.HasSuffix(name, ".sqlite") || strings.HasSuffix(name, ".sqlite3") || strings.HasSuffix(name, ".db-wal") || strings.HasSuffix(name, ".db-shm") { total += info.Size() } return nil }) return total } func (s *Service) handleSearch(c *gin.Context) { params := struct { Query string `form:"q"` Talker string `form:"talker"` Sender string `form:"sender"` Time string `form:"time"` Start string `form:"start"` End string `form:"end"` Limit int `form:"limit"` Offset int `form:"offset"` Format string `form:"format"` }{} if err := c.BindQuery(¶ms); err != nil { errors.Err(c, err) return } query := strings.TrimSpace(params.Query) talker := strings.TrimSpace(params.Talker) limit := params.Limit if limit <= 0 { limit = 20 } if limit > 200 { limit = 200 } offset := params.Offset if offset < 0 { offset = 0 } req := &model.SearchRequest{ Query: query, Talker: talker, Sender: strings.TrimSpace(params.Sender), Limit: limit, Offset: offset, } if params.Time != "" { start, end, ok := util.TimeRangeOf(params.Time) if !ok { errors.Err(c, errors.InvalidArg("time")) return } req.Start = start req.End = end } else { if params.Start != "" && params.End != "" { start, end, ok := util.TimeRangeOf(params.Start + "~" + params.End) if !ok { errors.Err(c, errors.InvalidArg("time")) return } req.Start = start req.End = end } else if params.Start != "" { start, end, ok := util.TimeRangeOf(params.Start) if !ok { errors.Err(c, errors.InvalidArg("start")) return } req.Start = start req.End = end } else if params.End != "" { start, end, ok := util.TimeRangeOf(params.End) if !ok { errors.Err(c, errors.InvalidArg("end")) return } req.Start = start req.End = end } } if !req.Start.IsZero() && !req.End.IsZero() && req.End.Before(req.Start) { req.Start, req.End = req.End, req.Start } resp, err := s.db.SearchMessages(req) if err != nil { errors.Err(c, err) return } if resp == nil { resp = &model.SearchResponse{Hits: []*model.SearchHit{}, Limit: limit, Offset: offset} } resp.Query = req.Query resp.Talker = req.Talker resp.Sender = req.Sender resp.Start = req.Start resp.End = req.End resp.Limit = limit resp.Offset = offset format := strings.ToLower(strings.TrimSpace(params.Format)) if format == "" { format = "json" } switch format { case "html": c.Writer.Header().Set("Content-Type", "text/html; charset=utf-8") writeChatlogHTMLHeader(c.Writer, "Search Result") c.Writer.WriteString("

搜索结果

") c.Writer.WriteString("
") if resp.Query != "" { c.Writer.WriteString("

关键词：" + template.HTMLEscapeString(resp.Query) + "

") } talkerLabel := "全部会话" if resp.Talker != "" { talkerLabel = template.HTMLEscapeString(resp.Talker) } c.Writer.WriteString("

会话：" + talkerLabel + "

") if resp.Sender != "" { c.Writer.WriteString("

发送者：" + template.HTMLEscapeString(resp.Sender) + "

") } timeLabel := "不限" if !resp.Start.IsZero() && !resp.End.IsZero() { timeLabel = resp.Start.Format("2006-01-02 15:04:05") + " ~ " + resp.End.Format("2006-01-02 15:04:05") } else if !resp.Start.IsZero() { timeLabel = ">= " + resp.Start.Format("2006-01-02 15:04:05") } else if !resp.End.IsZero() { timeLabel = "<= " + resp.End.Format("2006-01-02 15:04:05") } c.Writer.WriteString("

时间范围：" + template.HTMLEscapeString(timeLabel) + "

") c.Writer.WriteString(fmt.Sprintf("

命中条数：%d（本页 %d 条）

", resp.Total, len(resp.Hits))) c.Writer.WriteString("
") if len(resp.Hits) == 0 { c.Writer.WriteString("
暂无搜索结果
") } else { for idx, hit := range resp.Hits { if hit == nil || hit.Message == nil { continue } msg := hit.Message msg.SetContent("host", c.Request.Host) talkerDisplay := msg.Talker if msg.TalkerName != "" { talkerDisplay = fmt.Sprintf("%s (%s)", msg.TalkerName, msg.Talker) } senderDisplay := msg.Sender if msg.IsSelf { senderDisplay = "我" } if msg.SenderName != "" { senderDisplay = fmt.Sprintf("%s(%s)", msg.SenderName, msg.Sender) } avatarURL := template.HTMLEscapeString(s.composeAvatarURL(msg.Sender) + "?size=big") talkerText := template.HTMLEscapeString(talkerDisplay) senderText := template.HTMLEscapeString(senderDisplay) timeText := template.HTMLEscapeString(msg.Time.Format("2006-01-02 15:04:05")) c.Writer.WriteString("
\"avatar\"
") c.Writer.WriteString("
#" + fmt.Sprintf("%d", idx+1) + " · " + talkerText + "" + senderText + "" + timeText + "") if hit.Score > 0 { c.Writer.WriteString("score: " + fmt.Sprintf("%.4f", hit.Score) + "") } c.Writer.WriteString("
") c.Writer.WriteString("
" + messageHTMLPlaceholder(msg) + "
") c.Writer.WriteString("
") } } c.Writer.WriteString(previewHTMLSnippet) c.Writer.WriteString("") return case "text", "plain": c.Writer.Header().Set("Content-Type", "text/plain; charset=utf-8") c.Writer.Header().Set("Cache-Control", "no-cache") c.Writer.Header().Set("Connection", "keep-alive") fmt.Fprintf(c.Writer, "关键词: %s\n", resp.Query) talkerLabel := resp.Talker if talkerLabel == "" { talkerLabel = "全部会话" } fmt.Fprintf(c.Writer, "会话: %s\n", talkerLabel) if resp.Sender != "" { fmt.Fprintf(c.Writer, "发送者: %s\n", resp.Sender) } switch { case !resp.Start.IsZero() && !resp.End.IsZero(): fmt.Fprintf(c.Writer, "时间: %s ~ %s\n", resp.Start.Format("2006-01-02 15:04:05"), resp.End.Format("2006-01-02 15:04:05")) case !resp.Start.IsZero(): fmt.Fprintf(c.Writer, "时间: >= %s\n", resp.Start.Format("2006-01-02 15:04:05")) case !resp.End.IsZero(): fmt.Fprintf(c.Writer, "时间: <= %s\n", resp.End.Format("2006-01-02 15:04:05")) default: fmt.Fprintln(c.Writer, "时间: 不限") } fmt.Fprintf(c.Writer, "总命中: %d, 本页: %d\n", resp.Total, len(resp.Hits)) fmt.Fprintln(c.Writer, strings.Repeat("-", 60)) for idx, hit := range resp.Hits { if hit == nil || hit.Message == nil { continue } msg := hit.Message msg.SetContent("host", c.Request.Host) title := msg.Talker if msg.TalkerName != "" { title = fmt.Sprintf("%s (%s)", msg.TalkerName, msg.Talker) } sender := msg.Sender if msg.IsSelf { sender = "我" } if msg.SenderName != "" { sender = fmt.Sprintf("%s(%s)", msg.SenderName, msg.Sender) } fmt.Fprintf(c.Writer, "[%d] %s @ %s\n", idx+1, msg.Time.Format("2006-01-02 15:04:05"), title) fmt.Fprintf(c.Writer, "发送者: %s\n", sender) fmt.Fprintf(c.Writer, "%s\n", msg.PlainTextContent()) if snippet := strings.TrimSpace(hit.Snippet); snippet != "" { fmt.Fprintf(c.Writer, "Snippet: %s\n", snippet) } fmt.Fprintln(c.Writer, strings.Repeat("-", 60)) } return case "csv": c.Writer.Header().Set("Content-Type", "text/csv; charset=utf-8") c.Writer.Header().Set("Cache-Control", "no-cache") c.Writer.Header().Set("Connection", "keep-alive") c.Writer.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=search_%s.csv", time.Now().Format("20060102_150405"))) csvWriter := csv.NewWriter(c.Writer) csvWriter.Write([]string{"Seq", "Time", "Talker", "TalkerName", "Sender", "SenderName", "Content", "Snippet"}) for _, hit := range resp.Hits { if hit == nil || hit.Message == nil { continue } msg := hit.Message msg.SetContent("host", c.Request.Host) csvWriter.Write([]string{ fmt.Sprintf("%d", msg.Seq), msg.Time.Format("2006-01-02 15:04:05"), msg.Talker, msg.TalkerName, msg.Sender, msg.SenderName, msg.PlainTextContent(), strings.ReplaceAll(hit.Snippet, "\n", " "), }) } csvWriter.Flush() return case "json": c.JSON(http.StatusOK, resp) return default: c.JSON(http.StatusOK, resp) return } } func (s *Service) handleChatlog(c *gin.Context) { q := struct { Time string `form:"time"` Talker string `form:"talker"` Sender string `form:"sender"` Keyword string `form:"keyword"` Limit int `form:"limit"` Offset int `form:"offset"` Format string `form:"format"` }{} if err := c.BindQuery(&q); err != nil { errors.Err(c, err) return } start, end, ok := util.TimeRangeOf(q.Time) if !ok { errors.Err(c, errors.InvalidArg("time")) } if q.Limit < 0 { q.Limit = 0 } if q.Offset < 0 { q.Offset = 0 } format := strings.ToLower(strings.TrimSpace(q.Format)) if format == "" { format = "json" } // 1. 未指定 talker: 分组输出 if q.Talker == "" { sessionsResp, err := s.db.GetSessions("", 0, 0) if err != nil { errors.Err(c, err) return } type grouped struct { Talker string `json:"talker"` TalkerName string `json:"talkerName,omitempty"` Messages []*model.Message `json:"messages"` } groups := make([]*grouped, 0) for _, sess := range sessionsResp.Items { msgs, err := s.db.GetMessages(start, end, sess.UserName, q.Sender, q.Keyword, 0, 0) if err != nil || len(msgs) == 0 { continue } groups = append(groups, &grouped{Talker: sess.UserName, TalkerName: sess.NickName, Messages: msgs}) } switch format { case "html": c.Writer.Header().Set("Content-Type", "text/html; charset=utf-8") writeChatlogHTMLHeader(c.Writer, "Chatlog") c.Writer.WriteString(fmt.Sprintf("

All Messages %s ~ %s

", start.Format("2006-01-02 15:04:05"), end.Format("2006-01-02 15:04:05"))) for _, g := range groups { title := g.Talker if g.TalkerName != "" { title = fmt.Sprintf("%s (%s)", g.TalkerName, g.Talker) } c.Writer.WriteString("
" + template.HTMLEscapeString(title) + fmt.Sprintf(" - %d 条消息", len(g.Messages))) for _, m := range g.Messages { m.SetContent("host", c.Request.Host) senderDisplay := m.Sender if m.IsSelf { senderDisplay = "我" } if m.SenderName != "" { senderDisplay = template.HTMLEscapeString(m.SenderName) + "(" + template.HTMLEscapeString(senderDisplay) + ")" } else { senderDisplay = template.HTMLEscapeString(senderDisplay) } aurl := template.HTMLEscapeString(s.composeAvatarURL(m.Sender) + "?size=big") timeText := template.HTMLEscapeString(m.Time.Format("2006-01-02 15:04:05")) c.Writer.WriteString("
\"avatar\"
" + senderDisplay + "" + timeText + "
" + messageHTMLPlaceholder(m) + "
") } c.Writer.WriteString("
") } c.Writer.WriteString(previewHTMLSnippet) c.Writer.WriteString("") case "json": c.JSON(http.StatusOK, groups) case "csv": c.Writer.Header().Set("Content-Type", "text/csv; charset=utf-8") c.Writer.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=all_%s_%s.csv", start.Format("2006-01-02"), end.Format("2006-01-02"))) c.Writer.Header().Set("Cache-Control", "no-cache") c.Writer.Header().Set("Connection", "keep-alive") c.Writer.Flush() csvWriter := csv.NewWriter(c.Writer) csvWriter.Write([]string{"Talker", "TalkerName", "Time", "SenderName", "Sender", "Content"}) for _, g := range groups { for _, m := range g.Messages { csvWriter.Write([]string{g.Talker, g.TalkerName, m.Time.Format("2006-01-02 15:04:05"), m.SenderName, m.Sender, m.PlainTextContent()}) } } csvWriter.Flush() case "text", "plain": c.Writer.Header().Set("Content-Type", "text/plain; charset=utf-8") c.Writer.Header().Set("Cache-Control", "no-cache") c.Writer.Header().Set("Connection", "keep-alive") c.Writer.Flush() for _, g := range groups { header := g.Talker if g.TalkerName != "" { header = fmt.Sprintf("%s (%s)", g.TalkerName, g.Talker) } c.Writer.WriteString(header + "\n") for _, m := range g.Messages { sender := m.Sender if m.IsSelf { sender = "我" } if m.SenderName != "" { sender = m.SenderName + "(" + sender + ")" } c.Writer.WriteString(m.Time.Format("2006-01-02 15:04:05") + " " + sender + " " + m.PlainTextContent() + "\n") } c.Writer.WriteString("-----------------------------\n") } default: c.JSON(http.StatusOK, groups) } return } // 2. 指定 talker: 单会话消息 messages, err := s.db.GetMessages(start, end, q.Talker, q.Sender, q.Keyword, q.Limit, q.Offset) if err != nil { errors.Err(c, err) return } switch format { case "html": c.Writer.Header().Set("Content-Type", "text/html; charset=utf-8") writeChatlogHTMLHeader(c.Writer, "Chatlog") c.Writer.WriteString(fmt.Sprintf("

Messages %s ~ %s (%s)

", start.Format("2006-01-02 15:04:05"), end.Format("2006-01-02 15:04:05"), template.HTMLEscapeString(q.Talker))) for _, m := range messages { m.SetContent("host", c.Request.Host) c.Writer.WriteString("
") aurl := template.HTMLEscapeString(s.composeAvatarURL(m.Sender) + "?size=big") c.Writer.WriteString("\"avatar\"") c.Writer.WriteString("
") if m.SenderName != "" { c.Writer.WriteString(template.HTMLEscapeString(m.SenderName) + "(") } c.Writer.WriteString(template.HTMLEscapeString(m.Sender)) if m.SenderName != "" { c.Writer.WriteString(")") } timeText := template.HTMLEscapeString(m.Time.Format("2006-01-02 15:04:05")) c.Writer.WriteString("" + timeText + "
")
			c.Writer.WriteString(messageHTMLPlaceholder(m))
			c.Writer.WriteString("
") } c.Writer.WriteString(previewHTMLSnippet) c.Writer.WriteString("") case "csv": c.Writer.Header().Set("Content-Type", "text/csv; charset=utf-8") c.Writer.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s_%s_%s.csv", q.Talker, start.Format("2006-01-02"), end.Format("2006-01-02"))) c.Writer.Header().Set("Cache-Control", "no-cache") c.Writer.Header().Set("Connection", "keep-alive") c.Writer.Flush() csvWriter := csv.NewWriter(c.Writer) csvWriter.Write([]string{"Time", "SenderName", "Sender", "TalkerName", "Talker", "Content"}) for _, m := range messages { csvWriter.Write(m.CSV(c.Request.Host)) } csvWriter.Flush() case "json": c.JSON(http.StatusOK, messages) default: c.Writer.Header().Set("Content-Type", "text/plain; charset=utf-8") c.Writer.Header().Set("Cache-Control", "no-cache") c.Writer.Header().Set("Connection", "keep-alive") c.Writer.Flush() for _, m := range messages { c.Writer.WriteString(m.PlainText(strings.Contains(q.Talker, ","), util.PerfectTimeFormat(start, end), c.Request.Host) + "\n") } } } func (s *Service) handleContacts(c *gin.Context) { q := struct { Keyword string `form:"keyword"` Limit int `form:"limit"` Offset int `form:"offset"` Format string `form:"format"` }{} if err := c.BindQuery(&q); err != nil { errors.Err(c, err) return } // 关键字去空白；空关键字表示返回全部 q.Keyword = strings.TrimSpace(q.Keyword) list, err := s.db.GetContacts(q.Keyword, q.Limit, q.Offset) if err != nil { errors.Err(c, err) return } format := strings.ToLower(strings.TrimSpace(q.Format)) if format == "" { format = "json" } switch format { case "html": c.Writer.Header().Set("Content-Type", "text/html; charset=utf-8") c.Writer.WriteHeader(http.StatusOK) c.Writer.Write([]byte(`
`)) for _, contact := range list.Items { uname := template.HTMLEscapeString(contact.UserName) nick := template.HTMLEscapeString(contact.NickName) remark := template.HTMLEscapeString(contact.Remark) alias := template.HTMLEscapeString(contact.Alias) // compose avatar URL aurl := template.HTMLEscapeString(s.composeAvatarURL(contact.UserName)) c.Writer.WriteString(`
`) c.Writer.WriteString(``) c.Writer.WriteString(`
`) c.Writer.WriteString(`
` + nick + `
`) c.Writer.WriteString(`
` + uname) if remark != "" { c.Writer.WriteString(` · ` + remark) } if alias != "" { c.Writer.WriteString(` · alias:` + alias) } c.Writer.WriteString(`
`) } c.Writer.WriteString(`
`) return case "json": // fill avatar urls for _, item := range list.Items { item.AvatarURL = s.composeAvatarURL(item.UserName) } c.JSON(http.StatusOK, list) default: // csv if format == "csv" { // 浏览器访问时，会下载文件 c.Writer.Header().Set("Content-Type", "text/csv; charset=utf-8") } else { c.Writer.Header().Set("Content-Type", "text/plain; charset=utf-8") } c.Writer.Header().Set("Cache-Control", "no-cache") c.Writer.Header().Set("Connection", "keep-alive") c.Writer.Flush() c.Writer.WriteString("UserName,Alias,Remark,NickName,AvatarURL\n") for _, contact := range list.Items { avatarURL := s.composeAvatarURL(contact.UserName) c.Writer.WriteString(fmt.Sprintf("%s,%s,%s,%s,%s\n", contact.UserName, contact.Alias, contact.Remark, contact.NickName, avatarURL)) } c.Writer.Flush() } } // composeAvatarURL builds a relative URL that the server can serve for any username func (s *Service) composeAvatarURL(username string) string { if username == "" { return "" } return "/avatar/" + username } // handleAvatar serves avatar by username. For v3 returns redirect to remote URL; for v4 streams bytes. func (s *Service) handleAvatar(c *gin.Context) { username := c.Param("username") size := c.Query("size") // optional: small|big avatar, err := s.db.GetAvatar(username, size) if err != nil { errors.Err(c, err) return } if avatar == nil { errors.Err(c, errors.ErrAvatarNotFound) return } if avatar.URL != "" { // external URL, redirect c.Redirect(http.StatusFound, avatar.URL) return } // inline bytes ct := avatar.ContentType if ct == "" { ct = "image/jpeg" } c.Data(http.StatusOK, ct, avatar.Data) } func (s *Service) handleChatRooms(c *gin.Context) { q := struct { Keyword string `form:"keyword"` Limit int `form:"limit"` Offset int `form:"offset"` Format string `form:"format"` }{} if err := c.BindQuery(&q); err != nil { errors.Err(c, err) return } // 关键字去空白；空关键字表示返回全部 q.Keyword = strings.TrimSpace(q.Keyword) list, err := s.db.GetChatRooms(q.Keyword, q.Limit, q.Offset) if err != nil { errors.Err(c, err) return } format := strings.ToLower(strings.TrimSpace(q.Format)) if format == "" { format = "json" } switch format { case "json": // json c.JSON(http.StatusOK, list) default: // csv if format == "csv" { // 浏览器访问时，会下载文件 c.Writer.Header().Set("Content-Type", "text/csv; charset=utf-8") } else { c.Writer.Header().Set("Content-Type", "text/plain; charset=utf-8") } c.Writer.Header().Set("Cache-Control", "no-cache") c.Writer.Header().Set("Connection", "keep-alive") c.Writer.Flush() c.Writer.WriteString("Name,Remark,NickName,Owner,UserCount\n") for _, chatRoom := range list.Items { c.Writer.WriteString(fmt.Sprintf("%s,%s,%s,%s,%d\n", chatRoom.Name, chatRoom.Remark, chatRoom.NickName, chatRoom.Owner, len(chatRoom.Users))) } c.Writer.Flush() } } func (s *Service) handleSessions(c *gin.Context) { q := struct { Keyword string `form:"keyword"` Limit int `form:"limit"` Offset int `form:"offset"` Format string `form:"format"` }{} if err := c.BindQuery(&q); err != nil { errors.Err(c, err) return } sessions, err := s.db.GetSessions(q.Keyword, q.Limit, q.Offset) if err != nil { errors.Err(c, err) return } format := strings.ToLower(strings.TrimSpace(q.Format)) if format == "" { format = "json" } switch format { case "html": c.Writer.Header().Set("Content-Type", "text/html; charset=utf-8") c.Writer.WriteHeader(http.StatusOK) c.Writer.Write([]byte(`
`)) for _, session := range sessions.Items { // 转义 name := template.HTMLEscapeString(session.NickName) uname := template.HTMLEscapeString(session.UserName) content := template.HTMLEscapeString(session.Content) if len(content) > 400 { // 简单截断，避免过长 content = content[:400] + "..." } content = strings.ReplaceAll(content, "\r", "") content = strings.ReplaceAll(content, "\n", "\n") // 让 pre-wrap 生效 c.Writer.Write([]byte(`
` + name + `(` + uname + `)` + session.NTime.Format("2006-01-02 15:04:05") + `
` + content + `
`)) } c.Writer.Write([]byte(`
`)) c.Writer.Write([]byte(previewHTMLSnippet)) c.Writer.Flush() return case "csv": c.Writer.Header().Set("Content-Type", "text/csv; charset=utf-8") c.Writer.Header().Set("Cache-Control", "no-cache") c.Writer.Header().Set("Connection", "keep-alive") c.Writer.Flush() c.Writer.WriteString("UserName,NOrder,NickName,Content,NTime\n") for _, session := range sessions.Items { c.Writer.WriteString(fmt.Sprintf("%s,%d,%s,%s,%s\n", session.UserName, session.NOrder, session.NickName, strings.ReplaceAll(session.Content, "\n", "\\n"), session.NTime)) } c.Writer.Flush() case "json": // json c.JSON(http.StatusOK, sessions) default: c.Writer.Header().Set("Content-Type", "text/plain; charset=utf-8") c.Writer.Header().Set("Cache-Control", "no-cache") c.Writer.Header().Set("Connection", "keep-alive") c.Writer.Flush() for _, session := range sessions.Items { c.Writer.WriteString(session.PlainText(120)) c.Writer.WriteString("\n") } c.Writer.Flush() } } // handleDiary 返回指定日期内“我”参与的消息（日记），按 talker 分组。 // GET /api/v1/diary?date=YYYY-MM-DD&format=(html|json|csv|text) func (s *Service) handleDiary(c *gin.Context) { q := struct { Date string `form:"date"` Talker string `form:"talker"` Format string `form:"format"` }{} if err := c.BindQuery(&q); err != nil { errors.Err(c, err) return } dateStr := strings.TrimSpace(q.Date) if dateStr == "" { dateStr = time.Now().Format("2006-01-02") } parsed, err := time.ParseInLocation("2006-01-02", dateStr, time.Local) if err != nil { errors.Err(c, errors.InvalidArg("date")) return } start := time.Date(parsed.Year(), parsed.Month(), parsed.Day(), 0, 0, 0, 0, parsed.Location()) end := start.Add(24*time.Hour - time.Nanosecond) startDisplay := start.Format("2006-01-02 15:04:05") endDisplay := end.Format("2006-01-02 15:04:05") heading := fmt.Sprintf("%s 的聊天日记（%s ~ %s）", start.Format("2006-01-02"), startDisplay, endDisplay) // 获取会话（可选 talker 过滤） sessionsResp, err := s.db.GetSessions(q.Talker, 0, 0) if err != nil { errors.Err(c, err) return } type grouped struct { Talker string `json:"talker"` TalkerName string `json:"talkerName,omitempty"` Messages []*model.Message `json:"messages"` } groups := make([]*grouped, 0) for _, sess := range sessionsResp.Items { msgs, err := s.db.GetMessages(start, end, sess.UserName, "", "", 0, 0) if err != nil || len(msgs) == 0 { continue } hasSelf := false for _, m := range msgs { if m.IsSelf { hasSelf = true break } } if !hasSelf { continue } groups = append(groups, &grouped{Talker: sess.UserName, TalkerName: sess.NickName, Messages: msgs}) } format := strings.ToLower(strings.TrimSpace(q.Format)) if format == "" { format = "json" } switch format { case "html": c.Writer.Header().Set("Content-Type", "text/html; charset=utf-8") c.Writer.WriteString(`Diary`) c.Writer.WriteString(fmt.Sprintf("

%s

", template.HTMLEscapeString(heading))) for _, g := range groups { title := g.Talker if g.TalkerName != "" { title = fmt.Sprintf("%s (%s)", g.TalkerName, g.Talker) } c.Writer.WriteString("
" + template.HTMLEscapeString(title) + fmt.Sprintf(" - %d 条消息", len(g.Messages))) for _, m := range g.Messages { m.SetContent("host", c.Request.Host) senderDisplay := m.Sender if m.IsSelf { senderDisplay = "我" } if m.SenderName != "" { senderDisplay = template.HTMLEscapeString(m.SenderName) + "(" + template.HTMLEscapeString(senderDisplay) + ")" } else { senderDisplay = template.HTMLEscapeString(senderDisplay) } aurl := template.HTMLEscapeString(s.composeAvatarURL(m.Sender) + "?size=big") c.Writer.WriteString("
\"avatar\"
" + senderDisplay + "" + m.Time.Format("2006-01-02 15:04:05") + "
" + messageHTMLPlaceholder(m) + "
") } c.Writer.WriteString("
") } c.Writer.WriteString(previewHTMLSnippet) c.Writer.WriteString("") case "json": c.JSON(http.StatusOK, groups) case "csv": c.Writer.Header().Set("Content-Type", "text/csv; charset=utf-8") c.Writer.Header().Set("Cache-Control", "no-cache") c.Writer.Header().Set("Connection", "keep-alive") c.Writer.Flush() writer := csv.NewWriter(c.Writer) writer.Write([]string{"Talker", "TalkerName", "Time", "SenderName", "Sender", "Content"}) for _, g := range groups { for _, m := range g.Messages { writer.Write([]string{m.Talker, m.TalkerName, m.Time.Format("2006-01-02 15:04:05"), m.SenderName, m.Sender, m.PlainTextContent()}) } } writer.Flush() default: c.Writer.Header().Set("Content-Type", "text/plain; charset=utf-8") c.Writer.Header().Set("Cache-Control", "no-cache") c.Writer.Header().Set("Connection", "keep-alive") c.Writer.Flush() for _, g := range groups { if g.TalkerName != "" { c.Writer.WriteString(fmt.Sprintf("%s (%s)\n", g.TalkerName, g.Talker)) } else { c.Writer.WriteString(g.Talker + "\n") } for _, m := range g.Messages { senderDisplay := m.Sender if m.IsSelf { senderDisplay = "我" } if m.SenderName != "" { senderDisplay = m.SenderName + "(" + senderDisplay + ")" } c.Writer.WriteString(m.Time.Format("2006-01-02 15:04:05")) c.Writer.WriteString(" ") c.Writer.WriteString(senderDisplay) c.Writer.WriteString(" ") c.Writer.WriteString(m.PlainTextContent()) c.Writer.WriteString("\n") } c.Writer.WriteString("-----------------------------\n") } } } func (s *Service) handleMedia(c *gin.Context, _type string) { key := strings.TrimPrefix(c.Param("key"), "/") if key == "" { errors.Err(c, errors.InvalidArg(key)) return } keys := util.Str2List(key, ",") if len(keys) == 0 { errors.Err(c, errors.InvalidArg(key)) return } var _err error for _, k := range keys { if strings.Contains(k, "/") { if absolutePath, err := s.findPath(_type, k); err == nil { c.Redirect(http.StatusFound, "/data/"+absolutePath) return } } media, err := s.db.GetMedia(_type, k) if err != nil { _err = err continue } if c.Query("info") != "" { c.JSON(http.StatusOK, media) return } if media.Type == "voice" && c.Query("transcribe") != "" { s.handleVoiceTranscription(c, k, media) return } switch media.Type { case "voice": s.HandleVoice(c, media.Data) return default: c.Redirect(http.StatusFound, "/data/"+media.Path) return } } if _err != nil { errors.Err(c, _err) return } } func (s *Service) handleVoiceTranscription(c *gin.Context, key string, media *model.Media) { if s.speechTranscriber == nil { c.JSON(http.StatusServiceUnavailable, gin.H{"error": "speech transcription not enabled"}) return } if len(media.Data) == 0 { c.JSON(http.StatusInternalServerError, gin.H{"error": "voice data unavailable"}) return } ctx := c.Request.Context() if ctx == nil { ctx = context.Background() } var cancel context.CancelFunc if _, hasDeadline := ctx.Deadline(); !hasDeadline { ctx, cancel = context.WithTimeout(ctx, 2*time.Minute) } if cancel != nil { defer cancel() } opts := s.speechOptions if lang := strings.TrimSpace(c.Query("lang")); lang != "" { opts.Language = lang opts.LanguageSet = true } if translate := strings.TrimSpace(c.Query("translate")); translate != "" { switch strings.ToLower(translate) { case "1", "true", "yes", "on": opts.Translate = true opts.TranslateSet = true case "0", "false", "no", "off": opts.Translate = false opts.TranslateSet = true } } res, err := s.speechTranscriber.TranscribeSilk(ctx, media.Data, opts) if err != nil { if ctx.Err() != nil { c.JSON(http.StatusRequestTimeout, gin.H{"error": "transcription cancelled"}) return } log.Error().Err(err).Str("media_key", key).Msg("voice transcription failed") c.JSON(http.StatusInternalServerError, gin.H{"error": "transcription failed"}) return } if res == nil { c.JSON(http.StatusOK, gin.H{"key": key, "text": "", "language": opts.Language, "duration": 0}) return } c.JSON(http.StatusOK, gin.H{ "key": key, "text": res.Text, "language": res.Language, "duration": res.Duration.Seconds(), "segments": res.Segments, }) } func (s *Service) findPath(_type string, key string) (string, error) { absolutePath := filepath.Join(s.conf.GetDataDir(), key) if _, err := os.Stat(absolutePath); err == nil { return key, nil } switch _type { case "image": for _, suffix := range []string{"_h.dat", ".dat", "_t.dat"} { if _, err := os.Stat(absolutePath + suffix); err == nil { return key + suffix, nil } } case "video": for _, suffix := range []string{".mp4", "_thumb.jpg"} { if _, err := os.Stat(absolutePath + suffix); err == nil { return key + suffix, nil } } } return "", errors.ErrMediaNotFound } func (s *Service) handleMediaData(c *gin.Context) { relativePath := filepath.Clean(c.Param("path")) absolutePath := filepath.Join(s.conf.GetDataDir(), relativePath) if _, err := os.Stat(absolutePath); os.IsNotExist(err) { c.JSON(http.StatusNotFound, gin.H{ "error": "File not found", }) return } ext := strings.ToLower(filepath.Ext(absolutePath)) switch { case ext == ".dat": s.HandleDatFile(c, absolutePath) default: // 直接返回文件 c.File(absolutePath) } } func (s *Service) HandleDatFile(c *gin.Context, path string) { b, err := os.ReadFile(path) if err != nil { errors.Err(c, err) return } out, ext, err := dat2img.Dat2Image(b) if err != nil { c.File(path) return } switch ext { case "jpg", "jpeg": c.Data(http.StatusOK, "image/jpeg", out) case "png": c.Data(http.StatusOK, "image/png", out) case "gif": c.Data(http.StatusOK, "image/gif", out) case "bmp": c.Data(http.StatusOK, "image/bmp", out) case "mp4": c.Data(http.StatusOK, "video/mp4", out) default: c.Data(http.StatusOK, "image/jpg", out) // c.File(path) } } func (s *Service) HandleVoice(c *gin.Context, data []byte) { out, err := silk.Silk2MP3(data) if err != nil { c.Data(http.StatusOK, "audio/silk", data) return } c.Data(http.StatusOK, "audio/mp3", out) } // 统一占位符：将 PlainTextContent 里形如 ![标签](url) 或 [标签](url) 的模式转成超链接形式，仅显示 [标签]。 var ( placeholderPattern = regexp.MustCompile(`!?\[([^\]]+)\]\((https?://[^)]+)\)`) ) func messageHTMLPlaceholder(m *model.Message) string { content := m.PlainTextContent() return placeholderPattern.ReplaceAllStringFunc(content, func(s string) string { matches := placeholderPattern.FindStringSubmatch(s) if len(matches) != 3 { return template.HTMLEscapeString(s) } fullLabel := matches[1] url := matches[2] left := fullLabel rest := "" if p := strings.Index(fullLabel, "|"); p >= 0 { left = fullLabel[:p] rest = fullLabel[p+1:] } className := "media" if left == "动画表情" || left == "GIF表情" || strings.Contains(left, "表情") { className = "media anim" } if left == "语音" { className = "media voice-link" } var anchorText string if left == "链接" { // 保留完整形式 [链接|标题\n更多说明] escapedFull := template.HTMLEscapeString(fullLabel) escapedFull = strings.ReplaceAll(escapedFull, "\r", "") escapedFull = strings.ReplaceAll(escapedFull, "\n", "
") anchorText = "[" + escapedFull + "]" } else if left == "文件" && rest != "" { // 文件保留文件名 anchorText = "[文件]" + template.HTMLEscapeString(rest) } else { anchorText = "[" + template.HTMLEscapeString(left) + "]" } escapedURL := template.HTMLEscapeString(url) anchor := `` + anchorText + `` if left == "语音" { return `` + anchor + `` } return anchor }) }