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::management_canister::install_code(&ic_cdk::management_canister::InstallCodeArgs {
mode: ic_cdk::management_canister::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>,
pre_upgrade: Option<ic_cdk::management_canister::UpgradeFlags>,
) -> super::types::CanisterCallResult<()> {
let call_result =
ic_cdk::management_canister::install_code(&ic_cdk::management_canister::InstallCodeArgs {
mode: ic_cdk::management_canister::CanisterInstallMode::Upgrade(pre_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::management_canister::install_code(&ic_cdk::management_canister::InstallCodeArgs {
mode: ic_cdk::management_canister::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::management_canister::uninstall_code(
&ic_cdk::management_canister::UninstallCodeArgs { canister_id },
)
.await;
super::wrap_call_result(canister_id, "ic#uninstall_code", call_result)
}