From f59de39611a181fa80679dbfba74bb4b5979508a Mon Sep 17 00:00:00 2001
From: Ran Greenberg <rang@wix.com>
Date: Mon, 13 Mar 2017 15:32:40 +0200
Subject: [PATCH] fix number of images captured label on capture image

---
 src/CameraKitCameraScreen.js | 27 ++++++++++++++++++++++-----
 1 file changed, 22 insertions(+), 5 deletions(-)

diff --git a/src/CameraKitCameraScreen.js b/src/CameraKitCameraScreen.js
index 4980bb6..de74126 100644
--- a/src/CameraKitCameraScreen.js
+++ b/src/CameraKitCameraScreen.js
@@ -44,7 +44,8 @@ export default class CameraScreen extends Component {
       ratios: [],
       cameraOptions: {},
       ratioArrayPosition: -1,
-      imageCaptured: undefined
+      imageCaptured: undefined,
+      captured: false
     };
     this.onSetFlash = this.onSetFlash.bind(this);
     this.onSwitchCameraPressed = this.onSwitchCameraPressed.bind(this);
@@ -130,6 +131,17 @@ export default class CameraScreen extends Component {
     );
   }
 
+  numberOfImagesTaken() {
+    const numberTook = this.state.captureImages.length;
+    if (numberTook >= 2) {
+      return numberTook;
+    } else if (this.state.captured) {
+      return '1';
+    } else {
+      return '';
+    }
+  }
+
 
   renderCaptureButton() {
     if (this.props.captureButtonImage) {
@@ -141,10 +153,15 @@ export default class CameraScreen extends Component {
             <Image
               style={styles.captureButton}
               source={this.props.captureButtonImage}
-              resizeMode={'contain'}
-            />
+              resizeMode={'contain'}>
+
+              <Text style={styles.captureNumber}>
+                {this.numberOfImagesTaken()}
+              </Text>
+
+            </Image>
           </TouchableOpacity>
-        </View>
+        </View >
       );
     }
     return null;
@@ -227,7 +244,7 @@ export default class CameraScreen extends Component {
     const image = await this.camera.capture(true);
 
     if (image) {
-      this.setState({ imageCaptured: image, captureImages: _.concat(this.state.captureImages, image) });
+      this.setState({ captured: true, imageCaptured: image, captureImages: _.concat(this.state.captureImages, image) });
     }
 
   }