64 lines
2.5 KiB
Rust
64 lines
2.5 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)?;
|
||
|
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
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
}
|