Fix custom networks management (#351)

GemWallet · Jan 20, 2024 · 47cbf6e · 47cbf6e
1 parent 8f3a766
commit 47cbf6e
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 5 deletions.
diff --git a/packages/constants/src/network/network.constant.ts b/packages/constants/src/network/network.constant.ts
@@ -34,6 +34,7 @@ export const FAUCET_XAHAU_TESTNET = 'https://xahau-test.net/accounts';
 export interface NetworkNode {
   chain: Chain;
   name: Network;
+  customNetworkName?: string;
   server: string;
   nodes?: string[];
   description: string;

diff --git a/packages/extension/src/components/organisms/TransactionDetails/TransactionDetails.tsx b/packages/extension/src/components/organisms/TransactionDetails/TransactionDetails.tsx
@@ -34,7 +34,14 @@ export const TransactionDetails: FC<TransactionDetailsProps> = ({
   const mainToken = useMainToken();
 
   const currentNetwork = useMemo(() => {
-    return getNetwork(chainName, networkName as Network);
+    try {
+      return getNetwork(chainName, networkName as Network);
+    } catch (error) {
+      return {
+        ...getNetwork(chainName, 'Custom' as Network),
+        customNetworkName: networkName as string
+      };
+    }
   }, [chainName, networkName]);
 
   const expectedNetwork = useMemo(() => {

diff --git a/packages/extension/src/contexts/NetworkContext/NetworkContext.tsx b/packages/extension/src/contexts/NetworkContext/NetworkContext.tsx
@@ -68,7 +68,7 @@ const NetworkProvider: FC = ({ children }) => {
 
     const connectToNetwork = async () => {
       const network = loadNetwork();
-      setNetworkName(network.name);
+      setNetworkName(network.customNetworkName ?? network.name);
       if (chainName !== network.chain) {
         setChainName(network.chain);
       }
@@ -132,7 +132,7 @@ const NetworkProvider: FC = ({ children }) => {
     try {
       const loadedNetwork = loadNetwork();
       const ws = await connectToLedger(loadedNetwork.server);
-      setNetworkName(loadedNetwork.name);
+      setNetworkName(loadedNetwork.customNetworkName ?? loadedNetwork.name);
       setClient(ws);
       setIsConnectionFailed(false);
     } catch (err) {
@@ -214,7 +214,7 @@ const NetworkProvider: FC = ({ children }) => {
     try {
       await removeNetwork();
       const network = await loadNetwork();
-      setNetworkName(network.name);
+      setNetworkName(network.customNetworkName ?? network.name);
     } catch (err) {
       Sentry.captureException(err);
     }

diff --git a/packages/extension/src/utils/network.ts b/packages/extension/src/utils/network.ts
@@ -89,7 +89,8 @@ export const loadNetwork = () => {
     if ('chain' in parsedData && 'name' in parsedData && 'server' in parsedData) {
       return {
         chain: parsedData.chain,
-        name: parsedData.name,
+        name: 'Custom' as Network,
+        customNetworkName: parsedData.name,
         server: parsedData.server,
         description: 'Custom network'
       };