Skip to content

Commit

Permalink
run format
Browse files Browse the repository at this point in the history
  • Loading branch information
@bolsinga
bolsinga committed Oct 27, 2023
1 parent b7cf06a commit b35a5a2
Show file tree
Hide file tree
Showing 2 changed files with 78 additions and 77 deletions.
3 changes: 2 additions & 1 deletion Sources/MissingArtwork/LoadingState+Artwork.swift
View file
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,8 @@ extension NoArtworkError: LocalizedError {
switch self {
case .noneFound(let term):
return String(
localized: "Image Search was unable to find images for \"\(term)\". Unable to repair artwork without an image.",
localized:
"Image Search was unable to find images for \"\(term)\". Unable to repair artwork without an image.",
bundle: .module,
comment: "Error message when no Missing Artworks are found for search term.")
}
Expand Down
152 changes: 76 additions & 76 deletions Sources/MissingArtwork/MissingArtwork+MusicKit.swift
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,105 +5,105 @@
// Created by Greg Bolsinga on 3/19/23.
//

import MusicKit
import LoadingState
import LoadingState
import MusicKit

extension Song {
func matches(_ missingArtwork: MissingArtwork) -> Bool {
if let album = self.albums?.first {
switch missingArtwork {
case .ArtistAlbum(let artistName, let albumName, _):
return self.artistName == artistName && album.title == albumName
case .CompilationAlbum(let albumName, _):
return (album.isCompilation != nil || album.isCompilation ?? false)
&& album.title == albumName
}
extension Song {
func matches(_ missingArtwork: MissingArtwork) -> Bool {
if let album = self.albums?.first {
switch missingArtwork {
case .ArtistAlbum(let artistName, let albumName, _):
return self.artistName == artistName && album.title == albumName
case .CompilationAlbum(let albumName, _):
return (album.isCompilation != nil || album.isCompilation ?? false)
&& album.title == albumName
}
return false
}
return false
}
}

extension MissingArtwork {
public static func gatherMissingArtwork() async throws -> [MissingArtwork] {
var partial = [MissingArtwork: [Int: Int]]() // MissingItem : [discNumber: missingArtworkCount]
extension MissingArtwork {
public static func gatherMissingArtwork() async throws -> [MissingArtwork] {
var partial = [MissingArtwork: [Int: Int]]() // MissingItem : [discNumber: missingArtworkCount]

let request = MusicLibraryRequest<Song>()
let response = try await request.response()
let missingArtworkSongs = response.items.filter { $0.artwork == nil }
let request = MusicLibraryRequest<Song>()
let response = try await request.response()
let missingArtworkSongs = response.items.filter { $0.artwork == nil }

for song in missingArtworkSongs {
let refinedSong = try await song.with([.albums], preferredSource: .library)
if let album = refinedSong.albums?.first {
let missingArtwork =
(album.isCompilation ?? false)
? MissingArtwork.CompilationAlbum(album.title, .unknown)
: MissingArtwork.ArtistAlbum(refinedSong.artistName, album.title, .unknown)
for song in missingArtworkSongs {
let refinedSong = try await song.with([.albums], preferredSource: .library)
if let album = refinedSong.albums?.first {
let missingArtwork =
(album.isCompilation ?? false)
? MissingArtwork.CompilationAlbum(album.title, .unknown)
: MissingArtwork.ArtistAlbum(refinedSong.artistName, album.title, .unknown)

let discNumber = refinedSong.discNumber ?? 0
let discNumber = refinedSong.discNumber ?? 0

if let albumInfo = partial[missingArtwork] {
// We have tracked this missingArtwork already
if let trackCount = albumInfo[discNumber] {
// We have tracked this missingArtwork and discNumber
partial[missingArtwork]?[discNumber] = trackCount - 1
} else {
// We have tracked this missingArtwork but not this discNumber
let albumTrackCount = album.trackCount
partial[missingArtwork]?[discNumber] =
albumTrackCount == 0 ? -1 : albumTrackCount - 1
}
if let albumInfo = partial[missingArtwork] {
// We have tracked this missingArtwork already
if let trackCount = albumInfo[discNumber] {
// We have tracked this missingArtwork and discNumber
partial[missingArtwork]?[discNumber] = trackCount - 1
} else {
// We have not tracked this missingArtwork.
// We have tracked this missingArtwork but not this discNumber
let albumTrackCount = album.trackCount
partial[missingArtwork] = [
discNumber: albumTrackCount == 0 ? -1 : albumTrackCount - 1
]
partial[missingArtwork]?[discNumber] =
albumTrackCount == 0 ? -1 : albumTrackCount - 1
}
} else {
// We have not tracked this missingArtwork.
let albumTrackCount = album.trackCount
partial[missingArtwork] = [
discNumber: albumTrackCount == 0 ? -1 : albumTrackCount - 1
]
}
}
}

return partial.map { (key: MissingArtwork, albumInfo: [Int: Int]) in
let value = albumInfo.values.reduce(
0,
{ x, y in
x + y
})
return partial.map { (key: MissingArtwork, albumInfo: [Int: Int]) in
let value = albumInfo.values.reduce(
0,
{ x, y in
x + y
})

let availability: ArtworkAvailability = value < 0 ? .unknown : (value == 0 ? .none : .some)
var item: MissingArtwork
switch key {
case .ArtistAlbum(let artist, let album, _):
item = .ArtistAlbum(artist, album, availability)
case .CompilationAlbum(let album, _):
item = .CompilationAlbum(album, availability)
}
return item
let availability: ArtworkAvailability = value < 0 ? .unknown : (value == 0 ? .none : .some)
var item: MissingArtwork
switch key {
case .ArtistAlbum(let artist, let album, _):
item = .ArtistAlbum(artist, album, availability)
case .CompilationAlbum(let album, _):
item = .CompilationAlbum(album, availability)
}
return item
}
}

public func matchingPartialArtworkImage() async throws -> PlatformImage {
guard self.availability == .some else {
fatalError(
"Unable to get partial image for this MissingArtwork: \(String(describing: self))")
}
public func matchingPartialArtworkImage() async throws -> PlatformImage {
guard self.availability == .some else {
fatalError(
"Unable to get partial image for this MissingArtwork: \(String(describing: self))")
}

let request = MusicLibraryRequest<Song>()
let response = try await request.response()
let artworkSongs = response.items.filter { $0.artwork != nil }
let request = MusicLibraryRequest<Song>()
let response = try await request.response()
let artworkSongs = response.items.filter { $0.artwork != nil }

for artworkSong in artworkSongs {
if let artwork = artworkSong.artwork {
let refinedSong = try await artworkSong.with([.albums], preferredSource: .library)
if refinedSong.matches(self) {
var loadingState: LoadingState<PlatformImage> = .idle
await loadingState.load(artwork: artwork)
if let platformImage = loadingState.value {
return platformImage
}
for artworkSong in artworkSongs {
if let artwork = artworkSong.artwork {
let refinedSong = try await artworkSong.with([.albums], preferredSource: .library)
if refinedSong.matches(self) {
var loadingState: LoadingState<PlatformImage> = .idle
await loadingState.load(artwork: artwork)
if let platformImage = loadingState.value {
return platformImage
}
}
}

throw PartialArtworkImageError.noneFound
}

throw PartialArtworkImageError.noneFound
}
}

0 comments on commit b35a5a2

Please sign in to comment.