feat: add metadata embedding

lib/download.go

@@ -69,9 +69,9 @@ func (app *App) Download(url string, outputFile string, serviceString string, qu
 		return err
 	}
 
-	fileExists, err := FileExists(outputFile) 
+	fileExists, err := FileExists(outputFile)
 	if err != nil {
-		return err 
+		return err
 	}
 
 	if fileExists {
@@ -84,6 +84,11 @@ func (app *App) Download(url string, outputFile string, serviceString string, qu
 		return err
 	}
 
+	err = app.EmbedMetadata(outputFile, metadata)
+	if err != nil {
+		return err
+	}
+
 	return nil
 }
 

lib/metadata.go

@@ -3,6 +3,8 @@ package lib
 import (
 	"encoding/json"
 	"errors"
+
+	id3v2 "github.com/bogem/id3v2/v2"
 )
 
 func (app *App) GetTrackMetadata(url string) (TrackMetadata, error) {
@@ -40,3 +42,26 @@ func (app *App) GetTrackMetadata(url string) (TrackMetadata, error) {
 func (app *App) PrintMetadata(url string) error {
 	return errors.New("Unimplemented.")
 }
+
+func (app *App) EmbedMetadata(file string, metadata TrackMetadata) error {
+	tag, err := id3v2.Open(file, id3v2.Options{Parse: true})
+	if err != nil {
+		return err
+	}
+
+	artists, err := GetArtists(metadata)
+	if err != nil {
+		return err
+	}
+
+	tag.SetArtist(artists)
+	tag.SetTitle(metadata.Data.TrackUnion.Name)
+	tag.SetYear(string(metadata.Data.TrackUnion.AlbumOfTrack.Date.Year))
+	tag.SetAlbum(metadata.Data.TrackUnion.AlbumOfTrack.Name)
+
+	if err = tag.Save(); err != nil {
+		return err
+	}
+
+	return nil
+}

lib/utils.go

@@ -44,20 +44,30 @@ func ParseTrackId(url string) (string, error) {
 
 func BuildFileName(metadata TrackMetadata, extension string) (string, error) {
 	var result string
-	var artists string
+
+	artists, err := GetArtists(metadata)
+	if err != nil {
+		return result, err
+	}
+
+	result = fmt.Sprintf("%s - %s.%s", metadata.Data.TrackUnion.Name, artists, extension)
+
+	return result, nil
+}
+
+func GetArtists(metadata TrackMetadata) (string, error) {
+	var result string
 
 	firstArtistLen := len(metadata.Data.TrackUnion.FirstArtist.Items)
 	if firstArtistLen == 0 {
 		return result, errors.New("What? This should never happen.")
 	}
-	artists = metadata.Data.TrackUnion.FirstArtist.Items[firstArtistLen-1].Profile.Name
+	result = metadata.Data.TrackUnion.FirstArtist.Items[firstArtistLen-1].Profile.Name
 
 	for _, artist := range metadata.Data.TrackUnion.OtherArtists.Items {
-		artists += ", " + artist.Profile.Name
+		result += ", " + artist.Profile.Name
 	}
 
-	result = fmt.Sprintf("%s - %s.%s", metadata.Data.TrackUnion.Name, artists, extension)
-
 	return result, nil
 }