丰富艺术家和专辑接口

internal/repository/library.go

@@ -138,6 +138,58 @@ func (r *LibraryRepository) GetSongsByLibraryWithDetails(libraryID int) ([]SongD
 	return songs, nil
 }
 
+func (r *LibraryRepository) GetSongsByArtistWithDetails(artistID int) ([]SongDetail, error) {
+	rows, err := r.db.Query(`
+		SELECT s.id, s.title, a.name as artist_name, al.title as album_title, s.duration, mf.path
+		FROM songs s
+		INNER JOIN media_files mf ON s.media_file_id = mf.id
+		INNER JOIN artists a ON s.artist_id = a.id
+		INNER JOIN albums al ON s.album_id = al.id
+		WHERE s.artist_id = ?
+		ORDER BY s.title
+	`, artistID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	songs := []SongDetail{}
+	for rows.Next() {
+		var song SongDetail
+		if err := rows.Scan(&song.ID, &song.Title, &song.Artist, &song.Album, &song.Duration, &song.Path); err != nil {
+			return nil, err
+		}
+		songs = append(songs, song)
+	}
+	return songs, nil
+}
+
+func (r *LibraryRepository) GetSongsByAlbumWithDetails(albumID int) ([]SongDetail, error) {
+	rows, err := r.db.Query(`
+		SELECT s.id, s.title, a.name as artist_name, al.title as album_title, s.duration, mf.path
+		FROM songs s
+		INNER JOIN media_files mf ON s.media_file_id = mf.id
+		INNER JOIN artists a ON s.artist_id = a.id
+		INNER JOIN albums al ON s.album_id = al.id
+		WHERE s.album_id = ?
+		ORDER BY s.title
+	`, albumID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	songs := []SongDetail{}
+	for rows.Next() {
+		var song SongDetail
+		if err := rows.Scan(&song.ID, &song.Title, &song.Artist, &song.Album, &song.Duration, &song.Path); err != nil {
+			return nil, err
+		}
+		songs = append(songs, song)
+	}
+	return songs, nil
+}
+
 func (r *LibraryRepository) GetMediaFileByPath(path string) (model.MediaFile, error) {
 	row := r.db.QueryRow(`
 		SELECT id, path, library_id, 
@@ -304,6 +356,60 @@ func (r *LibraryRepository) GetAlbums() ([]model.Album, error) {
 	return albums, nil
 }
 
+func (r *LibraryRepository) GetAlbumsByArtist(artistID int) ([]model.Album, error) {
+	rows, err := r.db.Query("SELECT id, title, artist_id FROM albums WHERE artist_id = ?", artistID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	albums := []model.Album{}
+	for rows.Next() {
+		var album model.Album
+		if err := rows.Scan(&album.ID, &album.Title, &album.ArtistID); err != nil {
+			return nil, err
+		}
+		albums = append(albums, album)
+	}
+	return albums, nil
+}
+
+func (r *LibraryRepository) GetAlbumIDsByArtist(artistID int) ([]int, error) {
+	rows, err := r.db.Query("SELECT id FROM albums WHERE artist_id = ?", artistID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	ids := []int{}
+	for rows.Next() {
+		var id int
+		if err := rows.Scan(&id); err != nil {
+			return nil, err
+		}
+		ids = append(ids, id)
+	}
+	return ids, nil
+}
+
+func (r *LibraryRepository) GetSongIDsByArtist(artistID int) ([]int, error) {
+	rows, err := r.db.Query("SELECT id FROM songs WHERE artist_id = ?", artistID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	ids := []int{}
+	for rows.Next() {
+		var id int
+		if err := rows.Scan(&id); err != nil {
+			return nil, err
+		}
+		ids = append(ids, id)
+	}
+	return ids, nil
+}
+
 func (r *LibraryRepository) GetAlbum(id int) (model.Album, error) {
 	row := r.db.QueryRow("SELECT id, title, artist_id FROM albums WHERE id = ?", id)
 
@@ -313,3 +419,21 @@ func (r *LibraryRepository) GetAlbum(id int) (model.Album, error) {
 	}
 	return album, nil
 }
+
+func (r *LibraryRepository) GetSongIDsByAlbum(artistID int) ([]int, error) {
+	rows, err := r.db.Query("SELECT id FROM songs WHERE album_id = ?", artistID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	ids := []int{}
+	for rows.Next() {
+		var id int
+		if err := rows.Scan(&id); err != nil {
+			return nil, err
+		}
+		ids = append(ids, id)
+	}
+	return ids, nil
+}