arnef/godeez
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

feat(song): add FLAC tags

Browse Source
This commit is contained in:
Mathis Maquenne
2024-10-14 16:22:44 +02:00
parent 68f0b5c3e4
commit cfa53da6ae
9 changed files with 210 additions and 108 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/deezer/album.go
+15
View File
@@ -51,3 +51,18 @@ func GetAlbumData(id string) (*Album, error) {
return &album, nil
}
func (a *Album) GetCoverImage() ([]byte, error) {
url := fmt.Sprintf("https://e-cdn-images.dzcdn.net/images/cover/%s/500x500-000000-80-0-0.jpg", a.Data.CoverID)
resp, err := http.Get(url)
if err != nil {
return nil, err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return nil, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
}
return io.ReadAll(resp.Body)
}
internal/deezer/song.go
+5 -6
View File
@@ -20,12 +20,11 @@ type Song struct {
Composers []string `json:"composer"`
Authors []string `json:"author"`
} `json:"SNG_CONTRIBUTORS"`
Duration string `json:"DURATION"`
Gain string `json:"GAIN"`
ISRC string `json:"ISRC"`
ExplicitLyrics string `json:"EXPLICIT_LYRICS"`
TrackNumber string `json:"TRACK_NUMBER"`
TrackToken string `json:"TRACK_TOKEN"`
Duration string `json:"DURATION"`
Gain string `json:"GAIN"`
ISRC string `json:"ISRC"`
TrackNumber string `json:"TRACK_NUMBER"`
TrackToken string `json:"TRACK_TOKEN"`
}
func (s *Song) GetMediaData(quality string) (*Media, error) {
internal/deezer/tag.go
-101
View File
@@ -1,101 +0,0 @@
package deezer
import (
"fmt"
"io"
"net/http"
"path"
"strconv"
"strings"
"github.com/bogem/id3v2/v2"
)
func AddTags(album *Album, song *Song, pathFile string) error {
cover, err := getCoverImage(album.Data.CoverID)
if err != nil {
return err
}
ext := path.Ext(pathFile)
if ext == ".mp3" {
tag, err := id3v2.Open(pathFile, id3v2.Options{Parse: true})
if err != nil {
return err
}
defer tag.Close()
var year string
duration, _ := strconv.Atoi(song.Duration)
dateParts := strings.Split(album.Data.PhysicalReleaseDate, "-")
if len(dateParts) == 3 {
year = dateParts[0]
}
addTag(tag, "TALB", album.Data.Name)
addTag(tag, "TPE2", album.Data.Artist)
addTag(tag, "TPUB", album.Data.Label)
addTag(tag, "TDOR", album.Data.OriginalReleaseDate)
addTag(tag, "COMM", album.Data.ProducerLine)
if year != "" {
addTag(tag, "TYER", year)
}
addTag(tag, "TIT2", song.Title)
addTag(tag, "TPE1", strings.Join(song.Contributors.MainArtists, " / "))
addTag(tag, "TCOM", strings.Join(song.Contributors.Composers, " / "))
addTag(tag, "TEXT", strings.Join(song.Contributors.Authors, " / "))
addTag(tag, "TRCK", song.TrackNumber)
if duration > 0 {
addTag(tag, "TLEN", fmt.Sprintf("%d", duration*1000))
}
addTXXXTag(tag, "GAIN", song.Gain)
addTXXXTag(tag, "ISRC", song.ISRC)
addTXXXTag(tag, "Explicit lyrics", song.ExplicitLyrics)
frame := id3v2.PictureFrame{
Encoding: tag.DefaultEncoding(),
MimeType: "image/jpeg",
PictureType: id3v2.PTFrontCover,
Description: "Cover",
Picture: cover,
}
tag.AddAttachedPicture(frame)
return tag.Save()
}
return nil
}
func getCoverImage(id string) ([]byte, error) {
url := fmt.Sprintf("https://e-cdn-images.dzcdn.net/images/cover/%s/500x500-000000-80-0-0.jpg", id)
resp, err := http.Get(url)
if err != nil {
return nil, err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return nil, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
}
return io.ReadAll(resp.Body)
}
func addTag(tag *id3v2.Tag, name, value string) {
if value != "" {
tag.AddTextFrame(name, tag.DefaultEncoding(), value)
}
}
func addTXXXTag(tag *id3v2.Tag, description, value string) {
if value != "" {
udf := id3v2.UserDefinedTextFrame{
Encoding: tag.DefaultEncoding(),
Description: description,
Value: value,
}
tag.AddUserDefinedTextFrame(udf)
}
}
internal/tags/flac/tag.go
+86
View File
@@ -0,0 +1,86 @@
package flac
import (
"os"
"strings"
"github.com/go-flac/flacpicture/v2"
"github.com/go-flac/flacvorbis/v2"
"github.com/go-flac/go-flac/v2"
"github.com/mathismqn/godeez/internal/deezer"
)
func AddTags(album *deezer.Album, song *deezer.Song, cover []byte, path string) error {
file, err := flac.ParseFile(path)
if err != nil {
return err
}
cmts, idx, err := extractFLACComment(file)
if err != nil {
return err
}
if cmts == nil && idx > 0 {
cmts = flacvorbis.New()
}
addTag(cmts, "ALBUM", album.Data.Name)
addTag(cmts, "ALBUMARTIST", album.Data.Artist)
addTag(cmts, "PUBLISHER", album.Data.Label)
addTag(cmts, "ORIGINALDATE", album.Data.OriginalReleaseDate)
addTag(cmts, "DATE", album.Data.PhysicalReleaseDate)
addTag(cmts, "COMMENT", album.Data.ProducerLine)
addTag(cmts, "TITLE", song.Title)
addTag(cmts, "ARTIST", strings.Join(song.Contributors.MainArtists, " / "))
addTag(cmts, "COMPOSER", strings.Join(song.Contributors.Composers, " / "))
addTag(cmts, "LYRICIST", strings.Join(song.Contributors.Authors, " / "))
addTag(cmts, "TRACKNUMBER", song.TrackNumber)
addTag(cmts, "REPLAYGAIN_TRACK_GAIN", song.Gain)
addTag(cmts, "ISRC", song.ISRC)
cmtsmeta := cmts.Marshal()
if idx > 0 {
file.Meta[idx] = &cmtsmeta
} else {
file.Meta = append(file.Meta, &cmtsmeta)
}
picture, err := flacpicture.NewFromImageData(flacpicture.PictureTypeFrontCover, "Front cover", cover, "image/jpeg")
if err != nil {
return err
}
picturemeta := picture.Marshal()
file.Meta = append(file.Meta, &picturemeta)
return saveTags(file, path)
}
func addTag(cmts *flacvorbis.MetaDataBlockVorbisComment, name, value string) {
if value != "" {
cmts.Add(name, value)
}
}
func extractFLACComment(file *flac.File) (*flacvorbis.MetaDataBlockVorbisComment, int, error) {
var cmt *flacvorbis.MetaDataBlockVorbisComment
var cmtIdx int
var err error
for idx, meta := range file.Meta {
if meta.Type == flac.VorbisComment {
cmt, err = flacvorbis.ParseFromMetaDataBlock(*meta)
cmtIdx = idx
if err != nil {
return nil, 0, err
}
}
}
return cmt, cmtIdx, nil
}
func saveTags(file *flac.File, path string) error {
tempPath := path + ".tmp"
file.Save(tempPath)
return os.Rename(tempPath, path)
}
internal/tags/id3v2/tag.go
+60
View File
@@ -0,0 +1,60 @@
package id3v2
import (
"strings"
"github.com/bogem/id3v2/v2"
"github.com/mathismqn/godeez/internal/deezer"
)
func AddTags(album *deezer.Album, song *deezer.Song, cover []byte, path string) error {
tag, err := id3v2.Open(path, id3v2.Options{Parse: true})
if err != nil {
return err
}
defer tag.Close()
addTag(tag, "TALB", album.Data.Name)
addTag(tag, "TPE2", album.Data.Artist)
addTag(tag, "TPUB", album.Data.Label)
addTag(tag, "TDOR", album.Data.OriginalReleaseDate)
addTag(tag, "TYER", album.Data.PhysicalReleaseDate)
addTag(tag, "COMM", album.Data.ProducerLine)
addTag(tag, "TIT2", song.Title)
addTag(tag, "TPE1", strings.Join(song.Contributors.MainArtists, " / "))
addTag(tag, "TCOM", strings.Join(song.Contributors.Composers, " / "))
addTag(tag, "TEXT", strings.Join(song.Contributors.Authors, " / "))
addTag(tag, "TRCK", song.TrackNumber)
addTag(tag, "TLEN", song.Duration)
addTXXXTag(tag, "GAIN", song.Gain)
addTXXXTag(tag, "ISRC", song.ISRC)
frame := id3v2.PictureFrame{
Encoding: tag.DefaultEncoding(),
MimeType: "image/jpeg",
PictureType: id3v2.PTFrontCover,
Description: "Cover",
Picture: cover,
}
tag.AddAttachedPicture(frame)
return tag.Save()
}
func addTag(tag *id3v2.Tag, name, value string) {
if value != "" {
tag.AddTextFrame(name, tag.DefaultEncoding(), value)
}
}
func addTXXXTag(tag *id3v2.Tag, description, value string) {
if value != "" {
udf := id3v2.UserDefinedTextFrame{
Encoding: tag.DefaultEncoding(),
Description: description,
Value: value,
}
tag.AddUserDefinedTextFrame(udf)
}
}
internal/tags/tag.go
+33
View File
@@ -0,0 +1,33 @@
package tags
import (
"fmt"
"path"
"strconv"
"strings"
"github.com/mathismqn/godeez/internal/deezer"
"github.com/mathismqn/godeez/internal/tags/flac"
"github.com/mathismqn/godeez/internal/tags/id3v2"
)
func Add(album *deezer.Album, song *deezer.Song, pathFile string) error {
cover, err := album.GetCoverImage()
if err != nil {
return err
}
duration, _ := strconv.Atoi(song.Duration)
song.Duration = fmt.Sprintf("%d", duration*1000)
dateParts := strings.Split(album.Data.PhysicalReleaseDate, "-")
if len(dateParts) == 3 {
album.Data.PhysicalReleaseDate = dateParts[0]
}
ext := path.Ext(pathFile)
if ext == ".mp3" {
return id3v2.AddTags(album, song, cover, pathFile)
}
return flac.AddTags(album, song, cover, pathFile)
}