weee

internal/database/coverart.go

@@ -11,6 +11,7 @@ type CoverArt struct {
 	ID          string
 	AlbumID     sql.NullString
 	MediaFileID sql.NullString
+	ArtistID    sql.NullString
 	Source      string
 	SourcePath  string
 	MimeType    string
@@ -19,7 +20,7 @@ type CoverArt struct {
 	Height      int
 }
 
-const coverArtSelect = `SELECT id, album_id, media_file_id, source, source_path, mime_type, content_hash, width, height FROM cover_art`
+const coverArtSelect = `SELECT id, album_id, media_file_id, artist_id, source, source_path, mime_type, content_hash, width, height FROM cover_art`
 
 func GetCoverArt(db *sql.DB, id string) (*CoverArt, error) {
 	return scanCoverArt(db.QueryRow(coverArtSelect+` WHERE id = ?`, id))
@@ -31,7 +32,7 @@ func GetCoverArtForAlbum(db *sql.DB, albumID string) (*CoverArt, error) {
 
 func scanCoverArt(row *sql.Row) (*CoverArt, error) {
 	var c CoverArt
-	err := row.Scan(&c.ID, &c.AlbumID, &c.MediaFileID, &c.Source, &c.SourcePath, &c.MimeType, &c.ContentHash, &c.Width, &c.Height)
+	err := row.Scan(&c.ID, &c.AlbumID, &c.MediaFileID, &c.ArtistID, &c.Source, &c.SourcePath, &c.MimeType, &c.ContentHash, &c.Width, &c.Height)
 	if errors.Is(err, sql.ErrNoRows) {
 		return nil, ErrNotFound
 	}
@@ -71,6 +72,36 @@ func UpsertAlbumCoverArt(db *sql.DB, albumID, source, sourcePath, mediaFileID, m
 	return GetCoverArt(db, id)
 }
 
+// UpsertArtistCoverArt records (or replaces) an artist's photo and links
+// it as the artist's cover art. manual=true marks it as an admin
+// override (see database.SetArtistCoverArt); manual=false is the
+// auto-fetch path, which never overwrites an existing manual override.
+func UpsertArtistCoverArt(db *sql.DB, artistID, source, sourcePath, mimeType, contentHash string, width, height int, manual bool) (*CoverArt, error) {
+	var id string
+	err := db.QueryRow(`SELECT id FROM cover_art WHERE artist_id = ?`, artistID).Scan(&id)
+	switch {
+	case err == nil:
+		_, err = db.Exec(`UPDATE cover_art SET source=?, source_path=?, mime_type=?, content_hash=?, width=?, height=?, updated_at=strftime('%Y-%m-%dT%H:%M:%fZ','now') WHERE id=?`,
+			source, sourcePath, mimeType, contentHash, width, height, id)
+		if err != nil {
+			return nil, err
+		}
+	case errors.Is(err, sql.ErrNoRows):
+		id = idgen.New("ca")
+		_, err = db.Exec(`INSERT INTO cover_art (id, artist_id, source, source_path, mime_type, content_hash, width, height) VALUES (?, ?, ?, ?, ?, ?, ?, ?)`,
+			id, artistID, source, sourcePath, mimeType, contentHash, width, height)
+		if err != nil {
+			return nil, err
+		}
+	default:
+		return nil, err
+	}
+	if err := SetArtistCoverArt(db, artistID, sql.NullString{String: id, Valid: true}, manual); err != nil {
+		return nil, err
+	}
+	return GetCoverArt(db, id)
+}
+
 func nullEmptyStr(s string) any {
 	if s == "" {
 		return nil