Add source code of smart proxy contract to zip archive for download

poanetwork · Sep 3, 2018 · 4dc4362 · 4dc4362
1 parent 0034ba4
commit 4dc4362
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/src/components/stepFour/index.js b/src/components/stepFour/index.js
@@ -256,6 +256,9 @@ export class stepFour extends React.Component {
       }
     })
 
+    const fileName = crowdsaleStore.isMintedCappedCrowdsale ? 'MintedCappedProxy.sol' : 'DutchProxy.sol'
+    zip.file(fileName, this.getContractBySourceType('src'))
+
     zip.generateAsync({ type: DOWNLOAD_TYPE.blob }).then(content => {
       const downloadName = getDownloadName()
       download({ zip: content, filename: downloadName })
@@ -320,20 +323,24 @@ export class stepFour extends React.Component {
     )
   }
 
-  renderContractSource = sourceType => {
+  getContractBySourceType = sourceType => {
     const { crowdsaleStore, contractStore } = this.props
+    const parseContent = content => (isObservableArray(content) ? JSON.stringify(content.slice()) : content)
+
+    return crowdsaleStore.strategy === CROWDSALE_STRATEGIES.MINTED_CAPPED_CROWDSALE
+      ? parseContent(contractStore.MintedCappedProxy[sourceType])
+      : parseContent(contractStore.DutchProxy[sourceType])
+  }
+
+  renderContractSource = sourceType => {
     const sourceTypeName = {
       abi: 'ABI',
       bin: 'Creation Code',
       src: 'Source Code'
     }
-    const parseContent = content => (isObservableArray(content) ? JSON.stringify(content.slice()) : content)
 
     const label = `Crowdsale Proxy Contract ${sourceTypeName[sourceType]}`
-    const value =
-      crowdsaleStore.strategy === CROWDSALE_STRATEGIES.MINTED_CAPPED_CROWDSALE
-        ? parseContent(contractStore.MintedCappedProxy[sourceType])
-        : parseContent(contractStore.DutchProxy[sourceType])
+    const value = this.getContractBySourceType(sourceType)
 
     return <DisplayTextArea label={label} value={value} description={label} />
   }