97 lines
2.5 KiB
Go
97 lines
2.5 KiB
Go
package repository
|
|
|
|
import (
|
|
"butterfliu/internal/model"
|
|
"database/sql"
|
|
)
|
|
|
|
type SongRepository struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func NewSongRepository(db *sql.DB) *SongRepository {
|
|
return &SongRepository{db: db}
|
|
}
|
|
|
|
func (r *SongRepository) GetAll() ([]model.Song, error) {
|
|
rows, err := r.db.Query(`
|
|
SELECT id, title, artist_id, album_id, duration, media_file_id,
|
|
COALESCE(track_number, 0), COALESCE(disc_number, 0), COALESCE(genre, '')
|
|
FROM songs
|
|
`)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
songs := []model.Song{}
|
|
for rows.Next() {
|
|
var song model.Song
|
|
if err := rows.Scan(
|
|
&song.ID, &song.Title, &song.ArtistID, &song.AlbumID, &song.Duration, &song.MediaFileID,
|
|
&song.TrackNumber, &song.DiscNumber, &song.Genre,
|
|
); err != nil {
|
|
return nil, err
|
|
}
|
|
songs = append(songs, song)
|
|
}
|
|
return songs, nil
|
|
}
|
|
|
|
func (r *SongRepository) GetAllWithDetails() ([]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
|
|
ORDER BY s.title
|
|
`)
|
|
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 *SongRepository) GetWithDetails(id int) (model.SongDetail, error) {
|
|
var song model.SongDetail
|
|
err := r.db.QueryRow(`
|
|
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.id = ?
|
|
`, id).Scan(&song.ID, &song.Title, &song.Artist, &song.Album, &song.Duration, &song.Path)
|
|
if err != nil {
|
|
return model.SongDetail{}, err
|
|
}
|
|
return song, nil
|
|
}
|
|
|
|
func (r *SongRepository) Get(id int) (model.Song, error) {
|
|
var song model.Song
|
|
err := r.db.QueryRow(`
|
|
SELECT id, title, artist_id, album_id, duration, media_file_id,
|
|
COALESCE(track_number, 0), COALESCE(disc_number, 0), COALESCE(genre, '')
|
|
FROM songs WHERE id = ?
|
|
`, id).Scan(
|
|
&song.ID, &song.Title, &song.ArtistID, &song.AlbumID, &song.Duration, &song.MediaFileID,
|
|
&song.TrackNumber, &song.DiscNumber, &song.Genre,
|
|
)
|
|
if err != nil {
|
|
return model.Song{}, err
|
|
}
|
|
return song, nil
|
|
}
|