add nextblock, 0slot support

src/common/types.rs

@@ -0,0 +1,93 @@
+use std::sync::Arc;
+
+use solana_client::rpc_client::RpcClient;
+use solana_sdk::{commitment_config::CommitmentConfig, signature::Keypair};
+use serde::Deserialize;
+use crate::{constants::trade::{DEFAULT_BUY_TIP_FEE, DEFAULT_COMPUTE_UNIT_LIMIT, DEFAULT_COMPUTE_UNIT_PRICE, DEFAULT_SELL_TIP_FEE}, jito::FeeClient};
+
+#[derive(Debug, Clone, PartialEq)]
+pub enum FeeType {
+    Jito,
+    NextBlock,
+}
+
+#[derive(Debug, Clone)]
+pub struct Cluster {
+    pub rpc_url: String,
+    pub block_engine_url: String,
+    pub nextblock_url: String,
+    pub nextblock_auth_token: String,
+    pub zeroslot_url: String,
+    pub zeroslot_auth_token: String,
+    pub use_jito: bool,
+    pub use_nextblock: bool,
+    pub use_zeroslot: bool,
+    pub priority_fee: PriorityFee,
+    pub commitment: CommitmentConfig,
+}
+
+impl Cluster {
+    pub fn new(
+        rpc_url: String, 
+        block_engine_url: 
+        String, nextblock_url: 
+        String, nextblock_auth_token: 
+        String, zeroslot_url: String, 
+        zeroslot_auth_token: String, 
+        priority_fee: PriorityFee, 
+        commitment: CommitmentConfig, 
+        use_jito: bool, 
+        use_nextblock: bool, 
+        use_zeroslot: bool
+    ) -> Self {
+        Self { 
+            rpc_url, 
+            block_engine_url, 
+            nextblock_url, 
+            nextblock_auth_token, 
+            zeroslot_url, 
+            zeroslot_auth_token, 
+            priority_fee, 
+            commitment, 
+            use_jito, 
+            use_nextblock, 
+            use_zeroslot 
+        }
+    }
+}
+
+#[derive(Debug, Deserialize, Clone, Copy, PartialEq)]
+
+pub struct PriorityFee {
+    pub unit_limit: u32,
+    pub unit_price: u64,
+    pub buy_tip_fee: f64,
+    pub sell_tip_fee: f64,
+}
+
+impl Default for PriorityFee {
+    fn default() -> Self {
+        Self { 
+            unit_limit: DEFAULT_COMPUTE_UNIT_LIMIT, 
+            unit_price: DEFAULT_COMPUTE_UNIT_PRICE, 
+            buy_tip_fee: DEFAULT_BUY_TIP_FEE, 
+            sell_tip_fee: DEFAULT_SELL_TIP_FEE 
+        }
+    }
+}
+
+pub type SolanaRpcClient = solana_client::nonblocking::rpc_client::RpcClient;
+
+pub struct MethodArgs {
+    pub payer: Arc<Keypair>,
+    pub rpc: Arc<RpcClient>,
+    pub nonblocking_rpc: Arc<SolanaRpcClient>,
+    pub jito_client: Arc<FeeClient>,
+}
+
+impl MethodArgs {
+    pub fn new(payer: Arc<Keypair>, rpc: Arc<RpcClient>, nonblocking_rpc: Arc<SolanaRpcClient>, jito_client: Arc<FeeClient>) -> Self {
+        Self { payer, rpc, nonblocking_rpc, jito_client }
+    }
+}
+