carbon-lifinity-amm-v2-decoder 0.12.0

use crate::PROGRAM_ID;

use super::LifinityAmmV2Decoder;
pub mod deposit_all_token_types;
pub mod swap;
pub mod withdraw_all_token_types;

#[derive(
    carbon_core::InstructionType,
    serde::Serialize,
    serde::Deserialize,
    PartialEq,
    Eq,
    Debug,
    Clone,
    Hash,
)]
pub enum LifinityAmmV2Instruction {
    Swap(swap::Swap),
    DepositAllTokenTypes(deposit_all_token_types::DepositAllTokenTypes),
    WithdrawAllTokenTypes(withdraw_all_token_types::WithdrawAllTokenTypes),
}

impl carbon_core::instruction::InstructionDecoder<'_> for LifinityAmmV2Decoder {
    type InstructionType = LifinityAmmV2Instruction;

    fn decode_instruction(
        &self,
        instruction: &solana_instruction::Instruction,
    ) -> Option<carbon_core::instruction::DecodedInstruction<Self::InstructionType>> {
        if !instruction.program_id.eq(&PROGRAM_ID) {
            return None;
        }

        carbon_core::try_decode_instructions!(instruction,
            LifinityAmmV2Instruction::Swap => swap::Swap,
            LifinityAmmV2Instruction::DepositAllTokenTypes => deposit_all_token_types::DepositAllTokenTypes,
            LifinityAmmV2Instruction::WithdrawAllTokenTypes => withdraw_all_token_types::WithdrawAllTokenTypes,
        )
    }
}

#[cfg(test)]
mod tests {
    use carbon_core::{deserialize::ArrangeAccounts, instruction::InstructionDecoder};
    use solana_instruction::AccountMeta;

    use super::*;

