feat(rss): enhance media handling in RSS generation and rendering

Author: lin-snow

internal/model/embedding/embedding.go

@@ -31,8 +31,9 @@ func (EchoEmbedding) TableName() string { return "echo_embeddings" }
 
 // SearchResult 是一次向量检索命中的结果（含内容快照与距离）。
 //
-// Files / Extension 是命中 Echo 的图片附件与扩展分享，仅在检索命中后回查填充（见 copilot.enrichHits），
-// 用于前端在引用来源里展示缩略图与扩展类型标签——只随 SSE/会话给前端，不进向量索引、也不喂模型。
+// Files / Extension 是命中 Echo 的媒体附件（图片/视频/音频）与扩展分享，仅在检索命中后回查填充
+// （见 copilot.enrichHits），用于前端在引用来源里展示缩略图与类型标志——只随 SSE/会话给前端，
+// 不进向量索引、也不喂模型。
 type SearchResult struct {
 	EchoID      string                   `json:"echo_id"`
 	Content     string                   `json:"content"`

internal/service/common/common.go

@@ -16,6 +16,7 @@ import (
 	"github.com/lin-snow/ech0/internal/cache"
 	commonModel "github.com/lin-snow/ech0/internal/model/common"
 	userModel "github.com/lin-snow/ech0/internal/model/user"
+	"github.com/lin-snow/ech0/internal/storage"
 	"github.com/lin-snow/ech0/internal/util/egress"
 	mdUtil "github.com/lin-snow/ech0/internal/util/md"
 	timezoneUtil "github.com/lin-snow/ech0/internal/util/timezone"
@@ -109,15 +110,36 @@ func (s *CommonService) GenerateRSS(ctx *gin.Context) (string, error) {
 				title := msg.Username + " - " + createdAt.Format("2006-01-02")
 
 				if len(msg.EchoFiles) > 0 {
-					var imageContent []byte
+					var mediaContent []byte
 					for _, ef := range msg.EchoFiles {
-						imageContent = fmt.Appendf(
-							imageContent,
-							"<img src=\"%s\" alt=\"Image\" style=\"max-width:100%%;height:auto;\" />",
-							ef.File.URL,
-						)
+						if ef.File.URL == "" {
+							continue
+						}
+						// URL 进属性、文件名进链接文本都是可能来自 external 的用户可控字段，进入
+						// <summary type="html"> 前必须做 HTML 实体转义，阻断订阅器二次解码触发的
+						// stored XSS（与下方标签转义同一注入类，GHSA-3v85-fqvh-7rxf）。
+						url := stdhtml.EscapeString(ef.File.URL)
+						switch storage.NormalizeCategory(ef.File.Category) {
+						case storage.CategoryImage:
+							mediaContent = fmt.Appendf(mediaContent,
+								"<img src=\"%s\" alt=\"Image\" style=\"max-width:100%%;height:auto;\" />", url)
+						case storage.CategoryVideo:
+							// 内嵌 <a> 兜底：RSS 阅读器若剥离 <video> 标签，仍退化成可点链接，不丢内容。
+							mediaContent = fmt.Appendf(mediaContent,
+								"<video controls src=\"%s\" style=\"max-width:100%%;\"><a href=\"%s\">打开视频</a></video>", url, url)
+						case storage.CategoryAudio:
+							mediaContent = fmt.Appendf(mediaContent,
+								"<audio controls src=\"%s\"><a href=\"%s\">打开音频</a></audio>", url, url)
+						default:
+							// pdf / document / file / markdown：给一个可点的下载链接。
+							name := stdhtml.EscapeString(ef.File.Name)
+							if name == "" {
+								name = "下载文件"
+							}
+							mediaContent = fmt.Appendf(mediaContent, "<p>📎 <a href=\"%s\">%s</a></p>", url, name)
+						}
 					}
-					renderedContent = append(imageContent, renderedContent...)
+					renderedContent = append(mediaContent, renderedContent...)
 				}
 
 				if len(msg.Tags) > 0 {

internal/service/common/common_rss_test.go

@@ -165,6 +165,84 @@ func TestGenerateRSS_RendersEchoImages(t *testing.T) {
 	assert.Contains(t, atom, "look at this")
 }
 
+// TestGenerateRSS_RendersMediaByCategory 附件按 Category 分流渲染：
+// video → <video controls>、audio → <audio controls>（均内嵌 <a> 链接兜底），
+// 其它类型（pdf/file）→ 📎 下载链接；image 仍是 <img>。
+func TestGenerateRSS_RendersMediaByCategory(t *testing.T) {
+	repo := commonmock.NewMockCommonRepository(t)
+	svc := commonService.NewCommonService(repo, newFakeCache())
+
+	echos := []echoModel.Echo{
+		{
+			ID:        "echo-media",
+			Username:  "carol",
+			Content:   "mixed media",
+			CreatedAt: time.Now().UTC().Unix(),
+			EchoFiles: []echoModel.EchoFile{
+				{File: fileModel.File{Category: "image", URL: "http://example.com/files/pic.png"}},
+				{File: fileModel.File{Category: "video", URL: "http://example.com/files/clip.mp4"}},
+				{File: fileModel.File{Category: "audio", URL: "http://example.com/files/song.mp3"}},
+				{File: fileModel.File{Category: "pdf", URL: "http://example.com/files/doc.pdf", Name: "doc.pdf"}},
+			},
+		},
+	}
+
+	repo.EXPECT().GetAllEchos(mock.Anything, false).Return(echos, nil).Once()
+	repo.EXPECT().TrackRSSCacheKey(mock.Anything).Return().Once()
+
+	ctx := newRSSContext(t, "example.com")
+	atom, err := svc.GenerateRSS(ctx)
+	require.NoError(t, err)
+
+	// summary 内容会被 feeds 库再做一层 XML 转义（< → &lt;，" → &#34;），故断言转义后的形态。
+	// image → <img>
+	assert.Contains(t, atom, `&lt;img src=&#34;http://example.com/files/pic.png&#34;`, "图片应渲染为 <img>")
+	// video → <video controls> 且内嵌 <a> 链接兜底
+	assert.Contains(t, atom, `&lt;video controls src=&#34;http://example.com/files/clip.mp4&#34;`, "视频应渲染为 <video controls>")
+	assert.Contains(t, atom, `&lt;a href=&#34;http://example.com/files/clip.mp4&#34;&gt;打开视频&lt;/a&gt;`, "视频应内嵌链接兜底")
+	// audio → <audio controls> 且内嵌 <a> 链接兜底
+	assert.Contains(t, atom, `&lt;audio controls src=&#34;http://example.com/files/song.mp3&#34;`, "音频应渲染为 <audio controls>")
+	assert.Contains(t, atom, `&lt;a href=&#34;http://example.com/files/song.mp3&#34;&gt;打开音频&lt;/a&gt;`, "音频应内嵌链接兜底")
+	// pdf/其它 → 📎 下载链接（文件名作为链接文本）
+	assert.Contains(t, atom, `&lt;a href=&#34;http://example.com/files/doc.pdf&#34;&gt;doc.pdf&lt;/a&gt;`, "普通文件应渲染为下载链接")
+}
+
+// TestGenerateRSS_MediaFieldEscaping 绑定 GHSA-3v85-fqvh-7rxf 同类注入：
+// external 附件的 URL / 文件名是用户可控字段，进入 <summary type="html"> 前必须 HTML 实体转义，
+// 不得让原始引号/尖括号突破属性或标签上下文。
+func TestGenerateRSS_MediaFieldEscaping(t *testing.T) {
+	repo := commonmock.NewMockCommonRepository(t)
+	svc := commonService.NewCommonService(repo, newFakeCache())
+
+	echos := []echoModel.Echo{
+		{
+			ID:        "echo-evil",
+			Username:  "mallory",
+			Content:   "benign",
+			CreatedAt: time.Now().UTC().Unix(),
+			EchoFiles: []echoModel.EchoFile{
+				{File: fileModel.File{
+					Category: "file",
+					URL:      `http://x/"><script>alert(1)</script>`,
+					Name:     `<script>alert(2)</script>`,
+				}},
+			},
+		},
+	}
+
+	repo.EXPECT().GetAllEchos(mock.Anything, false).Return(echos, nil).Once()
+	repo.EXPECT().TrackRSSCacheKey(mock.Anything).Return().Once()
+
+	ctx := newRSSContext(t, "example.com")
+	atom, err := svc.GenerateRSS(ctx)
+	require.NoError(t, err)
+
+	// 不得出现由 URL/文件名注入的原始 <script>。
+	assert.NotContains(t, atom, "<script>", "URL/文件名注入的原始 script 标签不得出现")
+	// 单层转义形态（&lt;script&gt;）也不应出现——须先 HTML 实体转义再经 Atom 的 XML 序列化，呈双层转义。
+	assert.NotContains(t, atom, "&lt;script&gt;", "媒体字段必须先做 HTML 实体转义，杜绝单层转义形态")
+}
+
 // TestGenerateRSS_ReadThrough 读穿透：相同 host 第二次调用命中缓存，不再回源仓库。
 func TestGenerateRSS_ReadThrough(t *testing.T) {
 	repo := commonmock.NewMockCommonRepository(t)

internal/service/copilot/enrich.go

@@ -24,7 +24,7 @@ const maxImageBytes = 5 << 20 // 5MB
 
 // enrichHits 按命中顺序（最相关在前）回查 Echo（GetEchoById 带缓存），一次加载取齐三样：
 //   - exts：每条命中的 Extension 渲染文本（音乐/网站/位置等分享，常开，喂给模型理解）；
-//   - results[i].Files：命中 Echo 的图片附件元数据（常开，随 SSE 给前端展示缩略图，仅 URL 不含字节）；
+//   - results[i].Files：命中 Echo 的媒体附件元数据（图片/视频/音频，常开，随 SSE 给前端展示，仅元数据不含字节）；
 //   - images：配图的 base64 ImagePart，仅 multimodal 开启时收集，累计到 maxChatImages 即止（喂模型）。
 //
 // 不存进 embedding 索引、只在检索命中后回查，向量库保持纯文本干净。读取失败静默跳过（best-effort）。
@@ -47,12 +47,14 @@ func (s *CopilotService) enrichHits(
 
 		var files []fileModel.File
 		for _, ef := range echo.EchoFiles {
-			if !storage.NormalizeCategory(ef.File.Category).IsImageLike() {
-				continue
+			cat := storage.NormalizeCategory(ef.File.Category)
+			// 展示用：图片/视频/音频都带给前端（sources 里按类型展示缩略图或类型标志）。
+			switch cat {
+			case storage.CategoryImage, storage.CategoryVideo, storage.CategoryAudio:
+				files = append(files, ef.File) // 整条 File（含 storage_type/key/url/category 等）
 			}
-			files = append(files, ef.File) // 前端展示用：整条 File（含 storage_type/key/url 等）
-			// 多模态：再把图片字节读成 base64 喂给模型（受 maxChatImages 上限约束）。
-			if multimodal && s.storage != nil && len(images) < maxChatImages {
+			// 多模态：仅图片读成 base64 喂给模型（受 maxChatImages 上限约束）；视频/音频不入模型。
+			if cat.IsImageLike() && multimodal && s.storage != nil && len(images) < maxChatImages {
 				if part, ok := s.loadImagePart(ctx, ef.File); ok {
 					images = append(images, part)
 				}

internal/service/copilot/enrich_extra_test.go

@@ -5,14 +5,61 @@ package service
 
 import (
 	"context"
+	"reflect"
 	"testing"
 
 	echoModel "github.com/lin-snow/ech0/internal/model/echo"
+	embeddingModel "github.com/lin-snow/ech0/internal/model/embedding"
 	fileModel "github.com/lin-snow/ech0/internal/model/file"
 	settingModel "github.com/lin-snow/ech0/internal/model/setting"
 	"github.com/lin-snow/ech0/internal/storage"
 )
 
+// enrichEchoSvc 是 EchoService 的测试替身，只覆写 GetEchoById 返回带指定附件的 Echo，
+// 其余方法未实现（嵌入 nil 接口，未调用即不 panic）。enrichHits 只用到 GetEchoById。
+type enrichEchoSvc struct {
+	EchoService
+	echo *echoModel.Echo
+}
+
+func (f *enrichEchoSvc) GetEchoById(_ context.Context, _ string) (*echoModel.Echo, error) {
+	return f.echo, nil
+}
+
+// enrichHits：命中回查时图片/视频/音频都进 results.Files（供前端展示），pdf/file 等非媒体排除；
+// 多模态关闭时不产出 base64 图片。锁住「sources 带媒体类型标志、但不把非媒体塞给前端」的契约。
+func TestEnrichHits_MediaCategoriesToFiles(t *testing.T) {
+	echoFile := func(cat string) fileModel.EchoFile {
+		return fileModel.EchoFile{File: fileModel.File{Category: cat, URL: "https://f/" + cat}}
+	}
+	svc := &enrichEchoSvc{echo: &echoModel.Echo{
+		ID: "e1",
+		EchoFiles: []fileModel.EchoFile{
+			echoFile(string(storage.CategoryImage)),
+			echoFile(string(storage.CategoryVideo)),
+			echoFile(string(storage.CategoryAudio)),
+			echoFile(string(storage.CategoryPDF)),
+			echoFile(string(storage.CategoryFile)),
+		},
+	}}
+	s := &CopilotService{echoService: svc}
+
+	results := []embeddingModel.SearchResult{{EchoID: "e1"}}
+	_, images := s.enrichHits(context.Background(), results, false)
+
+	gotCats := make([]string, 0, len(results[0].Files))
+	for _, f := range results[0].Files {
+		gotCats = append(gotCats, f.Category)
+	}
+	want := []string{"image", "video", "audio"}
+	if !reflect.DeepEqual(gotCats, want) {
+		t.Fatalf("Files 类别 = %v, want %v（应带图片/视频/音频，排除 pdf/file）", gotCats, want)
+	}
+	if len(images) != 0 {
+		t.Fatalf("多模态关闭时不应产出 base64 图片，got %d", len(images))
+	}
+}
+
 // formatExtension：覆盖各扩展类型的渲染分支与缺字段/空值降级。
 func TestFormatExtension(t *testing.T) {
 	ext := func(typ string, kv map[string]any) *echoModel.EchoExtension {

web/src/typings/chat.d.ts

@@ -12,7 +12,7 @@ declare namespace App {
         username: string
         echo_created: number
         distance: number
-        // 命中 Echo 的图片附件（后端回查填充，整条 File，供前端展示缩略图）
+        // 命中 Echo 的媒体附件（图片/视频/音频，后端回查填充，整条 File，供前端展示缩略图/类型标志）
         files?: ChatSourceFile[]
         // 命中 Echo 的扩展分享（音乐/网站/位置…），仅用于在来源里展示一个类型标签
         extension?: Ech0.EchoExtension

web/src/views/chat/modules/ChatSources.vue

@@ -15,6 +15,11 @@
           <span aria-hidden="true">{{ src.ext.icon }}</span>
           {{ src.ext.label }}
         </span>
+        <!-- 媒体类型标志：视频/音频附件用图标 + 类型标签展示（图片走下方缩略图） -->
+        <span v-for="badge in src.mediaBadges" :key="badge.label" class="sources__ext">
+          <span aria-hidden="true">{{ badge.icon }}</span>
+          {{ badge.label }}
+        </span>
       </button>
       <!-- 命中 Echo 的配图缩略图：复用 getImageUrl 解析 local/S3/external 直链 -->
       <div v-if="src.images.length" class="sources__thumbs">
@@ -67,6 +72,13 @@ type DisplaySource = {
   empty: boolean
   images: App.Api.Ech0.FileObject[]
   ext?: { icon: string; label: string }
+  mediaBadges: { icon: string; label: string }[]
+}
+
+// 媒体类型 → 图标 emoji + i18n 标签 key（视频/音频；图片走缩略图不进这里）
+const MEDIA_META: Record<string, { icon: string; labelKey: string }> = {
+  video: { icon: '🎬', labelKey: 'editor.mediaTypeVideo' },
+  audio: { icon: '🎵', labelKey: 'editor.mediaTypeAudio' },
 }
 
 // Extension 类型 → 图标 emoji + i18n 标签 key（复用编辑器里既有的扩展类型文案）
@@ -96,7 +108,16 @@ const displaySources = computed<DisplaySource[]>(() =>
       .map((f) => ({ ...f, echo_id: src.echo_id }))
     const meta = src.extension ? EXT_META[src.extension.type] : undefined
     const ext = meta ? { icon: meta.icon, label: t(meta.labelKey) } : undefined
-    return { echoId: src.echo_id, day, text, empty, images, ext }
+    // 视频/音频类型标志：按类型去重（每条 Echo 至多各一个），图片不进这里
+    const mediaTypes = new Set<string>()
+    for (const f of src.files ?? []) {
+      if (f.category && f.category in MEDIA_META) mediaTypes.add(f.category)
+    }
+    const mediaBadges = [...mediaTypes].map((c) => ({
+      icon: MEDIA_META[c].icon,
+      label: t(MEDIA_META[c].labelKey),
+    }))
+    return { echoId: src.echo_id, day, text, empty, images, ext, mediaBadges }
   }),
 )