stellar-upgrader 1.1.4

Upgrader contract, responsible for upgrading contracts and migrate data in an atomic transaction.
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

use stellar_axelar_std::interfaces::UpgradableClient;
use stellar_axelar_std::{
    contract, contractimpl, ensure, soroban_sdk, symbol_short, Address, BytesN, Env, String,
    Symbol, Val,
};

use crate::error::ContractError;
use crate::interface::UpgraderInterface;

const MIGRATE: Symbol = symbol_short!("migrate");

#[contract]
pub struct Upgrader;

#[contractimpl]
impl Upgrader {
    pub fn __constructor(_env: Env) {}
}

#[contractimpl]
impl UpgraderInterface for Upgrader {
    fn upgrade(
        env: Env,
        contract_address: Address,
        new_version: String,
        new_wasm_hash: BytesN<32>,
        migration_data: stellar_axelar_std::Vec<Val>,
    ) -> Result<(), ContractError> {
        let contract_client = UpgradableClient::new(&env, &contract_address);

        contract_client.required_auths().iter().for_each(|addr| {
            addr.require_auth();
        });

        ensure!(
            contract_client.version() != new_version,
            ContractError::SameVersion
        );

        contract_client.upgrade(&new_wasm_hash);
        // The types of the arguments to the migrate function are unknown to this contract, so we need to call it with invoke_contract.
        // The migrate function's return value can be safely cast to Val no matter what it really is
        // because it will panic on failure anyway
        env.invoke_contract::<Val>(&contract_address, &MIGRATE, migration_data);

        ensure!(
            contract_client.version() == new_version,
            ContractError::UnexpectedNewVersion
        );
        Ok(())
    }
}