From c7df280d6877ff2891a3a795bc8a4bf1f49df0ba Mon Sep 17 00:00:00 2001
From: Anastasia Poupeney <anastasiasergienko@gmail.com>
Date: Tue, 19 Jun 2018 20:56:28 +0200
Subject: [PATCH] adding separate handle for the error screen and adding skip
 plus retry for gen check onboarding

---
 .../Onboarding/steps/GenuineCheck.js          | 63 ++++++++++++++-----
 src/reducers/onboarding.js                    |  2 +
 2 files changed, 49 insertions(+), 16 deletions(-)

diff --git a/src/components/Onboarding/steps/GenuineCheck.js b/src/components/Onboarding/steps/GenuineCheck.js
index 9b1c7b8e..36b93672 100644
--- a/src/components/Onboarding/steps/GenuineCheck.js
+++ b/src/components/Onboarding/steps/GenuineCheck.js
@@ -82,10 +82,12 @@ class GenuineCheck extends PureComponent<StepProps, State> {
     if (!item.pass) {
       this.setState(INITIAL_STATE)
       this.props.updateGenuineCheck({
-        isGenuineFail: true,
-        recoveryStepPass: false,
+        displayErrorScreen: true,
         pinStepPass: false,
+        recoveryStepPass: false,
+        isGenuineFail: false,
         isDeviceGenuine: false,
+        genuineCheckUnavailable: null,
       })
     }
   }
@@ -116,6 +118,7 @@ class GenuineCheck extends PureComponent<StepProps, State> {
         isGenuineFail: true,
         isDeviceGenuine: false,
         genuineCheckUnavailable: null,
+        displayErrorScreen: true,
       })
     })
   }
@@ -125,12 +128,13 @@ class GenuineCheck extends PureComponent<StepProps, State> {
       this.props.updateGenuineCheck({
         isDeviceGenuine: false,
         genuineCheckUnavailable: error,
+        displayErrorScreen: false,
       })
     })
   }
 
   redoGenuineCheck = () => {
-    this.props.updateGenuineCheck({ isGenuineFail: false })
+    this.props.updateGenuineCheck({ displayErrorScreen: false })
   }
 
   contactSupport = () => {
@@ -153,7 +157,7 @@ class GenuineCheck extends PureComponent<StepProps, State> {
     const { genuine } = onboarding
     const { cachedPinStepButton, cachedRecoveryStepButton, isGenuineCheckModalOpened } = this.state
 
-    if (genuine.isGenuineFail) {
+    if (genuine.displayErrorScreen) {
       return this.renderGenuineFail()
     }
 
@@ -226,6 +230,14 @@ class GenuineCheck extends PureComponent<StepProps, State> {
                     </Box>
                   ) : genuine.genuineCheckUnavailable ? (
                     <Box align="center" flow={1} color={colors.alertRed}>
+                      <FakeLink
+                        ff="Open Sans|Regular"
+                        fontSize={4}
+                        underline
+                        onClick={this.handleOpenGenuineCheckModal}
+                      >
+                        {t('app:common.retry')}
+                      </FakeLink>
                       <Box horizontal justify="center">
                         <Box justifyContent="center">
                           <IconCross size={12} />
@@ -234,9 +246,6 @@ class GenuineCheck extends PureComponent<StepProps, State> {
                           <TranslatedError error={genuine.genuineCheckUnavailable} />
                         </Box>
                       </Box>
-                      <FakeLink ff="Open Sans|Regular" fontSize={4} underline onClick={nextStep}>
-                        {t('app:common.skipThisStep')}
-                      </FakeLink>
                     </Box>
                   ) : (
                     <Button
@@ -252,15 +261,37 @@ class GenuineCheck extends PureComponent<StepProps, State> {
             </CardWrapper>
           </Box>
         </StepContainerInner>
-        <OnboardingFooter
-          horizontal
-          align="center"
-          flow={2}
-          t={t}
-          nextStep={nextStep}
-          prevStep={prevStep}
-          isContinueDisabled={!genuine.isDeviceGenuine}
-        />
+        {genuine.genuineCheckUnavailable ? (
+          <OnboardingFooterWrapper>
+            <Button padded outlineGrey onClick={() => prevStep()}>
+              {t('app:common.back')}
+            </Button>
+            <Box horizontal ml="auto">
+              <Button padded disabled={false} onClick={() => nextStep()} mx={2}>
+                {t('app:common.skipThisStep')}
+              </Button>
+              <Button
+                padded
+                onClick={nextStep}
+                isContinueDisabled={genuine.genuineCheckUnavailable}
+                primary
+              >
+                {t('app:common.continue')}
+              </Button>
+            </Box>
+          </OnboardingFooterWrapper>
+        ) : (
+          <OnboardingFooter
+            horizontal
+            align="center"
+            flow={2}
+            t={t}
+            nextStep={nextStep}
+            prevStep={prevStep}
+            isContinueDisabled={!genuine.isDeviceGenuine}
+          />
+        )}
+
         <GenuineCheckModal
           isOpened={isGenuineCheckModalOpened}
           onClose={this.handleCloseGenuineCheckModal}
diff --git a/src/reducers/onboarding.js b/src/reducers/onboarding.js
index 10cb0968..3412a4fc 100644
--- a/src/reducers/onboarding.js
+++ b/src/reducers/onboarding.js
@@ -24,6 +24,7 @@ export type OnboardingState = {
     isGenuineFail: boolean,
     isDeviceGenuine: boolean,
     genuineCheckUnavailable: ?Error,
+    displayErrorScreen: boolean,
   },
   isLedgerNano: boolean | null,
   flowType: string,
@@ -38,6 +39,7 @@ const state: OnboardingState = {
     isGenuineFail: false,
     isDeviceGenuine: false,
     genuineCheckUnavailable: null,
+    displayErrorScreen: false,
   },
   isLedgerNano: null,
   flowType: '',