    #[test]
    fn test_decode_swap() {
        let expected_ix = LifinityAmmV2Instruction::Swap(swap::Swap {
            amount_in: 99140,
            minimum_amount_out: 0,
        });
        let expected_accounts = vec![
            AccountMeta::new_readonly(
                solana_pubkey::Pubkey::from_str_const(
                    "Hc3DCcYyN7m2Wasf4Zhe8FMQ9qf9PiBiMJ9vqwSqB42h",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "EiEAydLqSKFqRPpuwYoVxEJ6h9UZh9tsTaHgs4f8b8Z5",
                ),
                false,
            ),
            AccountMeta::new_readonly(
                solana_pubkey::Pubkey::from_str_const(
                    "CapuXNQoDviLvU1PxFiizLgPNQCxrsag1uMeyk6zLVps",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "GQuvMWcBF1M2wgh2sbxkonq7FtBc6UNurtHjREMRAL1x",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "91bUbswo6Di8235jAPwim1At4cPZLbG2pkpneyqKg4NQ",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "D8F3PPxSuykAgyPPKwQdXDGGoRnUXzxowaheVJw5ATDC",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "GUicRosQyLJCYG8hjYcbiGKAVAmT1puQTVmJjFxJmdMK",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "2e6NAJy1qaKMq8PaswP2uzimMDvbr71Tbw38G6q9SNZ2",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "2EVZT2cFMvbqE9nSVidYVkrSouKfudcKG6R8AKiXoSY9",
                ),
                false,
            ),
            AccountMeta::new_readonly(
                solana_pubkey::Pubkey::from_str_const(
                    "TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA",
                ),
                false,
            ),
            AccountMeta::new_readonly(
                solana_pubkey::Pubkey::from_str_const(
                    "EPBJUVCmzvwkGPGcEuwKmXomfGt78Aozy6pj44x9xxDB",
                ),
                false,
            ),
            AccountMeta::new_readonly(
                solana_pubkey::Pubkey::from_str_const(
                    "EPBJUVCmzvwkGPGcEuwKmXomfGt78Aozy6pj44x9xxDB",
                ),
                false,
            ),
            AccountMeta::new_readonly(
                solana_pubkey::Pubkey::from_str_const(
                    "3ZDBff7jeQaksmGvmkRix36rU159EBDjYiPThvV8QVZM",
                ),
                false,
            ),
        ];
        let expected_arranged_accounts = swap::SwapInstructionAccounts {
            authority: solana_pubkey::Pubkey::from_str_const(
                "Hc3DCcYyN7m2Wasf4Zhe8FMQ9qf9PiBiMJ9vqwSqB42h",
            ),
            amm: solana_pubkey::Pubkey::from_str_const(
                "EiEAydLqSKFqRPpuwYoVxEJ6h9UZh9tsTaHgs4f8b8Z5",
            ),
            user_transfer_authority: solana_pubkey::Pubkey::from_str_const(
                "CapuXNQoDviLvU1PxFiizLgPNQCxrsag1uMeyk6zLVps",
            ),
            source_info: solana_pubkey::Pubkey::from_str_const(
                "GQuvMWcBF1M2wgh2sbxkonq7FtBc6UNurtHjREMRAL1x",
            ),
            destination_info: solana_pubkey::Pubkey::from_str_const(
                "91bUbswo6Di8235jAPwim1At4cPZLbG2pkpneyqKg4NQ",
            ),
            swap_source: solana_pubkey::Pubkey::from_str_const(
                "D8F3PPxSuykAgyPPKwQdXDGGoRnUXzxowaheVJw5ATDC",
            ),
            swap_destination: solana_pubkey::Pubkey::from_str_const(
                "GUicRosQyLJCYG8hjYcbiGKAVAmT1puQTVmJjFxJmdMK",
            ),
            pool_mint: solana_pubkey::Pubkey::from_str_const(
                "2e6NAJy1qaKMq8PaswP2uzimMDvbr71Tbw38G6q9SNZ2",
            ),
            fee_account: solana_pubkey::Pubkey::from_str_const(
                "2EVZT2cFMvbqE9nSVidYVkrSouKfudcKG6R8AKiXoSY9",
            ),
            token_program: solana_pubkey::Pubkey::from_str_const(
                "TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA",
            ),
            oracle_main_account: solana_pubkey::Pubkey::from_str_const(
                "EPBJUVCmzvwkGPGcEuwKmXomfGt78Aozy6pj44x9xxDB",
            ),
            oracle_sub_account: solana_pubkey::Pubkey::from_str_const(
                "EPBJUVCmzvwkGPGcEuwKmXomfGt78Aozy6pj44x9xxDB",
            ),
            oracle_pc_account: solana_pubkey::Pubkey::from_str_const(
                "3ZDBff7jeQaksmGvmkRix36rU159EBDjYiPThvV8QVZM",
            ),
        };

        let decoder = LifinityAmmV2Decoder;
        let instruction = carbon_test_utils::read_instruction("tests/fixtures/swap_ix.json")
            .expect("read fixture");
        let decoded = decoder
            .decode_instruction(&instruction)
            .expect("decode instruction");
        let decoded_arranged_accounts =
            swap::Swap::arrange_accounts(&instruction.accounts).expect("aranage accounts");

        assert_eq!(decoded.data, expected_ix);
        assert_eq!(decoded.program_id, PROGRAM_ID);
        assert_eq!(decoded.accounts, expected_accounts);
        assert_eq!(decoded_arranged_accounts, expected_arranged_accounts);
    }

