Add claim rewards action support

src/bin/claim_rewards.rs

@@ -0,0 +1,28 @@
+use alloy::signers::local::PrivateKeySigner;
+use hyperliquid_rust_sdk::{BaseUrl, ExchangeClient, ExchangeResponseStatus};
+use log::info;
+
+#[tokio::main]
+async fn main() {
+    env_logger::init();
+    // Key was randomly generated for testing and shouldn't be used with any real funds
+    let wallet: PrivateKeySigner =
+        "e908f86dbb4d55ac876378565aafeabc187f6690f046459397b17d9b9a19688e"
+            .parse()
+            .unwrap();
+
+    let exchange_client = ExchangeClient::new(None, wallet, Some(BaseUrl::Testnet), None, None)
+        .await
+        .unwrap();
+
+    let response = exchange_client.claim_rewards(None).await.unwrap();
+
+    match response {
+        ExchangeResponseStatus::Ok(exchange_response) => {
+            info!("Rewards claimed successfully: {:?}", exchange_response);
+        }
+        ExchangeResponseStatus::Err(e) => {
+            info!("Failed to claim rewards: {}", e);
+        }
+    }
+}

src/exchange/actions.rs

@@ -235,6 +235,10 @@ pub struct ScheduleCancel {
     pub time: Option<u64>,
 }
 
+#[derive(Serialize, Deserialize, Debug, Clone)]
+#[serde(rename_all = "camelCase")]
+pub struct ClaimRewards;
+
 impl Eip712 for ApproveBuilderFee {
     fn domain(&self) -> Eip712Domain {
         eip_712_domain(self.signature_chain_id)

src/exchange/exchange_client.rs

@@ -11,8 +11,9 @@ use serde::{ser::SerializeStruct, Deserialize, Serialize, Serializer};
 use crate::{
     exchange::{
         actions::{
-            ApproveAgent, ApproveBuilderFee, BulkCancel, BulkModify, BulkOrder, EvmUserModify,
-            ScheduleCancel, SetReferrer, UpdateIsolatedMargin, UpdateLeverage, UsdSend,
+            ApproveAgent, ApproveBuilderFee, BulkCancel, BulkModify, BulkOrder, ClaimRewards,
+            EvmUserModify, ScheduleCancel, SetReferrer, UpdateIsolatedMargin, UpdateLeverage,
+            UsdSend,
         },
         cancel::{CancelRequest, CancelRequestCloid, ClientCancelRequestCloid},
         modify::{ClientModifyRequest, ModifyRequest},
@@ -79,6 +80,7 @@ pub enum Actions {
     ApproveBuilderFee(ApproveBuilderFee),
     EvmUserModify(EvmUserModify),
     ScheduleCancel(ScheduleCancel),
+    ClaimRewards(ClaimRewards),
 }
 
 impl Actions {
@@ -805,6 +807,22 @@ impl ExchangeClient {
 
         self.post(action, signature, timestamp).await
     }
+
+    pub async fn claim_rewards(
+        &self,
+        wallet: Option<&PrivateKeySigner>,
+    ) -> Result<ExchangeResponseStatus> {
+        let wallet = wallet.unwrap_or(&self.wallet);
+        let timestamp = next_nonce();
+
+        let action = Actions::ClaimRewards(ClaimRewards {});
+        let connection_id = action.hash(timestamp, self.vault_address)?;
+        let action = serde_json::to_value(&action).map_err(|e| Error::JsonParse(e.to_string()))?;
+        let is_mainnet = self.http_client.is_mainnet();
+        let signature = sign_l1_action(wallet, connection_id, is_mainnet)?;
+
+        self.post(action, signature, timestamp).await
+    }
 }
 
 fn round_to_decimals(value: f64, decimals: u32) -> f64 {
@@ -1022,4 +1040,27 @@ mod tests {
 
         Ok(())
     }
+
+    #[test]
+    fn test_claim_rewards_action_hashing() -> Result<()> {
+        let wallet = get_wallet()?;
+        let action = Actions::ClaimRewards(ClaimRewards {});
+        let connection_id = action.hash(1583838, None)?;
+
+        // Test mainnet signature
+        let signature = sign_l1_action(&wallet, connection_id, true)?;
+        assert_eq!(
+            signature.to_string(),
+            "0xe13542800ba5ec821153401e1cafac484d1f861adbbb86c00b580ec2560c153248b8d9f0e004ecc86959c07d44b591861ebab2167b54651a81367e2c3d472d4e1c"
+        );
+
+        // Test testnet signature
+        let signature = sign_l1_action(&wallet, connection_id, false)?;
+        assert_eq!(
+            signature.to_string(),
+            "0x16de9b346ddd8e200492a2db45ec9104dcdfc7fbfdbcd85890a6063bdd56df2c44846714c261a431de7095ad52e07143346eb26d9e66c6aed4674f120a1048131c"
+        );
+
+        Ok(())
+    }
 }