feat: AuthorizationLock implementation

interface/src/lib.rs

@@ -7,18 +7,20 @@ use solana_sdk::{
 };
 pub use swig;
 use swig::actions::{
-    add_authority_v1::AddAuthorityV1Args, create_session_v1::CreateSessionV1Args,
-    create_sub_account_v1::CreateSubAccountV1Args, create_v1::CreateV1Args,
-    remove_authority_v1::RemoveAuthorityV1Args, sub_account_sign_v1::SubAccountSignV1Args,
-    toggle_sub_account_v1::ToggleSubAccountV1Args,
+    add_authority_v1::AddAuthorityV1Args, add_authorization_lock_v1::AddAuthorizationLockV1Args,
+    create_session_v1::CreateSessionV1Args, create_sub_account_v1::CreateSubAccountV1Args,
+    create_v1::CreateV1Args, remove_authority_v1::RemoveAuthorityV1Args,
+    remove_authorization_lock_v1::RemoveAuthorizationLockV1Args,
+    sub_account_sign_v1::SubAccountSignV1Args, toggle_sub_account_v1::ToggleSubAccountV1Args,
     withdraw_from_sub_account_v1::WithdrawFromSubAccountV1Args,
 };
 pub use swig_compact_instructions::*;
 use swig_state_x::{
     action::{
-        all::All, manage_authority::ManageAuthority, program::Program, program_scope::ProgramScope,
-        sol_limit::SolLimit, sol_recurring_limit::SolRecurringLimit, stake_all::StakeAll,
-        stake_limit::StakeLimit, stake_recurring_limit::StakeRecurringLimit,
+        all::All, manage_authority::ManageAuthority,
+        manage_authorization_locks::ManageAuthorizationLocks, program::Program,
+        program_scope::ProgramScope, sol_limit::SolLimit, sol_recurring_limit::SolRecurringLimit,
+        stake_all::StakeAll, stake_limit::StakeLimit, stake_recurring_limit::StakeRecurringLimit,
         sub_account::SubAccount, token_limit::TokenLimit,
         token_recurring_limit::TokenRecurringLimit, Action, Permission,
     },
@@ -39,6 +41,7 @@ pub enum ClientAction {
     ProgramScope(ProgramScope),
     All(All),
     ManageAuthority(ManageAuthority),
+    ManageAuthorizationLocks(ManageAuthorizationLocks),
     SubAccount(SubAccount),
     StakeLimit(StakeLimit),
     StakeRecurringLimit(StakeRecurringLimit),
@@ -60,6 +63,10 @@ impl ClientAction {
             ClientAction::ProgramScope(_) => (Permission::ProgramScope, ProgramScope::LEN),
             ClientAction::All(_) => (Permission::All, All::LEN),
             ClientAction::ManageAuthority(_) => (Permission::ManageAuthority, ManageAuthority::LEN),
+            ClientAction::ManageAuthorizationLocks(_) => (
+                Permission::ManageAuthorizationLocks,
+                ManageAuthorizationLocks::LEN,
+            ),
             ClientAction::SubAccount(_) => (Permission::SubAccount, SubAccount::LEN),
             ClientAction::StakeLimit(_) => (Permission::StakeLimit, StakeLimit::LEN),
             ClientAction::StakeRecurringLimit(_) => {
@@ -86,6 +93,7 @@ impl ClientAction {
             ClientAction::ProgramScope(action) => action.into_bytes(),
             ClientAction::All(action) => action.into_bytes(),
             ClientAction::ManageAuthority(action) => action.into_bytes(),
+            ClientAction::ManageAuthorizationLocks(action) => action.into_bytes(),
             ClientAction::SubAccount(action) => action.into_bytes(),
             ClientAction::StakeLimit(action) => action.into_bytes(),
             ClientAction::StakeRecurringLimit(action) => action.into_bytes(),
@@ -969,3 +977,65 @@ impl ToggleSubAccountInstruction {
         })
     }
 }
+
+pub struct AddAuthorizationLockInstruction;
+
+impl AddAuthorizationLockInstruction {
+    pub fn new(
+        swig_account: Pubkey,
+        authority: Pubkey,
+        payer: Pubkey,
+        acting_role_id: u32,
+        token_mint: [u8; 32],
+        amount: u64,
+        expiry_slot: u64,
+    ) -> anyhow::Result<Instruction> {
+        let accounts = vec![
+            AccountMeta::new(swig_account, false),
+            AccountMeta::new(payer, true),
+            AccountMeta::new_readonly(system_program::ID, false),
+            AccountMeta::new_readonly(authority, true),
+        ];
+
+        let args = AddAuthorizationLockV1Args::new(acting_role_id, token_mint, amount, expiry_slot);
+        let args_bytes = args
+            .into_bytes()
+            .map_err(|e| anyhow::anyhow!("Failed to serialize args {:?}", e))?;
+
+        Ok(Instruction {
+            program_id: program_id(),
+            accounts,
+            data: [args_bytes, &[3]].concat(),
+        })
+    }
+}
+
+pub struct RemoveAuthorizationLockInstruction;
+
+impl RemoveAuthorizationLockInstruction {
+    pub fn new(
+        swig_account: Pubkey,
+        authority: Pubkey,
+        payer: Pubkey,
+        acting_role_id: u32,
+        lock_index: u32,
+    ) -> anyhow::Result<Instruction> {
+        let accounts = vec![
+            AccountMeta::new(swig_account, false),
+            AccountMeta::new(payer, true),
+            AccountMeta::new_readonly(system_program::ID, false),
+            AccountMeta::new_readonly(authority, true),
+        ];
+
+        let args = RemoveAuthorizationLockV1Args::new(acting_role_id, lock_index);
+        let args_bytes = args
+            .into_bytes()
+            .map_err(|e| anyhow::anyhow!("Failed to serialize args {:?}", e))?;
+
+        Ok(Instruction {
+            program_id: program_id(),
+            accounts,
+            data: [args_bytes, &[3]].concat(),
+        })
+    }
+}