refactor(types): change account id to newtype and deprecate ValidAccountId

CHANGELOG.md

@@ -17,6 +17,9 @@
   * This removes the re-export of `wee_alloc` because if this feature is enabled, the allocator will already be set.
   * Deprecates `setup_alloc!` macro as this will be setup by default, as long as the `wee_alloc` feature is not specifically disabled. In this case, the allocator can be overriden to a custom one or set manually if intended.
 * Update `TreeMap` iterator implementation to avoid unnecessary storage reads. [PR 428](https://github.com/near/near-sdk-rs/pull/428).
+* Update `AccountId` to be a newtype with merged functionality from `ValidAccountId`
+  * Removes `ValidAccountId` to avoid having multiple types for account IDs
+  * This type will have `ValidAccountId`'s JSON (de)serialization and the borsh serialization will be equivalent to what it was previously
 
 ## `3.1.0` [04-06-2021]
 

HELP.md

@@ -320,15 +320,15 @@ impl Contract {
     }
 
     /// Change method. Changes the state, and then saves the new state internally.
-    pub fn set_owner_id(&mut self, new_owner_id: ValidAccountId) {
+    pub fn set_owner_id(&mut self, new_owner_id: AccountId) {
         self.owner_id = new_owner_id.into();
     }
 
     /// View method that "modifies" state, for code structure or computational
     /// efficiency reasons. Changes state in-memory, but does NOT save the new
     /// state. If called internally by a change method, WILL result in updated
     /// contract state.
-    pub fn update_stats(&self, account_id: ValidAccountId, score: U64) -> Account {
+    pub fn update_stats(&self, account_id: AccountId, score: U64) -> Account {
         let account = self.accounts.get(account_id).expect("account not found");
         account.total += score;
         account
@@ -427,7 +427,7 @@ E.g.
 ```rust
 #[near_bindgen]
 impl Contract {
-    pub fn withdraw_100(&mut self, receiver_id: ValidAccountId) -> Promise {
+    pub fn withdraw_100(&mut self, receiver_id: AccountId) -> Promise {
         Promise::new(receiver_id.into()).transfer(100)
     }
 }
@@ -641,8 +641,8 @@ impl Contract {
         self.status_updates.remove(&env::predecessor_account_id());
     }
 
-    pub fn get_status(&self, account_id: ValidAccountId) -> Option<String> {
-        self.status_updates.get(account_id.as_ref())
+    pub fn get_status(&self, account_id: AccountId) -> Option<String> {
+        self.status_updates.get(&account_id)
     }
 }
 ```

README.md

@@ -44,7 +44,7 @@ use near_sdk::{near_bindgen, env};
 #[near_bindgen]
 #[derive(Default, BorshDeserialize, BorshSerialize)]
 pub struct StatusMessage {
-    records: HashMap<String, String>,
+    records: HashMap<AccountId, String>,
 }
 
 #[near_bindgen]
@@ -54,7 +54,7 @@ impl StatusMessage {
         self.records.insert(account_id, message);
     }
 
-    pub fn get_status(&self, account_id: String) -> Option<String> {
+    pub fn get_status(&self, account_id: AccountId) -> Option<String> {
         self.records.get(&account_id).cloned()
     }
 }

examples/cross-contract-high-level/src/lib.rs

@@ -7,6 +7,7 @@ use near_sdk::{
     //    callback_vec,
     log,
     near_bindgen,
+    AccountId,
     Promise,
     PromiseOrValue,
 };
@@ -37,12 +38,12 @@ pub trait ExtCrossContract {
 #[ext_contract]
 pub trait ExtStatusMessage {
     fn set_status(&mut self, message: String);
-    fn get_status(&self, account_id: String) -> Option<String>;
+    fn get_status(&self, account_id: AccountId) -> Option<String>;
 }
 
 #[near_bindgen]
 impl CrossContract {
-    pub fn deploy_status_message(&self, account_id: String, amount: U128) {
+    pub fn deploy_status_message(&self, account_id: AccountId, amount: U128) {
         Promise::new(account_id)
             .create_account()
             .transfer(amount.0)
@@ -119,10 +120,10 @@ impl CrossContract {
     //        self.internal_merge(arrs.pop().unwrap(), arrs.pop().unwrap())
     //    }
 
-    pub fn simple_call(&mut self, account_id: String, message: String) {
+    pub fn simple_call(&mut self, account_id: AccountId, message: String) {
         ext_status_message::set_status(message, &account_id, 0, env::prepaid_gas() / 2);
     }
-    pub fn complex_call(&mut self, account_id: String, message: String) -> Promise {
+    pub fn complex_call(&mut self, account_id: AccountId, message: String) -> Promise {
         // 1) call status_message to record a message from the signer.
         // 2) call status_message to retrieve the message of the signer.
         // 3) return that message as its own result.
@@ -139,7 +140,7 @@ impl CrossContract {
         )
     }
 
-    pub fn transfer_money(&mut self, account_id: String, amount: u64) {
+    pub fn transfer_money(&mut self, account_id: AccountId, amount: u64) {
         Promise::new(account_id).transfer(amount as u128);
     }
 }

examples/cross-contract-high-level/tests/general.rs

@@ -29,7 +29,7 @@ fn init() -> (UserAccount, ContractAccount<CrossContractContract>) {
 fn test_sim_transfer() {
     let (master_account, contract) = init();
 
-    let status_id = "status".to_string();
+    let status_id: near_sdk::AccountId = "status".parse().unwrap();
     let status_amt = to_yocto("35");
     call!(
         master_account,

examples/cross-contract-low-level/src/lib.rs

@@ -1,7 +1,7 @@
 use near_sdk::borsh::{self, BorshDeserialize, BorshSerialize};
 use near_sdk::json_types::U128;
 use near_sdk::serde_json::{self, json};
-use near_sdk::{env, near_bindgen, PromiseResult};
+use near_sdk::{env, near_bindgen, AccountId, PromiseResult};
 
 near_sdk::setup_alloc!();
 
@@ -22,7 +22,7 @@ impl Default for CrossContract {
 
 #[near_bindgen]
 impl CrossContract {
-    pub fn deploy_status_message(&self, account_id: String, amount: U128) {
+    pub fn deploy_status_message(&self, account_id: AccountId, amount: U128) {
         let promise_idx = env::promise_batch_create(&account_id);
         env::promise_batch_action_create_account(promise_idx);
         env::promise_batch_action_transfer(promise_idx, amount.0);
@@ -103,16 +103,16 @@ impl CrossContract {
         result
     }
 
-    pub fn simple_call(&mut self, account_id: String, message: String) {
+    pub fn simple_call(&mut self, account_id: AccountId, message: String) {
         env::promise_create(
-            account_id.clone(),
+            account_id,
             b"set_status",
             json!({ "message": message }).to_string().as_bytes(),
             0,
             SINGLE_CALL_GAS,
         );
     }
-    pub fn complex_call(&mut self, account_id: String, message: String) {
+    pub fn complex_call(&mut self, account_id: AccountId, message: String) {
         // 1) call status_message to record a message from the signer.
         // 2) check that the promise succeed
         // 3) call status_message to retrieve the message of the signer.

examples/cross-contract-low-level/tests/general.rs

@@ -25,7 +25,7 @@ fn init(
         bytes: &TOKEN_WASM_BYTES,
         signer_account: master_account
     };
-    let alice = master_account.create_user("alice".to_string(), initial_balance);
+    let alice = master_account.create_user("alice".parse().unwrap(), initial_balance);
     (master_account, contract_account, alice)
 }
 
@@ -39,7 +39,7 @@ fn check_promise() {
     let (master_account, contract, _alice) = init(to_yocto("10000"));
     let res = view!(contract.promise_checked());
     assert_eq!(res.unwrap_json::<bool>(), false);
-    let status_id = "status".to_string();
+    let status_id: near_sdk::AccountId = "status".parse().unwrap();
     let status_amt = to_yocto("35");
     let res = call!(
         master_account,
@@ -64,7 +64,7 @@ fn test_sim_transfer() {
     // let transfer_amount = to_yocto("100");
     let initial_balance = to_yocto("100000");
     let (master_account, contract, _alice) = init(initial_balance);
-    let status_id = "status".to_string();
+    let status_id: near_sdk::AccountId = "status".parse().unwrap();
     let status_amt = to_yocto("35");
     let res = call!(
         master_account,

examples/fungible-token/examples/heavy.rs

@@ -19,12 +19,7 @@ const DEFI_ID: &str = "defi";
 
 fn init(
     initial_balance: u128,
-) -> (
-    UserAccount,
-    ContractAccount<FtContract>,
-    ContractAccount<DeFiContract>,
-    UserAccount,
-) {
+) -> (UserAccount, ContractAccount<FtContract>, ContractAccount<DeFiContract>, UserAccount) {
     let root = init_simulator(None);
     // uses default values for deposit and gas
     let ft = deploy!(
@@ -37,13 +32,13 @@ fn init(
         // User deploying the contract,
         signer_account: root,
         // init method
-        init_method: 
+        init_method:
           new_default_meta(
             root.account_id().try_into().unwrap(),
             initial_balance.into()
         )
     );
-    let alice = root.create_user("alice".to_string(), to_yocto("100"));
+    let alice = root.create_user(AccountId::new_unchecked("alice".to_string()), to_yocto("100"));
     register_user(&ft, &alice);
 
     let defi = deploy!(
@@ -88,9 +83,7 @@ fn transfer(
 
 fn main() {
     let iterations: u64 = if std::env::args().len() >= 2 {
-        (&std::env::args().collect::<Vec<String>>()[1])
-            .parse()
-            .unwrap()
+        (&std::env::args().collect::<Vec<String>>()[1]).parse().unwrap()
     } else {
         10
     };

examples/fungible-token/ft/src/lib.rs

@@ -21,7 +21,7 @@ use near_contract_standards::fungible_token::metadata::{
 use near_contract_standards::fungible_token::FungibleToken;
 use near_sdk::borsh::{self, BorshDeserialize, BorshSerialize};
 use near_sdk::collections::LazyOption;
-use near_sdk::json_types::{ValidAccountId, U128};
+use near_sdk::json_types::U128;
 use near_sdk::{
     env, log, near_bindgen, AccountId, Balance, BorshStorageKey, PanicOnDefault, PromiseOrValue,
 };
@@ -48,7 +48,7 @@ impl Contract {
     /// Initializes the contract with the given total supply owned by the given `owner_id` with
     /// default metadata (for example purposes only).
     #[init]
-    pub fn new_default_meta(owner_id: ValidAccountId, total_supply: U128) -> Self {
+    pub fn new_default_meta(owner_id: AccountId, total_supply: U128) -> Self {
         Self::new(
             owner_id,
             total_supply,
@@ -67,19 +67,15 @@ impl Contract {
     /// Initializes the contract with the given total supply owned by the given `owner_id` with
     /// the given fungible token metadata.
     #[init]
-    pub fn new(
-        owner_id: ValidAccountId,
-        total_supply: U128,
-        metadata: FungibleTokenMetadata,
-    ) -> Self {
+    pub fn new(owner_id: AccountId, total_supply: U128, metadata: FungibleTokenMetadata) -> Self {
         assert!(!env::state_exists(), "Already initialized");
         metadata.assert_valid();
         let mut this = Self {
             token: FungibleToken::new(StorageKey::FungibleToken),
             metadata: LazyOption::new(StorageKey::Metadata, Some(&metadata)),
         };
-        this.token.internal_register_account(owner_id.as_ref());
-        this.token.internal_deposit(owner_id.as_ref(), total_supply.into());
+        this.token.internal_register_account(&owner_id);
+        this.token.internal_deposit(&owner_id, total_supply.into());
         this
     }
 
@@ -105,14 +101,13 @@ impl FungibleTokenMetadataProvider for Contract {
 #[cfg(all(test, not(target_arch = "wasm32")))]
 mod tests {
     use near_sdk::test_utils::{accounts, VMContextBuilder};
-    use near_sdk::MockedBlockchain;
     use near_sdk::{testing_env, Balance};
 
     use super::*;
 
     const TOTAL_SUPPLY: Balance = 1_000_000_000_000_000;
 
-    fn get_context(predecessor_account_id: ValidAccountId) -> VMContextBuilder {
+    fn get_context(predecessor_account_id: AccountId) -> VMContextBuilder {
         let mut builder = VMContextBuilder::new();
         builder
             .current_account_id(accounts(0))

examples/fungible-token/test-contract-defi/src/lib.rs

@@ -3,7 +3,7 @@ Some hypothetical DeFi contract that will do smart things with the transferred t
 */
 use near_contract_standards::fungible_token::receiver::FungibleTokenReceiver;
 use near_sdk::borsh::{self, BorshDeserialize, BorshSerialize};
-use near_sdk::json_types::{ValidAccountId, U128};
+use near_sdk::json_types::U128;
 use near_sdk::{
     env, ext_contract, log, near_bindgen, setup_alloc, AccountId, Balance, Gas, PanicOnDefault,
     PromiseOrValue,
@@ -37,7 +37,7 @@ trait ValueReturnTrait {
 #[near_bindgen]
 impl DeFi {
     #[init]
-    pub fn new(fungible_token_account_id: ValidAccountId) -> Self {
+    pub fn new(fungible_token_account_id: AccountId) -> Self {
         assert!(!env::state_exists(), "Already initialized");
         Self { fungible_token_account_id: fungible_token_account_id.into() }
     }
@@ -50,7 +50,7 @@ impl FungibleTokenReceiver for DeFi {
     /// value_please will attempt to parse `msg` as an integer and return a U128 version of it
     fn ft_on_transfer(
         &mut self,
-        sender_id: ValidAccountId,
+        sender_id: AccountId,
         amount: U128,
         msg: String,
     ) -> PromiseOrValue<U128> {

examples/fungible-token/tests/sim/no_macros.rs

@@ -25,7 +25,7 @@ fn simulate_simple_transfer() {
         ft.account_id(),
         "ft_transfer",
         &json!({
-            "receiver_id": alice.valid_account_id(),
+            "receiver_id": alice.account_id(),
             "amount": U128::from(transfer_amount)
         })
         .to_string()
@@ -40,7 +40,7 @@ fn simulate_simple_transfer() {
             ft.account_id(),
             "ft_balance_of",
             &json!({
-                "account_id": root.valid_account_id()
+                "account_id": root.account_id()
             })
             .to_string()
             .into_bytes(),
@@ -51,7 +51,7 @@ fn simulate_simple_transfer() {
             ft.account_id(),
             "ft_balance_of",
             &json!({
-                "account_id": alice.valid_account_id()
+                "account_id": alice.account_id()
             })
             .to_string()
             .into_bytes(),