Update ImageView.swift

Shared/Views/ImageView.swift

@@ -6,70 +6,100 @@
 // Copyright (c) 2022 Jellyfin & Jellyfin Contributors
 //
 
+import Nuke
 import NukeUI
 import SwiftUI
+import UIKit
 
-// TODO: update multiple sources so that multiple blurhashes can be taken, clean up
+// TODO: Fix 100+ inits
 
-struct ImageView: View {
+struct ImageViewSource {
-
+    let url: URL?
-	@State
+    let blurHash: String?
-	private var sources: [URL]
+    
-	private var currentURL: URL? { sources.first }
+    init(url: URL? = nil, blurHash: String? = nil) {
-
+        self.url = url
-	private let blurhash: String
+        self.blurHash = blurHash
-	private let failureInitials: String
+    }
-
+}
-	init(src: URL, bh: String = "001fC^", failureInitials: String = "") {
+
-		self.sources = [src]
+struct DefaultFailureView: View {
-		self.blurhash = bh
+    
-		self.failureInitials = failureInitials
+    var body: some View {
-	}
+        Color.secondary
-
+    }
-	init(sources: [URL], bh: String = "001fC^", failureInitials: String = "") {
+}
-		assert(!sources.isEmpty, "Must supply at least one source")
+
-
+struct ImageView<FailureView: View>: View {
-		self.sources = sources
+
-		self.blurhash = bh
+    @State
-		self.failureInitials = failureInitials
+    private var sources: [ImageViewSource]
-	}
+    private var currentURL: URL? { sources.first?.url }
-
+    private var currentBlurHash: String? { sources.first?.blurHash }
-	// TODO: fix placeholder hash view
+    private var failureView: FailureView
-	@ViewBuilder
+    
-	private var placeholderView: some View {
+    init(_ source: URL?, blurHash: String? = nil, @ViewBuilder failureView: () -> FailureView) {
-		Image(uiImage: UIImage(blurHash: blurhash, size: CGSize(width: 12, height: 12)) ??
+        let imageViewSource = ImageViewSource(url: source, blurHash: blurHash)
-			UIImage(blurHash: "001fC^", size: CGSize(width: 12, height: 12))!)
+        _sources = State(initialValue: [imageViewSource])
-			.resizable()
+        self.failureView = failureView()
-	}
+    }
-
+
-	@ViewBuilder
+    init(_ source: ImageViewSource, @ViewBuilder failureView: () -> FailureView) {
-	private func failureImage() -> some View {
+        _sources = State(initialValue: [source])
-		ZStack {
+        self.failureView = failureView()
-			Rectangle()
+    }
-				.foregroundColor(Color(UIColor.darkGray))
+
-
+    init(_ sources: [ImageViewSource], @ViewBuilder failureView: () -> FailureView) {
-			Text(failureInitials)
+        _sources = State(initialValue: sources)
-				.font(.largeTitle)
+        self.failureView = failureView()
-				.foregroundColor(.secondary)
+    }
-				.accessibilityHidden(true)
+
-		}
+    @ViewBuilder
-	}
+    private var placeholderView: some View {
-
+        if let currentBlurHash = currentBlurHash {
-	var body: some View {
+            BlurHashView(blurHash: currentBlurHash)
-		if let u = currentURL {
+                .id(currentBlurHash)
-			LazyImage(source: u) { state in
+        } else {
-				if let image = state.image {
+            Color.secondary
-					image
+        }
-				} else if state.error != nil {
+    }
-					placeholderView.onAppear { sources.removeFirst() }
+
-				} else {
+    var body: some View {
-					placeholderView
+        
-				}
+        if let currentURL = currentURL {
-			}
+            LazyImage(source: currentURL) { state in
-			.pipeline(ImagePipeline(configuration: .withDataCache))
+                if let image = state.image {
-			.id(u)
+                    image
-		} else {
+                } else if state.error != nil {
-			failureImage()
+                    placeholderView.onAppear { sources.removeFirst() }
-		}
+                } else {
-	}
+                    placeholderView
+                }
+            }
+            .pipeline(ImagePipeline(configuration: .withDataCache))
+            .id(currentURL)
+        } else {
+            failureView
+        }
+    }
+}
+
+extension ImageView where FailureView == DefaultFailureView {
+    init(_ source: URL?, blurHash: String? = nil) {
+        let imageViewSource = ImageViewSource(url: source, blurHash: blurHash)
+        self.init(imageViewSource, failureView: { DefaultFailureView() })
+    }
+    
+    init(_ source: ImageViewSource) {
+        self.init(source, failureView: { DefaultFailureView() })
+    }
+
+    init(_ sources: [ImageViewSource]) {
+        self.init(sources, failureView: { DefaultFailureView() })
+    }
+    
+    init(sources: [URL]) {
+        let imageViewSources = sources.compactMap { ImageViewSource(url: $0, blurHash: nil) }
+        self.init(imageViewSources, failureView: { DefaultFailureView() })
+    }
 }