wip on autotagging

backend/database/sql/queries/audio_files.sql

@@ -2,6 +2,21 @@
 INSERT INTO audio_files (file_path, length_milliseconds, file_type_id, recording_id, sample_rate, bit_depth, channels, bitrate, file_size, basename, library_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
 RETURNING *;
 
+-- name: CreateAudioFileWithGroupKey :one
+INSERT INTO audio_files (
+  file_path, length_milliseconds, file_type_id, recording_id,
+  sample_rate, bit_depth, channels, bitrate, file_size, basename,
+  library_id, group_key, tag_status
+) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+RETURNING *;
+
+-- name: GetAudioFileGroupKey :one
+SELECT group_key FROM audio_files
+WHERE id = ? LIMIT 1;
+
+-- name: SetAudioFileGroupKey :exec
+UPDATE audio_files SET group_key = ? WHERE id = ?;
+
 -- name: GetAudioFile :one
 SELECT * FROM audio_files 
 WHERE id = ? LIMIT 1;

backend/database/sql/queries/libraries.sql

@@ -19,3 +19,6 @@ DELETE FROM libraries WHERE id = ?;
 
 -- name: CountLibraries :one
 SELECT COUNT(*) AS count FROM libraries;
+
+-- name: AckLibraryAutotagWarning :exec
+UPDATE libraries SET autotag_warning_acked = 1 WHERE id = ?;

backend/database/sql/queries/release_groups.sql

@@ -24,6 +24,13 @@ ON CONFLICT(name, album_artist_credit_id) DO UPDATE SET
   year = COALESCE(excluded.year, release_groups.year)
 RETURNING *;
 
+-- name: SetReleaseGroupOriginalYear :exec
+-- Set the release group's original-release-year (release-group's
+-- first-release-date from MusicBrainz).  Called from autotag apply
+-- when the user confirms a candidate; the file-tag year stays in
+-- the year column.
+UPDATE release_groups SET original_year = ? WHERE id = ?;
+
 -- name: UpdateReleaseGroup :exec
 UPDATE release_groups 
 SET name = ?
@@ -49,7 +56,12 @@ ORDER BY name;
 SELECT
     rg.id,
     rg.name,
-    rg.year,
+    -- year prefers original release year (MB first-release-date)
+    -- over the file-tag year so the UI surfaces the album's
+    -- original year by default.  release_year keeps the file-tag
+    -- year accessible.
+    COALESCE(rg.original_year, rg.year) AS year,
+    COALESCE(rg.year, 0) AS release_year,
     rg.mbid,
     COALESCE(ac.text, fallback_ac.text, '') as artist_name,
     COALESCE(ca.file_path, '') as cover_art_path
@@ -69,7 +81,12 @@ ORDER BY rg.name;
 SELECT
     rg.id,
     rg.name,
-    rg.year,
+    -- year prefers original release year (MB first-release-date)
+    -- over the file-tag year so the UI surfaces the album's
+    -- original year by default.  release_year keeps the file-tag
+    -- year accessible.
+    COALESCE(rg.original_year, rg.year) AS year,
+    COALESCE(rg.year, 0) AS release_year,
     rg.mbid,
     COALESCE(ac.text, fallback_ac.text, '') as artist_name,
     COALESCE(ca.file_path, '') as cover_art_path
@@ -96,7 +113,8 @@ ORDER BY rg.name;
 SELECT
     rg.id,
     rg.name,
-    rg.year,
+    COALESCE(rg.original_year, rg.year) AS year,
+    COALESCE(rg.year, 0) AS release_year,
     COALESCE(ac.text, fallback_ac.text, '') as artist_name,
     COALESCE(ca.file_path, '') as cover_art_path
 FROM release_groups rg
@@ -120,7 +138,8 @@ SELECT COUNT(*) FROM release_group_recordings WHERE release_group_id = ?;
 SELECT
     rg.id,
     rg.name,
-    rg.year,
+    COALESCE(rg.original_year, rg.year) AS year,
+    COALESCE(rg.year, 0) AS release_year,
     COALESCE(ac.text, fallback_ac.text, '') as artist_name,
     COALESCE(ca.file_path, '') as cover_art_path
 FROM release_groups rg

backend/database/sql/queries/tagging_items.sql

