Merge pull request #294 from lfglabs-dev/fix/nimbora-rewards

fix: nimbora rewards
lfglabs-dev · Oct 25, 2024 · 6c8c067 · 6c8c067
2 parents 09d2f0b + 6701aa6
commit 6c8c067
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/endpoints/defi/rewards.rs b/src/endpoints/defi/rewards.rs
@@ -96,6 +96,7 @@ async fn fetch_zklend_rewards(
                 .filter(|reward| !reward.claimed)
                 .map(|reward| CommonReward {
                     amount: reward.amount.value,
+                    displayed_amount: reward.amount.value,
                     proof: reward.proof,
                     reward_id: Some(reward.claim_id),
                     claim_contract: reward.claim_contract,
@@ -191,6 +192,7 @@ async fn fetch_nostra_rewards(
                     {
                         Some(CommonReward {
                             amount: doc.reward,
+                            displayed_amount: doc.reward,
                             proof: doc.proofs,
                             reward_id: None,
                             claim_contract: distributor,
@@ -245,7 +247,8 @@ async fn fetch_nimbora_rewards(
                 return Ok(vec![]);
             }
             let reward = CommonReward {
-                amount: amount - claimed_amount,
+                amount: amount,
+                displayed_amount: amount - claimed_amount,
                 proof: result.proof,
                 reward_id: None,
                 token_symbol: strk_symbol.clone(),
@@ -301,6 +304,7 @@ async fn fetch_ekubo_rewards(
                 {
                     Some(CommonReward {
                         amount: reward.claim.amount,
+                        displayed_amount: reward.claim.amount,
                         proof: reward.proof,
                         reward_id: Some(reward.claim.id),
                         claim_contract: reward.contract_address,
@@ -369,7 +373,7 @@ fn extract_rewards(common_rewards: &[CommonReward]) -> Vec<DefiReward> {
     common_rewards
         .iter()
         .map(|reward| DefiReward {
-            amount: reward.amount,
+            amount: reward.displayed_amount,
             token_symbol: reward.token_symbol.clone(),
         })
         .collect()

diff --git a/src/models.rs b/src/models.rs
@@ -470,6 +470,7 @@ pub enum RewardSource {
 #[derive(Serialize, Deserialize, Debug)]
 pub struct CommonReward {
     pub amount: FieldElement,
+    pub displayed_amount: FieldElement,
     pub proof: Vec<String>,
     pub reward_id: Option<u64>,
     pub claim_contract: FieldElement,