chore: Add view_account, view_code, view_block

workspaces/src/lib.rs

@@ -5,7 +5,7 @@ mod worker;
 
 pub mod prelude;
 
-pub use network::{Account, Contract, DevNetwork, Network};
+pub use network::{Account, AccountDetails, Block, Contract, DevNetwork, Network};
 pub use types::{AccountId, BlockHeight, CryptoHash, InMemorySigner};
 pub use worker::{
     mainnet, mainnet_archival, sandbox, testnet, with_mainnet, with_sandbox, with_testnet, Worker,

workspaces/src/network/account.rs

@@ -1,10 +1,11 @@
 use std::convert::TryInto;
 
 use near_crypto::KeyType;
+use near_primitives::views::AccountView;
 
 use crate::rpc::client::{DEFAULT_CALL_DEPOSIT, DEFAULT_CALL_FN_GAS};
 use crate::types::{AccountId, Balance, Gas, InMemorySigner, SecretKey};
-use crate::{Network, Worker};
+use crate::{CryptoHash, Network, Worker};
 
 use super::{CallExecution, CallExecutionDetails, ViewResultDetails};
 
@@ -75,6 +76,14 @@ impl Account {
             .await
     }
 
+    /// Views the current account's details such as balance and storage usage.
+    pub async fn view_account<T: Network>(
+        &self,
+        worker: &Worker<T>,
+    ) -> anyhow::Result<AccountDetails> {
+        worker.view_account(&self.id).await
+    }
+
     /// Create a new sub account. Returns a CreateAccountBuilder object that
     /// we can make use of to fill out the rest of the details. The sub account
     /// id will be in the form of: "{new_account_id}.{parent_account_id}"
@@ -165,6 +174,19 @@ impl Contract {
         worker.view(self.id(), function, args).await
     }
 
+    /// View the WASM code bytes of this contract.
+    pub async fn view_code<T: Network>(&self, worker: &Worker<T>) -> anyhow::Result<Vec<u8>> {
+        worker.view_code(self.id()).await
+    }
+
+    /// Views the current contract's details such as balance and storage usage.
+    pub async fn view_account<T: Network>(
+        &self,
+        worker: &Worker<T>,
+    ) -> anyhow::Result<AccountDetails> {
+        worker.view_account(self.id()).await
+    }
+
     /// Deletes the current contract, and returns the execution details of this
     /// transaction. The beneciary will receive the funds of the account deleted
     pub async fn delete_contract<T: Network>(
@@ -314,3 +336,24 @@ where
         })
     }
 }
+
+/// Details of an Account or Contract. This is an non-exhaustive list of items
+/// that the account stores in the blockchain state.
+#[non_exhaustive]
+pub struct AccountDetails {
+    pub amount: Balance,
+    pub locked: Balance,
+    pub code_hash: CryptoHash,
+    pub storage_usage: u64,
+}
+
+impl From<AccountView> for AccountDetails {
+    fn from(account: AccountView) -> Self {
+        Self {
+            amount: account.amount,
+            locked: account.locked,
+            code_hash: CryptoHash(account.code_hash.0),
+            storage_usage: account.storage_usage,
+        }
+    }
+}

workspaces/src/network/block.rs

@@ -0,0 +1,59 @@
+use near_primitives::views::{BlockHeaderView, BlockView};
+
+use crate::{BlockHeight, CryptoHash};
+
+/// Struct containing information on block coming from the network
+pub struct Block {
+    header: BlockHeader,
+}
+
+impl Block {
+    /// The block timestamp in nanoseconds.
+    pub fn timestamp(&self) -> u64 {
+        self.header.timestamp_nanosec
+    }
+
+    /// Current height of this block.
+    pub fn height(&self) -> BlockHeight {
+        self.header.height
+    }
+
+    /// The hash of the block itself.
+    pub fn hash(&self) -> &CryptoHash {
+        &self.header.hash
+    }
+
+    /// The id of an epoch this block belongs to.
+    pub fn epoch_id(&self) -> &CryptoHash {
+        &self.header.epoch_id
+    }
+}
+
+/// The block header info. This is a non-exhaustive list of items that
+/// could be present in a block header. More can be added in the future.
+#[non_exhaustive]
+struct BlockHeader {
+    height: BlockHeight,
+    epoch_id: CryptoHash,
+    hash: CryptoHash,
+    timestamp_nanosec: u64,
+}
+
+impl From<BlockView> for Block {
+    fn from(block_view: BlockView) -> Self {
+        Self {
+            header: block_view.header.into(),
+        }
+    }
+}
+
+impl From<BlockHeaderView> for BlockHeader {
+    fn from(header_view: BlockHeaderView) -> Self {
+        Self {
+            height: header_view.height,
+            epoch_id: CryptoHash(header_view.epoch_id.0),
+            hash: CryptoHash(header_view.hash.0),
+            timestamp_nanosec: header_view.timestamp_nanosec,
+        }
+    }
+}

