86 lines
2.9 KiB
Rust
86 lines
2.9 KiB
Rust
use crate::error::*;
|
|
use crate::state::lockup::*;
|
|
use crate::state::registrar::Registrar;
|
|
use crate::state::voter::Voter;
|
|
use anchor_lang::prelude::*;
|
|
use anchor_lang::solana_program::sysvar::instructions as tx_instructions;
|
|
use spl_governance::addins::voter_weight::VoterWeightAccountType;
|
|
use std::mem::size_of;
|
|
|
|
pub const VOTER_WEIGHT_RECORD: [u8; 19] = *b"voter-weight-record";
|
|
|
|
#[derive(Accounts)]
|
|
#[instruction(voter_bump: u8, voter_weight_record_bump: u8)]
|
|
pub struct CreateVoter<'info> {
|
|
pub registrar: Box<Account<'info, Registrar>>,
|
|
|
|
#[account(
|
|
init,
|
|
seeds = [registrar.key().as_ref(), b"voter".as_ref(), voter_authority.key().as_ref()],
|
|
bump = voter_bump,
|
|
payer = payer,
|
|
space = 8 + size_of::<Voter>(),
|
|
)]
|
|
pub voter: AccountLoader<'info, Voter>,
|
|
pub voter_authority: Signer<'info>,
|
|
|
|
#[account(
|
|
init,
|
|
seeds = [VOTER_WEIGHT_RECORD.as_ref(), registrar.key().as_ref(), voter_authority.key().as_ref()],
|
|
bump = voter_weight_record_bump,
|
|
payer = payer,
|
|
space = 150,
|
|
)]
|
|
pub voter_weight_record: Account<'info, VoterWeightRecord>,
|
|
|
|
#[account(mut)]
|
|
pub payer: Signer<'info>,
|
|
|
|
pub system_program: Program<'info, System>,
|
|
pub rent: Sysvar<'info, Rent>,
|
|
|
|
#[account(address = tx_instructions::ID)]
|
|
pub instructions: UncheckedAccount<'info>,
|
|
}
|
|
|
|
/// Creates a new voter account. There can only be a single voter per
|
|
/// user wallet.
|
|
pub fn create_voter(
|
|
ctx: Context<CreateVoter>,
|
|
voter_bump: u8,
|
|
voter_weight_record_bump: u8,
|
|
) -> Result<()> {
|
|
msg!("--------create_voter--------");
|
|
// Forbid creating voter accounts from CPI. The goal is to make automation
|
|
// impossible that weakens some of the limitations intentionally imposed on
|
|
// locked tokens.
|
|
{
|
|
let ixns = ctx.accounts.instructions.to_account_info();
|
|
let current_index = tx_instructions::load_current_index_checked(&ixns)? as usize;
|
|
let current_ixn = tx_instructions::load_instruction_at_checked(current_index, &ixns)?;
|
|
require!(
|
|
current_ixn.program_id == *ctx.program_id,
|
|
ErrorCode::ForbiddenCpi
|
|
);
|
|
}
|
|
|
|
// Load accounts.
|
|
let registrar = &ctx.accounts.registrar;
|
|
let voter = &mut ctx.accounts.voter.load_init()?;
|
|
let voter_weight_record = &mut ctx.accounts.voter_weight_record;
|
|
|
|
// Init the voter.
|
|
voter.voter_bump = voter_bump;
|
|
voter.voter_weight_record_bump = voter_weight_record_bump;
|
|
voter.voter_authority = ctx.accounts.voter_authority.key();
|
|
voter.registrar = ctx.accounts.registrar.key();
|
|
|
|
// Init the voter weight record.
|
|
voter_weight_record.account_type = VoterWeightAccountType::VoterWeightRecord;
|
|
voter_weight_record.realm = registrar.realm;
|
|
voter_weight_record.governing_token_mint = registrar.realm_governing_token_mint;
|
|
voter_weight_record.governing_token_owner = ctx.accounts.voter_authority.key();
|
|
|
|
Ok(())
|
|
}
|