Add clear storage on manage section

poanetwork · Aug 22, 2018 · 7dfd810 · 7dfd810
1 parent 32b1fe3
commit 7dfd810
Show file tree

Hide file tree

Showing 3 changed files with 57 additions and 45 deletions.
diff --git a/src/components/Home/index.js b/src/components/Home/index.js
@@ -4,7 +4,7 @@ import CrowdsalesList from '../Common/CrowdsalesList'
 import { Loader } from '../Common/Loader'
 import { loadRegistryAddresses } from '../../utils/blockchainHelpers'
 import { ModalContainer } from '../Common/ModalContainer'
-import { toast } from '../../utils/utils'
+import { toast, clearStorage } from '../../utils/utils'
 import { TOAST, NAVIGATION_STEPS, DOWNLOAD_STATUS } from '../../utils/constants'
 import { inject, observer } from 'mobx-react'
 import { checkWeb3, getNetworkVersion } from '../../utils/blockchainHelpers'
@@ -55,23 +55,23 @@ export class Home extends Component {
     try {
       await loadRegistryAddresses()
       this.setState({
-        loading: false,
         showModal: true
       })
     } catch (e) {
       logger.error('There was a problem loading the crowdsale addresses from the registry', e)
-      this.setState({
-        loading: false
-      })
     }
+
+    this.setState({
+      loading: false
+    })
   }
 
   goNextStep = async () => {
     // Clear local storage if there is no incomplete deployment, and reload
     if (storage.has('DeploymentStore') && storage.get('DeploymentStore').deploymentStep) {
       this.props.history.push('/')
     } else {
-      this.clearStorage()
+      clearStorage(this.props)
       await this.reloadStorage()
       this.props.history.push('1')
     }
@@ -99,41 +99,6 @@ export class Home extends Component {
     }
   }
 
-  clearStorage() {
-    // Generate of stores to clear
-    const toArray = ({
-      generalStore,
-      contractStore,
-      crowdsaleStore,
-      gasPriceStore,
-      deploymentStore,
-      reservedTokenStore,
-      stepTwoValidationStore,
-      tierStore,
-      tokenStore
-    }) => {
-      return [
-        generalStore,
-        contractStore,
-        crowdsaleStore,
-        gasPriceStore,
-        deploymentStore,
-        reservedTokenStore,
-        stepTwoValidationStore,
-        tierStore,
-        tokenStore
-      ]
-    }
-
-    const storesToClear = toArray(this.props)
-    for (let storeToClear of storesToClear) {
-      if (typeof storeToClear.reset === 'function') {
-        logger.log('Store to be cleared:', storeToClear.constructor.name)
-        storeToClear.reset()
-      }
-    }
-  }
-
   onClick = crowdsaleAddress => {
     this.props.history.push(`manage/${crowdsaleAddress}`)
   }

diff --git a/src/components/manage/index.js b/src/components/manage/index.js
@@ -23,7 +23,7 @@ import {
   checkWeb3,
   isAddressValid
 } from '../../utils/blockchainHelpers'
-import { isExecIDValid, isNetworkIDValid, toast, toBigNumber } from '../../utils/utils'
+import { isExecIDValid, isNetworkIDValid, toast, toBigNumber, clearStorage } from '../../utils/utils'
 import { getCrowdsaleAssets } from '../../stores/utils'
 import { getFieldsToUpdate, processTier, updateTierAttribute } from './utils'
 import { Loader } from '../Common/Loader'
@@ -39,7 +39,18 @@ import logdown from 'logdown'
 
 const logger = logdown('TW:manage')
 
-@inject('crowdsaleStore', 'web3Store', 'tierStore', 'contractStore', 'generalStore', 'tokenStore', 'gasPriceStore')
+@inject(
+  'crowdsaleStore',
+  'web3Store',
+  'tierStore',
+  'contractStore',
+  'reservedTokenStore',
+  'stepTwoValidationStore',
+  'generalStore',
+  'tokenStore',
+  'gasPriceStore',
+  'deploymentStore'
+)
 @observer
 export class Manage extends Component {
   constructor(props) {
@@ -65,8 +76,9 @@ export class Manage extends Component {
     setTimeout(() => window.scrollTo(0, 0), 500)
   }
 
-  componentWillMount() {
-    this.preparePage()
+  async componentWillMount() {
+    clearStorage(this.props)
+    await this.preparePage()
   }
 
   preparePage = async () => {

diff --git a/src/utils/utils.js b/src/utils/utils.js
@@ -193,3 +193,38 @@ export const objectKeysToLowerCase = input => {
     return newObj
   }, {})
 }
+
+export const clearStorage = props => {
+  // Generate of stores to clear
+  const toArray = ({
+    generalStore,
+    contractStore,
+    crowdsaleStore,
+    gasPriceStore,
+    deploymentStore,
+    reservedTokenStore,
+    stepTwoValidationStore,
+    tierStore,
+    tokenStore
+  }) => {
+    return [
+      generalStore,
+      contractStore,
+      crowdsaleStore,
+      gasPriceStore,
+      deploymentStore,
+      reservedTokenStore,
+      stepTwoValidationStore,
+      tierStore,
+      tokenStore
+    ]
+  }
+
+  const storesToClear = toArray(props)
+  for (let storeToClear of storesToClear) {
+    if (typeof storeToClear.reset === 'function') {
+      logger.log('Store to be cleared:', storeToClear.constructor.name)
+      storeToClear.reset()
+    }
+  }
+}