Merge pull request #14 from yhkaplan/feature/fix-prevent-from-duplica…

…ted-call prevent from duplicated calling delegate methods
yhkaplan · Aug 5, 2020 · 18ebe35 · 18ebe35
2 parents 6f91e04 + 4e59f70
commit 18ebe35
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/Sources/CreditCardScanner/CameraView.swift b/Sources/CreditCardScanner/CameraView.swift
@@ -199,6 +199,7 @@ extension CameraView: AVCaptureVideoDataOutputSampleBufferDelegate {
 
         guard let pixelBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else {
             delegate?.didError(with: CreditCardScannerError(kind: .capture))
+            delegate = nil
             return
         }
 
@@ -212,6 +213,7 @@ extension CameraView: AVCaptureVideoDataOutputSampleBufferDelegate {
         guard let fullCameraImage = cgImage,
             let croppedImage = fullCameraImage.cropping(to: regionOfInterest) else {
             delegate?.didError(with: CreditCardScannerError(kind: .capture))
+            delegate = nil
             return
         }
 

diff --git a/Sources/CreditCardScanner/ImageAnalyzer.swift b/Sources/CreditCardScanner/ImageAnalyzer.swift
@@ -46,6 +46,7 @@ final class ImageAnalyzer {
         } catch {
             let e = CreditCardScannerError(kind: .photoProcessing, underlyingError: error)
             delegate?.didFinishAnalyzation(with: .failure(e))
+            delegate = nil
         }
     }
 
@@ -124,6 +125,7 @@ final class ImageAnalyzer {
 
         if strongSelf.selectedCard.number != nil {
             strongSelf.delegate?.didFinishAnalyzation(with: .success(strongSelf.selectedCard))
+            strongSelf.delegate = nil
         }
     }
 }