Wallet Smart Contract

Icon Link钱包智能合约

ABI 声明是与 ABI 实现分开的一个项目。代码的项目结构应该组织如下,其中 wallet_abi 被视为外部库:

.
├── wallet_abi
│ ├── Forc.toml
│ └── src
│ └── main.sw
└── wallet_smart_contract
├── Forc.toml
└── src
└── main.sw

在使用外部库时,指定依赖项的来源也很重要,应该在项目的 Forc.toml 文件中进行声明。在 wallet_smart_contract 项目内,需要声明如下:

[dependencies]
wallet_abi = { path = "../wallet_abi/" }

Icon LinkABI 声明

library;
 
abi Wallet {
    #[storage(read, write), payable]
    fn receive_funds();
 
    #[storage(read, write)]
    fn send_funds(amount_to_send: u64, recipient_address: Address);
}

Icon LinkABI 实现

contract;
 
use std::{asset::transfer, call_frames::msg_asset_id, context::msg_amount,};
 
use wallet_abi::Wallet;
const OWNER_ADDRESS = Address::from(0x8900c5bec4ca97d4febf9ceb4754a60d782abbf3cd815836c1872116f203f861);
 
storage {
    balance: u64 = 0,
}
 
impl Wallet for Contract {
    #[storage(read, write), payable]
    fn receive_funds() {
        if msg_asset_id() == AssetId::base() {
            // If we received the base asset then keep track of the balance.
            // Otherwise, we're receiving other native assets and don't care
            // about our balance of coins.
            storage.balance.write(storage.balance.read() + msg_amount());
        }
    }
 
    #[storage(read, write)]
    fn send_funds(amount_to_send: u64, recipient_address: Address) {
        let sender = msg_sender().unwrap();
        match sender {
            Identity::Address(addr) => assert(addr == OWNER_ADDRESS),
            _ => revert(0),
        };
 
        let current_balance = storage.balance.read();
        assert(current_balance >= amount_to_send);
 
        storage.balance.write(current_balance - amount_to_send);
 
        // Note: `transfer()` is not a call and thus not an
        // interaction. Regardless, this code conforms to
        // checks-effects-interactions to avoid re-entrancy.
        transfer(
            Identity::Address(recipient_address),
            AssetId::base(),
            amount_to_send,
        );
    }
}