crema-cli 0.1.0

Blockchain, Crema for Solana
Documentation
use std::str::FromStr;
use std::sync::Arc;

use clap::ArgMatches;
use solana_clap_utils::keypair::DefaultSigner;
use solana_client::rpc_client::RpcClient;
use solana_program::pubkey::Pubkey;
use solana_remote_wallet::remote_wallet::RemoteWalletManager;
use spl_associated_token_account::get_associated_token_address;

use crate::check_and_update_err;
use crate::command::{CliCommand, CliCommandInfo, CliConfig, CliError, ProcessResult};
use crate::contract::instructions::collect_partner_fee::new_collect_partner_fee;
use crate::contract::state::clmmpools::Clmmpool;
use crate::utils::send::send_tx;

pub fn parse_collect_partner_fee<'a>(matches: &'a ArgMatches, default_signer: &DefaultSigner, wallet_manager: &mut Option<Arc<RemoteWalletManager>>) -> Result<CliCommandInfo<'a>, CliError> {
    let partner = matches.value_of("partner");
    let clmmpool = matches.value_of("clmmpool");

    Ok(CliCommandInfo {
        command: CliCommand::PairCollectPartnerFee {
            partner: Pubkey::from_str(partner.unwrap()).unwrap(),
            clmmpool: Pubkey::from_str(clmmpool.unwrap()).unwrap(),
        },
        signers: vec![check_and_update_err!(default_signer.signer_from_path(matches, wallet_manager), CliError::RpcRequestError("owner key is invalid".to_string()))?],
    })
}

pub fn process_collect_partner_fee(
    rpc_client: &RpcClient,
    config: &CliConfig,
    partner: &Pubkey,
    clmmpool: &Pubkey,
) -> ProcessResult {
    let clmmpool_info = Clmmpool::get_info(rpc_client, clmmpool);
    let token_a_ata = get_associated_token_address(&config.pubkey().unwrap(), &clmmpool_info.token_a);
    let token_b_ata = get_associated_token_address(&config.pubkey().unwrap(), &clmmpool_info.token_b);
    let token_a_partner_fee_vault = get_associated_token_address(&partner, &clmmpool_info.token_a);
    let token_b_partner_fee_vault = get_associated_token_address(&partner, &clmmpool_info.token_b);

    let ixs = [
        new_collect_partner_fee(
            partner,
            clmmpool,
            &token_a_ata,
            &token_b_ata,
            &token_a_partner_fee_vault,
            &token_b_partner_fee_vault,
            config.pubkey().unwrap(),
        )
    ];

    let res = send_tx(rpc_client, config, &ixs)?;

    Ok("signers : ".to_owned() + res.to_string().as_str())
}