use crate::error::ClientResult;
use crate::common::pumpswap::{
    logs_data::{
        PumpSwapInstruction,
        BuyEvent, SellEvent, CreatePoolEvent, DepositEvent, WithdrawEvent,
        DisableEvent, UpdateAdminEvent, UpdateFeeConfigEvent, discriminators
    },
    logs_events::PumpSwapEvent
};

use solana_sdk::pubkey::Pubkey;
use solana_sdk::instruction::CompiledInstruction;
use std::time::{SystemTime, UNIX_EPOCH};

/// 处理PumpSwap日志并调用回调函数
pub async fn process_pumpswap_logs<F>(
    signature: &str,
    logs: Vec<String>,
    slot: Option<u64>,
    callback: F,
) -> ClientResult<()>
where
    F: Fn(&str, PumpSwapEvent) + Send + Sync,
{
    let events = PumpSwapEvent::parse_logs(&logs);
    for mut event in events {
        // 设置签名和slot
        match &mut event {
            PumpSwapEvent::Buy(e) => {
                e.signature = signature.to_string();
                if let Some(s) = slot {
                    e.slot = s;
                }
            },
            PumpSwapEvent::Sell(e) => {
                e.signature = signature.to_string();
                if let Some(s) = slot {
                    e.slot = s;
                }
            },
            PumpSwapEvent::CreatePool(e) => {
                e.signature = signature.to_string();
                if let Some(s) = slot {
                    e.slot = s;
                }
            },
            PumpSwapEvent::Deposit(e) => {
                e.signature = signature.to_string();
                if let Some(s) = slot {
                    e.slot = s;
                }
            },
            PumpSwapEvent::Withdraw(e) => {
                e.signature = signature.to_string();
                if let Some(s) = slot {
                    e.slot = s;
                }
            },
            PumpSwapEvent::Disable(e) => {
                e.signature = signature.to_string();
                if let Some(s) = slot {
                    e.slot = s;
                }
            },
            PumpSwapEvent::UpdateAdmin(e) => {
                e.signature = signature.to_string();
                if let Some(s) = slot {
                    e.slot = s;
                }
            },
            PumpSwapEvent::UpdateFeeConfig(e) => {
                e.signature = signature.to_string();
                if let Some(s) = slot {
                    e.slot = s;
                }
            },
            _ => {}
        }
        callback(signature, event);
    }
    Ok(())
}

/// 获取当前时间戳
fn current_timestamp() -> i64 {
    SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .expect("Time went backwards")
        .as_secs() as i64
}

