feat: optimize Bonk protocol with caching and wSOL management

- Add instruction/PDA caching for Bonk operations
- Implement reusable wSOL management module
- Support pre-computed pool addresses in BonkParams
- Optimize memory allocation with SmallVec and fixed arrays
- Reduce code duplication across protocols

src/trading/common/mod.rs

@@ -3,10 +3,12 @@ pub mod transaction_builder;
 pub mod compute_budget_manager;
 pub mod address_lookup_manager;
 pub mod utils;
+pub mod wsol_manager;
 
 // Re-export commonly used functions
 pub use nonce_manager::*;
 pub use transaction_builder::*;
 pub use compute_budget_manager::*;
 pub use address_lookup_manager::*;
-pub use utils::*;
+pub use utils::*;
+pub use wsol_manager::*;

src/trading/common/wsol_manager.rs

@@ -0,0 +1,65 @@
+use crate::common::fast_fn::create_associated_token_account_idempotent_fast;
+use smallvec::SmallVec;
+use solana_sdk::{instruction::Instruction, pubkey::Pubkey};
+use solana_system_interface::instruction::transfer;
+use spl_token::instruction::close_account;
+
+#[inline]
+pub fn handle_wsol(payer: &Pubkey, amount_in: u64) -> SmallVec<[Instruction; 3]> {
+    let wsol_token_account =
+        crate::common::fast_fn::get_associated_token_address_with_program_id_fast(
+            &payer,
+            &crate::constants::WSOL_TOKEN_ACCOUNT,
+            &crate::constants::TOKEN_PROGRAM,
+        );
+
+    let mut insts = SmallVec::<[Instruction; 3]>::new();
+    insts.extend([
+        create_associated_token_account_idempotent_fast(
+            &payer,
+            &payer,
+            &crate::constants::WSOL_TOKEN_ACCOUNT,
+            &crate::constants::TOKEN_PROGRAM,
+        ),
+        transfer(&payer, &wsol_token_account, amount_in),
+        spl_token::instruction::sync_native(&crate::constants::TOKEN_PROGRAM, &wsol_token_account)
+            .unwrap(),
+    ]);
+
+    insts
+}
+
+pub fn close_wsol(payer: &Pubkey) -> Instruction {
+    let wsol_token_account =
+        crate::common::fast_fn::get_associated_token_address_with_program_id_fast(
+            &payer,
+            &crate::constants::WSOL_TOKEN_ACCOUNT,
+            &crate::constants::TOKEN_PROGRAM,
+        );
+    crate::common::fast_fn::get_cached_instruction(
+        crate::common::fast_fn::InstructionCacheKey::CloseWsolAccount {
+            payer: *payer,
+            wsol_token_account,
+        },
+        || {
+            close_account(
+                &crate::constants::TOKEN_PROGRAM,
+                &wsol_token_account,
+                &payer,
+                &payer,
+                &[],
+            )
+            .unwrap()
+        },
+    )
+}
+
+#[inline]
+pub fn create_wsol_ata(payer: &Pubkey) -> Instruction {
+    create_associated_token_account_idempotent_fast(
+        &payer,
+        &payer,
+        &crate::constants::WSOL_TOKEN_ACCOUNT,
+        &crate::constants::TOKEN_PROGRAM,
+    )
+}