Handle images from URLs

Author: Arclite

Modules/ShareContents/Sources/Extensions/NSItemProviderExtensions.swift

@@ -5,6 +5,12 @@ import Foundation
 import UniformTypeIdentifiers
 
 extension NSItemProvider {
+    func contains(_ uniformType: UTType) -> Bool {
+        return registeredContentTypes.contains {
+            $0.conforms(to: uniformType)
+        }
+    }
+
     @MainActor func loadData(for uniformType: UTType) async throws -> Data {
         return try await withCheckedThrowingContinuation { continuation in
             _ = loadDataRepresentation(for: uniformType) { data, error in
@@ -18,4 +24,18 @@ extension NSItemProvider {
             }
         }
     }
+
+    @MainActor func loadItem<ItemClass: NSItemProviderReading & Sendable>(for uniformType: UTType) async throws -> ItemClass {
+        return try await withCheckedThrowingContinuation { continuation in
+            _ = loadObject(ofClass: ItemClass.self) { maybeItem, error in
+                if let item = maybeItem as? ItemClass {
+                    continuation.resume(returning: item)
+                } else if let error {
+                    continuation.resume(throwing: error)
+                } else {
+                    continuation.resume(throwing: ShareError.cannotLoadData)
+                }
+            }
+        }
+    }
 }

Modules/ShareContents/Sources/MobileExtensionInputHandler.swift

@@ -43,19 +43,29 @@ import BarcPurchasing
             throw ShareError.noInputProviders
         }
 
-        let imageProviders = inputProviders.filter { inputProvider in
-            inputProvider.registeredContentTypes.contains { type in
-                type.conforms(to: .image)
-            }
-        }
-        guard let imageProvider = imageProviders.first else {
+        guard let imageProvider = inputProviders.first(where: { inputProvider in
+            inputProvider.contains(.image) || inputProvider.contains(.url)
+        }) else {
             throw ShareError.noImageProviders
         }
 
-        let data = try await imageProvider.loadData(for: .image)
-        let cgImage = try CGImage.image(from: data)
+        let cgImage = try await loadImage(from: imageProvider)
         guard let codeValue = try await imageReader.codeValue(in: cgImage) else { throw ShareError.noCodeInImage }
         return codeValue
     }
+
+    private func loadImage(from imageProvider: NSItemProvider) async throws -> CGImage {
+        let data: Data
+        if imageProvider.contains(.image) {
+            data = try await imageProvider.loadData(for: .image)
+        } else if imageProvider.contains(.url) {
+            let url: NSURL = try await imageProvider.loadItem(for: .url)
+            (data, _) = try await URLSession.shared.data(from: url as URL)
+        } else {
+            throw ShareError.unexpectedContentType
+        }
+
+        return try CGImage.image(from: data)
+    }
 }
 #endif

Modules/ShareContents/Sources/ShareError.swift

@@ -11,5 +11,6 @@ enum ShareError: Error {
     case noImageProviders
     case noInputProviders
     case notImplementedOnPlatform
+    case unexpectedContentType
     case userIsNotUnleashed
 }

Modules/WidgetContents/Sources/CodeDisplayView.swift

@@ -32,7 +32,6 @@ struct CodeDisplayView: View {
 
     @ViewBuilder
     private var codeShape: some View {
-        let _ = print("code \(code.name) has value: \(String(describing: code.value))")
         switch code.value {
         case .code128, .code39, .codabar, .ean:
             BarcodeShape()

Products/Share/Info.plist

@@ -20,6 +20,8 @@
 		<dict>
 			<key>NSExtensionActivationRule</key>
 			<dict>
+				<key>NSExtensionActivationSupportsWebURLWithMaxCount</key>
+				<integer>1</integer>
 				<key>NSExtensionActivationSupportsImageWithMaxCount</key>
 				<integer>1</integer>
 			</dict>