/// 从指令中解析PumpSwap指令
pub fn parse_pumpswap_instruction(instruction: &CompiledInstruction, accounts: &[Pubkey]) -> Option<PumpSwapInstruction> {
    if instruction.data.len() < 8 {
        return None;
    }

    let discriminator = &instruction.data[..8];
    let data = &instruction.data[8..];

    match discriminator {
        d if d == discriminators::BUY_IX => {
            // buy指令参数: base_amount_out: u64, max_quote_amount_in: u64
            // 账户顺序：pool, user, global_config, base_mint, quote_mint, user_base_token_account, 
            // user_quote_token_account, pool_base_token_account, pool_quote_token_account, 
            // protocol_fee_recipient, protocol_fee_recipient_token_account, ...
            if data.len() < 16 || accounts.len() < 11 {
                return None;
            }
            let base_amount_out = u64::from_le_bytes(data[0..8].try_into().ok()?);
            let max_quote_amount_in = u64::from_le_bytes(data[8..16].try_into().ok()?);
            
            Some(PumpSwapInstruction::Buy(BuyEvent {
                base_amount_out,
                max_quote_amount_in,
                pool: accounts[0],
                user: accounts[1],
                user_base_token_account: accounts[5],
                user_quote_token_account: accounts[6],
                protocol_fee_recipient: accounts[9],
                protocol_fee_recipient_token_account: accounts[10],
                timestamp: current_timestamp(),
                ..Default::default()
            }))
        },
        d if d == discriminators::SELL_IX => {
            // sell指令参数: base_amount_in: u64, min_quote_amount_out: u64
            // 账户顺序：pool, user, global_config, base_mint, quote_mint, user_base_token_account,
            // user_quote_token_account, pool_base_token_account, pool_quote_token_account,
            // protocol_fee_recipient, protocol_fee_recipient_token_account, ...
            if data.len() < 16 || accounts.len() < 11 {
                return None;
            }
            let base_amount_in = u64::from_le_bytes(data[0..8].try_into().ok()?);
            let min_quote_amount_out = u64::from_le_bytes(data[8..16].try_into().ok()?);
            
            Some(PumpSwapInstruction::Sell(SellEvent {
                base_amount_in,
                min_quote_amount_out,
                pool: accounts[0],
                user: accounts[1],
                user_base_token_account: accounts[5],
                user_quote_token_account: accounts[6],
                protocol_fee_recipient: accounts[9],
                protocol_fee_recipient_token_account: accounts[10],
                timestamp: current_timestamp(),
                ..Default::default()
            }))
        },
        d if d == discriminators::CREATE_POOL_IX => {
            // create_pool指令参数: index: u16, base_amount_in: u64, quote_amount_in: u64
            // 账户顺序：pool, global_config, creator, base_mint, quote_mint, lp_mint,
            // user_base_token_account, user_quote_token_account, user_pool_token_account,
            // pool_base_token_account, pool_quote_token_account, ...
            if data.len() < 18 || accounts.len() < 11 {
                return None;
            }
            let index = u16::from_le_bytes(data[0..2].try_into().ok()?);
            let base_amount_in = u64::from_le_bytes(data[2..10].try_into().ok()?);
            let quote_amount_in = u64::from_le_bytes(data[10..18].try_into().ok()?);
            
            Some(PumpSwapInstruction::CreatePool(CreatePoolEvent {
                index,
                base_amount_in,
                quote_amount_in,
                pool: accounts[0],
                creator: accounts[2],
                base_mint: accounts[3],
                quote_mint: accounts[4],
                lp_mint: accounts[5],
                user_base_token_account: accounts[6],
                user_quote_token_account: accounts[7],
                timestamp: current_timestamp(),
                ..Default::default()
            }))
        },
        d if d == discriminators::DEPOSIT_IX => {
            // deposit指令参数: lp_token_amount_out: u64, max_base_amount_in: u64, max_quote_amount_in: u64
            // 账户顺序：pool, global_config, user, base_mint, quote_mint, lp_mint,
            // user_base_token_account, user_quote_token_account, user_pool_token_account,
            // pool_base_token_account, pool_quote_token_account, ...
            if data.len() < 24 || accounts.len() < 11 {
                return None;
            }
            let lp_token_amount_out = u64::from_le_bytes(data[0..8].try_into().ok()?);
            let max_base_amount_in = u64::from_le_bytes(data[8..16].try_into().ok()?);
            let max_quote_amount_in = u64::from_le_bytes(data[16..24].try_into().ok()?);
            
            Some(PumpSwapInstruction::Deposit(DepositEvent {
                lp_token_amount_out,
                max_base_amount_in,
                max_quote_amount_in,
                pool: accounts[0],
                user: accounts[2],
                user_base_token_account: accounts[6],
                user_quote_token_account: accounts[7],
                user_pool_token_account: accounts[8],
                timestamp: current_timestamp(),
                ..Default::default()
            }))
        },
        d if d == discriminators::WITHDRAW_IX => {
            // withdraw指令参数: lp_token_amount_in: u64, min_base_amount_out: u64, min_quote_amount_out: u64
            // 账户顺序：pool, global_config, user, base_mint, quote_mint, lp_mint,
            // user_base_token_account, user_quote_token_account, user_pool_token_account,
            // pool_base_token_account, pool_quote_token_account, ...
            if data.len() < 24 || accounts.len() < 11 {
                return None;
            }
            let lp_token_amount_in = u64::from_le_bytes(data[0..8].try_into().ok()?);
            let min_base_amount_out = u64::from_le_bytes(data[8..16].try_into().ok()?);
            let min_quote_amount_out = u64::from_le_bytes(data[16..24].try_into().ok()?);
            
            Some(PumpSwapInstruction::Withdraw(WithdrawEvent {
                lp_token_amount_in,
                min_base_amount_out,
                min_quote_amount_out,
                pool: accounts[0],
                user: accounts[2],
                user_base_token_account: accounts[6],
                user_quote_token_account: accounts[7],
                user_pool_token_account: accounts[8],
                timestamp: current_timestamp(),
                ..Default::default()
            }))
        },
        d if d == discriminators::DISABLE_IX => {
            // disable指令参数: disable_create_pool: bool, disable_deposit: bool, disable_withdraw: bool, disable_buy: bool, disable_sell: bool
            // 账户顺序：admin, global_config, event_authority, program
            if data.len() < 5 || accounts.len() < 2 {
                return None;
            }
            let disable_create_pool = data[0] != 0;
            let disable_deposit = data[1] != 0;
            let disable_withdraw = data[2] != 0;
            let disable_buy = data[3] != 0;
            let disable_sell = data[4] != 0;
            
            Some(PumpSwapInstruction::Disable(DisableEvent {
                disable_create_pool,
                disable_deposit,
                disable_withdraw,
                disable_buy,
                disable_sell,
                admin: accounts[0],
                timestamp: current_timestamp(),
                ..Default::default()
            }))
        },
        d if d == discriminators::UPDATE_ADMIN_IX => {
            // update_admin指令参数: 无
            // 账户顺序：admin, global_config, new_admin, event_authority, program
            if accounts.len() < 3 {
                return None;
            }
            Some(PumpSwapInstruction::UpdateAdmin(UpdateAdminEvent {
                old_admin: accounts[0],
                new_admin: accounts[2],
                timestamp: current_timestamp(),
                ..Default::default()
            }))
        },
        d if d == discriminators::UPDATE_FEE_CONFIG_IX => {
            // update_fee_config指令参数: lp_fee_basis_points: u64, protocol_fee_basis_points: u64, protocol_fee_recipients: [pubkey; 8]
            // 账户顺序：admin, global_config, event_authority, program
            if data.len() < 272 || accounts.len() < 2 { // 8 + 8 + 32*8 = 272 bytes
                return None;
            }
            let lp_fee_basis_points = u64::from_le_bytes(data[0..8].try_into().ok()?);
            let protocol_fee_basis_points = u64::from_le_bytes(data[8..16].try_into().ok()?);
            
            let mut protocol_fee_recipients = [Pubkey::default(); 8];
            for i in 0..8 {
                let start = 16 + i * 32;
                let end = start + 32;
                if let Ok(pubkey_bytes) = data[start..end].try_into() {
                    protocol_fee_recipients[i] = Pubkey::new_from_array(pubkey_bytes);
                }
            }
            
            Some(PumpSwapInstruction::UpdateFeeConfig(UpdateFeeConfigEvent {
                admin: accounts[0],
                new_lp_fee_basis_points: lp_fee_basis_points,
                new_protocol_fee_basis_points: protocol_fee_basis_points,
                new_protocol_fee_recipients: protocol_fee_recipients,
                timestamp: current_timestamp(),
                ..Default::default()
            }))
        },
        _ => Some(PumpSwapInstruction::Other),
    }
}