feat(neuron-ui): add a condition of isMainnet for displaying addresses

nervosnetwork · Sep 18, 2019 · 5be0335 · 5be0335
1 parent d55a415
commit 5be0335
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/packages/neuron-ui/src/components/Addresses/index.tsx b/packages/neuron-ui/src/components/Addresses/index.tsx
@@ -27,10 +27,12 @@ const Addresses = ({
     loadings: { addressList: isLoading },
   },
   wallet: { addresses = [], id: walletID },
-  settings: { showAddressBook = false },
+  chain: { networkID },
+  settings: { showAddressBook = false, networks = [] },
   history,
   dispatch,
 }: React.PropsWithoutRef<StateWithDispatch & RouteComponentProps>) => {
+  const isMainnet = (networks.find(n => n.id === networkID) || {}).type === 'mainnet'
   const [showMainnetAddress, setShowMainnetAddress] = useState(false)
   const [t] = useTranslation()
   useEffect(() => {
@@ -211,11 +213,13 @@ const Addresses = ({
   return (
     <>
       <Stack verticalAlign="center" horizontalAlign="start" tokens={{ childrenGap: 15 }}>
-        <DefaultButton
-          text={t(`addresses.display-${showMainnetAddress ? 'testnet' : 'mainnet'}-addresses`)}
-          onClick={() => setShowMainnetAddress(!showMainnetAddress)}
-        />
-        {showMainnetAddress ? (
+        {!isMainnet ? (
+          <DefaultButton
+            text={t(`addresses.display-${showMainnetAddress ? 'testnet' : 'mainnet'}-addresses`)}
+            onClick={() => setShowMainnetAddress(!showMainnetAddress)}
+          />
+        ) : null}
+        {showMainnetAddress && !isMainnet ? (
           <Text variant="medium" style={{ color: semanticColors.errorText }}>
             {t('addresses.mainnet-address-caution')}
           </Text>

diff --git a/packages/neuron-ui/src/types/App/index.d.ts b/packages/neuron-ui/src/types/App/index.d.ts
@@ -94,6 +94,7 @@ declare namespace State {
   interface NetworkProperty {
     name: string
     remote: string
+    type?: 'mainnet' | 'testnet'
   }
 
   interface Network extends NetworkProperty {