Blame view

App/camera/ALCameraLib/Utilities/SingleImageSaver.swift 2.47 KB
1341bf603   Trịnh Văn Quân   version 1.1
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)
      }
  }