fix StakeToSBTPlugin install event (#215)

starcoinorg · Nov 17, 2022 · 77a968a · 77a968a
1 parent a98de9e
commit 77a968a
Show file tree

Hide file tree

Showing 6 changed files with 45 additions and 28 deletions.
diff --git a/integration-tests/daospaceplugin/stake_to_sbt_plugin.exp b/integration-tests/daospaceplugin/stake_to_sbt_plugin.exp
@@ -2,13 +2,13 @@ processed 17 tasks
 
 task 5 'run'. lines 52-60:
 {
-  "gas_used": 1887227,
+  "gas_used": 1958252,
   "status": "Executed"
 }
 
 task 6 'run'. lines 62-98:
 {
-  "gas_used": 1022609,
+  "gas_used": 1159804,
   "status": "Executed"
 }
 
@@ -26,25 +26,25 @@ task 9 'run'. lines 121-139:
 
 task 10 'run'. lines 141-157:
 {
-  "gas_used": 458467,
+  "gas_used": 475573,
   "status": "Executed"
 }
 
 task 12 'run'. lines 161-174:
 {
-  "gas_used": 417670,
+  "gas_used": 554865,
   "status": "Executed"
 }
 
 task 13 'run'. lines 176-192:
 {
-  "gas_used": 592341,
+  "gas_used": 609447,
   "status": "Executed"
 }
 
 task 15 'run'. lines 196-211:
 {
-  "gas_used": 261857,
+  "gas_used": 278963,
   "status": "Executed"
 }
 

diff --git a/integration-tests/starcoin_dao/starcoin_onchain_config.exp b/integration-tests/starcoin_dao/starcoin_onchain_config.exp
@@ -2,25 +2,25 @@ processed 42 tasks
 
 task 11 'run'. lines 23-37:
 {
-  "gas_used": 959229,
+  "gas_used": 1096577,
   "status": "Executed"
 }
 
 task 12 'run'. lines 39-53:
 {
-  "gas_used": 959229,
+  "gas_used": 976335,
   "status": "Executed"
 }
 
 task 13 'run'. lines 55-69:
 {
-  "gas_used": 959229,
+  "gas_used": 976335,
   "status": "Executed"
 }
 
 task 14 'run'. lines 71-85:
 {
-  "gas_used": 959229,
+  "gas_used": 976335,
   "status": "Executed"
 }
 

diff --git a/integration-tests/starcoin_dao/starcoin_treasury_withdraw.exp b/integration-tests/starcoin_dao/starcoin_treasury_withdraw.exp
@@ -2,25 +2,25 @@ processed 44 tasks
 
 task 11 'run'. lines 23-37:
 {
-  "gas_used": 959229,
+  "gas_used": 1096577,
   "status": "Executed"
 }
 
 task 12 'run'. lines 39-53:
 {
-  "gas_used": 959229,
+  "gas_used": 976335,
   "status": "Executed"
 }
 
 task 13 'run'. lines 55-69:
 {
-  "gas_used": 959229,
+  "gas_used": 976335,
   "status": "Executed"
 }
 
 task 14 'run'. lines 71-85:
 {
-  "gas_used": 959229,
+  "gas_used": 976335,
   "status": "Executed"
 }
 

diff --git a/integration-tests/starcoin_dao/starcoin_upgrade_module.exp b/integration-tests/starcoin_dao/starcoin_upgrade_module.exp
@@ -2,25 +2,25 @@ processed 45 tasks
 
 task 11 'run'. lines 23-37:
 {
-  "gas_used": 959229,
+  "gas_used": 1096577,
   "status": "Executed"
 }
 
 task 12 'run'. lines 39-53:
 {
-  "gas_used": 959229,
+  "gas_used": 976335,
   "status": "Executed"
 }
 
 task 13 'run'. lines 55-69:
 {
-  "gas_used": 959229,
+  "gas_used": 976335,
   "status": "Executed"
 }
 
 task 14 'run'. lines 71-85:
 {
-  "gas_used": 959229,
+  "gas_used": 976335,
   "status": "Executed"
 }
 

diff --git a/sources/daospace/DAOSpace.move b/sources/daospace/DAOSpace.move
@@ -967,6 +967,10 @@ module StarcoinFramework::DAOSpace {
         EventUtil::init_event<EventT>(&dao_signer);
     }
 
+    public fun exist_plugin_event<DAOT: store, PluginT: store, EventT: store+drop>(): bool {
+        EventUtil::exist_event<EventT>(dao_address<DAOT>())
+    }
+
     public fun emit_plugin_event<
         DAOT: store,
         PluginT: store,

diff --git a/sources/daospaceplugin/StakeToSBTPlugin.move b/sources/daospaceplugin/StakeToSBTPlugin.move
@@ -117,19 +117,16 @@ module StarcoinFramework::StakeToSBTPlugin {
 
     /// Accept token with token type by given DAO
     public fun accept_token_by_dao<DAOT: store, TokenT: store>(_witness: &DAOT) {
-        install_event<DAOT>();
         accept_token(AcceptTokenCap<DAOT, TokenT> {});
     }
 
-    public fun install_event<DAOT: store>() {
-        let witness = StakeToSBTPlugin {};
-        let plugin_event_cap =
-            DAOSpace::acquire_plugin_event_cap<DAOT, StakeToSBTPlugin>(&witness);
-
-        DAOSpace::init_plugin_event<DAOT, StakeToSBTPlugin, SBTTokenAcceptedEvent>(&plugin_event_cap);
-        DAOSpace::init_plugin_event<DAOT, StakeToSBTPlugin, SBTWeightChangedEvent>(&plugin_event_cap);
-        DAOSpace::init_plugin_event<DAOT, StakeToSBTPlugin, SBTStakeEvent>(&plugin_event_cap);
-        DAOSpace::init_plugin_event<DAOT, StakeToSBTPlugin, SBTUnstakeEvent>(&plugin_event_cap);
+    fun install_event<DAOT: store, EventT: store+drop>() {
+        if (!DAOSpace::exist_plugin_event<DAOT, StakeToSBTPlugin, EventT>()) {
+            let witness = StakeToSBTPlugin {};
+            let plugin_event_cap =
+                DAOSpace::acquire_plugin_event_cap<DAOT, StakeToSBTPlugin>(&witness);
+            DAOSpace::init_plugin_event<DAOT, StakeToSBTPlugin, EventT>(&plugin_event_cap);
+        }
     }
 
     /// Set sbt weight by given DAO
@@ -161,6 +158,7 @@ module StarcoinFramework::StakeToSBTPlugin {
             weight_vec: Vector::empty<LockWeight<DAOT, TokenT>>()
         });
 
+        install_event<DAOT, SBTTokenAcceptedEvent>();
         let witness = StakeToSBTPlugin {};
         let plugin_event_cap =
             DAOSpace::acquire_plugin_event_cap<DAOT, StakeToSBTPlugin>(&witness);
@@ -224,6 +222,7 @@ module StarcoinFramework::StakeToSBTPlugin {
             });
         stake_list.next_id = id;
 
+        install_event<DAOT, SBTStakeEvent>();
         let witness = StakeToSBTPlugin {};
         let plugin_event_cap =
             DAOSpace::acquire_plugin_event_cap<DAOT, StakeToSBTPlugin>(&witness);
@@ -301,6 +300,7 @@ module StarcoinFramework::StakeToSBTPlugin {
 
         Account::deposit<TokenT>(member, unstake_item(member, poped_item));
 
+        install_event<DAOT, SBTUnstakeEvent>();
         let witness = StakeToSBTPlugin {};
         let plugin_event_cap =
             DAOSpace::acquire_plugin_event_cap<DAOT, StakeToSBTPlugin>(&witness);
@@ -431,7 +431,20 @@ module StarcoinFramework::StakeToSBTPlugin {
             });
         };
 
+        install_event<DAOT, SBTWeightChangedEvent>();
         let witness = StakeToSBTPlugin {};
+        let plugin_event_cap =
+            DAOSpace::acquire_plugin_event_cap<DAOT, StakeToSBTPlugin>(&witness);
+        DAOSpace::emit_plugin_event<DAOT, StakeToSBTPlugin, SBTWeightChangedEvent>(
+            &plugin_event_cap,
+            SBTWeightChangedEvent {
+                dao_id: DAOSpace::dao_id(DAOSpace::dao_address<DAOT>()),
+                token_code: Token::token_code<TokenT>(),
+                lock_time,
+                weight,
+            }
+        );
+
         let modify_config_cap =
             DAOSpace::acquire_modify_config_cap<DAOT, StakeToSBTPlugin>(&witness);