    #[test]
    fn test_decode_deposit_all_token_types() {
        let expected_ix = LifinityAmmV2Instruction::DepositAllTokenTypes(
            deposit_all_token_types::DepositAllTokenTypes {
                pool_token_amount: 2457402,
                maximum_token_a_amount: 1000000,
                maximum_token_b_amount: 64575,
            },
        );
        let expected_accounts = vec![
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "4ruXyJT6rgHERQeVmMCc5pJDC5wgLujuMjTQgY562sdh",
                ),
                false,
            ),
            AccountMeta::new_readonly(
                solana_pubkey::Pubkey::from_str_const(
                    "2p4AA6xU2gxVvCVMVMAwpmyF49zBfp9MD4Ef8JeGVeMY",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "CbYf9QNrkVgNRCMTDiVdvzMqSzXh8AAgnrKAoTfEACdh",
                ),
                true,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "oVzCGsiinVTxPNmWFzi4QvGJ1iyhzUHj148Jc7SPuyj",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "74NpLshFLkuVyFKq2L3yBU2V8mcExSGy5E8tjCpMfVvU",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "7GawBqVSriYXQYCTr5XygeRNTeHamRWeHmVFiuf6wLfK",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "7oWW46LfAzrzWaZxV8CURBKpWwR8cc3ZpaXMFi6CH8sC",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "Dt1EBunetoSDPSEj2EB5m5bc93bt75ZB8Yy9vmaJ2ng",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "3hmRWv4Qik4rEzut8kqjxChN3n7Ms1MLKFKLqcoWkbGg",
                ),
                false,
            ),
            AccountMeta::new_readonly(
                solana_pubkey::Pubkey::from_str_const(
                    "TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA",
                ),
                false,
            ),
        ];
        let expected_arranged_accounts =
            deposit_all_token_types::DepositAllTokenTypesInstructionAccounts {
                amm: solana_pubkey::Pubkey::from_str_const(
                    "4ruXyJT6rgHERQeVmMCc5pJDC5wgLujuMjTQgY562sdh",
                ),
                authority: solana_pubkey::Pubkey::from_str_const(
                    "2p4AA6xU2gxVvCVMVMAwpmyF49zBfp9MD4Ef8JeGVeMY",
                ),
                user_transfer_authority_info: solana_pubkey::Pubkey::from_str_const(
                    "CbYf9QNrkVgNRCMTDiVdvzMqSzXh8AAgnrKAoTfEACdh",
                ),
                source_a_info: solana_pubkey::Pubkey::from_str_const(
                    "oVzCGsiinVTxPNmWFzi4QvGJ1iyhzUHj148Jc7SPuyj",
                ),
                source_b_info: solana_pubkey::Pubkey::from_str_const(
                    "74NpLshFLkuVyFKq2L3yBU2V8mcExSGy5E8tjCpMfVvU",
                ),
                token_a: solana_pubkey::Pubkey::from_str_const(
                    "7GawBqVSriYXQYCTr5XygeRNTeHamRWeHmVFiuf6wLfK",
                ),
                token_b: solana_pubkey::Pubkey::from_str_const(
                    "7oWW46LfAzrzWaZxV8CURBKpWwR8cc3ZpaXMFi6CH8sC",
                ),
                pool_mint: solana_pubkey::Pubkey::from_str_const(
                    "Dt1EBunetoSDPSEj2EB5m5bc93bt75ZB8Yy9vmaJ2ng",
                ),
                destination: solana_pubkey::Pubkey::from_str_const(
                    "3hmRWv4Qik4rEzut8kqjxChN3n7Ms1MLKFKLqcoWkbGg",
                ),
                token_program: solana_pubkey::Pubkey::from_str_const(
                    "TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA",
                ),
            };

        let decoder = LifinityAmmV2Decoder;
        let instruction =
            carbon_test_utils::read_instruction("tests/fixtures/deposit_all_token_types_ix.json")
                .expect("read fixture");
        let decoded = decoder
            .decode_instruction(&instruction)
            .expect("decode instruction");
        let decoded_arranged_accounts =
            deposit_all_token_types::DepositAllTokenTypes::arrange_accounts(&instruction.accounts)
                .expect("arrange accounts");

        assert_eq!(decoded.data, expected_ix);
        assert_eq!(decoded.program_id, PROGRAM_ID);
        assert_eq!(decoded.accounts, expected_accounts);
        assert_eq!(decoded_arranged_accounts, expected_arranged_accounts);
    }

    #[test]
    fn test_decode_withdraw_all_token_types() {
        let expected_ix = LifinityAmmV2Instruction::WithdrawAllTokenTypes(
            withdraw_all_token_types::WithdrawAllTokenTypes {
                pool_token_amount: 2000655039581,
                minimum_token_a_amount: 1076172871830,
                minimum_token_b_amount: 752578501683,
            },
        );
        let expected_accounts = vec![
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "DrRd8gYMJu9XGxLhwTCPdHNLXCKHsxJtMpbn62YqmwQe",
                ),
                false,
            ),
            AccountMeta::new_readonly(
                solana_pubkey::Pubkey::from_str_const(
                    "7GmDCbu7bYiWJvFaNUyPNiM8PjvvBcmyBcZY1qSsAGi2",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "71hhezkHQ2dhmPySsHVCCkLggfWzPFEBdfEjbn4NCXMG",
                ),
                true,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "A9x5SwN9vbg3YDxZySDWaHuVoQQ8YSJDrmjFZ3yMksqW",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "EVGW4q1iFjDmtxtHr3NoPi5iVKAxwEjohsusMrinDxr6",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "53EkU98Vbv2TQPwGG6t2asCynzFjCX5AnvaabbXafaed",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "FGYgFJSxZTGzaLwzUL9YZqK2yUZ8seofCwGq8BPEw4o8",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "Gf5ucgWGfJ8NLvVaKJVen9CYsPFRK4eAsZWT8zDjARkA",
                ),
                false,
            ),
            AccountMeta::new(
                solana_pubkey::Pubkey::from_str_const(
                    "8vPGw7d6a4rmejVg2uXMs8kvEnVy7WgN1oNk9mRpdfGP",
                ),
                false,
            ),
            AccountMeta::new_readonly(
                solana_pubkey::Pubkey::from_str_const(
                    "TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA",
                ),
                false,
            ),
        ];
        let expected_arranged_accounts =
            withdraw_all_token_types::WithdrawAllTokenTypesInstructionAccounts {
                amm: solana_pubkey::Pubkey::from_str_const(
                    "DrRd8gYMJu9XGxLhwTCPdHNLXCKHsxJtMpbn62YqmwQe",
                ),
                authority: solana_pubkey::Pubkey::from_str_const(
                    "7GmDCbu7bYiWJvFaNUyPNiM8PjvvBcmyBcZY1qSsAGi2",
                ),
                user_transfer_authority_info: solana_pubkey::Pubkey::from_str_const(
                    "71hhezkHQ2dhmPySsHVCCkLggfWzPFEBdfEjbn4NCXMG",
                ),
                source_info: solana_pubkey::Pubkey::from_str_const(
                    "A9x5SwN9vbg3YDxZySDWaHuVoQQ8YSJDrmjFZ3yMksqW",
                ),
                token_a: solana_pubkey::Pubkey::from_str_const(
                    "EVGW4q1iFjDmtxtHr3NoPi5iVKAxwEjohsusMrinDxr6",
                ),
                token_b: solana_pubkey::Pubkey::from_str_const(
                    "53EkU98Vbv2TQPwGG6t2asCynzFjCX5AnvaabbXafaed",
                ),
                pool_mint: solana_pubkey::Pubkey::from_str_const(
                    "FGYgFJSxZTGzaLwzUL9YZqK2yUZ8seofCwGq8BPEw4o8",
                ),
                dest_token_a_info: solana_pubkey::Pubkey::from_str_const(
                    "Gf5ucgWGfJ8NLvVaKJVen9CYsPFRK4eAsZWT8zDjARkA",
                ),
                dest_token_b_info: solana_pubkey::Pubkey::from_str_const(
                    "8vPGw7d6a4rmejVg2uXMs8kvEnVy7WgN1oNk9mRpdfGP",
                ),
                token_program: solana_pubkey::Pubkey::from_str_const(
                    "TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA",
                ),
            };

        let decoder = LifinityAmmV2Decoder;
        let instruction =
            carbon_test_utils::read_instruction("tests/fixtures/withdraw_all_token_types_ix.json")
                .expect("read fixture");
        let decoded = decoder
            .decode_instruction(&instruction)
            .expect("decode instruction");
        let decoded_arranged_accounts =
            withdraw_all_token_types::WithdrawAllTokenTypes::arrange_accounts(
                &instruction.accounts,
            )
            .expect("arrange accounts");

        assert_eq!(decoded.data, expected_ix);
        assert_eq!(decoded.program_id, PROGRAM_ID);
        assert_eq!(decoded.accounts, expected_accounts);
        assert_eq!(decoded_arranged_accounts, expected_arranged_accounts);
    }
}