use crate::identity::CanisterId;
pub type CanisterCodeWasm = Vec<u8>;
pub type CanisterInitArg = Vec<u8>;
pub async fn install_code(
canister_id: CanisterId,
wasm_module: CanisterCodeWasm,
arg: CanisterInitArg,
) -> super::types::CanisterCallResult<()> {
let call_result = ic_cdk::api::management_canister::main::install_code(
ic_cdk::api::management_canister::main::InstallCodeArgument {
mode: ic_cdk::api::management_canister::main::CanisterInstallMode::Install,
canister_id,
wasm_module,
arg,
},
)
.await;
super::wrap_call_result(canister_id, "ic#install_code#install", call_result)
}
pub async fn upgrade_code(
canister_id: CanisterId,
wasm_module: CanisterCodeWasm,
arg: Option<CanisterInitArg>,
) -> super::types::CanisterCallResult<()> {
let call_result = ic_cdk::api::management_canister::main::install_code(
ic_cdk::api::management_canister::main::InstallCodeArgument {
mode: ic_cdk::api::management_canister::main::CanisterInstallMode::Upgrade,
canister_id,
wasm_module,
arg: arg.unwrap_or_default(),
},
)
.await;
super::wrap_call_result(canister_id, "ic#install_code#upgrade", call_result)
}
pub async fn reinstall_code(
canister_id: CanisterId,
wasm_module: CanisterCodeWasm,
arg: CanisterInitArg,
) -> super::types::CanisterCallResult<()> {
let call_result = ic_cdk::api::management_canister::main::install_code(
ic_cdk::api::management_canister::main::InstallCodeArgument {
mode: ic_cdk::api::management_canister::main::CanisterInstallMode::Reinstall,
canister_id,
wasm_module,
arg,
},
)
.await;
super::wrap_call_result(canister_id, "ic#install_code#reinstall", call_result)
}
pub async fn uninstall_code(canister_id: CanisterId) -> super::types::CanisterCallResult<()> {
let call_result = ic_cdk::api::management_canister::main::uninstall_code(
ic_cdk::api::management_canister::main::CanisterIdRecord { canister_id },
)
.await;
super::wrap_call_result(canister_id, "ic#uninstall_code", call_result)
}