feat: update custodian in upgrade script

agustinustheo · Aug 15, 2024 · ab19ba8 · ab19ba8
1 parent 4f13ff0
commit ab19ba8
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/icp_subaccount_indexer/src/lib.rs b/src/icp_subaccount_indexer/src/lib.rs
@@ -699,6 +699,15 @@ fn reconstruct_network() {
 #[ic_cdk::post_upgrade]
 async fn post_upgrade() {
     ic_cdk::println!("running post_upgrade...");
+
+    let custodian_principal =
+        Principal::from_text("custodian_principal").expect("Invalid custodian principal");
+
+    CUSTODIAN_PRINCIPAL.with(|principal_ref| {
+        let stored_principal = StoredPrincipal::new(custodian_principal);
+        let _ = principal_ref.borrow_mut().set(stored_principal);
+    });
+
     reconstruct_subaccounts();
     reconstruct_network();
 }