Make getAddressType consistent between renders

confio · Oct 2, 2024 · 7df2983 · 7df2983
1 parent 1bb07f0
commit 7df2983
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/lib/chain.ts b/lib/chain.ts
@@ -1,8 +1,10 @@
 //TODO - query from chain
-export function getAddressType(_address: string) {
-  if (Math.random() < 0.5) {
-    return "account";
+export function getAddressType(address: string) {
+  const unprefixedAddress = address.slice(7);
+
+  if (unprefixedAddress.includes("i") || unprefixedAddress.includes("u")) {
+    return "validator";
   }
 
-  return Math.random() < 0.5 ? "contract" : "validator";
+  return unprefixedAddress.includes("a") ? "account" : "contract";
 }