use crate::*;
#[derive(Accounts)]
pub struct GaugeCommitVote<'info> {
pub gaugemeister: Account<'info, Gaugemeister>,
pub gauge: Account<'info, Gauge>,
pub gauge_voter: Account<'info, GaugeVoter>,
pub gauge_vote: Account<'info, GaugeVote>,
#[account(mut)]
pub epoch_gauge: Account<'info, EpochGauge>,
#[account(mut)]
pub epoch_gauge_voter: Account<'info, EpochGaugeVoter>,
#[account(
init,
seeds = [
b"EpochGaugeVote".as_ref(),
gauge_vote.key().as_ref(),
epoch_gauge_voter.voting_epoch.to_le_bytes().as_ref(),
],
bump,
space = 8 + EpochGaugeVote::LEN,
payer = payer
)]
pub epoch_gauge_vote: Account<'info, EpochGaugeVote>,
#[account(mut)]
pub payer: Signer<'info>,
pub system_program: Program<'info, System>,
}
impl<'info> GaugeCommitVote<'info> {
fn vote_shares_for_next_epoch(&self) -> Option<u64> {
if self.gauge_vote.weight == 0 {
return Some(0);
}
let power: u64 = self.epoch_gauge_voter.voting_power;
let total_shares = ::u128::mul_div_u64(
power,
self.gauge_vote.weight.into(),
self.gauge_voter.total_weight.into(),
)?;
msg!("power: {}, shares: {}", power, total_shares);
Some(total_shares)
}
}
pub fn handler(ctx: Context<GaugeCommitVote>) -> Result<()> {
let next_vote_shares = unwrap_int!(ctx.accounts.vote_shares_for_next_epoch());
if next_vote_shares == 0 {
return Ok(());
}
let epoch_gauge = &mut ctx.accounts.epoch_gauge;
let epoch_voter = &mut ctx.accounts.epoch_gauge_voter;
let epoch_vote = &mut ctx.accounts.epoch_gauge_vote;
epoch_voter.allocated_power =
unwrap_int!(epoch_voter.allocated_power.checked_add(next_vote_shares));
epoch_vote.allocated_power = next_vote_shares;
epoch_gauge.total_power = unwrap_int!(epoch_gauge.total_power.checked_add(next_vote_shares));
emit!(CommitGaugeVoteEvent {
gaugemeister: ctx.accounts.gauge.gaugemeister,
gauge: ctx.accounts.gauge.key(),
quarry: ctx.accounts.gauge.quarry,
gauge_voter_owner: ctx.accounts.gauge_voter.owner,
vote_shares_for_next_epoch: next_vote_shares,
voting_epoch: epoch_voter.voting_epoch,
updated_allocated_power: epoch_voter.allocated_power,
updated_total_power: epoch_gauge.total_power,
});
Ok(())
}
impl<'info> Validate<'info> for GaugeCommitVote<'info> {
fn validate(&self) -> Result<()> {
assert_keys_eq!(self.gaugemeister, self.gauge.gaugemeister);
assert_keys_eq!(self.gauge, self.gauge_vote.gauge);
assert_keys_eq!(self.gauge_voter, self.gauge_vote.gauge_voter);
assert_keys_eq!(self.epoch_gauge.gauge, self.gauge);
assert_keys_eq!(self.epoch_gauge_voter.gauge_voter, self.gauge_voter);
invariant!(!self.gauge.is_disabled, CannotCommitGaugeDisabled);
invariant!(
self.epoch_gauge_voter.weight_change_seqno == self.gauge_voter.weight_change_seqno,
WeightSeqnoChanged
);
let voting_epoch = self.gaugemeister.voting_epoch()?;
invariant!(
self.epoch_gauge_voter.voting_epoch == voting_epoch,
EpochGaugeNotVoting
);
Ok(())
}
}
#[event]
pub struct CommitGaugeVoteEvent {
#[index]
pub gaugemeister: Pubkey,
#[index]
pub gauge: Pubkey,
#[index]
pub quarry: Pubkey,
#[index]
pub gauge_voter_owner: Pubkey,
pub voting_epoch: u32,
pub vote_shares_for_next_epoch: u64,
pub updated_allocated_power: u64,
pub updated_total_power: u64,
}