提取cover服务

2026-04-08 20:44:04 +08:00
parent 55c017f60d
commit 4bc160284d
6 changed files with 88 additions and 60 deletions
--- a/internal/controller/library.go
+++ b/internal/controller/library.go
@@ -12,17 +12,18 @@ import (
 )

 type LibraryController struct {
-	service    *service.LibraryService
-	scanMutex  sync.Mutex
-	isScanning bool
+	libraryService *service.LibraryService
+	coverService   *service.CoverService
+	scanMutex      sync.Mutex
+	isScanning     bool
 }

-func NewLibraryController(service *service.LibraryService) *LibraryController {
-	return &LibraryController{service: service}
+func NewLibraryController(libraryService *service.LibraryService, coverSvc *service.CoverService) *LibraryController {
+	return &LibraryController{libraryService: libraryService, coverService: coverSvc}
 }

 func (c *LibraryController) GetAll(w http.ResponseWriter, r *http.Request) {
-	libraries, err := c.service.GetAll()
+	libraries, err := c.libraryService.GetAll()
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -43,7 +44,7 @@ func (c *LibraryController) Create(w http.ResponseWriter, r *http.Request) {
 		jsonError(w, "name and path are required", http.StatusBadRequest)
 		return
 	}
-	_, err := c.service.Create(req.Name, req.Path)
+	_, err := c.libraryService.Create(req.Name, req.Path)
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -58,7 +59,7 @@ func (c *LibraryController) GetByID(w http.ResponseWriter, r *http.Request) {
 		jsonError(w, err.Error(), http.StatusBadRequest)
 		return
 	}
-	l, err := c.service.GetByID(id)
+	l, err := c.libraryService.GetByID(id)
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -80,7 +81,7 @@ func (c *LibraryController) UpdateName(w http.ResponseWriter, r *http.Request) {
 		return
 	}

-	err = c.service.UpdateName(id, name)
+	err = c.libraryService.UpdateName(id, name)
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -102,7 +103,7 @@ func (c *LibraryController) UpdatePath(w http.ResponseWriter, r *http.Request) {
 		return
 	}

-	err = c.service.UpdatePath(id, path)
+	err = c.libraryService.UpdatePath(id, path)
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -137,7 +138,7 @@ func (c *LibraryController) Scan(w http.ResponseWriter, r *http.Request) {
 			c.scanMutex.Unlock()
 		}()

-		report, err := c.service.Scan(id)
+		report, err := c.libraryService.Scan(id)
 		if err != nil {
 			log.Printf("Scan failed for library %d: %v", id, err)
 			return
@@ -163,7 +164,7 @@ func (c *LibraryController) Delete(w http.ResponseWriter, r *http.Request) {
 		jsonError(w, err.Error(), http.StatusBadRequest)
 		return
 	}
-	err = c.service.Delete(id)
+	err = c.libraryService.Delete(id)
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -178,7 +179,7 @@ func (c *LibraryController) GetSongs(w http.ResponseWriter, r *http.Request) {
 		jsonError(w, err.Error(), http.StatusBadRequest)
 		return
 	}
-	songs, err := c.service.GetSongs(id)
+	songs, err := c.libraryService.GetSongs(id)
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -193,7 +194,7 @@ func (c *LibraryController) GetSongsByArtist(w http.ResponseWriter, r *http.Requ
 		jsonError(w, err.Error(), http.StatusBadRequest)
 		return
 	}
-	songs, err := c.service.GetSongsByArtist(id)
+	songs, err := c.libraryService.GetSongsByArtist(id)
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -208,7 +209,7 @@ func (c *LibraryController) GetSongsByAlbum(w http.ResponseWriter, r *http.Reque
 		jsonError(w, err.Error(), http.StatusBadRequest)
 		return
 	}
-	songs, err := c.service.GetSongsByAlbum(id)
+	songs, err := c.libraryService.GetSongsByAlbum(id)
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -217,7 +218,7 @@ func (c *LibraryController) GetSongsByAlbum(w http.ResponseWriter, r *http.Reque
 }

 func (c *LibraryController) GetArtists(w http.ResponseWriter, r *http.Request) {
-	artists, err := c.service.GetArtists()
+	artists, err := c.libraryService.GetArtists()
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -232,7 +233,7 @@ func (c *LibraryController) GetArtist(w http.ResponseWriter, r *http.Request) {
 		jsonError(w, err.Error(), http.StatusBadRequest)
 		return
 	}
-	artist, err := c.service.GetArtist(id)
+	artist, err := c.libraryService.GetArtist(id)
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -241,7 +242,7 @@ func (c *LibraryController) GetArtist(w http.ResponseWriter, r *http.Request) {
 }

 func (c *LibraryController) GetAlbums(w http.ResponseWriter, r *http.Request) {
-	albums, err := c.service.GetAlbums()
+	albums, err := c.libraryService.GetAlbums()
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -256,7 +257,7 @@ func (c *LibraryController) GetAlbumsByArtist(w http.ResponseWriter, r *http.Req
 		jsonError(w, err.Error(), http.StatusBadRequest)
 		return
 	}
-	albums, err := c.service.GetAlbumsByArtistWithDetail(id)
+	albums, err := c.libraryService.GetAlbumsByArtistWithDetail(id)
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -271,7 +272,7 @@ func (c *LibraryController) GetAlbum(w http.ResponseWriter, r *http.Request) {
 		jsonError(w, err.Error(), http.StatusBadRequest)
 		return
 	}
-	album, err := c.service.GetAlbum(id)
+	album, err := c.libraryService.GetAlbum(id)
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -286,7 +287,7 @@ func (c *LibraryController) GetAlbumCover(w http.ResponseWriter, r *http.Request
 		jsonError(w, err.Error(), http.StatusBadRequest)
 		return
 	}
-	coverPath, err := c.service.GetAlbumCover(id)
+	coverPath, err := c.coverService.GetAlbumCover(id)
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
--- a/internal/controller/song.go
+++ b/internal/controller/song.go
@@ -9,15 +9,16 @@ import (
 )

 type SongController struct {
-	service *service.SongService
+	songService  *service.SongService
+	coverService *service.CoverService
 }

-func NewSongController(service *service.SongService) *SongController {
-	return &SongController{service: service}
+func NewSongController(songService *service.SongService, coverService *service.CoverService) *SongController {
+	return &SongController{songService: songService, coverService: coverService}
 }

 func (c *SongController) GetAllWithDetails(w http.ResponseWriter, r *http.Request) {
-	songs, err := c.service.GetAllWithDetails()
+	songs, err := c.songService.GetAllWithDetails()
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -32,7 +33,7 @@ func (c *SongController) GetByIDWithDetails(w http.ResponseWriter, r *http.Reque
 		http.Error(w, "无效的歌曲 ID", http.StatusBadRequest)
 		return
 	}
-	song, err := c.service.GetByIDWithDetails(id)
+	song, err := c.songService.GetByIDWithDetails(id)
 	if err != nil {
 		jsonError(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -48,7 +49,7 @@ func (c *SongController) Stream(w http.ResponseWriter, r *http.Request) {
 		return
 	}

-	mediaFile, err := c.service.GetMediaFile(id)
+	mediaFile, err := c.songService.GetMediaFile(id)
 	if err != nil {
 		http.Error(w, "未找到该媒体文件", http.StatusNotFound)
 		return
@@ -64,7 +65,7 @@ func (c *SongController) Cover(w http.ResponseWriter, r *http.Request) {
 		http.Error(w, "无效的歌曲 ID", http.StatusBadRequest)
 		return
 	}
-	coverPath, err := c.service.GetCover(id)
+	coverPath, err := c.coverService.GetSongCover(id)
 	if err != nil {
 		http.Error(w, "获取封面异常", http.StatusNotFound)
 		return
--- a/internal/service/cover_svc.go
+++ b/internal/service/cover_svc.go
@@ -0,0 +1,54 @@
+package service
+
+import (
+	"butterfliu/config"
+	"butterfliu/internal/model"
+	"butterfliu/internal/repository"
+	"butterfliu/internal/util"
+	"errors"
+	"path"
+	"strconv"
+)
+
+type CoverService struct {
+	mediaRepo *repository.MediaRepository
+	songRepo  *repository.SongRepository
+	albumRepo *repository.AlbumRepository
+}
+
+func NewCoverService(mediaRepo *repository.MediaRepository, songRepo *repository.SongRepository, albumRepo *repository.AlbumRepository) *CoverService {
+	return &CoverService{
+		mediaRepo: mediaRepo,
+		songRepo:  songRepo,
+		albumRepo: albumRepo,
+	}
+}
+
+func (s *CoverService) getMediaFile(id int) (model.MediaFile, error) {
+	song, err := s.songRepo.Get(id)
+	if err != nil {
+		return model.MediaFile{}, err
+	}
+	return s.mediaRepo.Get(song.MediaFileID)
+}
+
+func (s *CoverService) GetSongCover(id int) (string, error) {
+	file, err := s.getMediaFile(id)
+	if err != nil {
+		return "", err
+	}
+	util.ExtractCover(file.Path, id)
+	conf := config.LoadConfig()
+	return path.Join(conf.GetCachePath("cover"), strconv.Itoa(id)+".jpg"), nil
+}
+
+func (s *CoverService) GetAlbumCover(id int) (string, error) {
+	songs, err := s.albumRepo.GetSongIDs(id)
+	if err != nil {
+		return "", err
+	}
+	if len(songs) == 0 {
+		return "", errors.New("no songs found for album")
+	}
+	return s.GetSongCover(songs[0])
+}
--- a/internal/service/library.go
+++ b/internal/service/library.go
@@ -1,16 +1,13 @@
 package service

 import (
-	"butterfliu/config"
 	"butterfliu/internal/model"
 	"butterfliu/internal/repository"
 	"butterfliu/internal/scanner"
 	"errors"
 	"fmt"
 	"log"
-	"path"
 	"path/filepath"
-	"strconv"
 	"strings"
 )

@@ -23,7 +20,7 @@ type ScanReport struct {
 }

 type LibraryService struct {
-	libRepo   *repository.LibraryRepository
+	libRepo    *repository.LibraryRepository
 	artistRepo *repository.ArtistRepository
 	albumRepo  *repository.AlbumRepository
 	songRepo   *repository.SongRepository
@@ -169,18 +166,6 @@ func (s *LibraryService) GetAlbum(id int) (model.AlbumDetail, error) {
 	}, nil
 }

-func (s *LibraryService) GetAlbumCover(id int) (string, error) {
-	songs, err := s.albumRepo.GetSongIDs(id)
-	if err != nil {
-		return "", err
-	}
-	if len(songs) == 0 {
-		return "", errors.New("no songs found for album")
-	}
-	conf := config.LoadConfig()
-	return path.Join(conf.GetCachePath("cover"), strconv.Itoa(songs[0])+".jpg"), nil
-}
-
 func (s *LibraryService) Scan(id int) (*ScanReport, error) {
 	lib, err := s.libRepo.GetByID(id)
 	if err != nil {
--- a/internal/service/song_svc.go
+++ b/internal/service/song_svc.go
@@ -1,12 +1,8 @@
 package service

 import (
-	"butterfliu/config"
 	"butterfliu/internal/model"
 	"butterfliu/internal/repository"
-	"butterfliu/internal/util"
-	"path"
-	"strconv"
 )

 type SongService struct {
@@ -37,13 +33,3 @@ func (s *SongService) GetMediaFile(id int) (model.MediaFile, error) {
 	}
 	return s.mediaRepo.Get(song.MediaFileID)
 }
-
-func (s *SongService) GetCover(id int) (string, error) {
-	file, err := s.GetMediaFile(id)
-	if err != nil {
-		return "", err
-	}
-	util.ExtractCover(file.Path, id)
-	conf := config.LoadConfig()
-	return path.Join(conf.GetCachePath("cover"), strconv.Itoa(id)+".jpg"), nil
-}