derive-rs 0.1.11

use std::collections::HashMap;

use alloy::primitives::TxHash;

use crate::{
    actions::{
        ActionData, DepositArgs, DepositManager, ModuleType, SpotTransferArgs, SpotTransferData,
        TransferPositionsArgs, WithdrawArgs, WithdrawData,
    },
    models::{
        Direction, Instrument, PrivateTransferSpotResponse, PrivateWithdrawResponse,
        TransferPositionsResponse,
    },
    types::ClientError,
    ws_client::WsClient,
};

impl Direction {
    pub fn opposite(&self) -> Self {
        match self {
            Direction::Buy => Direction::Sell,
            Direction::Sell => Direction::Buy,
        }
    }
}
pub struct FundMovementsNamespace<'a> {
    pub ws_client: &'a WsClient,
}
impl<'a> FundMovementsNamespace<'a> {
    pub async fn withdraw(
        &self,
        withdraw_args: WithdrawArgs,
    ) -> Result<PrivateWithdrawResponse, ClientError> {
        let subaccount_id = self.ws_client.subaccount_id.unwrap();
        let signer = self.ws_client.wallet.clone().unwrap();

        let wallet = self.ws_client.derive_wallet.clone().unwrap();
        let env = &self.ws_client.environment;

        let erc20_details = self.ws_client
            .erc20_cache
            .get(&withdraw_args.asset)
            .expect("ERC20 asset details not found in cache. Please ensure the asset is supported and cached.")
            .clone();

        let data = WithdrawData::from_args(withdraw_args.clone(), erc20_details.clone())?;
        let action = ActionData::new(
            data,
            subaccount_id,
            signer.address(),
            &wallet.parse().expect("Couldnt parse wallet address"),
            &self.ws_client.environment,
            ModuleType::Withdraw,
        )?;

        let params =
            action.populate_withdraw_params(&signer, withdraw_args.clone(), env, subaccount_id)?;

        self.ws_client
            .rpc()
            .transfers_withdrawals()
            .withdraw(params)
            .await
        // Ok(SignableRequest::new())
    }

    pub async fn deposit(&self, deposit_args: DepositArgs) -> Result<Vec<TxHash>, ClientError> {
        let env = &self.ws_client.environment;
        let private_key = self
            .ws_client
            .wallet
            .clone()
            .expect("Must have set wallet to deposit");
        let wallet = self
            .ws_client
            .derive_wallet
            .clone()
            .expect("Must have set smart contract wallet address to deposit");
        let erc20_cache = self.ws_client.erc20_cache.clone();
        let deposit_manager =
            DepositManager::new(&deposit_args, &private_key, &wallet, env, erc20_cache)?;
        let hashes = deposit_manager.deposit().await?;
        Ok(hashes)
    }
    pub async fn transfer_spot(
        &self,
        args: SpotTransferArgs,
    ) -> Result<PrivateTransferSpotResponse, ClientError> {
        let signer = self.ws_client.wallet.clone().unwrap();

        let wallet = self.ws_client.derive_wallet.clone().unwrap();
        let env = &self.ws_client.environment;

        let erc20_details = self.ws_client
            .erc20_cache
            .get(&args.asset)
            .expect("ERC20 asset details not found in cache. Please ensure the asset is supported and cached.")
            .clone();

        let asset = self.ws_client
            .assets_cache
            .get(&args.asset)
            .expect("Asset details not found in cache. Please ensure the asset is supported and cached.")
            .clone();

        let data = SpotTransferData::from_args(args.clone(), erc20_details.clone(), asset.clone())?;
        let action = ActionData::new(
            data,
            args.subaccount_id,
            signer.address(),
            &wallet.parse().expect("Couldnt parse wallet address"),
            &self.ws_client.environment,
            ModuleType::SpotTransfer,
        )?;

        let params = action.populate_transfer_spot_params(&signer, args.clone(), env, &asset)?;

        println!("{}", serde_json::to_string_pretty(&params).unwrap());

        self.ws_client
            .rpc()
            .transfers_withdrawals()
            .transfer_spot(params)
            .await
    }

    pub async fn transfer_positions(
        &self,
        args: TransferPositionsArgs,
    ) -> Result<TransferPositionsResponse, ClientError> {
        let signer = self.ws_client.wallet.clone().unwrap();

        let env = &self.ws_client.environment;

        let required_instruments: Vec<String> = args
            .legs
            .iter()
            .map(|leg| leg.instrument_name.clone())
            .collect();

        let instruments: HashMap<String, Instrument> = self
            .ws_client
            .instruments_cache
            .iter()
            .filter_map(|entry| {
                let instrument = entry.value();
                if required_instruments.contains(&instrument.instrument_name) {
                    Some((instrument.instrument_name.clone(), instrument.clone()))
                } else {
                    None
                }
            })
            .collect::<HashMap<String, Instrument>>();

        let params = ActionData::populate_transfer_positions(
            &signer,
            args.clone(),
            env,
            &self.ws_client.derive_wallet.clone().unwrap(),
            &instruments,
        )?;

        self.ws_client
            .rpc()
            .transfers_withdrawals()
            .transfer_positions(params)
            .await
    }
}