add fns to get reward token and payer

BibliothecaDAO · May 19, 2024 · e44cd34 · e44cd34
1 parent 4688f70
commit e44cd34
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/contracts/src/components/strealm.cairo b/contracts/src/components/strealm.cairo
@@ -3,6 +3,8 @@ use openzeppelin::token::erc721::interface::{IERC721Dispatcher};
 use starknet::ContractAddress;
 #[starknet::interface]
 trait IStRealm<TState> {
+    fn get_reward_token(self: @TState) -> ContractAddress;
+    fn get_reward_payer(self: @TState) -> ContractAddress;
     fn get_stream(self: @TState, owner: ContractAddress) -> Stream;
     fn get_flow(self: @TState, flow_id: u32) -> Flow;
     fn get_latest_flow_id(self: @TState) -> u32;
@@ -125,6 +127,14 @@ mod StRealmComponent {
             self._reward_balance(starknet::get_caller_address())
         }
 
+        fn get_reward_token(self: @ComponentState<TContractState>) -> ContractAddress {
+            self.StRealm_reward_token.read()
+        }
+
+        fn get_reward_payer(self: @ComponentState<TContractState>) -> ContractAddress {
+            self.StRealm_reward_payer.read()
+        }
+
 
         //
         // Mutables