From 25fa75bb9b27087dea0df3fe3eba789d3c4e7aa9 Mon Sep 17 00:00:00 2001
From: Miraculous Owonubi <omiraculous@gmail.com>
Date: Wed, 27 Nov 2024 16:55:43 +0100
Subject: [PATCH] nuke proxy when erasing context config

---
 contracts/context-config/src/sys.rs | 12 ++++++++++--
 contracts/proxy-lib/src/lib.rs      |  1 -
 contracts/proxy-lib/src/mutate.rs   | 21 +++++++++++++++++++++
 3 files changed, 31 insertions(+), 3 deletions(-)

diff --git a/contracts/context-config/src/sys.rs b/contracts/context-config/src/sys.rs
index 00fb9121f..e59b2e1da 100644
--- a/contracts/context-config/src/sys.rs
+++ b/contracts/context-config/src/sys.rs
@@ -5,7 +5,7 @@ use calimero_context_config::repr::Repr;
 use calimero_context_config::types::{Application, ContextId, ContextIdentity, SignerId};
 use calimero_context_config::{SystemRequest, Timestamp};
 use near_sdk::store::{IterableMap, IterableSet};
-use near_sdk::{env, near};
+use near_sdk::{env, near, Gas, NearToken, Promise};
 
 use crate::{parse_input, Config, ContextConfigs, ContextConfigsExt};
 
@@ -36,9 +36,17 @@ impl ContextConfigs {
         for (_, context) in self.contexts.drain() {
             let _ignored = context.application.into_inner();
             context.members.into_inner().clear();
-            let _ignored = context.proxy.into_inner();
+            let proxy = context.proxy.into_inner();
+
+            let _is_sent_on_drop = Promise::new(proxy).function_call(
+                "nuke".to_owned(),
+                vec![],
+                NearToken::default(),
+                Gas::from_tgas(1),
+            );
         }
 
+        self.next_proxy_id = 0;
         self.proxy_code.set(None);
 
         env::log_str(&format!(
diff --git a/contracts/proxy-lib/src/lib.rs b/contracts/proxy-lib/src/lib.rs
index 1207cfc28..7425d2783 100644
--- a/contracts/proxy-lib/src/lib.rs
+++ b/contracts/proxy-lib/src/lib.rs
@@ -102,7 +102,6 @@ impl ProxyContract {
         &self,
         proposal_id: Repr<ProposalId>,
     ) -> Option<Vec<Repr<SignerId>>> {
-        let approvals_for_proposal = self.approvals.get(&proposal_id);
         let approvals = self.approvals.get(&proposal_id)?;
         Some(approvals.iter().flat_map(|a| a.rt()).collect())
     }
diff --git a/contracts/proxy-lib/src/mutate.rs b/contracts/proxy-lib/src/mutate.rs
index ae415a974..8aec79d60 100644
--- a/contracts/proxy-lib/src/mutate.rs
+++ b/contracts/proxy-lib/src/mutate.rs
@@ -40,6 +40,7 @@ impl ProxyContract {
         }
     }
 }
+
 #[near]
 impl ProxyContract {
     #[private]
@@ -226,6 +227,26 @@ impl ProxyContract {
             )
     }
 
+    fn erase(&mut self) {
+        // if this is going to be exposed, it should be a proposal
+        self.proposals.clear();
+        self.approvals.clear();
+        self.num_proposals_pk.clear();
+        self.context_storage.clear();
+    }
+
+    pub fn nuke(&mut self) -> Promise {
+        require!(
+            env::predecessor_account_id() == self.context_config_account_id,
+            "Only the context config contract can nuke the proxy"
+        );
+
+        self.erase();
+
+        Promise::new(env::current_account_id())
+            .delete_account(self.context_config_account_id.clone())
+    }
+
     #[private]
     pub fn update_contract_callback(
         &mut self,