61 lines
1.9 KiB
Rust
61 lines
1.9 KiB
Rust
use crate::error::*;
|
|
use crate::state::*;
|
|
use anchor_lang::prelude::*;
|
|
|
|
#[derive(Accounts)]
|
|
pub struct ResetLockup<'info> {
|
|
pub registrar: AccountLoader<'info, Registrar>,
|
|
|
|
// checking the PDA address it just an extra precaution,
|
|
// the other constraints must be exhaustive
|
|
#[account(
|
|
mut,
|
|
seeds = [registrar.key().as_ref(), b"voter".as_ref(), voter_authority.key().as_ref()],
|
|
bump = voter.load()?.voter_bump,
|
|
has_one = voter_authority,
|
|
has_one = registrar)]
|
|
pub voter: AccountLoader<'info, Voter>,
|
|
pub voter_authority: Signer<'info>,
|
|
}
|
|
|
|
/// Resets a lockup to start at the current slot timestamp and to last for
|
|
/// `periods`, which must be >= the number of periods left on the lockup.
|
|
/// This will re-lock any non-withdrawn vested funds.
|
|
pub fn reset_lockup(
|
|
ctx: Context<ResetLockup>,
|
|
deposit_entry_index: u8,
|
|
kind: LockupKind,
|
|
periods: u32,
|
|
) -> Result<()> {
|
|
let registrar = &ctx.accounts.registrar.load()?;
|
|
let voter = &mut ctx.accounts.voter.load_mut()?;
|
|
let curr_ts = registrar.clock_unix_timestamp();
|
|
|
|
let source = voter.active_deposit_mut(deposit_entry_index)?;
|
|
|
|
// Must not decrease duration or strictness
|
|
require_gte!(
|
|
(periods as u64).checked_mul(kind.period_secs()).unwrap(),
|
|
source.lockup.seconds_left(curr_ts),
|
|
VsrError::InvalidLockupPeriod
|
|
);
|
|
require_gte!(
|
|
kind.strictness(),
|
|
source.lockup.kind.strictness(),
|
|
VsrError::InvalidLockupKind
|
|
);
|
|
|
|
// Don't re-lock clawback deposits. Users must withdraw and create a new one.
|
|
require!(
|
|
!source.allow_clawback,
|
|
VsrError::InvalidChangeToClawbackDepositEntry
|
|
);
|
|
|
|
// Change the deposit entry.
|
|
let d_entry = voter.active_deposit_mut(deposit_entry_index)?;
|
|
d_entry.amount_initially_locked_native = d_entry.amount_deposited_native;
|
|
d_entry.lockup = Lockup::new_from_periods(kind, curr_ts, curr_ts, periods)?;
|
|
|
|
Ok(())
|
|
}
|