fix: add missing SetTransactionFee and WithdrawFee governance actions

- Add SetTransactionFee (8) and WithdrawFee (9) to GovernanceAction enum
- Implement parsing logic for both actions matching Solidity implementation
- Add TODO comments for event emissions to match EVM contract behavior
- Ensure instruction parsing compatibility between Stylus and Solidity

Co-Authored-By: [email protected] <[email protected]>

Author: devin-ai-integration[bot]

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

@@ -16,6 +16,8 @@ pub enum GovernanceAction {
     RequestGovernanceDataSourceTransfer,
     SetWormholeAddress,
     SetFeeInToken,
+    SetTransactionFee,
+    WithdrawFee,
 }
 
 impl TryFrom<u8> for GovernanceAction {
@@ -31,6 +33,8 @@ impl TryFrom<u8> for GovernanceAction {
             5 => Ok(GovernanceAction::RequestGovernanceDataSourceTransfer),
             6 => Ok(GovernanceAction::SetWormholeAddress),
             7 => Ok(GovernanceAction::SetFeeInToken),
+            8 => Ok(GovernanceAction::SetTransactionFee),
+            9 => Ok(GovernanceAction::WithdrawFee),
             _ => Err(PythReceiverError::InvalidGovernanceAction),
         }
     }
@@ -313,6 +317,71 @@ pub fn parse_instruction(payload: Vec<u8>) -> Result<GovernanceInstruction, Pyth
                 address: Address::from(address_bytes),
             })
         }
+        GovernanceAction::SetTransactionFee => {
+            if payload.len() < cursor + 16 {
+                return Err(PythReceiverError::InvalidGovernanceMessage);
+            }
+            let fee_value_bytes = payload
+                .get(cursor..cursor + 8)
+                .ok_or(PythReceiverError::InvalidGovernanceMessage)?;
+
+            let value = u64::from_be_bytes(
+                fee_value_bytes
+                    .try_into()
+                    .map_err(|_| PythReceiverError::InvalidGovernanceMessage)?,
+            );
+
+            cursor += 8;
+
+            let expo_bytes = payload
+                .get(cursor..cursor + 8)
+                .ok_or(PythReceiverError::InvalidGovernanceMessage)?;
+            let expo = u64::from_be_bytes(
+                expo_bytes
+                    .try_into()
+                    .map_err(|_| PythReceiverError::InvalidGovernanceMessage)?,
+            );
+
+            cursor += 8;
+            GovernancePayload::SetTransactionFee(SetTransactionFee { value, expo })
+        }
+        GovernanceAction::WithdrawFee => {
+            if payload.len() < cursor + 28 {
+                return Err(PythReceiverError::InvalidGovernanceMessage);
+            }
+            
+            let mut target_address_bytes = [0u8; 20];
+            target_address_bytes.copy_from_slice(&payload[cursor..cursor + 20]);
+            cursor += 20;
+
+            let fee_value_bytes = payload
+                .get(cursor..cursor + 8)
+                .ok_or(PythReceiverError::InvalidGovernanceMessage)?;
+
+            let value = u64::from_be_bytes(
+                fee_value_bytes
+                    .try_into()
+                    .map_err(|_| PythReceiverError::InvalidGovernanceMessage)?,
+            );
+
+            cursor += 8;
+
+            let expo_bytes = payload
+                .get(cursor..cursor + 8)
+                .ok_or(PythReceiverError::InvalidGovernanceMessage)?;
+            let expo = u64::from_be_bytes(
+                expo_bytes
+                    .try_into()
+                    .map_err(|_| PythReceiverError::InvalidGovernanceMessage)?,
+            );
+
+            cursor += 8;
+            GovernancePayload::WithdrawFee(WithdrawFee {
+                value,
+                expo,
+                target_address: Address::from(target_address_bytes),
+            })
+        }
     };
 
     if cursor != payload.len() {

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

@@ -599,14 +599,15 @@ impl PythReceiver {
 
         self.single_update_fee_in_wei.set(new_fee);
 
-        // TODO: HANDLE EVENT EMISSION
+        // TODO: Emit FeeSet event with old_fee and new_fee
     }
 
     fn set_valid_period(&mut self, valid_time_period_seconds: u64) {
         let _old_valid_period = self.valid_time_period_seconds.get();
         self.valid_time_period_seconds.set(U256::from(valid_time_period_seconds));
 
-        // TODO: HANDLE EVENT EMISSION
+        // TODO: Emit ValidPeriodSet event with old_valid_period and new_valid_period
+        // emit ValidPeriodSet(old_valid_period, valid_time_period_seconds);
     }
 
     fn set_wormhole_address(
@@ -719,6 +720,8 @@ impl PythReceiver {
         let _old_fee = self.transaction_fee_in_wei.get();
 
         self.transaction_fee_in_wei.set(new_fee);
+
+        // TODO: Emit TransactionFeeSet event with old_fee and new_fee
     }
 
     fn withdraw_fee(&mut self, value: u64, expo: u64, target_address: Address) -> Result<(), PythReceiverError> {
@@ -733,6 +736,8 @@ impl PythReceiver {
         self.vm().transfer_eth(target_address, fee_to_withdraw)
             .map_err(|_| PythReceiverError::InsufficientFee)?;
 
+        // TODO: Emit FeeWithdrawn event with target_address and fee_to_withdraw
+
         Ok(())
     }
 }