//! This code was AUTOGENERATED using the codama library.
//! Please DO NOT EDIT THIS FILE, instead use visitors
//! to add features, then rerun codama to update it.
//!
//! <https://github.com/codama-idl/codama>
//!

use {
    crate::generated::types::{Epoch, UnixTimestamp},
    borsh::{BorshDeserialize, BorshSerialize},
    solana_pubkey::Pubkey,
};

pub const SET_LOCKUP_DISCRIMINATOR: u32 = 6;

/// Accounts.
#[derive(Debug)]
pub struct SetLockup {
    /// Initialized stake account
    pub stake: solana_pubkey::Pubkey,
    /// Lockup authority or withdraw authority
    pub authority: solana_pubkey::Pubkey,
}

impl SetLockup {
    pub fn instruction(&self, args: SetLockupInstructionArgs) -> solana_instruction::Instruction {
        self.instruction_with_remaining_accounts(args, &[])
    }
    #[allow(clippy::arithmetic_side_effects)]
    #[allow(clippy::vec_init_then_push)]
    pub fn instruction_with_remaining_accounts(
        &self,
        args: SetLockupInstructionArgs,
        remaining_accounts: &[solana_instruction::AccountMeta],
    ) -> solana_instruction::Instruction {
        let mut accounts = Vec::with_capacity(2 + remaining_accounts.len());
        accounts.push(solana_instruction::AccountMeta::new(self.stake, false));
        accounts.push(solana_instruction::AccountMeta::new_readonly(
            self.authority,
            true,
        ));
        accounts.extend_from_slice(remaining_accounts);
        let mut data = SetLockupInstructionData::new().try_to_vec().unwrap();
        let mut args = args.try_to_vec().unwrap();
        data.append(&mut args);

        solana_instruction::Instruction {
            program_id: crate::SPHERENET_STAKE_INTERFACE_ID,
            accounts,
            data,
        }
    }
}

#[derive(BorshSerialize, BorshDeserialize, Clone, Debug, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct SetLockupInstructionData {
    discriminator: u32,
}

impl SetLockupInstructionData {
    pub fn new() -> Self {
        Self { discriminator: 6 }
    }

    pub(crate) fn try_to_vec(&self) -> Result<Vec<u8>, std::io::Error> {
        borsh::to_vec(self)
    }
}

impl Default for SetLockupInstructionData {
    fn default() -> Self {
        Self::new()
    }
}

#[derive(BorshSerialize, BorshDeserialize, Clone, Debug, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct SetLockupInstructionArgs {
    pub unix_timestamp: Option<UnixTimestamp>,
    pub epoch: Option<Epoch>,
    pub custodian: Option<Pubkey>,
}

impl SetLockupInstructionArgs {
    pub(crate) fn try_to_vec(&self) -> Result<Vec<u8>, std::io::Error> {
        borsh::to_vec(self)
    }
}

/// Instruction builder for `SetLockup`.
///
/// ### Accounts:
///
///   0. `[writable]` stake
///   1. `[signer]` authority
#[derive(Clone, Debug, Default)]
pub struct SetLockupBuilder {
    stake: Option<solana_pubkey::Pubkey>,
    authority: Option<solana_pubkey::Pubkey>,
    unix_timestamp: Option<UnixTimestamp>,
    epoch: Option<Epoch>,
    custodian: Option<Pubkey>,
    __remaining_accounts: Vec<solana_instruction::AccountMeta>,
}

impl SetLockupBuilder {
    pub fn new() -> Self {
        Self::default()
    }
    /// Initialized stake account
    #[inline(always)]
    pub fn stake(&mut self, stake: solana_pubkey::Pubkey) -> &mut Self {
        self.stake = Some(stake);
        self
    }
    /// Lockup authority or withdraw authority
    #[inline(always)]
    pub fn authority(&mut self, authority: solana_pubkey::Pubkey) -> &mut Self {
        self.authority = Some(authority);
        self
    }
    /// `[optional argument]`
    #[inline(always)]
    pub fn unix_timestamp(&mut self, unix_timestamp: UnixTimestamp) -> &mut Self {
        self.unix_timestamp = Some(unix_timestamp);
        self
    }
    /// `[optional argument]`
    #[inline(always)]
    pub fn epoch(&mut self, epoch: Epoch) -> &mut Self {
        self.epoch = Some(epoch);
        self
    }
    /// `[optional argument]`
    #[inline(always)]
    pub fn custodian(&mut self, custodian: Pubkey) -> &mut Self {
        self.custodian = Some(custodian);
        self
    }
    /// Add an additional account to the instruction.
    #[inline(always)]
    pub fn add_remaining_account(&mut self, account: solana_instruction::AccountMeta) -> &mut Self {
        self.__remaining_accounts.push(account);
        self
    }
    /// Add additional accounts to the instruction.
    #[inline(always)]
    pub fn add_remaining_accounts(
        &mut self,
        accounts: &[solana_instruction::AccountMeta],
    ) -> &mut Self {
        self.__remaining_accounts.extend_from_slice(accounts);
        self
    }
    #[allow(clippy::clone_on_copy)]
    pub fn instruction(&self) -> solana_instruction::Instruction {
        let accounts = SetLockup {
            stake: self.stake.expect("stake is not set"),
            authority: self.authority.expect("authority is not set"),
        };
        let args = SetLockupInstructionArgs {
            unix_timestamp: self.unix_timestamp.clone(),
            epoch: self.epoch.clone(),
            custodian: self.custodian.clone(),
        };

        accounts.instruction_with_remaining_accounts(args, &self.__remaining_accounts)
    }
}