@@ -0,0 +1,236 @@
+-- name: UpsertTaggingItemOnTrackAdd :exec
+INSERT INTO tagging_items (
+  group_key, library_id, track_count,
+  album_name, album_artist, disc_number, status
+)
+VALUES (?, ?, 1, ?, ?, ?, 'pending')
+ON CONFLICT(group_key) DO UPDATE SET
+  track_count = tagging_items.track_count + 1,
+  album_name = CASE
+    WHEN tagging_items.album_name = '' THEN excluded.album_name
+    ELSE tagging_items.album_name
+  END,
+  album_artist = CASE
+    WHEN tagging_items.album_artist = '' THEN excluded.album_artist
+    ELSE tagging_items.album_artist
+  END;
+
+-- name: DecrementTaggingItemTrackCount :exec
+UPDATE tagging_items
+SET track_count = track_count - 1
+WHERE group_key = ?;
+
+-- name: DeleteTaggingItemIfEmpty :exec
+DELETE FROM tagging_items
+WHERE group_key = ? AND track_count <= 0;
+
+-- name: GetTaggingItem :one
+SELECT * FROM tagging_items
+WHERE group_key = ?
+LIMIT 1;
+
+-- name: CountPendingTaggingItems :one
+SELECT COUNT(*) FROM tagging_items
+WHERE status = 'pending'
+  AND (CAST(@library_id AS INTEGER) = 0 OR library_id = @library_id);
+
+-- name: ListPendingTaggingItemsAlphabetical :many
+SELECT
+  ti.group_key,
+  ti.library_id,
+  COALESCE(lb.name, '') AS library_name,
+  ti.track_count,
+  ti.album_name,
+  ti.album_artist,
+  ti.disc_number,
+  ti.best_match_release_mbid,
+  ti.score,
+  ti.last_checked_at,
+  ti.status,
+  ti.created_at
+FROM tagging_items ti
+LEFT JOIN libraries lb ON lb.id = ti.library_id
+WHERE (CAST(@library_id AS INTEGER) = 0 OR ti.library_id = @library_id)
+  AND (CAST(@status_filter AS TEXT) = 'all' OR ti.status = @status_filter)
+  AND ti.cleared_at IS NULL
+ORDER BY LOWER(ti.album_artist), LOWER(ti.album_name), ti.disc_number
+LIMIT @row_limit OFFSET @row_offset;
+
+-- name: ListPendingTaggingItemsByScore :many
+SELECT
+  ti.group_key,
+  ti.library_id,
+  COALESCE(lb.name, '') AS library_name,
+  COALESCE(lb.path, '') AS library_path,
+  CAST(COALESCE((SELECT af.file_path FROM audio_files af WHERE af.group_key = ti.group_key LIMIT 1), '') AS TEXT) AS sample_file_path,
+  ti.track_count,
+  ti.album_name,
+  ti.album_artist,
+  ti.disc_number,
+  ti.best_match_release_mbid,
+  ti.score,
+  ti.last_checked_at,
+  ti.status,
+  ti.created_at
+FROM tagging_items ti
+LEFT JOIN libraries lb ON lb.id = ti.library_id
+WHERE (CAST(@library_id AS INTEGER) = 0 OR ti.library_id = @library_id)
+  AND (CAST(@status_filter AS TEXT) = 'all' OR ti.status = @status_filter)
+  AND ti.cleared_at IS NULL
+ORDER BY ti.score IS NULL, ti.score DESC, LOWER(ti.album_artist), LOWER(ti.album_name)
+LIMIT @row_limit OFFSET @row_offset;
+
+-- name: ClearCompletedTaggingItems :exec
+UPDATE tagging_items
+SET cleared_at = CURRENT_TIMESTAMP
+WHERE status = 'confirmed'
+  AND cleared_at IS NULL
+  AND (CAST(@library_id AS INTEGER) = 0 OR library_id = @library_id);
+
+-- name: GetPendingFolderDetail :one
+SELECT
+  ti.group_key,
+  ti.library_id,
+  COALESCE(lb.name, '') AS library_name,
+  COALESCE(lb.path, '') AS library_path,
+  CAST(COALESCE((SELECT af.file_path FROM audio_files af WHERE af.group_key = ti.group_key LIMIT 1), '') AS TEXT) AS sample_file_path,
+  ti.track_count,
+  ti.album_name,
+  ti.album_artist,
+  ti.disc_number,
+  ti.best_match_release_mbid,
+  ti.score,
+  ti.last_checked_at,
+  ti.status,
+  ti.created_at
+FROM tagging_items ti
+LEFT JOIN libraries lb ON lb.id = ti.library_id
+WHERE ti.group_key = ?
+LIMIT 1;
+
+-- name: ListPendingTaggingItemsByRecent :many
+SELECT
+  ti.group_key,
+  ti.library_id,
+  COALESCE(lb.name, '') AS library_name,
+  ti.track_count,
+  ti.album_name,
+  ti.album_artist,
+  ti.disc_number,
+  ti.best_match_release_mbid,
+  ti.score,
+  ti.last_checked_at,
+  ti.status,
+  ti.created_at
+FROM tagging_items ti
+LEFT JOIN libraries lb ON lb.id = ti.library_id
+WHERE (CAST(@library_id AS INTEGER) = 0 OR ti.library_id = @library_id)
+  AND (CAST(@status_filter AS TEXT) = 'all' OR ti.status = @status_filter)
+ORDER BY ti.created_at DESC, ti.group_key
+LIMIT @row_limit OFFSET @row_offset;
+
+-- name: ListAudioFilesInTaggingGroup :many
+SELECT
+  af.id,
+  af.file_path,
+  af.basename,
+  af.length_milliseconds,
+  af.tag_status,
+  COALESCE(r.track_number, 0) AS track_number,
+  COALESCE(r.disc_number, 0) AS disc_number,
+  COALESCE(r.name, '') AS title,
+  COALESCE(ac.text, '') AS artist_name,
+  COALESCE(r.mbid, '') AS recording_mbid
+FROM audio_files af
+LEFT JOIN recordings r ON af.recording_id = r.id
+LEFT JOIN artist_credit ac ON r.artist_credit_id = ac.id
+WHERE af.group_key = ?
+ORDER BY COALESCE(r.disc_number, 0),
+         COALESCE(r.track_number, 0),
+         af.file_path;
+
+-- name: ListLocalReleaseGroupCandidates :many
+-- Returns one row per (release_group, track) combination for any
+-- local release_group that has an MBID.  Callers group these in Go
+-- and filter by normalized album-name match.  Joined case-insensitive
+-- on name to pre-filter cheaply; Go does the real normalization.
+SELECT
+  rg.id AS release_group_id,
+  rg.mbid AS release_group_mbid,
+  rg.name AS album_name,
+  COALESCE(rg.year, 0) AS year,
+  COALESCE(ac.text, '') AS artist_credit,
+  COALESCE(rgr.track_number, 0) AS track_number,
+  COALESCE(rgr.disc_number, 0) AS disc_number,
+  COALESCE(r.name, '') AS track_title,
+  COALESCE(r.mbid, '') AS recording_mbid,
+  COALESCE(local_af.length_milliseconds, 0) AS length_milliseconds
+FROM release_groups rg
+JOIN release_group_recordings rgr ON rgr.release_group_id = rg.id
+JOIN recordings r ON r.id = rgr.recording_id
+LEFT JOIN artist_credit ac ON rg.album_artist_credit_id = ac.id
+LEFT JOIN audio_files local_af ON local_af.recording_id = r.id
+WHERE rg.mbid IS NOT NULL
+  AND rg.mbid != ''
+  AND r.mbid IS NOT NULL
+  AND r.mbid != ''
+  AND rg.name = ? COLLATE NOCASE
+ORDER BY rg.id, rgr.disc_number, rgr.track_number;
+
+-- name: SetTaggingItemBestMatch :exec
+UPDATE tagging_items
+SET best_match_release_mbid = ?,
+    score = ?,
+    status = ?,
+    last_checked_at = CURRENT_TIMESTAMP
+WHERE group_key = ?;
+
+-- name: SetTaggingItemScore :exec
+UPDATE tagging_items
+SET best_match_release_mbid = ?,
+    score = ?,
+    last_checked_at = CURRENT_TIMESTAMP
+WHERE group_key = ?;
+
+-- name: SetTaggingItemStatus :exec
+UPDATE tagging_items
+SET status = ?,
+    last_checked_at = CURRENT_TIMESTAMP
+WHERE group_key = ?;
+
+-- name: SetAudioFileTagStatus :exec
+UPDATE audio_files SET tag_status = ? WHERE id = ?;
+
+-- name: SetRecordingMBID :exec
+UPDATE recordings SET mbid = ? WHERE id = ?;
+
+-- name: SetReleaseGroupMBID :exec
+UPDATE release_groups SET mbid = ? WHERE id = ?;
+
+-- name: GetRecordingReleaseGroupID :one
+SELECT COALESCE(rgr.release_group_id, 0) AS release_group_id
+FROM release_group_recordings rgr
+WHERE rgr.recording_id = ?
+LIMIT 1;
+
+-- name: GetNextPendingTaggingItem :one
+SELECT
+  ti.group_key,
+  ti.library_id,
+  COALESCE(lb.name, '') AS library_name,
+  ti.track_count,
+  ti.album_name,
+  ti.album_artist,
+  ti.disc_number,
+  ti.best_match_release_mbid,
+  ti.score,
+  ti.last_checked_at,
+  ti.status,
+  ti.created_at
+FROM tagging_items ti
+LEFT JOIN libraries lb ON lb.id = ti.library_id
+WHERE ti.status = 'pending'
+  AND (CAST(@library_id AS INTEGER) = 0 OR ti.library_id = @library_id)
+  AND ti.group_key > @after_group_key
+ORDER BY ti.group_key
+LIMIT 1;