add image extension

UrbanCompass · Jul 21, 2017 · 9469ca3 · 9469ca3
1 parent 63ead6e
commit 9469ca3
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/Snail/Extensions/URLSessionExtension.swift b/Snail/Extensions/URLSessionExtension.swift
@@ -52,4 +52,17 @@ extension URLSession {
         task.resume()
         return observer
     }
+
+    public func image(request: URLRequest) -> Observable<(UIImage, URLResponse)> {
+        let observer = Replay<(UIImage, URLResponse)>(1)
+        data(request: request).subscribe(queue: .main, onNext: { data, response in
+            guard let image = UIImage(data: data) else {
+                observer.on(.error(ErrorType.invalidData))
+                return
+            }
+            observer.on(.next(image, response))
+            observer.on(.done)
+        }, onError: { observer.on(.error($0)) })
+        return observer
+    }
 }