feat(sdk-rs): add get_spending_limit

Squads-Protocol · Oct 15, 2023 · 8f97564 · 8f97564
1 parent 03bc569
commit 8f97564
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/sdk/rs/Cargo.toml b/sdk/rs/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "squads-multisig"
-version = "0.0.16"
+version = "0.0.17"
 description = "An SDK for building automated programs on Solana"
 edition = "2021"
 license = "MIT OR Apache-2.0"

diff --git a/sdk/rs/src/client.rs b/sdk/rs/src/client.rs
@@ -20,7 +20,6 @@ pub use squads_multisig_program::instructions::ProposalCreateArgs;
 pub use squads_multisig_program::instructions::ProposalVoteArgs;
 pub use squads_multisig_program::instructions::SpendingLimitUseArgs;
 pub use squads_multisig_program::instructions::VaultTransactionCreateArgs;
-use squads_multisig_program::Multisig;
 
 use crate::anchor_lang::prelude::Pubkey;
 use crate::anchor_lang::AccountDeserialize;
@@ -29,6 +28,7 @@ use crate::anchor_lang::{
 };
 use crate::error::ClientError;
 use crate::solana_program::instruction::AccountMeta;
+use crate::state::{Multisig, SpendingLimit};
 use crate::ClientResult;
 
 /// Gets a `Multisig` account from the chain.
@@ -41,6 +41,20 @@ pub async fn get_multisig(rpc_client: &RpcClient, multisig_key: &Pubkey) -> Clie
     Ok(multisig)
 }
 
+/// Gets a `SpendingLimit` account from the chain.
+pub async fn get_spending_limit(
+    rpc_client: &RpcClient,
+    spending_limit_key: &Pubkey,
+) -> ClientResult<SpendingLimit> {
+    let spending_limit_account = rpc_client.get_account(spending_limit_key).await?;
+
+    let spending_limit =
+        SpendingLimit::try_deserialize(&mut spending_limit_account.data.as_slice())
+            .map_err(|_| ClientError::DeserializationError)?;
+
+    Ok(spending_limit)
+}
+
 /// Creates a new multisig config transaction.
 /// Example:
 /// ```

diff --git a/sdk/rs/src/lib.rs b/sdk/rs/src/lib.rs
@@ -28,7 +28,7 @@ pub mod state {
     pub use squads_multisig_program::state::{
         Batch, ConfigAction, ConfigTransaction, Member, Multisig, MultisigCompiledInstruction,
         MultisigMessageAddressTableLookup, Period, Permission, Permissions, Proposal,
-        ProposalStatus, VaultTransactionMessage,
+        ProposalStatus, SpendingLimit, VaultTransactionMessage,
     };
     pub use squads_multisig_program::SmallVec;
 }