Cleanup resize, remove print staments, add extra check (#272)

Author: guibescos

program/rust/src/rust_oracle.rs

@@ -120,15 +120,14 @@ pub fn resize_price_account(
         check_id(system_program.key),
         OracleError::InvalidSystemAccount.into(),
     )?;
-    //throw an error if not a price account
-    //need to makre sure it goes out of scope immediatly to avoid mutable borrow errors
+    // Check that it is a valid initialized price account
     {
         load_checked::<PriceAccount>(price_account_info, PC_VERSION)?;
     }
     let account_len = price_account_info.try_data_len()?;
     match account_len {
         PRICE_T_SIZE => {
-            //ensure account is still rent exempt after resizing
+            // Ensure account is still rent exempt after resizing
             let rent: Rent = Default::default();
             let lamports_needed: u64 = rent
                 .minimum_balance(size_of::<PriceAccountWrapper>())
@@ -141,15 +140,19 @@ pub fn resize_price_account(
                     lamports_needed,
                 )?;
             }
-            //resize
-            //we do not need to zero initialize since this is the first time this memory
-            //is allocated
+            // We do not need to zero allocate because we won't access the data in the same
+            // instruction
             price_account_info.realloc(size_of::<PriceAccountWrapper>(), false)?;
-            //The load below would fail if the account was not a price account, reverting the whole
-            // transaction
+
+            // Check that everything is ok
+            check_valid_signable_account(
+                program_id,
+                price_account_info,
+                size_of::<PriceAccountWrapper>(),
+            )?;
             let mut price_account =
                 load_checked::<PriceAccountWrapper>(price_account_info, PC_VERSION)?;
-            //Initialize Time Machine
+            // Initialize Time Machine
             price_account.initialize_time_machine()?;
             Ok(())
         }

program/rust/src/time_machine_types.rs

@@ -10,7 +10,6 @@ use bytemuck::{
     Pod,
     Zeroable,
 };
-use solana_program::msg;
 
 
 #[derive(Debug, Clone, Copy)]
@@ -35,12 +34,12 @@ pub struct PriceAccountWrapper {
 }
 impl PriceAccountWrapper {
     pub fn initialize_time_machine(&mut self) -> Result<(), OracleError> {
-        msg!("implement me");
+        // TO DO
         Ok(())
     }
 
     pub fn add_price_to_time_machine(&mut self) -> Result<(), OracleError> {
-        msg!("implement me");
+        // TO DO
         Ok(())
     }
 }