loam-subcontract-core 0.7.8

Core Subcontract for redeployable contracts
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

use loam_sdk::{
    soroban_sdk::{self, contracttype, env, symbol_short, Address, BytesN, Lazy, Symbol},
    subcontract,
};

#[contracttype(export = false)]
#[derive(Default)]
pub struct Admin(Kind);

fn admin_key() -> Symbol {
    symbol_short!("ADMIN")
}

impl Lazy for Admin {
    fn get_lazy() -> Option<Self> {
        env().storage().instance().get(&admin_key())
    }

    fn set_lazy(self) {
        env().storage().instance().set(&admin_key(), &self);
    }
}

/// Work around not having `Option` in `contracttype`
#[contracttype(export = false)]
#[derive(Default)]
pub enum Kind {
    Address(Address),
    #[default]
    None,
}

impl IsCore for Admin {
    fn admin_get(&self) -> Option<Address> {
        match &self.0 {
            Kind::Address(address) => Some(address.clone()),
            Kind::None => None,
        }
    }

    fn admin_set(&mut self, new_admin: Address) {
        if let Admin(Kind::Address(admin)) = &self {
            admin.require_auth();
        }
        self.0 = Kind::Address(new_admin);
    }

    fn redeploy(&self, wasm_hash: BytesN<32>) {
        self.admin_get().unwrap().require_auth();
        env().deployer().update_current_contract_wasm(wasm_hash);
    }
}

#[subcontract]
pub trait IsCore {
    /// Get current admin
    fn admin_get(&self) -> Option<loam_sdk::soroban_sdk::Address>;
    /// Transfer to new admin
    /// Should be called in the same transaction as deploying the contract to ensure that
    /// a different account try to become admin
    fn admin_set(&mut self, new_admin: loam_sdk::soroban_sdk::Address);

    /// Admin can redeploy the contract with given hash.
    fn redeploy(&self, wasm_hash: loam_sdk::soroban_sdk::BytesN<32>);
}