Rust 智能合约养成日记（10-2）Sputnik DAO::Factory合约解读

pub struct SputnikDAOFactory {    factory_manager: FactoryManager,    daos: UnorderedSet<AccountId>,}

1     #[payable]2     pub fn create(&mut self, name: AccountId, args: Base64VecU8) {3         let account_id: AccountId = format!("{}.{}", name, env::current_account_id())4             .parse()5             .unwrap();6         let callback_args = serde_json::to_vec(&json!({7             "account_id": account_id,8             "attached_deposit": U128(env::attached_deposit()),9             "predecessor_account_id": env::predecessor_account_id()10        }))11        .expect("Failed to serialize");12        self.factory_manager.create_contract(13            self.get_default_code_hash(),14            account_id,15            "new",16            &args.0,17            "on_create",18            &callback_args,19        );20    }

{    "config": {        "name": "multicall",        "purpose": "governance for near-multicall",        "metadata": ""    },    "policy": [        "multicall.near"    ]}

1     pub fn create_contract(2         &self,3         code_hash: Base58CryptoHash,4         account_id: AccountId,5         new_method: &str,6         args: &[u8],7         callback_method: &str,8         callback_args: &[u8],9     ) {10        let code_hash: CryptoHash = code_hash.into();11        let attached_deposit = env::attached_deposit();12        let factory_account_id = env::current_account_id().as_bytes().to_vec();13        let account_id = account_id.as_bytes().to_vec();14        unsafe {15            // Check that such contract exists.16            assert_eq!(17                sys::storage_has_key(code_hash.len() as _, code_hash.as_ptr() as _),18                1,19                "Contract doesn't exist"20            );21            // Load input (wasm code) into register 0.22            sys::storage_read(code_hash.len() as _, code_hash.as_ptr() as _, 0);23            // schedule a Promise tx to account_id24            let promise_id =25                sys::promise_batch_create(account_id.len() as _, account_id.as_ptr() as _);26            // create account first.27            sys::promise_batch_action_create_account(promise_id);28            // transfer attached deposit.29            sys::promise_batch_action_transfer(promise_id, &attached_deposit as *const u128 as _);30            // deploy contract (code is taken from register 0).31            sys::promise_batch_action_deploy_contract(promise_id, u64::MAX as _, 0);32            // call `new` with given arguments.33            sys::promise_batch_action_function_call(34                promise_id,35                new_method.len() as _,36                new_method.as_ptr() as _,37                args.len() as _,38                args.as_ptr() as _,39                &NO_DEPOSIT as *const u128 as _,40                CREATE_CALL_GAS.0,41            );42            // attach callback to the factory.43            let _ = sys::promise_then(44                promise_id,45                factory_account_id.len() as _,46                factory_account_id.as_ptr() as _,47                callback_method.len() as _,48                callback_method.as_ptr() as _,49                callback_args.len() as _,50                callback_args.as_ptr() as _,51                &NO_DEPOSIT as *const u128 as _,52                ON_CREATE_CALL_GAS.0,53            );54            sys::promise_return(promise_id);55        }56    }

    pub struct Config {        /// Name of the DAO.        pub name: String,        /// Purpose of this DAO.        pub purpose: String,        /// Generic metadata. Can be used by specific UI to store additional data.        /// This is not used by anything in the contract.        pub metadata: Base64VecU8,    }

    pub struct Policy {        /// List of roles and permissions for them in the current policy.        pub roles: Vec<RolePermission>,        /// Default vote policy. Used when given proposal kind doesn't have special policy.        pub default_vote_policy: VotePolicy,        /// Proposal bond.        pub proposal_bond: U128,        /// Expiration period for proposals.        pub proposal_period: U64,        /// Bond for claiming a bounty.        pub bounty_bond: U128,        /// Period in which giving up on bounty is not punished.        pub bounty_forgiveness_period: U64,    }

    #[private]    pub fn on_create(        &mut self,        account_id: AccountId,        attached_deposit: U128,        predecessor_account_id: AccountId,    ) -> bool {        if near_sdk::is_promise_success() {            self.daos.insert(&account_id);            true        } else {            Promise::new(predecessor_account_id).transfer(attached_deposit.0);            false        }    }

    /// Tries to update given account created by this factory to the specified code.    pub fn update(&self, account_id: AccountId, code_hash: Base58CryptoHash) {        let caller_id = env::predecessor_account_id();        assert!(            caller_id == self.get_owner() || caller_id == account_id,            "Must be updated by the factory owner or the DAO itself"        );        assert!(            self.daos.contains(&account_id),            "Must be contract created by factory"        );        self.factory_manager            .update_contract(account_id, code_hash, "update");    }

    /// Forces update on the given contract.    /// Contract must support update by factory for this via permission check.    pub fn update_contract(        &self,        account_id: AccountId,        code_hash: Base58CryptoHash,        method_name: &str,    ) {        let code_hash: CryptoHash = code_hash.into();        let account_id = account_id.as_bytes().to_vec();        unsafe {            // Check that such contract exists.            assert!(env::storage_has_key(&code_hash), "Contract doesn't exist");            // Load the hash from storage.            sys::storage_read(code_hash.len() as _, code_hash.as_ptr() as _, 0);            // Create a promise toward given account.            let promise_id =                sys::promise_batch_create(account_id.len() as _, account_id.as_ptr() as _);            // Call `update` method, which should also handle migrations.            sys::promise_batch_action_function_call(                promise_id,                method_name.len() as _,                method_name.as_ptr() as _,                u64::MAX as _,                0,                &NO_DEPOSIT as *const u128 as _,                (env::prepaid_gas() - env::used_gas() - GAS_UPDATE_LEFTOVER).0,);            sys::promise_return(promise_id);        }    }

        assert!(            caller_id == self.get_owner() || caller_id == account_id,            "Must be updated by the factory owner or the DAO itself"        );

1.  #[no_mangle]2.  pub fn update() {3.     env::setup_panic_hook();4.     let factory_info = internal_get_factory_info();5.     let current_id = env::current_account_id();6.     assert!(7.         env::predecessor_account_id() == current_id8.             || (env::predecessor_account_id() == factory_info.factory_id9.                 && factory_info.auto_update ),   10.         "{}",11.        ERR_MUST_BE_SELF_OR_FACTORY12.   );13.    ......

    #[init]    pub fn new(config: Config, policy: VersionedPolicy) -> Self {        let this = Self {            config: LazyOption::new(StorageKeys::Config, Some(&config)),            policy: LazyOption::new(StorageKeys::Policy, Some(&policy.upgrade())),            staking_id: None,            total_delegation_amount: 0,            delegations: LookupMap::new(StorageKeys::Delegations),            last_proposal_id: 0,            proposals: LookupMap::new(StorageKeys::Proposals),            last_bounty_id: 0,            bounties: LookupMap::new(StorageKeys::Bounties),            bounty_claimers: LookupMap::new(StorageKeys::BountyClaimers),            bounty_claims_count: LookupMap::new(StorageKeys::BountyClaimCounts),            blobs: LookupMap::new(StorageKeys::Blobs),            locked_amount: 0,        };        internal_set_factory_info(&FactoryInfo {            factory_id: env::predecessor_account_id(),            auto_update: true,  // 这里 factory_info.auto_update该值被默认设置为 true.        });        this    }

    fn assert_owner(&self) {        assert_eq!(            self.get_owner(),            env::predecessor_account_id(),            "Must be owner"        );    }