feat: add shared library and use shared library in zap-program

Author: bangyro

Cargo.lock

@@ -1,5 +1,5 @@
 [workspace]
-members = ["programs/*", "libs/*"]
+members = ["programs/*", "libs/*", "protocol-zap"]
 resolver = "2"
 
 [profile.release]
@@ -15,3 +15,4 @@ codegen-units = 1
 anchor-lang = "0.31.1"
 anchor-spl = "0.31.1"
 bytemuck = "1.20.0"
+ruint = "1.3.0"

Cargo.toml

@@ -26,4 +26,5 @@ dlmm = { path = "../../libs/dlmm" }
 damm-v2 = { git = "https://github.com/MeteoraAg/damm-v2", features = [
     "cpi",
 ], rev = "d9cef5aaec9cfa1a6d3b0b28d213c87c285127de", package = "cp-amm" }
-ruint = "1.3.0"
+ruint = { workspace = true }
+protocol-zap = { path = "../../protocol-zap" }

programs/zap/Cargo.toml

@@ -1,25 +1,10 @@
 use anchor_lang::constant;
-use anchor_lang::{prelude::Pubkey, pubkey};
+use anchor_lang::prelude::Pubkey;
 
-#[constant]
-pub const DAMM_V2: Pubkey = pubkey!("cpamdpZCGKUy5JxQXB4dcpGPiikHawvSWAd6mEn1sGG");
-// https://github.com/MeteoraAg/zap-program/blob/main/idls/damm_v2.json#L3512-L3521
-#[constant]
-pub const DAMM_V2_SWAP_DISC: [u8; 8] = [248, 198, 158, 145, 225, 117, 135, 200];
-
-#[constant]
-pub const JUP_V6: Pubkey = pubkey!("JUP6LkbZbjS1jKKwapdHNy74zcZ3tLUZoi5QNyVTaV4");
-// https://github.com/MeteoraAg/zap-program/blob/main/idls/jup_v6.json#L14-L23
-#[constant]
-pub const JUP_V6_ROUTE_DISC: [u8; 8] = [229, 23, 203, 151, 122, 227, 173, 42];
-// https://github.com/MeteoraAg/zap-program/blob/main/idls/jup_v6.json#L257-L266
-#[constant]
-pub const JUP_V6_SHARED_ACCOUNT_ROUTE_DISC: [u8; 8] = [193, 32, 155, 51, 65, 214, 156, 129];
-
-pub const DLMM: Pubkey = pubkey!("LBUZKhRxPF3XUpBCjp4YzTKgLccjZhTSDM9YuVaPwxo");
-// https://github.com/MeteoraAg/zap-program/blob/main/idls/dlmm.json#L3413-L3422
-#[constant]
-pub const DLMM_SWAP2_DISC: [u8; 8] = [65, 75, 63, 76, 235, 91, 91, 136];
+use protocol_zap::constants::{
+    DAMM_V2, DAMM_V2_SWAP_DISC, DLMM, DLMM_SWAP2_DISC, JUP_V6, JUP_V6_ROUTE_DISC,
+    JUP_V6_SHARED_ACCOUNT_ROUTE_DISC,
+};
 
 #[constant]
 pub const WHITELISTED_AMM_PROGRAMS: [(Pubkey, [u8; 8]); 4] = [

programs/zap/src/constants.rs

@@ -1,46 +1,12 @@
-use std::cmp::min;
-
 use anchor_lang::{
     prelude::*,
     solana_program::{instruction::Instruction, program::invoke},
 };
 use anchor_spl::token_interface::TokenAccount;
+use protocol_zap::ZapOutParameters;
 
 use crate::{constants::WHITELISTED_AMM_PROGRAMS, error::ZapError, safe_math::SafeMath};
 
-#[derive(AnchorSerialize, AnchorDeserialize)]
-pub struct ZapOutParameters {
-    pub percentage: u8,
-    pub offset_amount_in: u16,
-    pub pre_user_token_balance: u64,
-    pub max_swap_amount: u64, // avoid the issue someone send token to user token account when user zap out
-    pub payload_data: Vec<u8>,
-}
-
-impl ZapOutParameters {
-    fn validate(&self) -> Result<()> {
-        require!(
-            self.percentage <= 100 && self.percentage > 0,
-            ZapError::InvalidZapOutParameters
-        );
-
-        Ok(())
-    }
-
-    fn get_swap_amount(&self, balance_change_amount: u64) -> Result<u64> {
-        let swap_amount = if self.percentage == 100 {
-            balance_change_amount
-        } else {
-            let amount = u128::from(balance_change_amount)
-                .safe_mul(self.percentage.into())?
-                .safe_div(100)?;
-            u64::try_from(amount).map_err(|_| ZapError::TypeCastFailed)?
-        };
-
-        Ok(min(swap_amount, self.max_swap_amount))
-    }
-}
-
 pub fn is_support_amm_program(amm_program: &Pubkey, discriminator: &[u8]) -> bool {
     WHITELISTED_AMM_PROGRAMS
         .iter()

programs/zap/src/instructions/ix_zap_out.rs

@@ -16,6 +16,7 @@ pub mod utils;
 use dlmm::types::RemainingAccountsInfo;
 pub use utils::*;
 declare_id!("zapvX9M3uf5pvy4wRPAbQgdQsM1xmuiFnkfHKPvwMiz");
+use protocol_zap::ZapOutParameters;
 
 #[program]
 pub mod zap {

programs/zap/src/lib.rs

@@ -0,0 +1,11 @@
+[package]
+name = "protocol-zap"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+anchor-lang = { workspace = true }
+anchor-spl = { workspace = true }
+const-crypto = "0.3.0"
+jupiter = { path = "../libs/jupiter" }
+ruint = { workspace = true }

protocol-zap/Cargo.toml

@@ -0,0 +1,71 @@
+use anchor_lang::prelude::*;
+use anchor_spl::associated_token::ID as ASSOCIATED_TOKEN_PROGRAM_ID;
+use anchor_spl::token::ID as TOKEN_PROGRAM_ID;
+use const_crypto::ed25519;
+
+pub const JUP_V6_SHARED_ACCOUNT_ROUTE_AMOUNT_IN_REVERSE_OFFSET: usize = 1 + 2 + 8 + 8; // Due to jupiter parameters have dynamic length type (vec), we have to do parameters_data.length - JUP_V6_SHARED_ACCOUNT_ROUTE_AMOUNT_IN_REVERSE_OFFSET
+pub const JUP_V6_SHARED_ACCOUNT_ROUTE_SOURCE_ACCOUNT_INDEX: usize = 3;
+pub const JUP_V6_SHARED_ACCOUNT_ROUTE_DESTINATION_ACCOUNT_INDEX: usize = 6;
+
+pub const JUP_V6_ROUTE_AMOUNT_IN_REVERSE_OFFSET: usize = 1 + 2 + 8 + 8;
+pub const JUP_V6_ROUTE_SOURCE_ACCOUNT_INDEX: usize = 2;
+pub const JUP_V6_ROUTE_DESTINATION_ACCOUNT_INDEX: usize = 4;
+
+pub const DLMM_SWAP2_AMOUNT_IN_OFFSET: u16 = 8;
+pub const DLMM_SWAP2_SOURCE_ACCOUNT_INDEX: usize = 4;
+pub const DLMM_SWAP2_DESTINATION_ACCOUNT_INDEX: usize = 5;
+
+pub const DAMM_V2_SWAP_AMOUNT_IN_OFFSET: u16 = 8;
+pub const DAMM_V2_SWAP_SOURCE_ACCOUNT_INDEX: usize = 2;
+pub const DAMM_V2_SWAP_DESTINATION_ACCOUNT_INDEX: usize = 3;
+
+pub const JUP_V6: Pubkey = pubkey!("JUP6LkbZbjS1jKKwapdHNy74zcZ3tLUZoi5QNyVTaV4");
+// https://github.com/MeteoraAg/zap-program/blob/4b67bfc64e5a023a1b74386be8b82c3908934a0b/idls/jupiter.json#L161
+pub const JUP_V6_ROUTE_DISC: [u8; 8] = [229, 23, 203, 151, 122, 227, 173, 42];
+// https://github.com/MeteoraAg/zap-program/blob/4b67bfc64e5a023a1b74386be8b82c3908934a0b/idls/jupiter.json#L270
+pub const JUP_V6_SHARED_ACCOUNT_ROUTE_DISC: [u8; 8] = [193, 32, 155, 51, 65, 214, 156, 129];
+
+pub const DLMM: Pubkey = pubkey!("LBUZKhRxPF3XUpBCjp4YzTKgLccjZhTSDM9YuVaPwxo");
+// https://github.com/MeteoraAg/zap-program/blob/4b67bfc64e5a023a1b74386be8b82c3908934a0b/idls/dlmm.json#L5242-L5251
+pub const DLMM_SWAP2_DISC: [u8; 8] = [65, 75, 63, 76, 235, 91, 91, 136];
+
+pub const DAMM_V2: Pubkey = pubkey!("cpamdpZCGKUy5JxQXB4dcpGPiikHawvSWAd6mEn1sGG");
+// https://github.com/MeteoraAg/zap-program/blob/4b67bfc64e5a023a1b74386be8b82c3908934a0b/idls/damm_v2.json#L2154
+pub const DAMM_V2_SWAP_DISC: [u8; 8] = [248, 198, 158, 145, 225, 117, 135, 200];
+
+pub const ZAP: Pubkey = pubkey!("zapvX9M3uf5pvy4wRPAbQgdQsM1xmuiFnkfHKPvwMiz");
+pub const ZAP_OUT_DISC: [u8; 8] = [155, 108, 185, 112, 104, 210, 161, 64];
+
+pub const USDC_ADDRESS: Pubkey =
+    Pubkey::from_str_const("EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v");
+
+pub const SOL_ADDRESS: Pubkey =
+    Pubkey::from_str_const("So11111111111111111111111111111111111111112");
+
+pub const MINTS_DISALLOWED_TO_ZAP_OUT: [Pubkey; 2] = [USDC_ADDRESS, SOL_ADDRESS];
+
+pub const TREASURY: Pubkey = pubkey!("6aYhxiNGmG8AyU25rh2R7iFu4pBrqnQHpNUGhmsEXRcm");
+
+pub const TREASURY_USDC_ADDRESS: Pubkey = Pubkey::new_from_array(
+    ed25519::derive_program_address(
+        &[
+            &TREASURY.to_bytes(),
+            &TOKEN_PROGRAM_ID.to_bytes(),
+            &USDC_ADDRESS.to_bytes(),
+        ],
+        &ASSOCIATED_TOKEN_PROGRAM_ID.to_bytes(),
+    )
+    .0,
+);
+
+pub const TREASURY_SOL_ADDRESS: Pubkey = Pubkey::new_from_array(
+    ed25519::derive_program_address(
+        &[
+            &TREASURY.to_bytes(),
+            &TOKEN_PROGRAM_ID.to_bytes(),
+            &SOL_ADDRESS.to_bytes(),
+        ],
+        &ASSOCIATED_TOKEN_PROGRAM_ID.to_bytes(),
+    )
+    .0,
+);

protocol-zap/src/constants.rs

@@ -0,0 +1,27 @@
+use crate::{
+    constants::{
+        DAMM_V2_SWAP_AMOUNT_IN_OFFSET, DAMM_V2_SWAP_DESTINATION_ACCOUNT_INDEX,
+        DAMM_V2_SWAP_SOURCE_ACCOUNT_INDEX,
+    },
+    RawZapOutAmmInfo, ZapInfoProcessor, ZapOutParameters,
+};
+use anchor_lang::prelude::*;
+
+pub struct ZapDammV2InfoProcessor;
+
+impl ZapInfoProcessor for ZapDammV2InfoProcessor {
+    fn validate_payload(&self, _payload: &[u8]) -> Result<()> {
+        Ok(())
+    }
+
+    fn extract_raw_zap_out_amm_info(
+        &self,
+        _zap_params: &ZapOutParameters,
+    ) -> Result<RawZapOutAmmInfo> {
+        Ok(RawZapOutAmmInfo {
+            source_index: DAMM_V2_SWAP_SOURCE_ACCOUNT_INDEX,
+            destination_index: DAMM_V2_SWAP_DESTINATION_ACCOUNT_INDEX,
+            amount_in_offset: DAMM_V2_SWAP_AMOUNT_IN_OFFSET,
+        })
+    }
+}

protocol-zap/src/damm_v2_zap.rs

@@ -0,0 +1,27 @@
+use crate::{
+    constants::{
+        DLMM_SWAP2_AMOUNT_IN_OFFSET, DLMM_SWAP2_DESTINATION_ACCOUNT_INDEX,
+        DLMM_SWAP2_SOURCE_ACCOUNT_INDEX,
+    },
+    RawZapOutAmmInfo, ZapInfoProcessor, ZapOutParameters,
+};
+use anchor_lang::prelude::*;
+
+pub struct ZapDlmmInfoProcessor;
+
+impl ZapInfoProcessor for ZapDlmmInfoProcessor {
+    fn validate_payload(&self, _payload: &[u8]) -> Result<()> {
+        Ok(())
+    }
+
+    fn extract_raw_zap_out_amm_info(
+        &self,
+        _zap_params: &ZapOutParameters,
+    ) -> Result<RawZapOutAmmInfo> {
+        Ok(RawZapOutAmmInfo {
+            source_index: DLMM_SWAP2_SOURCE_ACCOUNT_INDEX,
+            destination_index: DLMM_SWAP2_DESTINATION_ACCOUNT_INDEX,
+            amount_in_offset: DLMM_SWAP2_AMOUNT_IN_OFFSET,
+        })
+    }
+}