SingleImageSaver.swift
2.47 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
//
// SingleImageSavingInteractor.swift
// ALCameraViewController
//
// Created by Alex Littlejohn on 2016/02/16.
// Copyright © 2016 zero. All rights reserved.
//
import UIKit
import Photos
public typealias SingleImageSaverSuccess = (PHAsset) -> Void
public typealias SingleImageSaverFailure = (NSError) -> Void
public class SingleImageSaver {
private let errorDomain = "com.zero.singleImageSaver"
private var success: SingleImageSaverSuccess?
private var failure: SingleImageSaverFailure?
private var image: UIImage?
public init() { }
public func onSuccess(_ success: @escaping SingleImageSaverSuccess) -> Self {
self.success = success
return self
}
public func onFailure(_ failure: @escaping SingleImageSaverFailure) -> Self {
self.failure = failure
return self
}
public func setImage(_ image: UIImage) -> Self {
self.image = image
return self
}
public func save() -> Self {
_ = PhotoLibraryAuthorizer { error in
if error == nil {
self._save()
} else {
self.failure?(error!)
}
}
return self
}
private func _save() {
guard let image = image else {
self.invokeFailure()
return
}
var assetIdentifier: PHObjectPlaceholder?
PHPhotoLibrary.shared()
.performChanges({
let request = PHAssetChangeRequest.creationRequestForAsset(from: image)
assetIdentifier = request.placeholderForCreatedAsset
}) { finished, error in
guard let assetIdentifier = assetIdentifier, finished else {
self.invokeFailure()
return
}
self.fetch(assetIdentifier)
}
}
private func fetch(_ assetIdentifier: PHObjectPlaceholder) {
let assets = PHAsset.fetchAssets(withLocalIdentifiers: [assetIdentifier.localIdentifier], options: nil)
DispatchQueue.main.async {
guard let asset = assets.firstObject else {
self.invokeFailure()
return
}
self.success?(asset)
}
}
private func invokeFailure() {
let error = errorWithKey("error.cant-fetch-photo", domain: errorDomain)
failure?(error)
}
}