Add library music items count

rryam · Jul 14, 2022 · a3b6510 · a3b6510
1 parent bfec4e5
commit a3b6510
Show file tree

Hide file tree

Showing 5 changed files with 81 additions and 0 deletions.
diff --git a/Sources/MusadoraKit/Library/LibraryAlbum.swift b/Sources/MusadoraKit/Library/LibraryAlbum.swift
@@ -6,6 +6,7 @@
 //
 
 import MusicKit
+import MediaPlayer
 
 public extension MusadoraKit {
   /// Fetch an album from the user's library by using its identifier.
@@ -67,4 +68,20 @@ public extension MusadoraKit {
       return response.items
     }
   }
+
+  static var libraryAlbumsCount: Int {
+    get async throws {
+      if #available(iOS 16, tvOS 16.0, watchOS 9.0, *) {
+        let request = MusicLibraryRequest<Album>()
+        let response = try await request.response()
+        return response.items.count
+      } else {
+        if let items = MPMediaQuery.albums().items {
+          return items.count
+        } else {
+          throw MediaPlayError.notFound(for: "albums")
+        }
+      }
+    }
+  }
 }
diff --git a/Sources/MusadoraKit/Library/LibraryArtists.swift b/Sources/MusadoraKit/Library/LibraryArtists.swift
@@ -6,6 +6,7 @@
 //
 
 import MusicKit
+import MediaPlayer
 
 public extension MusadoraKit {
   /// Fetch an artist from the user's library by using its identifier.
@@ -42,4 +43,20 @@ public extension MusadoraKit {
     let response = try await request.response()
     return response.items
   }
+
+  static var libraryArtistsCount: Int {
+    get async throws {
+      if #available(iOS 16, tvOS 16.0, watchOS 9.0, *) {
+        let request = MusicLibraryRequest<Artist>()
+        let response = try await request.response()
+        return response.items.count
+      } else {
+        if let items = MPMediaQuery.artists().items {
+          return items.count
+        } else {
+          throw MediaPlayError.notFound(for: "artists")
+        }
+      }
+    }
+  }
 }
diff --git a/Sources/MusadoraKit/Library/LibraryPlaylist.swift b/Sources/MusadoraKit/Library/LibraryPlaylist.swift
@@ -6,6 +6,7 @@
 //
 
 import MusicKit
+import MediaPlayer
 
 public extension MusadoraKit {
   /// Fetch a playlist from the user's library by using its identifier.
@@ -42,4 +43,20 @@ public extension MusadoraKit {
     let response = try await request.response()
     return response.items
   }
+
+  static var libraryPlaylistsCount: Int {
+    get async throws {
+      if #available(iOS 16, tvOS 16.0, watchOS 9.0, *) {
+        let request = MusicLibraryRequest<Playlist>()
+        let response = try await request.response()
+        return response.items.count
+      } else {
+        if let items = MPMediaQuery.playlists().items {
+          return items.count
+        } else {
+          throw MediaPlayError.notFound(for: "playlists")
+        }
+      }
+    }
+  }
 }
diff --git a/Sources/MusadoraKit/Library/LibrarySong.swift b/Sources/MusadoraKit/Library/LibrarySong.swift
@@ -6,6 +6,7 @@
 //
 
 import MusicKit
+import MediaPlayer
 
 public extension MusadoraKit {
   /// Fetch a song from the user's library by using its identifier.
@@ -42,4 +43,20 @@ public extension MusadoraKit {
     let response = try await request.response()
     return response.items
   }
+
+  static var librarySongsCount: Int {
+    get async throws {
+      if #available(iOS 16, tvOS 16.0, watchOS 9.0, *) {
+        let request = MusicLibraryRequest<Song>()
+        let response = try await request.response()
+        return response.items.count
+      } else {
+        if let items = MPMediaQuery.songs().items {
+          return items.count
+        } else {
+          throw MediaPlayError.notFound(for: "songs")
+        }
+      }
+    }
+  }
 }
diff --git a/Sources/MusadoraKit/Models/MusadoraKitError.swift b/Sources/MusadoraKit/Models/MusadoraKitError.swift
@@ -24,6 +24,19 @@ extension RatingsError: CustomStringConvertible {
   }
 }
 
+public enum MediaPlayError: Error, Equatable {
+  case notFound(for: String)
+}
+
+extension MediaPlayError: CustomStringConvertible {
+  public var description: String {
+    switch self {
+      case let .notFound(item):
+        return "Not able to count the music items for \(item)."
+    }
+  }
+}
+
 public enum MusadoraKitError: Error, Equatable {
   case notFound(for: String)
   case typeMissing