/// `set_lockup` CPI accounts.
pub struct SetLockupCpiAccounts<'a, 'b> {
    /// Initialized stake account
    pub stake: &'b solana_account_info::AccountInfo<'a>,
    /// Lockup authority or withdraw authority
    pub authority: &'b solana_account_info::AccountInfo<'a>,
}

/// `set_lockup` CPI instruction.
pub struct SetLockupCpi<'a, 'b> {
    /// The program to invoke.
    pub __program: &'b solana_account_info::AccountInfo<'a>,
    /// Initialized stake account
    pub stake: &'b solana_account_info::AccountInfo<'a>,
    /// Lockup authority or withdraw authority
    pub authority: &'b solana_account_info::AccountInfo<'a>,
    /// The arguments for the instruction.
    pub __args: SetLockupInstructionArgs,
}

impl<'a, 'b> SetLockupCpi<'a, 'b> {
    pub fn new(
        program: &'b solana_account_info::AccountInfo<'a>,
        accounts: SetLockupCpiAccounts<'a, 'b>,
        args: SetLockupInstructionArgs,
    ) -> Self {
        Self {
            __program: program,
            stake: accounts.stake,
            authority: accounts.authority,
            __args: args,
        }
    }
    #[inline(always)]
    pub fn invoke(&self) -> solana_program_error::ProgramResult {
        self.invoke_signed_with_remaining_accounts(&[], &[])
    }
    #[inline(always)]
    pub fn invoke_with_remaining_accounts(
        &self,
        remaining_accounts: &[(&'b solana_account_info::AccountInfo<'a>, bool, bool)],
    ) -> solana_program_error::ProgramResult {
        self.invoke_signed_with_remaining_accounts(&[], remaining_accounts)
    }
    #[inline(always)]
    pub fn invoke_signed(&self, signers_seeds: &[&[&[u8]]]) -> solana_program_error::ProgramResult {
        self.invoke_signed_with_remaining_accounts(signers_seeds, &[])
    }
    #[allow(clippy::arithmetic_side_effects)]
    #[allow(clippy::clone_on_copy)]
    #[allow(clippy::vec_init_then_push)]
    pub fn invoke_signed_with_remaining_accounts(
        &self,
        signers_seeds: &[&[&[u8]]],
        remaining_accounts: &[(&'b solana_account_info::AccountInfo<'a>, bool, bool)],
    ) -> solana_program_error::ProgramResult {
        let mut accounts = Vec::with_capacity(2 + remaining_accounts.len());
        accounts.push(solana_instruction::AccountMeta::new(*self.stake.key, false));
        accounts.push(solana_instruction::AccountMeta::new_readonly(
            *self.authority.key,
            true,
        ));
        remaining_accounts.iter().for_each(|remaining_account| {
            accounts.push(solana_instruction::AccountMeta {
                pubkey: *remaining_account.0.key,
                is_signer: remaining_account.1,
                is_writable: remaining_account.2,
            })
        });
        let mut data = SetLockupInstructionData::new().try_to_vec().unwrap();
        let mut args = self.__args.try_to_vec().unwrap();
        data.append(&mut args);

        let instruction = solana_instruction::Instruction {
            program_id: crate::SPHERENET_STAKE_INTERFACE_ID,
            accounts,
            data,
        };
        let mut account_infos = Vec::with_capacity(3 + remaining_accounts.len());
        account_infos.push(self.__program.clone());
        account_infos.push(self.stake.clone());
        account_infos.push(self.authority.clone());
        remaining_accounts
            .iter()
            .for_each(|remaining_account| account_infos.push(remaining_account.0.clone()));

        if signers_seeds.is_empty() {
            solana_cpi::invoke(&instruction, &account_infos)
        } else {
            solana_cpi::invoke_signed(&instruction, &account_infos, signers_seeds)
        }
    }
}

