93 lines
2.8 KiB
Rust
93 lines
2.8 KiB
Rust
|
//! The `genesis_block` module is a library for generating the chain's genesis block.
|
||
|
|
||
|
use solana_sdk::hash::{hash, Hash};
|
||
|
use solana_sdk::pubkey::Pubkey;
|
||
|
use solana_sdk::signature::{Keypair, KeypairUtil};
|
||
|
use std::fs::File;
|
||
|
use std::io::Write;
|
||
|
use std::path::Path;
|
||
|
|
||
|
#[derive(Serialize, Deserialize, Debug)]
|
||
|
pub struct GenesisBlock {
|
||
|
pub bootstrap_leader_id: Pubkey,
|
||
|
pub bootstrap_leader_tokens: u64,
|
||
|
pub mint_id: Pubkey,
|
||
|
pub tokens: u64,
|
||
|
}
|
||
|
|
||
|
impl GenesisBlock {
|
||
|
#[allow(clippy::new_ret_no_self)]
|
||
|
pub fn new(tokens: u64) -> (Self, Keypair) {
|
||
|
let mint_keypair = Keypair::new();
|
||
|
(
|
||
|
Self {
|
||
|
bootstrap_leader_id: Pubkey::default(),
|
||
|
bootstrap_leader_tokens: 0,
|
||
|
mint_id: mint_keypair.pubkey(),
|
||
|
tokens,
|
||
|
},
|
||
|
mint_keypair,
|
||
|
)
|
||
|
}
|
||
|
|
||
|
pub fn new_with_leader(
|
||
|
tokens: u64,
|
||
|
bootstrap_leader_id: Pubkey,
|
||
|
bootstrap_leader_tokens: u64,
|
||
|
) -> (Self, Keypair) {
|
||
|
let mint_keypair = Keypair::new();
|
||
|
(
|
||
|
Self {
|
||
|
bootstrap_leader_id,
|
||
|
bootstrap_leader_tokens,
|
||
|
mint_id: mint_keypair.pubkey(),
|
||
|
tokens,
|
||
|
},
|
||
|
mint_keypair,
|
||
|
)
|
||
|
}
|
||
|
|
||
|
pub fn last_id(&self) -> Hash {
|
||
|
let serialized = serde_json::to_string(self).unwrap();
|
||
|
hash(&serialized.into_bytes())
|
||
|
}
|
||
|
|
||
|
pub fn load(ledger_path: &str) -> Result<Self, std::io::Error> {
|
||
|
let file = File::open(&Path::new(ledger_path).join("genesis.json"))?;
|
||
|
let genesis_block = serde_json::from_reader(file)?;
|
||
|
Ok(genesis_block)
|
||
|
}
|
||
|
|
||
|
pub fn write(&self, ledger_path: &str) -> Result<(), std::io::Error> {
|
||
|
let serialized = serde_json::to_string(self)?;
|
||
|
let mut file = File::create(&Path::new(ledger_path).join("genesis.json"))?;
|
||
|
file.write_all(&serialized.into_bytes())
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#[cfg(test)]
|
||
|
mod tests {
|
||
|
use super::*;
|
||
|
|
||
|
#[test]
|
||
|
fn test_genesis_block_new() {
|
||
|
let (genesis_block, mint) = GenesisBlock::new(10_000);
|
||
|
assert_eq!(genesis_block.tokens, 10_000);
|
||
|
assert_eq!(genesis_block.mint_id, mint.pubkey());
|
||
|
assert_eq!(genesis_block.bootstrap_leader_id, Pubkey::default());
|
||
|
assert_eq!(genesis_block.bootstrap_leader_tokens, 0);
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
fn test_genesis_block_new_with_leader() {
|
||
|
let leader_keypair = Keypair::new();
|
||
|
let (genesis_block, mint) =
|
||
|
GenesisBlock::new_with_leader(20_000, leader_keypair.pubkey(), 123);
|
||
|
|
||
|
assert_eq!(genesis_block.tokens, 20_000);
|
||
|
assert_eq!(genesis_block.mint_id, mint.pubkey());
|
||
|
assert_eq!(genesis_block.bootstrap_leader_id, leader_keypair.pubkey());
|
||
|
assert_eq!(genesis_block.bootstrap_leader_tokens, 123);
|
||
|
}
|
||
|
}
|