67 lines
2.7 KiB
Rust
67 lines
2.7 KiB
Rust
/// A macro is exposed so that we can embed the program ID.
|
|
#[macro_export]
|
|
macro_rules! vote_weight_record {
|
|
($id:expr) => {
|
|
/// Anchor wrapper for the SPL governance program's VoterWeightRecord type.
|
|
#[derive(Clone)]
|
|
pub struct VoterWeightRecord(spl_governance::addins::voter_weight::VoterWeightRecord);
|
|
|
|
impl anchor_lang::AccountDeserialize for VoterWeightRecord {
|
|
fn try_deserialize(buf: &mut &[u8]) -> std::result::Result<Self, ProgramError> {
|
|
let mut data = buf;
|
|
let vwr: spl_governance::addins::voter_weight::VoterWeightRecord =
|
|
anchor_lang::AnchorDeserialize::deserialize(&mut data)
|
|
.map_err(|_| anchor_lang::__private::ErrorCode::AccountDidNotDeserialize)?;
|
|
if vwr.account_type != spl_governance::addins::voter_weight::VoterWeightAccountType::VoterWeightRecord {
|
|
return Err(anchor_lang::__private::ErrorCode::AccountDidNotSerialize.into());
|
|
}
|
|
Ok(VoterWeightRecord(vwr))
|
|
}
|
|
|
|
fn try_deserialize_unchecked(
|
|
buf: &mut &[u8],
|
|
) -> std::result::Result<Self, ProgramError> {
|
|
let mut data = buf;
|
|
let vwr: spl_governance::addins::voter_weight::VoterWeightRecord =
|
|
anchor_lang::AnchorDeserialize::deserialize(&mut data)
|
|
.map_err(|_| anchor_lang::__private::ErrorCode::AccountDidNotDeserialize)?;
|
|
if vwr.account_type != spl_governance::addins::voter_weight::VoterWeightAccountType::Uninitialized {
|
|
return Err(anchor_lang::__private::ErrorCode::AccountDidNotSerialize.into());
|
|
}
|
|
Ok(VoterWeightRecord(vwr))
|
|
}
|
|
}
|
|
|
|
impl anchor_lang::AccountSerialize for VoterWeightRecord {
|
|
fn try_serialize<W: std::io::Write>(
|
|
&self,
|
|
writer: &mut W,
|
|
) -> std::result::Result<(), ProgramError> {
|
|
anchor_lang::AnchorSerialize::serialize(&self.0, writer)
|
|
.map_err(|_| anchor_lang::__private::ErrorCode::AccountDidNotSerialize)?;
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl anchor_lang::Owner for VoterWeightRecord {
|
|
fn owner() -> Pubkey {
|
|
$id
|
|
}
|
|
}
|
|
|
|
impl std::ops::Deref for VoterWeightRecord {
|
|
type Target = spl_governance::addins::voter_weight::VoterWeightRecord;
|
|
|
|
fn deref(&self) -> &Self::Target {
|
|
&self.0
|
|
}
|
|
}
|
|
|
|
impl std::ops::DerefMut for VoterWeightRecord {
|
|
fn deref_mut(&mut self) -> &mut Self::Target {
|
|
&mut self.0
|
|
}
|
|
}
|
|
};
|
|
}
|