From 70382cd7a0e4d874bcdb17ee1e66ce6e39279f09 Mon Sep 17 00:00:00 2001
From: thephez <thephez@users.noreply.github.com>
Date: Tue, 5 Aug 2025 08:56:00 -0400
Subject: [PATCH 1/5] feat: enable proofs on site and in tests for all identity
 queries

---
 packages/wasm-sdk/index.html                  | 171 +++++++++++++-----
 .../tests/query-execution.spec.js             |  20 +-
 2 files changed, 137 insertions(+), 54 deletions(-)

diff --git a/packages/wasm-sdk/index.html b/packages/wasm-sdk/index.html
index c962e55c76b..a8252f65253 100644
--- a/packages/wasm-sdk/index.html
+++ b/packages/wasm-sdk/index.html
@@ -1076,17 +1076,25 @@ <h2>Results</h2>
       prefetch_trusted_quorums_testnet,
       // Identity queries
       get_identity_keys,
+      get_identity_keys_with_proof_info,
       get_identity_nonce,
       get_identity_nonce_with_proof_info,
       get_identity_contract_nonce,
       get_identity_contract_nonce_with_proof_info,
       get_identity_balance,
+      get_identity_balance_with_proof_info,
       get_identities_balances,
+      get_identities_balances_with_proof_info,
       get_identity_balance_and_revision,
+      get_identity_balance_and_revision_with_proof_info,
       get_identity_by_public_key_hash,
+      get_identity_by_public_key_hash_with_proof_info,
       get_identities_contract_keys,
+      get_identities_contract_keys_with_proof_info,
       get_identity_by_non_unique_public_key_hash,
+      get_identity_by_non_unique_public_key_hash_with_proof_info,
       get_identity_token_balances,
+      get_identity_token_balances_with_proof_info,
       // Data contract queries
       get_data_contract_history,
       get_data_contracts,
@@ -1125,7 +1133,9 @@ <h2>Results</h2>
       get_identities_token_balances,
       get_identities_token_balances_with_proof_info,
       get_identity_token_infos,
+      get_identity_token_infos_with_proof_info,
       get_identities_token_infos,
+      get_identities_token_infos_with_proof_info,
       get_token_statuses,
       get_token_statuses_with_proof_info,
       get_token_direct_purchase_prices,
@@ -3769,26 +3779,48 @@ <h2>Results</h2>
             throw new Error('Key IDs are required when using specific key request type');
           }
           
