sol-trade-sdk/examples/wsol_wrapper/src/main.rs

use sol_trade_sdk::{
    constants::trade::trade::{DEFAULT_CU_LIMIT, DEFAULT_CU_PRICE},
    swqos::{settings::SwqosSettings, SwqosConfig},
    SolanaTrade,
};
use solana_sdk::{commitment_config::CommitmentConfig, signature::Keypair};
use std::sync::Arc;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    println!("🔄 WSOL Wrapper Example");
    println!("This example demonstrates how to wrap SOL to WSOL and unwrap WSOL back to SOL");

    // Initialize SolanaTrade client
    let solana_trade = create_solana_trade_client().await?;

    // Example 1: Wrap SOL to WSOL
    println!("\n📦 Example 1: Wrapping SOL to WSOL");
    let wrap_amount = 1_000_000; // 0.001 SOL in lamports
    println!("Wrapping {} lamports (0.001 SOL) to WSOL...", wrap_amount);

    match solana_trade.wrap_sol_to_wsol(wrap_amount).await {
        Ok(signature) => {
            println!("✅ Successfully wrapped SOL to WSOL!");
            println!("Transaction signature: {}", signature);
            println!("Explorer: https://solscan.io/tx/{}", signature);
        }
        Err(e) => {
            println!("❌ Failed to wrap SOL to WSOL: {}", e);
        }
    }

    // Wait a moment before unwrapping
    println!("\n⏳ Waiting 3 seconds before unwrapping...");
    tokio::time::sleep(tokio::time::Duration::from_secs(3)).await;

    // Example 2: Close WSOL account and unwrap all remaining balance
    println!("\n🔒 Example 2: Closing WSOL account and unwrapping remaining balance");
    println!("Closing WSOL account and unwrapping all remaining balance to SOL...");

    match solana_trade.close_wsol().await {
        Ok(signature) => {
            println!("✅ Successfully closed WSOL account and unwrapped remaining balance!");
            println!("Transaction signature: {}", signature);
            println!("Explorer: https://solscan.io/tx/{}", signature);
        }
        Err(e) => {
            println!("❌ Failed to close WSOL account: {}", e);
        }
    }

    println!("\n🎉 WSOL Wrapper example completed!");
    Ok(())
}

/// Create and initialize SolanaTrade client
async fn create_solana_trade_client() -> Result<SolanaTrade, Box<dyn std::error::Error>> {
    println!("🚀 Initializing SolanaTrade client...");
    let payer = Keypair::from_base58_string("use_your_payer_keypair_here");
    let rpc_url = "https://api.mainnet-beta.solana.com".to_string();
    let commitment = CommitmentConfig::confirmed();
    let swqos_settings: Vec<SwqosSettings> = vec![SwqosSettings::new(
        SwqosConfig::Default(rpc_url.clone()),
        DEFAULT_CU_LIMIT,
        DEFAULT_CU_PRICE,
        0.0,
        0.0,
    )];
    let solana_trade = SolanaTrade::new(Arc::new(payer), rpc_url, commitment, swqos_settings).await;
    println!("✅ SolanaTrade client initialized successfully!");
    Ok(solana_trade)
}