add modify_total_fee

ref-exchange/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "ref-exchange"
-version = "1.7.1"
+version = "1.7.2"
 authors = ["Illia Polosukhin <[email protected]>"]
 edition = "2018"
 publish = false

ref-exchange/release_notes.md

@@ -1,5 +1,11 @@
 # Release Notes
 
+### Version 1.7.2
+```
+9KuaBbp9FT1g17YCaCsaezGxWDuZkKrjPsXu1RCN8Ux7
+```
+1. add modify_total_fee;
+
 ### Version 1.7.1
 ```
 6ZU3rmDwEs988pvYWyDxg6DJm7fP1F3FnAwHGo698ATw

ref-exchange/src/owner.rs

@@ -6,7 +6,6 @@ use near_contract_standards::fungible_token::core_impl::ext_fungible_token;
 use crate::*;
 use crate::rated_swap::rate::{global_register_rate, global_unregister_rate};
 use crate::utils::{FEE_DIVISOR, MAX_ADMIN_FEE_BPS, GAS_FOR_BASIC_OP};
-use crate::legacy::ContractV1;
 
 #[near_bindgen]
 impl Contract {
@@ -199,6 +198,19 @@ impl Contract {
         self.admin_fee_bps = admin_fee_bps;
     }
 
+    #[payable]
+    pub fn modify_total_fee(&mut self, pool_id: u64, total_fee: u32) {
+        assert_one_yocto();
+        assert!(self.is_owner_or_guardians(), "{}", ERR100_NOT_ALLOWED);
+        assert!(total_fee < FEE_DIVISOR, "{}", ERR62_FEE_ILLEGAL);
+        let mut pool = self.pools.get(pool_id).expect(ERR85_NO_POOL);
+        env::log(
+            format!("Modify total_fee pool_id {} from {} to {}", pool_id, pool.get_fee(), total_fee).as_bytes()
+        );
+        pool.modify_total_fee(total_fee);
+        self.pools.replace(pool_id, &pool);
+    }
+
     /// Remove exchange fee liquidity to owner's inner account.
     /// Owner's inner account storage should be prepared in advance.
     #[payable]

ref-exchange/src/pool.rs

@@ -35,6 +35,14 @@ impl Pool {
         }
     }
 
+    pub fn modify_total_fee(&mut self, total_fee: u32) {
+        match self {
+            Pool::SimplePool(pool) => pool.modify_total_fee(total_fee),
+            Pool::StableSwapPool(pool) => pool.modify_total_fee(total_fee),
+            Pool::RatedSwapPool(pool) => pool.modify_total_fee(total_fee),
+        }
+    }
+
     /// Adds liquidity into underlying pool.
     /// Updates amounts to amount kept in the pool.
     pub fn add_liquidity(

ref-exchange/src/rated_swap/mod.rs

@@ -86,6 +86,10 @@ impl RatedSwapPool {
         }
     }
 
+    pub fn modify_total_fee(&mut self, total_fee: u32) {
+        self.total_fee = total_fee;
+    }
+
     pub fn get_rates(&self) -> Vec<u128> {
         self.token_account_ids
         .iter()

ref-exchange/src/simple_pool.rs

@@ -64,6 +64,10 @@ impl SimplePool {
         }
     }
 
+    pub fn modify_total_fee(&mut self, total_fee: u32) {
+        self.total_fee = total_fee;
+    }
+
     /// See if the given account has been registered as a LP
     pub fn share_has_registered(&self, account_id: &AccountId) -> bool {
         self.shares.contains_key(account_id)

ref-exchange/src/stable_swap/mod.rs

@@ -77,6 +77,10 @@ impl StableSwapPool {
         }
     }
 
+    pub fn modify_total_fee(&mut self, total_fee: u32) {
+        self.total_fee = total_fee;
+    }
+
     pub fn get_amounts(&self) ->Vec<u128> {
         let mut amounts = self.c_amounts.clone();
         for (index, value) in self.token_decimals.iter().enumerate() {

ref-exchange/tests/test_migrate.rs

@@ -9,7 +9,7 @@ use crate::common::utils::*;
 pub mod common;
 
 near_sdk_sim::lazy_static_include::lazy_static_include_bytes! {
-    PREV_EXCHANGE_WASM_BYTES => "../releases/ref_exchange_release_v170.wasm",
+    PREV_EXCHANGE_WASM_BYTES => "../releases/ref_exchange_release_v171.wasm",
     EXCHANGE_WASM_BYTES => "../res/ref_exchange.wasm",
 }
 
@@ -47,7 +47,7 @@ fn test_upgrade() {
     .assert_success();
     let metadata = get_metadata(&pool);
     // println!("{:#?}", metadata);
-    assert_eq!(metadata.version, "1.7.1".to_string());
+    assert_eq!(metadata.version, "1.7.2".to_string());
     assert_eq!(metadata.admin_fee_bps, 5);
     assert_eq!(metadata.state, RunningState::Running);