/// Instruction builder for `SetLockup` via CPI.
///
/// ### Accounts:
///
///   0. `[writable]` stake
///   1. `[signer]` authority
#[derive(Clone, Debug)]
pub struct SetLockupCpiBuilder<'a, 'b> {
    instruction: Box<SetLockupCpiBuilderInstruction<'a, 'b>>,
}

impl<'a, 'b> SetLockupCpiBuilder<'a, 'b> {
    pub fn new(program: &'b solana_account_info::AccountInfo<'a>) -> Self {
        let instruction = Box::new(SetLockupCpiBuilderInstruction {
            __program: program,
            stake: None,
            authority: None,
            unix_timestamp: None,
            epoch: None,
            custodian: None,
            __remaining_accounts: Vec::new(),
        });
        Self { instruction }
    }
    /// Initialized stake account
    #[inline(always)]
    pub fn stake(&mut self, stake: &'b solana_account_info::AccountInfo<'a>) -> &mut Self {
        self.instruction.stake = Some(stake);
        self
    }
    /// Lockup authority or withdraw authority
    #[inline(always)]
    pub fn authority(&mut self, authority: &'b solana_account_info::AccountInfo<'a>) -> &mut Self {
        self.instruction.authority = Some(authority);
        self
    }
    /// `[optional argument]`
    #[inline(always)]
    pub fn unix_timestamp(&mut self, unix_timestamp: UnixTimestamp) -> &mut Self {
        self.instruction.unix_timestamp = Some(unix_timestamp);
        self
    }
    /// `[optional argument]`
    #[inline(always)]
    pub fn epoch(&mut self, epoch: Epoch) -> &mut Self {
        self.instruction.epoch = Some(epoch);
        self
    }
    /// `[optional argument]`
    #[inline(always)]
    pub fn custodian(&mut self, custodian: Pubkey) -> &mut Self {
        self.instruction.custodian = Some(custodian);
        self
    }
    /// Add an additional account to the instruction.
    #[inline(always)]
    pub fn add_remaining_account(
        &mut self,
        account: &'b solana_account_info::AccountInfo<'a>,
        is_writable: bool,
        is_signer: bool,
    ) -> &mut Self {
        self.instruction
            .__remaining_accounts
            .push((account, is_writable, is_signer));
        self
    }
    /// Add additional accounts to the instruction.
    ///
    /// Each account is represented by a tuple of the `AccountInfo`, a `bool` indicating whether the account is writable or not,
    /// and a `bool` indicating whether the account is a signer or not.
    #[inline(always)]
    pub fn add_remaining_accounts(
        &mut self,
        accounts: &[(&'b solana_account_info::AccountInfo<'a>, bool, bool)],
    ) -> &mut Self {
        self.instruction
            .__remaining_accounts
            .extend_from_slice(accounts);
        self
    }
    #[inline(always)]
    pub fn invoke(&self) -> solana_program_error::ProgramResult {
        self.invoke_signed(&[])
    }
    #[allow(clippy::clone_on_copy)]
    #[allow(clippy::vec_init_then_push)]
    pub fn invoke_signed(&self, signers_seeds: &[&[&[u8]]]) -> solana_program_error::ProgramResult {
        let args = SetLockupInstructionArgs {
            unix_timestamp: self.instruction.unix_timestamp.clone(),
            epoch: self.instruction.epoch.clone(),
            custodian: self.instruction.custodian.clone(),
        };
        let instruction = SetLockupCpi {
            __program: self.instruction.__program,

            stake: self.instruction.stake.expect("stake is not set"),

            authority: self.instruction.authority.expect("authority is not set"),
            __args: args,
        };
        instruction.invoke_signed_with_remaining_accounts(
            signers_seeds,
            &self.instruction.__remaining_accounts,
        )
    }
}

#[derive(Clone, Debug)]
struct SetLockupCpiBuilderInstruction<'a, 'b> {
    __program: &'b solana_account_info::AccountInfo<'a>,
    stake: Option<&'b solana_account_info::AccountInfo<'a>>,
    authority: Option<&'b solana_account_info::AccountInfo<'a>>,
    unix_timestamp: Option<UnixTimestamp>,
    epoch: Option<Epoch>,
    custodian: Option<Pubkey>,
    /// Additional instruction accounts `(AccountInfo, is_writable, is_signer)`.
    __remaining_accounts: Vec<(&'b solana_account_info::AccountInfo<'a>, bool, bool)>,
}
