Merge pull request #100 from TosiDrop/airdrop

translate both testnet and mainnet address

client/src/pages/Rewards/index.tsx

@@ -28,11 +28,7 @@ import "./index.scss";
 import { useCallback } from "react";
 import { useDispatch } from "react-redux";
 import { showModal } from "src/reducers/modalSlice";
-import {
-    Address,
-    BaseAddress,
-    RewardAddress,
-} from "@emurgo/cardano-serialization-lib-asmjs";
+import { getStakeKey } from "./utils/common.function";
 
 let Buffer = require("buffer").Buffer;
 
@@ -84,28 +80,8 @@ function Rewards({ connectedWallet, wrongNetwork }: Params) {
                  * we want the stake address
                  * if it is cardano address, get the staking address
                  */
-
-                let address = searchAddress;
-
-                if (searchAddress.includes("addr")) {
-                    const addressObject = Address.from_bech32(searchAddress);
-                    const stakeAddress =
-                        BaseAddress.from_address(addressObject);
-                    let rewardAddressBytes = new Uint8Array(29);
-                    rewardAddressBytes.set([0xe0], 0);
-
-                    if (stakeAddress == null) return;
-                    rewardAddressBytes.set(
-                        stakeAddress.stake_cred().to_bytes().slice(4, 32),
-                        1
-                    );
-                    let rewardAddress = RewardAddress.from_address(
-                        Address.from_bytes(rewardAddressBytes)
-                    );
-
-                    if (rewardAddress == null) return;
-                    address = rewardAddress?.to_address().to_bech32();
-                }
+                let address = getStakeKey(searchAddress);
+                if (address == null) throw new Error();
 
                 const rewards = await getRewards(address);
 
@@ -125,14 +101,16 @@ function Rewards({ connectedWallet, wrongNetwork }: Params) {
                     setRewardsLoader(false);
                 }
             } catch (ex: any) {
-                if (ex?.response?.status === 404) {
-                    dispatch(
-                        showModal({
-                            text: "Account not found.",
-                            type: ModalTypes.info,
-                        })
-                    );
-                    setRewardsLoader(false);
+                switch (true) {
+                    case ex?.response?.status === 404:
+                    default:
+                        dispatch(
+                            showModal({
+                                text: "Account not found.",
+                                type: ModalTypes.info,
+                            })
+                        );
+                        setRewardsLoader(false);
                 }
             }
         }

client/src/pages/Rewards/utils/common.function.ts

@@ -0,0 +1,43 @@
+import {
+    Address,
+    BaseAddress,
+    RewardAddress,
+} from "@emurgo/cardano-serialization-lib-asmjs";
+
+export const getStakeKey = (searchAddress: string): string | null => {
+    const prefix = searchAddress.slice(0, 5);
+    const addressObject = Address.from_bech32(searchAddress);
+    const baseAddress = BaseAddress.from_address(addressObject);
+    let rewardAddressBytes = new Uint8Array(29);
+
+    switch (true) {
+        /**
+         * check if testnet address
+         */
+        case prefix === "addr_":
+            rewardAddressBytes.set([0xe0], 0);
+            break;
+        /**
+         * check if mainnet address
+         */
+        case prefix === "addr1":
+            rewardAddressBytes.set([0xe1], 0);
+            break;
+        /**
+         * check if not receive address
+         */
+        default:
+            return searchAddress;
+    }
+
+    if (baseAddress == null) return null;
+    rewardAddressBytes.set(baseAddress.stake_cred().to_bytes().slice(4, 32), 1);
+
+    let rewardAddress = RewardAddress.from_address(
+        Address.from_bytes(rewardAddressBytes)
+    );
+
+    if (rewardAddress == null) return null;
+
+    return rewardAddress.to_address().to_bech32();
+};