聊天增加图片功能

backend/internal/modules/file/service.go

@@ -41,10 +41,10 @@ func (s *Service) Upload(req uploadRequest) (*UploadDTO, error) {
 	if contentType == "" {
 		contentType = req.Header.Header.Get("Content-Type")
 	}
-	if !allowedContentTypes[contentType] {
+	scene := normalizeScene(req.Scene)
+	if !allowedContentTypes[contentType] || (scene == "chat" && !strings.HasPrefix(contentType, "image/")) {
 		return nil, ErrInvalidFile
 	}
-	scene := normalizeScene(req.Scene)
 	key, err := s.storage.Put(req.Context, scene, req.Header, req.Reader, contentType)
 	if err != nil {
 		return nil, err
@@ -65,7 +65,7 @@ func (s *Service) Upload(req uploadRequest) (*UploadDTO, error) {
 func normalizeScene(scene string) string {
 	scene = strings.TrimSpace(strings.ToLower(scene))
 	switch scene {
-	case "listing", "handoff", "dispute", "realname", "avatar", "home-banner":
+	case "listing", "handoff", "dispute", "realname", "avatar", "home-banner", "chat":
 		return scene
 	default:
 		return "misc"