finished get_price_unsafe, get_price_no_older_than

Author: ayushboss

target_chains/stylus/contracts/pyth-receiver/src/error.rs

@@ -0,0 +1,19 @@
+use alloc::vec::Vec;
+
+pub enum PythReceiverError {
+    PriceUnavailable
+}
+
+impl core::fmt::Debug for PythReceiverError {
+    fn fmt(&self, _: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
+        Ok(())
+    }
+}
+
+impl From<PythReceiverError> for Vec<u8> {
+    fn from(error: PythReceiverError) -> Vec<u8> {
+        vec![match error {
+            PythReceiverError::PriceUnavailable => 1,
+        }]
+    }
+}

target_chains/stylus/contracts/pyth-receiver/src/lib.rs

@@ -6,13 +6,15 @@
 extern crate alloc;
 
 mod structs;
+mod error;
 
 use alloc::vec::Vec;
 use stylus_sdk::{alloy_primitives::{U256, U64, I32, I64, FixedBytes},
                 prelude::*, 
                 storage::{StorageAddress, StorageVec, StorageMap, StorageUint, StorageBool, StorageU256}};
 
 use structs::{DataSourceStorage, PriceInfoReturn, PriceInfoStorage};
+use error::{PythReceiverError};
 
 #[storage]
 #[entrypoint]
@@ -31,12 +33,27 @@ pub struct PythReceiver {
 
 #[public]
 impl PythReceiver {
-    pub fn get_price_unsafe(&self, _id: [u8; 32]) -> PriceInfoReturn {
-        (U64::ZERO, I32::ZERO, I64::ZERO, U64::ZERO, I64::ZERO, U64::ZERO)
+    pub fn get_price_unsafe(&self, _id: [u8; 32]) -> Result<PriceInfoReturn, PythReceiverError> {
+        let id_fb = FixedBytes::<32>::from(_id);
+        
+        let price_info = self.latest_price_info.get(id_fb).unwrap_or(PythReceiverError::PriceUnavailable)?;
+
+        Ok((
+            price_info.publish_time,
+            price_info.expo,
+            price_info.price,
+            price_info.conf,
+            price_info.ema_price,
+            price_info.ema_conf,
+        ))
     }
 
-    pub fn get_price_no_older_than(&self, _id: [u8; 32], _age: u64) -> PriceInfoReturn {
-        (U64::ZERO, I32::ZERO, I64::ZERO, U64::ZERO, I64::ZERO, U64::ZERO)
+    pub fn get_price_no_older_than(&self, _id: [u8; 32], _age: u64) -> Result<PriceInfoReturn, PythReceiverError> {
+        let price_info = self.get_price_unsafe(_id)?;
+        if !self.is_no_older_than(price_info.0, _age) {
+            return Err(PythReceiverError::PriceUnavailable);
+        }
+        Ok(price_info)
     }
 
     pub fn get_ema_price_unsafe(&self, _id: [u8; 32]) -> PriceInfoReturn {
@@ -108,4 +125,11 @@ impl PythReceiver {
     ) -> Vec<PriceInfoReturn> {
         Vec::new()
     }
+
+    fn is_no_older_than(&self, publish_time: U64, max_age: u64) -> bool {
+        let current_u64: u64 = self.vm().block_timestamp();
+        let publish_time_u64: u64 = publish_time.to::<u64>();
+        
+        current_u64.saturating_sub(publish_time_u64) <= max_age
+    }
 }