workspaces/src/network/mod.rs

@@ -1,4 +1,5 @@
 mod account;
+mod block;
 mod info;
 mod mainnet;
 mod result;
@@ -17,7 +18,8 @@ use crate::rpc::patch::ImportContractBuilder;
 use crate::types::{AccountId, KeyType, SecretKey};
 use crate::Worker;
 
-pub use crate::network::account::{Account, Contract};
+pub use crate::network::account::{Account, AccountDetails, Contract};
+pub use crate::network::block::Block;
 pub use crate::network::mainnet::Mainnet;
 pub use crate::network::result::{CallExecution, CallExecutionDetails, ViewResultDetails};
 pub use crate::network::sandbox::Sandbox;

workspaces/src/rpc/client.rs

@@ -15,7 +15,8 @@ use near_primitives::transaction::{
 };
 use near_primitives::types::{Balance, BlockId, Finality, Gas, StoreKey};
 use near_primitives::views::{
-    AccessKeyView, AccountView, ContractCodeView, FinalExecutionOutcomeView, QueryRequest,
+    AccessKeyView, AccountView, BlockView, ContractCodeView, FinalExecutionOutcomeView,
+    QueryRequest,
 };
 
 use crate::network::ViewResultDetails;
@@ -238,6 +239,18 @@ impl Client {
         }
     }
 
+    pub(crate) async fn view_block(&self, block_id: Option<BlockId>) -> anyhow::Result<BlockView> {
+        let block_reference = block_id
+            .map(Into::into)
+            .unwrap_or_else(|| Finality::None.into());
+
+        let block_view = self
+            .query(&methods::block::RpcBlockRequest { block_reference })
+            .await?;
+
+        Ok(block_view)
+    }
+
     pub(crate) async fn deploy(
         &self,
         signer: &InMemorySigner,

workspaces/src/worker/impls.rs

@@ -4,15 +4,15 @@ use async_trait::async_trait;
 use near_primitives::types::{Balance, StoreKey};
 
 use crate::network::{
-    Account, AllowDevAccountCreation, CallExecution, CallExecutionDetails, Contract, NetworkClient,
-    NetworkInfo, StatePatcher, TopLevelAccountCreator, ViewResultDetails,
+    Account, AllowDevAccountCreation, Block, CallExecution, CallExecutionDetails, Contract,
+    NetworkClient, NetworkInfo, StatePatcher, TopLevelAccountCreator, ViewResultDetails,
 };
 use crate::network::{Info, Sandbox};
 use crate::rpc::client::{Client, DEFAULT_CALL_DEPOSIT, DEFAULT_CALL_FN_GAS};
 use crate::rpc::patch::ImportContractBuilder;
 use crate::types::{AccountId, Gas, InMemorySigner, SecretKey};
 use crate::worker::Worker;
-use crate::Network;
+use crate::{AccountDetails, Network};
 
 impl<T> Clone for Worker<T> {
     fn clone(&self) -> Self {
@@ -119,6 +119,12 @@ where
             .await
     }
 
+    /// View the WASM code bytes of a contract on the network.
+    pub async fn view_code(&self, contract_id: &AccountId) -> anyhow::Result<Vec<u8>> {
+        let code_view = self.client().view_code(contract_id.clone(), None).await?;
+        Ok(code_view.code)
+    }
+
     pub async fn view_state(
         &self,
         contract_id: &AccountId,
@@ -127,6 +133,10 @@ where
         self.client().view_state(contract_id.clone(), prefix).await
     }
 
+    pub async fn view_latest_block(&self) -> anyhow::Result<Block> {
+        self.client().view_block(None).await.map(Into::into)
+    }
+
     pub async fn transfer_near(
         &self,
         signer: &InMemorySigner,
@@ -150,6 +160,13 @@ where
             .await
             .map(Into::into)
     }
+
+    pub async fn view_account(&self, account_id: &AccountId) -> anyhow::Result<AccountDetails> {
+        self.client()
+            .view_account(account_id.clone(), None)
+            .await
+            .map(Into::into)
+    }
 }
 
 impl Worker<Sandbox> {