-          result = await get_identity_keys(
-            sdk,
-            values.identityId,
-            values.keyRequestType,
-            keyIds,
-            searchPurposeMap,
-            values.limit,
-            values.offset
-          );
+          if (useProofs) {
+            result = await get_identity_keys_with_proof_info(
+              sdk,
+              values.identityId,
+              values.keyRequestType,
+              keyIds,
+              values.limit,
+              values.offset
+            );
+          } else {
+            result = await get_identity_keys(
+              sdk,
+              values.identityId,
+              values.keyRequestType,
+              keyIds,
+              searchPurposeMap,
+              values.limit,
+              values.offset
+            );
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getIdentitiesContractKeys') {
           const purposes = values.purposes ? values.purposes.map(p => parseInt(p)) : undefined;
-          result = await get_identities_contract_keys(
-            sdk,
-            values.identitiesIds,
-            values.contractId,
-            values.documentTypeName || undefined,
-            purposes
-          );
-          // Result is already a JS object from serde_wasm_bindgen
+          if (useProofs) {
+            result = await get_identities_contract_keys_with_proof_info(
+              sdk,
+              values.identitiesIds,
+              values.contractId,
+              values.documentTypeName || undefined,
+              purposes
+            );
+            // Result is already a JS object from serde_wasm_bindgen
+          } else {
+            result = await get_identities_contract_keys(
+              sdk,
+              values.identitiesIds,
+              values.contractId,
+              values.documentTypeName || undefined,
+              purposes
+            );
+            // Result is already a JS object from serde_wasm_bindgen
+          }
         } else if (queryType === 'getIdentityNonce') {
           if (useProofs) {
             result = await get_identity_nonce_with_proof_info(sdk, values.identityId);
@@ -3802,27 +3834,58 @@ <h2>Results</h2>
             result = await get_identity_contract_nonce(sdk, values.identityId, values.contractId);
           }
         } else if (queryType === 'getIdentityBalance') {
-          result = await get_identity_balance(sdk, values.id);
+          if (useProofs) {
+            result = await get_identity_balance_with_proof_info(sdk, values.id);
+          } else {
+            result = await get_identity_balance(sdk, values.id);
+          }
           // Result is already an object with balance field
         } else if (queryType === 'getIdentitiesBalances') {
-          result = await get_identities_balances(sdk, values.ids);
+          if (useProofs) {
+            result = await get_identities_balances_with_proof_info(sdk, values.ids);
+          } else {
+            result = await get_identities_balances(sdk, values.ids);
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getIdentityBalanceAndRevision') {
-          result = await get_identity_balance_and_revision(sdk, values.id);
+          if (useProofs) {
+            result = await get_identity_balance_and_revision_with_proof_info(sdk, values.id);
+          } else {
+            result = await get_identity_balance_and_revision(sdk, values.id);
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getIdentityByPublicKeyHash') {
-          result = await get_identity_by_public_key_hash(sdk, values.publicKeyHash);
-          result = result.toJSON();
+          if (useProofs) {
+            result = await get_identity_by_public_key_hash_with_proof_info(sdk, values.publicKeyHash);
+            // Result is already a JS object from serde_wasm_bindgen
+          } else {
+            result = await get_identity_by_public_key_hash(sdk, values.publicKeyHash);
+            result = result.toJSON();
+          }
         } else if (queryType === 'getIdentityByNonUniquePublicKeyHash') {
-          result = await get_identity_by_non_unique_public_key_hash(
-            sdk,
-            values.publicKeyHash,
-            values.startAfter || undefined
-          );
-          // Result is already a JS array
+          if (useProofs) {
+            result = await get_identity_by_non_unique_public_key_hash_with_proof_info(
+              sdk,
+              values.publicKeyHash,
+              values.startAfter || undefined
+            );
+            // Result is already a JS object from serde_wasm_bindgen
+          } else {
+            result = await get_identity_by_non_unique_public_key_hash(
+              sdk,
+              values.publicKeyHash,
+              values.startAfter || undefined
+            );
+            // Result is already a JS array
+          }
         } else if (queryType === 'getIdentityTokenBalances') {
-          result = await get_identity_token_balances(sdk, values.identityId, values.tokenIds);
-          // Result is already a JS object from serde_wasm_bindgen
+          if (useProofs) {
+            result = await get_identity_token_balances_with_proof_info(sdk, values.identityId, values.tokenIds);
+            // Result is already a JS object from serde_wasm_bindgen
+          } else {
+            result = await get_identity_token_balances(sdk, values.identityId, values.tokenIds);
+            // Result is already a JS object from serde_wasm_bindgen
+          }
         }
         // Data contract queries
         else if (queryType === 'getDataContract' && values.id) {
@@ -4128,21 +4191,41 @@ <h2>Results</h2>
           }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getIdentityTokenInfos') {
-          result = await get_identity_token_infos(
-            sdk,
-            values.identityId,
-            values.tokenIds,
-            values.limit,
-            values.offset
-          );
-          // Result is already a JS object from serde_wasm_bindgen
+          if (useProofs) {
+            result = await get_identity_token_infos_with_proof_info(
+              sdk,
+              values.identityId,
+              values.tokenIds,
+              values.limit,
+              values.offset
+            );
+            // Result is already a JS object from serde_wasm_bindgen
+          } else {
+            result = await get_identity_token_infos(
+              sdk,
+              values.identityId,
+              values.tokenIds,
+              values.limit,
+              values.offset
+            );
+            // Result is already a JS object from serde_wasm_bindgen
+          }
         } else if (queryType === 'getIdentitiesTokenInfos') {
-          result = await get_identities_token_infos(
-            sdk,
-            values.identityIds,
-            values.tokenId
-          );
-          // Result is already a JS object from serde_wasm_bindgen
+          if (useProofs) {
+            result = await get_identities_token_infos_with_proof_info(
+              sdk,
+              values.identityIds,
+              values.tokenId
+            );
+            // Result is already a JS object from serde_wasm_bindgen
+          } else {
+            result = await get_identities_token_infos(
+              sdk,
+              values.identityIds,
+              values.tokenId
+            );
+            // Result is already a JS object from serde_wasm_bindgen
+          }
         } else if (queryType === 'getTokenStatuses') {
           if (useProofs) {
             result = await get_token_statuses_with_proof_info(sdk, values.tokenIds);
diff --git a/packages/wasm-sdk/test/ui-automation/tests/query-execution.spec.js b/packages/wasm-sdk/test/ui-automation/tests/query-execution.spec.js
index 0eb2041bcd9..216b648d7c6 100644
--- a/packages/wasm-sdk/test/ui-automation/tests/query-execution.spec.js
+++ b/packages/wasm-sdk/test/ui-automation/tests/query-execution.spec.js
@@ -573,19 +573,19 @@ test.describe('WASM SDK Query Execution Tests', () => {
     // Complete set of all available identity queries with correct proof support
     const testQueries = [
       { name: 'getIdentity', hasProofSupport: true, validateFn: validateIdentityResult },
-      { name: 'getIdentityBalance', hasProofSupport: false, validateFn: (result) => validateNumericResult(result, 'balance') },
-      { name: 'getIdentityKeys', hasProofSupport: false, validateFn: validateKeysResult },
+      { name: 'getIdentityBalance', hasProofSupport: true, validateFn: (result) => validateNumericResult(result, 'balance') },
+      { name: 'getIdentityKeys', hasProofSupport: true, validateFn: validateKeysResult },
       { name: 'getIdentityNonce', hasProofSupport: true, validateFn: (result) => validateNumericResult(result, 'nonce') },
       { name: 'getIdentityContractNonce', hasProofSupport: true, validateFn: (result) => validateNumericResult(result, 'nonce') },
-      { name: 'getIdentityByPublicKeyHash', hasProofSupport: false, validateFn: validateIdentityResult },
-      { name: 'getIdentitiesContractKeys', hasProofSupport: false, validateFn: validateKeysResult },
-      { name: 'getIdentitiesBalances', hasProofSupport: false, validateFn: validateBalancesResult },
-      { name: 'getIdentityBalanceAndRevision', hasProofSupport: false, validateFn: validateBalanceAndRevisionResult },
-      { name: 'getIdentityByNonUniquePublicKeyHash', hasProofSupport: false, validateFn: validateIdentitiesResult },
-      { name: 'getIdentityTokenBalances', hasProofSupport: false, validateFn: validateTokenBalanceResult },
+      { name: 'getIdentityByPublicKeyHash', hasProofSupport: true, validateFn: validateIdentityResult },
+      { name: 'getIdentitiesContractKeys', hasProofSupport: true, validateFn: validateKeysResult },
+      { name: 'getIdentitiesBalances', hasProofSupport: true, validateFn: validateBalancesResult },
+      { name: 'getIdentityBalanceAndRevision', hasProofSupport: true, validateFn: validateBalanceAndRevisionResult },
+      { name: 'getIdentityByNonUniquePublicKeyHash', hasProofSupport: true, validateFn: validateIdentitiesResult },
+      { name: 'getIdentityTokenBalances', hasProofSupport: true, validateFn: validateTokenBalanceResult },
       { name: 'getIdentitiesTokenBalances', hasProofSupport: true, validateFn: validateTokenBalanceResult },
-      { name: 'getIdentityTokenInfos', hasProofSupport: false, validateFn: validateTokenInfoResult },
-      { name: 'getIdentitiesTokenInfos', hasProofSupport: false, validateFn: validateTokenInfoResult }
+      { name: 'getIdentityTokenInfos', hasProofSupport: true, validateFn: validateTokenInfoResult },
+      { name: 'getIdentitiesTokenInfos', hasProofSupport: true, validateFn: validateTokenInfoResult }
     ];
 
     testQueries.forEach(({ name, hasProofSupport, validateFn }) => {

From a2f792d9ec5b2877148b305f22aa391079425fcc Mon Sep 17 00:00:00 2001
From: thephez <thephez@users.noreply.github.com>
Date: Tue, 5 Aug 2025 09:13:12 -0400
Subject: [PATCH 2/5] feat: enable proofs for more queries and tests

---
 packages/wasm-sdk/index.html                  | 69 ++++++++++++++-----
 .../tests/query-execution.spec.js             |  6 +-
 2 files changed, 52 insertions(+), 23 deletions(-)

diff --git a/packages/wasm-sdk/index.html b/packages/wasm-sdk/index.html
index a8252f65253..989c4a9a4a1 100644
--- a/packages/wasm-sdk/index.html
+++ b/packages/wasm-sdk/index.html
@@ -1097,7 +1097,9 @@ <h2>Results</h2>
       get_identity_token_balances_with_proof_info,
       // Data contract queries
       get_data_contract_history,
+      get_data_contract_history_with_proof_info,
       get_data_contracts,
+      get_data_contracts_with_proof_info,
       // Document queries
       get_documents,
       get_documents_with_proof_info,
@@ -1149,6 +1151,7 @@ <h2>Results</h2>
       get_contested_resource_voters_for_identity,
       get_contested_resource_identity_votes,
       get_vote_polls_by_end_date,
+      get_vote_polls_by_end_date_with_proof_info,
       // Group queries
       get_group_info,
       get_group_info_with_proof_info,
@@ -3897,17 +3900,33 @@ <h2>Results</h2>
             result = result.toJSON();
           }
         } else if (queryType === 'getDataContractHistory') {
-          result = await get_data_contract_history(
-            sdk,
-            values.id,
-            values.limit,
-            values.offset,
-            values.startAtMs
-          );
-          // Result is already a JS object from serde_wasm_bindgen
+          if (useProofs) {
+            result = await get_data_contract_history_with_proof_info(
+              sdk,
+              values.id,
+              values.limit,
+              values.offset,
+              values.startAtMs
+            );
+            // Result is already a JS object from serde_wasm_bindgen
+          } else {
+            result = await get_data_contract_history(
+              sdk,
+              values.id,
+              values.limit,
+              values.offset,
+              values.startAtMs
+            );
+            // Result is already a JS object from serde_wasm_bindgen
+          }
         } else if (queryType === 'getDataContracts') {
-          result = await get_data_contracts(sdk, values.ids);
-          // Result is already a JS object from serde_wasm_bindgen
+          if (useProofs) {
+            result = await get_data_contracts_with_proof_info(sdk, values.ids);
+            // Result is already a JS object from serde_wasm_bindgen
+          } else {
+            result = await get_data_contracts(sdk, values.ids);
+            // Result is already a JS object from serde_wasm_bindgen
+          }
         }
         // Document queries
         else if (queryType === 'getDocuments') {
@@ -4305,15 +4324,27 @@ <h2>Results</h2>
           );
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getVotePollsByEndDate') {
-          result = await get_vote_polls_by_end_date(
-            sdk,
-            values.startTimeMs,
-            values.endTimeMs,
-            values.limit,
-            values.offset,
-            values.orderAscending
-          );
-          // Result is already a JS object from serde_wasm_bindgen
+          if (useProofs) {
+            result = await get_vote_polls_by_end_date_with_proof_info(
+              sdk,
+              values.startTimeMs,
+              values.endTimeMs,
+              values.limit,
+              values.offset,
+              values.orderAscending
+            );
+            // Result is already a JS object from serde_wasm_bindgen
+          } else {
+            result = await get_vote_polls_by_end_date(
+              sdk,
+              values.startTimeMs,
+              values.endTimeMs,
+              values.limit,
+              values.offset,
+              values.orderAscending
+            );
+            // Result is already a JS object from serde_wasm_bindgen
+          }
         }
         // Epoch/Block queries
         else if (queryType === 'getFinalizedEpochInfos') {
diff --git a/packages/wasm-sdk/test/ui-automation/tests/query-execution.spec.js b/packages/wasm-sdk/test/ui-automation/tests/query-execution.spec.js
index 216b648d7c6..dfb5c582564 100644
--- a/packages/wasm-sdk/test/ui-automation/tests/query-execution.spec.js
+++ b/packages/wasm-sdk/test/ui-automation/tests/query-execution.spec.js
@@ -307,8 +307,7 @@ test.describe('WASM SDK Query Execution Tests', () => {
       }
     });
 
-    // Skip this test - proof support not yet implemented in WASM SDK for getDataContracts
-    test.skip('should execute getDataContracts query with proof info', async () => {
+    test('should execute getDataContracts query with proof info', async () => {
       const { result, proofEnabled } = await executeQueryWithProof(
         wasmSdkPage, 
         parameterInjector, 
@@ -334,8 +333,7 @@ test.describe('WASM SDK Query Execution Tests', () => {
       }
     });
 
-    // Skip this test - proof support not yet implemented in WASM SDK for getDataContractHistory
-    test.skip('should execute getDataContractHistory query with proof info', async () => {
+    test('should execute getDataContractHistory query with proof info', async () => {
       const { result, proofEnabled } = await executeQueryWithProof(
         wasmSdkPage, 
         parameterInjector, 

From 92af0cec5814ad3231846ddd6c90cfe7e4933952 Mon Sep 17 00:00:00 2001
From: thephez <thephez@users.noreply.github.com>
Date: Tue, 5 Aug 2025 09:44:02 -0400
Subject: [PATCH 3/5] chore: handle lack of proof support for getIdentityKeys
 search

  1. Identified the root cause: The Rust proof function doesn't support "search" key request type and lacks the
  search_purpose_map parameter
  2. Implemented graceful fallback: When keyRequestType === 'search' and proofs are enabled, the code now falls back to the
  non-proof version with a console warning
---
 packages/wasm-sdk/index.html | 31 +++++++++++++++++++++++--------
 1 file changed, 23 insertions(+), 8 deletions(-)

diff --git a/packages/wasm-sdk/index.html b/packages/wasm-sdk/index.html
index 989c4a9a4a1..d6a2537e513 100644
--- a/packages/wasm-sdk/index.html
+++ b/packages/wasm-sdk/index.html
@@ -3783,14 +3783,29 @@ <h2>Results</h2>
           }
           
           if (useProofs) {
-            result = await get_identity_keys_with_proof_info(
-              sdk,
-              values.identityId,
-              values.keyRequestType,
-              keyIds,
-              values.limit,
-              values.offset
-            );
+            if (values.keyRequestType === 'search') {
+              // Search keys with proof not yet supported in WASM SDK, fallback to non-proof version
+              console.warn('Search keys with proof not yet supported, using non-proof version');
+              result = await get_identity_keys(
+                sdk,
+                values.identityId,
+                values.keyRequestType,
+                keyIds,
+                searchPurposeMap,
+                values.limit,
+                values.offset
+              );
+            } else {
+              // 'all' and 'specific' key types support proof
+              result = await get_identity_keys_with_proof_info(
+                sdk,
+                values.identityId,
+                values.keyRequestType,
+                keyIds,
+                values.limit,
+                values.offset
+              );
+            }
           } else {
             result = await get_identity_keys(
               sdk,

From 4af5e30e81b6be674c9d4da7476a9ed7f3af1395 Mon Sep 17 00:00:00 2001
From: thephez <thephez@users.noreply.github.com>
Date: Tue, 5 Aug 2025 13:12:34 -0400
Subject: [PATCH 4/5] feat: enable proofs for most remaining queries

---
 packages/wasm-sdk/index.html | 277 +++++++++++++++++++++++++----------
 1 file changed, 201 insertions(+), 76 deletions(-)

diff --git a/packages/wasm-sdk/index.html b/packages/wasm-sdk/index.html
index d6a2537e513..933409c874a 100644
--- a/packages/wasm-sdk/index.html
+++ b/packages/wasm-sdk/index.html
@@ -1117,6 +1117,7 @@ <h2>Results</h2>
       get_epochs_info,
       get_epochs_info_with_proof_info,
       get_finalized_epoch_infos,
+      get_finalized_epoch_infos_with_proof_info,
       get_current_epoch,
       get_current_epoch_with_proof_info,
       get_evonodes_proposed_epoch_blocks_by_ids,
@@ -1141,15 +1142,22 @@ <h2>Results</h2>
       get_token_statuses,
       get_token_statuses_with_proof_info,
       get_token_direct_purchase_prices,
+      get_token_direct_purchase_prices_with_proof_info,
       get_token_contract_info,
+      get_token_contract_info_with_proof_info,
       get_token_perpetual_distribution_last_claim,
+      get_token_perpetual_distribution_last_claim_with_proof_info,
       get_token_total_supply,
       get_token_total_supply_with_proof_info,
       // Voting/Contested Resource queries
       get_contested_resources,
+      get_contested_resources_with_proof_info,
       get_contested_resource_vote_state,
+      get_contested_resource_vote_state_with_proof_info,
       get_contested_resource_voters_for_identity,
+      get_contested_resource_voters_for_identity_with_proof_info,
       get_contested_resource_identity_votes,
+      get_contested_resource_identity_votes_with_proof_info,
       get_vote_polls_by_end_date,
       get_vote_polls_by_end_date_with_proof_info,
       // Group queries
@@ -1158,7 +1166,9 @@ <h2>Results</h2>
       get_group_infos,
       get_group_infos_with_proof_info,
       get_group_actions,
+      get_group_actions_with_proof_info,
       get_group_action_signers,
+      get_group_action_signers_with_proof_info,
       // DPNS functions
       dpns_convert_to_homograph_safe,
       dpns_is_valid_username,
@@ -4112,7 +4122,11 @@ <h2>Results</h2>
         }
         // Protocol/Version queries
         else if (queryType === 'getProtocolVersionUpgradeState') {
-          result = await get_protocol_version_upgrade_state(sdk);
+          if (useProofs) {
+            result = await get_protocol_version_upgrade_state_with_proof_info(sdk);
+          } else {
+            result = await get_protocol_version_upgrade_state(sdk);
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getProtocolVersionUpgradeVoteStatus') {
           result = await get_protocol_version_upgrade_vote_status(
@@ -4148,12 +4162,21 @@ <h2>Results</h2>
           }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getFinalizedEpochInfos') {
-          result = await get_finalized_epoch_infos(
-            sdk,
-            values.startEpoch,
-            values.count,
-            values.ascending
-          );
+          if (useProofs) {
+            result = await get_finalized_epoch_infos_with_proof_info(
+              sdk,
+              values.startEpoch,
+              values.count,
+              values.ascending
+            );
+          } else {
+            result = await get_finalized_epoch_infos(
+              sdk,
+              values.startEpoch,
+              values.count,
+              values.ascending
+            );
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getEvonodesProposedEpochBlocksByIds') {
           result = await get_evonodes_proposed_epoch_blocks_by_ids(
@@ -4268,17 +4291,33 @@ <h2>Results</h2>
           }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getTokenDirectPurchasePrices') {
-          result = await get_token_direct_purchase_prices(sdk, values.tokenIds);
+          if (useProofs) {
+            result = await get_token_direct_purchase_prices_with_proof_info(sdk, values.tokenIds);
+          } else {
+            result = await get_token_direct_purchase_prices(sdk, values.tokenIds);
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getTokenContractInfo') {
-          result = await get_token_contract_info(sdk, values.dataContractId);
+          if (useProofs) {
+            result = await get_token_contract_info_with_proof_info(sdk, values.dataContractId);
+          } else {
+            result = await get_token_contract_info(sdk, values.dataContractId);
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getTokenPerpetualDistributionLastClaim') {
-          result = await get_token_perpetual_distribution_last_claim(
-            sdk,
-            values.identityId,
-            values.tokenId
-          );
+          if (useProofs) {
+            result = await get_token_perpetual_distribution_last_claim_with_proof_info(
+              sdk,
+              values.identityId,
+              values.tokenId
+            );
+          } else {
+            result = await get_token_perpetual_distribution_last_claim(
+              sdk,
+              values.identityId,
+              values.tokenId
+            );
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getTokenTotalSupply') {
           if (useProofs) {
@@ -4291,52 +4330,104 @@ <h2>Results</h2>
         // Voting/Contested Resource queries
         else if (queryType === 'getContestedResources') {
           const startAtValue = values.startAtValue ? new TextEncoder().encode(values.startAtValue) : undefined;
-          result = await get_contested_resources(
-            sdk,
-            values.documentTypeName,
-            values.dataContractId,
-            values.indexName,
-            values.resultType,
-            values.allowIncludeLockedAndAbstainingVoteTally,
-            startAtValue,
-            values.limit,
-            values.offset,
-            values.orderAscending
-          );
+          if (useProofs) {
+            result = await get_contested_resources_with_proof_info(
+              sdk,
+              values.documentTypeName,
+              values.dataContractId,
+              values.indexName,
+              values.resultType,
+              values.allowIncludeLockedAndAbstainingVoteTally,
+              startAtValue,
+              values.limit,
+              values.offset,
+              values.orderAscending
+            );
+          } else {
+            result = await get_contested_resources(
+              sdk,
+              values.documentTypeName,
+              values.dataContractId,
+              values.indexName,
+              values.resultType,
+              values.allowIncludeLockedAndAbstainingVoteTally,
+              startAtValue,
+              values.limit,
+              values.offset,
+              values.orderAscending
+            );
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getContestedResourceVoteState') {
-          result = await get_contested_resource_vote_state(
-            sdk,
-            values.dataContractId,
-            values.documentTypeName,
-            values.indexName,
-            values.resultType,
-            values.allowIncludeLockedAndAbstainingVoteTally,
-            values.startAtIdentifierInfo ? JSON.stringify(values.startAtIdentifierInfo) : undefined,
-            values.count,
-            values.orderAscending
-          );
+          if (useProofs) {
+            result = await get_contested_resource_vote_state_with_proof_info(
+              sdk,
+              values.dataContractId,
+              values.documentTypeName,
+              values.indexName,
+              values.resultType,
+              values.allowIncludeLockedAndAbstainingVoteTally,
+              values.startAtIdentifierInfo ? JSON.stringify(values.startAtIdentifierInfo) : undefined,
+              values.count,
+              values.orderAscending
+            );
+          } else {
+            result = await get_contested_resource_vote_state(
+              sdk,
+              values.dataContractId,
+              values.documentTypeName,
+              values.indexName,
+              values.resultType,
+              values.allowIncludeLockedAndAbstainingVoteTally,
+              values.startAtIdentifierInfo ? JSON.stringify(values.startAtIdentifierInfo) : undefined,
+              values.count,
+              values.orderAscending
+            );
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getContestedResourceVotersForIdentity') {
-          result = await get_contested_resource_voters_for_identity(
-            sdk,
-            values.dataContractId,
-            values.documentTypeName,
-            values.indexName,
-            values.contestantId,
-            values.startAtIdentifierInfo ? JSON.stringify(values.startAtIdentifierInfo) : undefined,
-            values.count,
-            values.orderAscending
-          );
+          if (useProofs) {
+            result = await get_contested_resource_voters_for_identity_with_proof_info(
+              sdk,
+              values.dataContractId,
+              values.documentTypeName,
+              values.indexName,
+              values.contestantId,
+              values.startAtIdentifierInfo ? JSON.stringify(values.startAtIdentifierInfo) : undefined,
+              values.count,
+              values.orderAscending
+            );
+          } else {
+            result = await get_contested_resource_voters_for_identity(
+              sdk,
+              values.dataContractId,
+              values.documentTypeName,
+              values.indexName,
+              values.contestantId,
+              values.startAtIdentifierInfo ? JSON.stringify(values.startAtIdentifierInfo) : undefined,
+              values.count,
+              values.orderAscending
+            );
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getContestedResourceIdentityVotes') {
-          result = await get_contested_resource_identity_votes(
-            sdk,
-            values.identityId,
-            values.limit,
-            values.offset,
-            values.orderAscending
-          );
+          if (useProofs) {
+            result = await get_contested_resource_identity_votes_with_proof_info(
+              sdk,
+              values.identityId,
+              values.limit,
+              values.offset,
+              values.orderAscending
+            );
+          } else {
+            result = await get_contested_resource_identity_votes(
+              sdk,
+              values.identityId,
+              values.limit,
+              values.offset,
+              values.orderAscending
+            );
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getVotePollsByEndDate') {
           if (useProofs) {
@@ -4389,7 +4480,11 @@ <h2>Results</h2>
         }
         // Group queries
         else if (queryType === 'getGroupInfo') {
-          result = await get_group_info(sdk, values.contractId, values.groupContractPosition);
+          if (useProofs) {
+            result = await get_group_info_with_proof_info(sdk, values.contractId, values.groupContractPosition);
+          } else {
+            result = await get_group_info(sdk, values.contractId, values.groupContractPosition);
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getGroupInfos') {
           let startAtInfo = null;
@@ -4399,12 +4494,21 @@ <h2>Results</h2>
               included: values.startGroupContractPositionIncluded || false
             };
           }
-          result = await get_group_infos(
-            sdk,
-            values.contractId,
-            startAtInfo,
-            values.count
-          );
+          if (useProofs) {
+            result = await get_group_infos_with_proof_info(
+              sdk,
+              values.contractId,
+              startAtInfo,
+              values.count
+            );
+          } else {
+            result = await get_group_infos(
+              sdk,
+              values.contractId,
+              startAtInfo,
+              values.count
+            );
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getGroupActions') {
           let startAtInfo = null;
@@ -4414,23 +4518,44 @@ <h2>Results</h2>
               included: values.startActionIdIncluded || false
             };
           }
-          result = await get_group_actions(
-            sdk,
-            values.contractId,
-            values.groupContractPosition,
-            values.status,
-            startAtInfo,
-            values.count
-          );
+          if (useProofs) {
+            result = await get_group_actions_with_proof_info(
+              sdk,
+              values.contractId,
+              values.groupContractPosition,
+              values.status,
+              startAtInfo,
+              values.count
+            );
+          } else {
+            result = await get_group_actions(
+              sdk,
+              values.contractId,
+              values.groupContractPosition,
+              values.status,
+              startAtInfo,
+              values.count
+            );
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getGroupActionSigners') {
-          result = await get_group_action_signers(
-            sdk,
-            values.contractId,
-            values.groupContractPosition,
-            values.status,
-            values.actionId
-          );
+          if (useProofs) {
+            result = await get_group_action_signers_with_proof_info(
+              sdk,
+              values.contractId,
+              values.groupContractPosition,
+              values.status,
+              values.actionId
+            );
+          } else {
+            result = await get_group_action_signers(
+              sdk,
+              values.contractId,
+              values.groupContractPosition,
+              values.status,
+              values.actionId
+            );
+          }
           // Result is already a JS object from serde_wasm_bindgen
         } else {
           // Placeholder for unimplemented queries

From 56724befc79a0a4e66432bdf3f8279854d6e2be7 Mon Sep 17 00:00:00 2001
From: thephez <thephez@users.noreply.github.com>
Date: Tue, 5 Aug 2025 13:13:53 -0400
Subject: [PATCH 5/5] fix: make proofs work for some dpns queries

Also handle the proof response data consistently
---
 packages/wasm-sdk/index.html | 69 +++++++++++++++++++++++++++---------
 1 file changed, 52 insertions(+), 17 deletions(-)

diff --git a/packages/wasm-sdk/index.html b/packages/wasm-sdk/index.html
index 933409c874a..4690923ad45 100644
--- a/packages/wasm-sdk/index.html
+++ b/packages/wasm-sdk/index.html
@@ -1107,6 +1107,7 @@ <h2>Results</h2>
       get_document_with_proof_info,
       get_dpns_username,
       get_dpns_usernames,
+      get_dpns_usernames_with_proof_info,
       get_dpns_username_by_name,
       get_dpns_username_by_name_with_proof_info,
       // Protocol/Version queries
@@ -4009,11 +4010,19 @@ <h2>Results</h2>
           }
           // Result is already a JS object from serde_wasm_bindgen
         } else if (queryType === 'getDpnsUsername') {
-          result = await get_dpns_usernames(
-            sdk,
-            values.identityId,
-            values.limit || 10  // Default to 10 if not specified
-          );
+          if (useProofs) {
+            result = await get_dpns_usernames_with_proof_info(
+              sdk,
+              values.identityId,
+              values.limit || 10  // Default to 10 if not specified
+            );
+          } else {
+            result = await get_dpns_usernames(
+              sdk,
+              values.identityId,
+              values.limit || 10  // Default to 10 if not specified
+            );
+          }
           // Result is an array of usernames
         } else if (queryType === 'dpnsCheckAvailability') {
           // Handle DPNS availability check
@@ -4085,19 +4094,34 @@ <h2>Results</h2>
           ]);
           
           // Execute the document query
-          const documents = await get_documents(
-            sdk,
-            DPNS_CONTRACT_ID,
-            DPNS_DOCUMENT_TYPE,
-            whereClause,
-            orderBy,
-            limit,
-            null, // startAfter
-            null  // startAt
-          );
+          let documents;
+          if (useProofs) {
+            documents = await get_documents_with_proof_info(
+              sdk,
+              DPNS_CONTRACT_ID,
+              DPNS_DOCUMENT_TYPE,
+              whereClause,
+              orderBy,
+              limit,
+              null, // startAfter
+              null  // startAt
+            );
+          } else {
+            documents = await get_documents(
+              sdk,
+              DPNS_CONTRACT_ID,
+              DPNS_DOCUMENT_TYPE,
+              whereClause,
+              orderBy,
+              limit,
+              null, // startAfter
+              null  // startAt
+            );
+          }
           
           // Transform the results to show username and owner identity
-          const searchResults = documents.map(doc => {
+          const documentsArray = useProofs ? documents.data : documents;
+          const searchResults = documentsArray.map(doc => {
             // Access the data field which contains the DPNS document fields
             const data = doc.data || doc;
             
@@ -4110,7 +4134,7 @@ <h2>Results</h2>
             };
           });
           
-          result = {
+          const processedResult = {
             prefix: values.prefix,
             totalResults: searchResults.length,
             limit: limit,
@@ -4119,6 +4143,17 @@ <h2>Results</h2>
               ? `Found ${searchResults.length} name(s) starting with "${values.prefix}"` 
               : `No names found starting with "${values.prefix}"`
           };
+          
+          if (useProofs) {
+            // Create proof response structure with processed data for split view
+            result = {
+              data: processedResult,
+              metadata: documents.metadata,
+              proof: documents.proof
+            };
+          } else {
+            result = processedResult;
+          }
         }
         // Protocol/Version queries
         else if (queryType === 'getProtocolVersionUpgradeState') {