Lowercase WasmMemoryPersistence

Author: adamspofford-dfinity

ic-utils/src/interfaces/management_canister/builders.rs

@@ -524,9 +524,11 @@ impl<'agent, 'canister: 'agent> IntoFuture for CreateCanisterBuilder<'agent, 'ca
 pub enum WasmMemoryPersistence {
     /// Retain the main memory across upgrades.
     /// Used for enhanced orthogonal persistence, as implemented in Motoko
+    #[serde(rename = "keep")]
     Keep,
     /// Reinitialize the main memory on upgrade.
     /// Default behavior without enhanced orthogonal persistence.
+    #[serde(rename = "replace")]
     Replace,
 }
 
@@ -539,6 +541,16 @@ pub struct CanisterUpgradeOptions {
     pub wasm_memory_persistence: Option<WasmMemoryPersistence>,
 }
 
+impl Default for CanisterUpgradeOptions {
+    /// Default: `skip_pre_upgrade = true`, `wasm_memory_persistence = Replace`
+    fn default() -> Self {
+        Self {
+            skip_pre_upgrade: Some(false),
+            wasm_memory_persistence: Some(WasmMemoryPersistence::Replace),
+        }
+    }
+}
+
 /// The install mode of the canister to install. If a canister is already installed,
 /// using [`InstallMode::Install`] will be an error. [`InstallMode::Reinstall`] overwrites
 /// the module, and [`InstallMode::Upgrade`] performs an Upgrade step.
@@ -555,6 +567,22 @@ pub enum InstallMode {
     Upgrade(Option<CanisterUpgradeOptions>),
 }
 
+impl InstallMode {
+    /// Fills in all `Option` fields with their default values.
+    pub fn canonicalize(&self) -> InstallMode {
+        match self {
+            Self::Upgrade(u) => {
+                let mut opts = u.unwrap_or_default();
+                opts.skip_pre_upgrade.get_or_insert(false);
+                opts.wasm_memory_persistence
+                    .get_or_insert(WasmMemoryPersistence::Replace);
+                Self::Upgrade(Some(opts))
+            }
+            x => *x,
+        }
+    }
+}
+
 /// A prepared call to `install_code`.
 #[derive(Debug, Clone, CandidType, Deserialize)]
 pub struct CanisterInstall {