feat(server/m7-379): extract MusicBrainz IDs from audio tags during scan
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,59 @@
|
|||||||
|
package library
|
||||||
|
|
||||||
|
// MusicBrainz tag keys vary by audio format. dhowden/tag's Raw() map exposes
|
||||||
|
// the format-specific entries as-is — so we look up multiple candidates and
|
||||||
|
// take the first non-empty match.
|
||||||
|
//
|
||||||
|
// References:
|
||||||
|
// - ID3v2 (MP3): TXXX frames keyed by description.
|
||||||
|
// - Vorbis (FLAC/OGG): all-caps comment names.
|
||||||
|
// - MP4 (M4A): iTunes freeform atoms ("----:com.apple.iTunes:...").
|
||||||
|
|
||||||
|
var albumMBIDKeys = []string{
|
||||||
|
"TXXX:MusicBrainz Album Id",
|
||||||
|
"MUSICBRAINZ_ALBUMID",
|
||||||
|
"----:com.apple.iTunes:MusicBrainz Album Id",
|
||||||
|
}
|
||||||
|
|
||||||
|
var artistMBIDKeys = []string{
|
||||||
|
"TXXX:MusicBrainz Artist Id",
|
||||||
|
"MUSICBRAINZ_ARTISTID",
|
||||||
|
"----:com.apple.iTunes:MusicBrainz Artist Id",
|
||||||
|
}
|
||||||
|
|
||||||
|
// extractMBIDs reads MusicBrainz album and artist IDs from a tag.Metadata
|
||||||
|
// Raw() map. Returns ("", "") when the keys are absent. Multi-value keys
|
||||||
|
// (some formats list multiple artist IDs for collaborations) take only the
|
||||||
|
// first value — Minstrel uses the primary release artist for MBCAA lookup.
|
||||||
|
func extractMBIDs(raw map[string]interface{}) (albumMBID, artistMBID string) {
|
||||||
|
albumMBID = firstStringValue(raw, albumMBIDKeys)
|
||||||
|
artistMBID = firstStringValue(raw, artistMBIDKeys)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func firstStringValue(raw map[string]interface{}, keys []string) string {
|
||||||
|
for _, k := range keys {
|
||||||
|
v, ok := raw[k]
|
||||||
|
if !ok || v == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
switch x := v.(type) {
|
||||||
|
case string:
|
||||||
|
if x != "" {
|
||||||
|
return x
|
||||||
|
}
|
||||||
|
case []string:
|
||||||
|
if len(x) > 0 && x[0] != "" {
|
||||||
|
return x[0]
|
||||||
|
}
|
||||||
|
case []interface{}:
|
||||||
|
if len(x) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if s, ok := x[0].(string); ok && s != "" {
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
@@ -0,0 +1,69 @@
|
|||||||
|
package library
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
func TestExtractMBIDs_ID3v2(t *testing.T) {
|
||||||
|
raw := map[string]interface{}{
|
||||||
|
"TXXX:MusicBrainz Album Id": "abc-123",
|
||||||
|
"TXXX:MusicBrainz Artist Id": "def-456",
|
||||||
|
}
|
||||||
|
a, ar := extractMBIDs(raw)
|
||||||
|
if a != "abc-123" || ar != "def-456" {
|
||||||
|
t.Errorf("got (%q, %q), want (\"abc-123\", \"def-456\")", a, ar)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestExtractMBIDs_Vorbis(t *testing.T) {
|
||||||
|
raw := map[string]interface{}{
|
||||||
|
"MUSICBRAINZ_ALBUMID": "abc",
|
||||||
|
"MUSICBRAINZ_ARTISTID": "def",
|
||||||
|
}
|
||||||
|
a, ar := extractMBIDs(raw)
|
||||||
|
if a != "abc" || ar != "def" {
|
||||||
|
t.Errorf("got (%q, %q)", a, ar)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestExtractMBIDs_MP4(t *testing.T) {
|
||||||
|
raw := map[string]interface{}{
|
||||||
|
"----:com.apple.iTunes:MusicBrainz Album Id": "abc",
|
||||||
|
"----:com.apple.iTunes:MusicBrainz Artist Id": "def",
|
||||||
|
}
|
||||||
|
a, ar := extractMBIDs(raw)
|
||||||
|
if a != "abc" || ar != "def" {
|
||||||
|
t.Errorf("got (%q, %q)", a, ar)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestExtractMBIDs_MissingKeysReturnEmpty(t *testing.T) {
|
||||||
|
a, ar := extractMBIDs(map[string]interface{}{})
|
||||||
|
if a != "" || ar != "" {
|
||||||
|
t.Errorf("got (%q, %q), want empty", a, ar)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestExtractMBIDs_MultiValueArtistTakesFirst(t *testing.T) {
|
||||||
|
// Some FLAC tags list collaborators as multi-value MUSICBRAINZ_ARTISTID.
|
||||||
|
// Minstrel uses the primary release artist; first wins.
|
||||||
|
raw := map[string]interface{}{
|
||||||
|
"MUSICBRAINZ_ARTISTID": []string{"primary-mbid", "collab-mbid"},
|
||||||
|
}
|
||||||
|
_, ar := extractMBIDs(raw)
|
||||||
|
if ar != "primary-mbid" {
|
||||||
|
t.Errorf("got %q, want %q", ar, "primary-mbid")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestExtractMBIDs_NilEntryIgnored(t *testing.T) {
|
||||||
|
raw := map[string]interface{}{
|
||||||
|
"TXXX:MusicBrainz Album Id": nil,
|
||||||
|
"TXXX:MusicBrainz Artist Id": "real-id",
|
||||||
|
}
|
||||||
|
a, ar := extractMBIDs(raw)
|
||||||
|
if a != "" {
|
||||||
|
t.Errorf("nil album entry should be ignored, got %q", a)
|
||||||
|
}
|
||||||
|
if ar != "real-id" {
|
||||||
|
t.Errorf("artist = %q", ar)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -131,6 +131,7 @@ func (s *Scanner) scanFile(ctx context.Context, q *dbq.Queries, path string, sta
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("tag read: %w", err)
|
return fmt.Errorf("tag read: %w", err)
|
||||||
}
|
}
|
||||||
|
albumMBID, artistMBID := extractMBIDs(meta.Raw())
|
||||||
|
|
||||||
artistName := meta.Artist()
|
artistName := meta.Artist()
|
||||||
if artistName == "" {
|
if artistName == "" {
|
||||||
@@ -145,11 +146,11 @@ func (s *Scanner) scanFile(ctx context.Context, q *dbq.Queries, path string, sta
|
|||||||
trackTitle = strings.TrimSuffix(filepath.Base(path), filepath.Ext(path))
|
trackTitle = strings.TrimSuffix(filepath.Base(path), filepath.Ext(path))
|
||||||
}
|
}
|
||||||
|
|
||||||
artist, err := s.resolveArtist(ctx, q, artistName)
|
artist, err := s.resolveArtist(ctx, q, artistName, artistMBID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("artist: %w", err)
|
return fmt.Errorf("artist: %w", err)
|
||||||
}
|
}
|
||||||
album, err := s.resolveAlbum(ctx, q, artist.ID, albumTitle, meta.Year())
|
album, err := s.resolveAlbum(ctx, q, artist.ID, albumTitle, meta.Year(), albumMBID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("album: %w", err)
|
return fmt.Errorf("album: %w", err)
|
||||||
}
|
}
|
||||||
@@ -197,7 +198,7 @@ func (s *Scanner) scanFile(ctx context.Context, q *dbq.Queries, path string, sta
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Scanner) resolveArtist(ctx context.Context, q *dbq.Queries, name string) (dbq.Artist, error) {
|
func (s *Scanner) resolveArtist(ctx context.Context, q *dbq.Queries, name, mbid string) (dbq.Artist, error) {
|
||||||
existing, err := q.GetArtistByName(ctx, name)
|
existing, err := q.GetArtistByName(ctx, name)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
return existing, nil
|
return existing, nil
|
||||||
@@ -205,13 +206,18 @@ func (s *Scanner) resolveArtist(ctx context.Context, q *dbq.Queries, name string
|
|||||||
if !errors.Is(err, pgx.ErrNoRows) {
|
if !errors.Is(err, pgx.ErrNoRows) {
|
||||||
return dbq.Artist{}, err
|
return dbq.Artist{}, err
|
||||||
}
|
}
|
||||||
return q.UpsertArtist(ctx, dbq.UpsertArtistParams{
|
params := dbq.UpsertArtistParams{
|
||||||
Name: name,
|
Name: name,
|
||||||
SortName: sortKey(name),
|
SortName: sortKey(name),
|
||||||
})
|
}
|
||||||
|
if mbid != "" {
|
||||||
|
m := mbid
|
||||||
|
params.Mbid = &m
|
||||||
|
}
|
||||||
|
return q.UpsertArtist(ctx, params)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Scanner) resolveAlbum(ctx context.Context, q *dbq.Queries, artistID pgtype.UUID, title string, year int) (dbq.Album, error) {
|
func (s *Scanner) resolveAlbum(ctx context.Context, q *dbq.Queries, artistID pgtype.UUID, title string, year int, mbid string) (dbq.Album, error) {
|
||||||
existing, err := q.GetAlbumByArtistAndTitle(ctx, dbq.GetAlbumByArtistAndTitleParams{ArtistID: artistID, Title: title})
|
existing, err := q.GetAlbumByArtistAndTitle(ctx, dbq.GetAlbumByArtistAndTitleParams{ArtistID: artistID, Title: title})
|
||||||
if err == nil {
|
if err == nil {
|
||||||
return existing, nil
|
return existing, nil
|
||||||
@@ -233,6 +239,10 @@ func (s *Scanner) resolveAlbum(ctx context.Context, q *dbq.Queries, artistID pgt
|
|||||||
s.logger.Warn("library scan: dropping invalid release year",
|
s.logger.Warn("library scan: dropping invalid release year",
|
||||||
"year", year, "album", title)
|
"year", year, "album", title)
|
||||||
}
|
}
|
||||||
|
if mbid != "" {
|
||||||
|
m := mbid
|
||||||
|
params.Mbid = &m
|
||||||
|
}
|
||||||
return q.UpsertAlbum(ctx, params)
|
return q.UpsertAlbum(ctx, params)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user