gemachain_stake_program/
config.rs

1//! config for staking
2//!  carries variables that the stake program cares about
3use bincode::deserialize;
4use gemachain_config_program::{create_config_account, get_config_data};
5use gemachain_sdk::{
6    account::{AccountSharedData, ReadableAccount, WritableAccount},
7    genesis_config::GenesisConfig,
8    instruction::InstructionError,
9    keyed_account::KeyedAccount,
10    stake::config::{self, Config},
11};
12
13#[deprecated(
14    since = "1.8.0",
15    note = "Please use `gemachain_sdk::stake::config` or `gemachain_program::stake::config` instead"
16)]
17pub use gemachain_sdk::stake::config::*;
18
19pub fn from<T: ReadableAccount>(account: &T) -> Option<Config> {
20    get_config_data(account.data())
21        .ok()
22        .and_then(|data| deserialize(data).ok())
23}
24
25pub fn from_keyed_account(account: &KeyedAccount) -> Result<Config, InstructionError> {
26    if !config::check_id(account.unsigned_key()) {
27        return Err(InstructionError::InvalidArgument);
28    }
29    from(&*account.try_account_ref()?).ok_or(InstructionError::InvalidArgument)
30}
31
32pub fn create_account(carats: u64, config: &Config) -> AccountSharedData {
33    create_config_account(vec![], config, carats)
34}
35
36pub fn add_genesis_account(genesis_config: &mut GenesisConfig) -> u64 {
37    let mut account = create_config_account(vec![], &Config::default(), 0);
38    let carats = genesis_config.rent.minimum_balance(account.data().len());
39
40    account.set_carats(carats.max(1));
41
42    genesis_config.add_account(config::id(), account);
43
44    carats
45}
46
47#[cfg(test)]
48mod tests {
49    use super::*;
50    use gemachain_sdk::pubkey::Pubkey;
51    use std::cell::RefCell;
52
53    #[test]
54    fn test() {
55        let account = RefCell::new(create_account(0, &Config::default()));
56        assert_eq!(from(&account.borrow()), Some(Config::default()));
57        assert_eq!(
58            from_keyed_account(&KeyedAccount::new(&Pubkey::default(), false, &account)),
59            Err(InstructionError::InvalidArgument)
60        );
61    }
62}