From 0db4ec2d9aba1a81fa3f4c6db5e9de79c8de4c10 Mon Sep 17 00:00:00 2001
From: Peterjah <ps@massa.net>
Date: Thu, 12 Dec 2024 22:25:38 +0100
Subject: [PATCH] get owned token from enumerable contract

---
 frontend/src/utils/write-mns-sc.tsx | 79 ++++++++++++++++++++++++-----
 1 file changed, 67 insertions(+), 12 deletions(-)

diff --git a/frontend/src/utils/write-mns-sc.tsx b/frontend/src/utils/write-mns-sc.tsx
index ec69e58..9b6f844 100644
--- a/frontend/src/utils/write-mns-sc.tsx
+++ b/frontend/src/utils/write-mns-sc.tsx
@@ -4,9 +4,12 @@ import {
   Client,
   EOperationStatus,
   ICallData,
+  IDatastoreEntryInput,
   MAX_GAS_CALL,
   bytesToStr,
+  bytesToU256,
   bytesToU64,
+  strToBytes,
   toMAS,
 } from '@massalabs/massa-web3';
 import { ToastContent, toast } from '@massalabs/react-ui-kit';
@@ -20,6 +23,7 @@ import { OperationToast } from '../lib/connectMassaWallets/components/OperationT
 import { logSmartContractEvents } from '../lib/connectMassaWallets/utils';
 import { ExplorerApiClient } from '../ExplorerApiClient';
 import { useAccountStore } from '../lib/connectMassaWallets/store';
+import { isEqual } from 'lodash';
 
 interface ToasterMessage {
   pending: string;
@@ -352,21 +356,72 @@ export function useWriteMNS(client?: Client) {
 
     let list: DnsUserEntryListResult[] = [];
 
-    let domains = await explorerApi.getDomainOwnedByAddress(params.address);
-    if (domains && domains.length > 0) {
-      let dnsInfos = await explorerApi.getDomainsInfo(domains);
+    const addressInfo = await client?.publicApi().getAddresses([SC_ADDRESS]);
+    const filter = [
+      ...strToBytes('ownedTokens'),
+      ...strToBytes(params.address),
+    ];
+    const ownedKeys = addressInfo?.[0].candidate_datastore_keys.filter(
+      (key) => {
+        return !filter.length || isEqual(key.slice(0, filter.length), filter);
+      },
+    );
 
-      for (const domain in dnsInfos) {
-        if (Object.prototype.hasOwnProperty.call(dnsInfos, domain)) {
-          list.push({
-            domain: domain,
-            targetAddress: dnsInfos[domain].target_address,
-            tokenId: dnsInfos[domain].tokenId,
-          });
-        }
+    if (ownedKeys) {
+      const tokenIdsBytes = ownedKeys.map((key) => key.slice(filter.length));
+      const tokenIds = tokenIdsBytes.map((val) =>
+        bytesToU256(Uint8Array.from(val)),
+      );
+
+      const DOMAIN_SEPARATOR_KEY = [0x42];
+      const DOMAIN_KEY_PREFIX = [0x3];
+      const TARGET_KEY_PREFIX = [0x02];
+
+      const domainsDataStoreEntries: IDatastoreEntryInput[] = tokenIdsBytes.map(
+        (tokenIdBytes) => ({
+          address: SC_ADDRESS,
+          key: Uint8Array.from([
+            ...DOMAIN_SEPARATOR_KEY,
+            ...DOMAIN_KEY_PREFIX,
+            ...tokenIdBytes,
+          ]),
+        }),
+      );
+
+      const domainsRes = await client!
+        .publicApi()
+        .getDatastoreEntries(domainsDataStoreEntries);
+      const domains = domainsRes.map((val) => bytesToStr(val.candidate_value!));
+
+      const targetDataStoreEntries: IDatastoreEntryInput[] = domains.map(
+        (domain) => ({
+          address: SC_ADDRESS,
+          key: Uint8Array.from([
+            ...DOMAIN_SEPARATOR_KEY,
+            ...TARGET_KEY_PREFIX,
+            ...strToBytes(domain),
+          ]),
+        }),
+      );
+      const targetsRes = await client!
+        .publicApi()
+        .getDatastoreEntries(targetDataStoreEntries);
+      const targets = targetsRes.map((val) => bytesToStr(val.candidate_value!));
+
+      if (
+        targets.length !== domains.length ||
+        domains.length !== tokenIds.length ||
+        tokenIds.length !== targets.length
+      ) {
+        console.error('Inconsistent data for owned MNS');
       }
-    }
 
+      list = domains.map((domain, index) => ({
+        domain,
+        targetAddress: targets[index],
+        tokenId: tokenIds[index],
+      }));
+    }
     setList(list);
     setListSpinning(false);
     return list;