Add OfflinePrivilegeCheck to bootstrap process

tidal-music · Oct 7, 2024 · f37e898 · f37e898
1 parent 2e91425
commit f37e898
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 5 deletions.
diff --git a/Sources/Player/PlaybackEngine/Internal/PlayerItemLoader.swift b/Sources/Player/PlaybackEngine/Internal/PlayerItemLoader.swift
@@ -5,11 +5,18 @@ import Foundation
 
 final class PlayerItemLoader {
 	private let offlineStorage: OfflineStorage?
+	private let offlinePlaybackPrivilegeCheck: (() -> Bool)?
 	private let playbackInfoFetcher: PlaybackInfoFetcher
 	private var playerLoader: PlayerLoader
 
-	init(with offlineStorage: OfflineStorage?, _ playbackInfoFetcher: PlaybackInfoFetcher, and playerLoader: PlayerLoader) {
+	init(
+		with offlineStorage: OfflineStorage?,
+		_ offlinePlaybackPrivilegeCheck: (() -> Bool)?,
+		_ playbackInfoFetcher: PlaybackInfoFetcher,
+		and playerLoader: PlayerLoader
+	) {
 		self.offlineStorage = offlineStorage
+		self.offlinePlaybackPrivilegeCheck = offlinePlaybackPrivilegeCheck
 		self.playbackInfoFetcher = playbackInfoFetcher
 		self.playerLoader = playerLoader
 	}
@@ -42,7 +49,8 @@ private extension PlayerItemLoader {
 
 		if PlayerWorld.developmentFeatureFlagProvider.isOffliningEnabled,
 		   let offlineEntry = try? offlineStorage?.get(key: mediaProduct.productId),
-		   let offlinedMediaProduct = PlayableOfflinedMediaProduct(from: offlineEntry)
+		   let offlinedMediaProduct = PlayableOfflinedMediaProduct(from: offlineEntry),
+		   offlinePlaybackPrivilegeCheck?() ?? true == true
 		{
 			return try await (metadata(of: offlinedMediaProduct), playerLoader.load(offlinedMediaProduct))
 		}

diff --git a/Sources/Player/PlaybackEngine/PlayerEngine.swift b/Sources/Player/PlaybackEngine/PlayerEngine.swift
@@ -108,6 +108,7 @@ final class PlayerEngine {
 		_ playerEventSender: PlayerEventSender,
 		_ networkMonitor: NetworkMonitor,
 		_ offlineStorage: OfflineStorage?,
+		_ offlinePlaybackPrivilegeCheck: (() -> Bool)?,
 		_ playerLoader: PlayerLoader,
 		_ featureFlagProvider: FeatureFlagProvider,
 		_ notificationsHandler: NotificationsHandler?
@@ -129,7 +130,12 @@ final class PlayerEngine {
 		self.notificationsHandler = notificationsHandler
 		self.featureFlagProvider = featureFlagProvider
 
-		playerItemLoader = PlayerItemLoader(with: offlineStorage, playbackInfoFetcher, and: playerLoader)
+		playerItemLoader = PlayerItemLoader(
+			with: offlineStorage,
+			offlinePlaybackPrivilegeCheck,
+			playbackInfoFetcher,
+			and: playerLoader
+		)
 
 		state = .IDLE
 		currentItem = nil

diff --git a/Sources/Player/Player.swift b/Sources/Player/Player.swift
@@ -112,7 +112,8 @@ public extension Player {
 		credentialsProvider: CredentialsProvider,
 		eventSender: EventSender,
 		userClientIdSupplier: (() -> Int)? = nil,
-		shouldAddLogging: Bool = false
+		shouldAddLogging: Bool = false,
+		offlinePlaybackPrivilegeCheck: (() -> Bool)? = nil
 	) -> Player? {
 		if shared != nil {
 			return nil
@@ -444,7 +445,8 @@ private extension Player {
 		_ notificationsHandler: NotificationsHandler?,
 		_ featureFlagProvider: FeatureFlagProvider,
 		_ externalPlayers: [GenericMediaPlayer.Type],
-		_ credentialsProvider: CredentialsProvider
+		_ credentialsProvider: CredentialsProvider,
+		_ offlinePlaybackPrivilegeCheck: (() -> Bool)? = nil
 	) -> PlayerEngine {
 		let internalPlayerLoader = InternalPlayerLoader(
 			with: configuration,
@@ -465,6 +467,7 @@ private extension Player {
 			playerEventSender,
 			networkMonitor,
 			offlineStorage,
+			offlinePlaybackPrivilegeCheck,
 			internalPlayerLoader,
 			featureFlagProvider,
 			notificationsHandler