restaking: record the last received height when collecting rewards in…

… withdrawal request (#318)

When requesting for withdrawal, the pending rewards are transferred to
the user but the height at which the rewards were claimed was not
updated. This fixes that.

---------

Co-authored-by: Michal Nazarewicz <mina86@mina86.com>

solana/restaking/programs/restaking/src/lib.rs

@@ -176,12 +176,14 @@ pub mod restaking {
          * Get the rewards from guest blockchain.
          */
 
-        let (rewards, _current_height) = chain.calculate_rewards(
+        let (rewards, current_height) = chain.calculate_rewards(
             vault_params.last_received_rewards_height,
             *validator_key,
             vault_params.stake_amount,
         )?;
 
+        vault_params.last_received_rewards_height = current_height;
+
         let bump = ctx.bumps.staking_params;
         let seeds =
             [STAKING_PARAMS_SEED, TEST_SEED, core::slice::from_ref(&bump)];