feat: token support

Cargo.toml

@@ -28,7 +28,8 @@ members = [
     "packages/simple-signer",
     "packages/rs-json-schema-compatibility-validator",
     "packages/check-features",
-    "packages/wallet-utils-contract"
+    "packages/wallet-utils-contract",
+    "packages/token-history-contract"
 ]
 [workspace.package]
 

package.json

@@ -65,7 +65,8 @@
     "packages/masternode-reward-shares-contract",
     "packages/dash-spv",
     "packages/wasm-dpp",
-    "packages/withdrawals-contract"
+    "packages/withdrawals-contract",
+    "packages/token-history-contract"
   ],
   "resolutions": {
     "elliptic": "6.5.7",

packages/dapi-grpc/protos/platform/v0/platform.proto

@@ -55,6 +55,8 @@ service Platform {
   rpc getPathElements(GetPathElementsRequest) returns (GetPathElementsResponse);
   rpc getStatus(GetStatusRequest) returns (GetStatusResponse);
   rpc getCurrentQuorumsInfo(GetCurrentQuorumsInfoRequest) returns (GetCurrentQuorumsInfoResponse);
+  rpc getIdentityTokenBalances(GetIdentityTokenBalancesRequest) returns (GetIdentityTokenBalancesResponse);
+  rpc getIdentitiesTokenBalances(GetIdentitiesTokenBalancesRequest) returns (GetIdentitiesTokenBalancesResponse);
 }
 
 // Proof message includes cryptographic proofs for validating responses
@@ -1215,3 +1217,69 @@ message GetCurrentQuorumsInfoResponse {
   }
   oneof version { GetCurrentQuorumsInfoResponseV0 v0 = 1; }
 }
+
+message GetIdentityTokenBalancesRequest {
+  message GetIdentityTokenBalancesRequestV0 {
+    bytes identity_id = 1;            // ID of the identity
+    repeated bytes token_ids = 2;     // List of token IDs
+    bool prove = 3;                   // Flag to request a proof as the response
+  }
+  oneof version {
+    GetIdentityTokenBalancesRequestV0 v0 = 1;
+  }
+}
+
+message GetIdentityTokenBalancesResponse {
+  message GetIdentityTokenBalancesResponseV0 {
+    message TokenBalanceEntry {
+      bytes token_id = 1;     // Token ID
+      optional uint64 balance = 2;     // Token balance for the contract
+    }
+
+    message TokenBalances {
+      repeated TokenBalanceEntry token_balances = 1;  // List of token balances
+    }
+
+    oneof result {
+      TokenBalances token_balances = 1;  // Actual token balances
+      Proof proof = 2;                   // Proof of the token balances, if requested
+    }
+    ResponseMetadata metadata = 3;  // Metadata about the blockchain state
+  }
+  oneof version {
+    GetIdentityTokenBalancesResponseV0 v0 = 1;
+  }
+}
+
+message GetIdentitiesTokenBalancesRequest {
+  message GetIdentitiesTokenBalancesRequestV0 {
+    bytes token_id = 1;               // Token ID
+    repeated bytes identity_ids = 2;  // List of identity IDs
+    bool prove = 3;                   // Flag to request a proof as the response
+  }
+  oneof version {
+    GetIdentitiesTokenBalancesRequestV0 v0 = 1;
+  }
+}
+
+message GetIdentitiesTokenBalancesResponse {
+  message GetIdentitiesTokenBalancesResponseV0 {
+    message IdentityTokenBalanceEntry {
+      bytes identity_id = 1;  // Identity ID
+      optional uint64 balance = 2;     // Token balance for the identity
+    }
+
+    message IdentityTokenBalances {
+      repeated IdentityTokenBalanceEntry identity_token_balances = 1;  // List of identity token balances
+    }
+
+    oneof result {
+      IdentityTokenBalances identity_token_balances = 1;  // Actual identity token balances
+      Proof proof = 2;                                    // Proof of the balances, if requested
+    }
+    ResponseMetadata metadata = 3;  // Metadata about the blockchain state
+  }
+  oneof version {
+    GetIdentitiesTokenBalancesResponseV0 v0 = 1;
+  }
+}

packages/data-contracts/Cargo.toml

@@ -17,3 +17,4 @@ dashpay-contract = { path = "../dashpay-contract" }
 feature-flags-contract = { path = "../feature-flags-contract" }
 platform-value = { path = "../rs-platform-value" }
 wallet-utils-contract = { path = "../wallet-utils-contract" }
+token-history-contract = { path = "../token-history-contract" }

packages/data-contracts/src/error.rs

@@ -119,3 +119,20 @@ impl From<wallet_utils_contract::Error> for Error {
         }
     }
 }
+
+impl From<token_history_contract::Error> for Error {
+    fn from(e: token_history_contract::Error) -> Self {
+        match e {
+            token_history_contract::Error::UnknownVersionMismatch {
+                method,
+                known_versions,
+                received,
+            } => Error::UnknownVersionMismatch {
+                method,
+                known_versions,
+                received,
+            },
+            token_history_contract::Error::InvalidSchemaJson(e) => Error::InvalidSchemaJson(e),
+        }
+    }
+}

packages/data-contracts/src/lib.rs

@@ -9,6 +9,7 @@ pub use feature_flags_contract;
 pub use masternode_reward_shares_contract;
 use platform_value::Identifier;
 use platform_version::version::PlatformVersion;
+pub use token_history_contract;
 pub use wallet_utils_contract;
 pub use withdrawals_contract;
 
@@ -21,6 +22,7 @@ pub enum SystemDataContract {
     DPNS = 3,
     Dashpay = 4,
     WalletUtils = 5,
+    TokenHistory = 6,
 }
 
 pub struct DataContractSource {
@@ -40,6 +42,7 @@ impl SystemDataContract {
             SystemDataContract::DPNS => dpns_contract::ID_BYTES,
             SystemDataContract::Dashpay => dashpay_contract::ID_BYTES,
             SystemDataContract::WalletUtils => wallet_utils_contract::ID_BYTES,
+            SystemDataContract::TokenHistory => token_history_contract::ID_BYTES,
         };
         Identifier::new(bytes)
     }
@@ -92,6 +95,13 @@ impl SystemDataContract {
                 definitions: wallet_utils_contract::load_definitions(platform_version)?,
                 document_schemas: wallet_utils_contract::load_documents_schemas(platform_version)?,
             },
+            SystemDataContract::TokenHistory => DataContractSource {
+                id_bytes: token_history_contract::ID_BYTES,
+                owner_id_bytes: token_history_contract::OWNER_ID_BYTES,
+                version: platform_version.system_data_contracts.wallet as u32,
+                definitions: token_history_contract::load_definitions(platform_version)?,
+                document_schemas: token_history_contract::load_documents_schemas(platform_version)?,
+            },
         };
 
         Ok(data)