54 lines
1.5 KiB
Rust
54 lines
1.5 KiB
Rust
|
use libp2p::identity::{self, Keypair};
|
||
|
|
use serde::{Deserialize, Serialize};
|
||
|
|
|
||
|
|
#[derive(Debug, Deserialize, Serialize)]
|
||
|
|
pub struct NodeConfig {
|
||
|
|
#[serde(with = "keypair")]
|
||
|
|
secret: Keypair,
|
||
|
|
}
|
||
|
|
|
||
|
|
mod keypair {
|
||
|
|
use base64::{prelude::BASE64_STANDARD, Engine};
|
||
|
|
use libp2p::identity::Keypair;
|
||
|
|
use serde::{Deserialize, Deserializer, Serializer};
|
||
|
|
|
||
|
|
pub fn serialize<S>(keypair: &Keypair, serializer: S) -> Result<S::Ok, S::Error>
|
||
|
|
where S: Serializer
|
||
|
|
{
|
||
|
|
let vec = keypair.to_protobuf_encoding().unwrap();
|
||
|
|
let base64 = BASE64_STANDARD.encode(vec);
|
||
|
|
serializer.serialize_str(&base64)
|
||
|
|
}
|
||
|
|
pub fn deserialize<'de, D>(deserializer: D) -> Result<Keypair, D::Error>
|
||
|
|
where D: Deserializer<'de>
|
||
|
|
{
|
||
|
|
let base64 = String::deserialize(deserializer)?;
|
||
|
|
let vec = BASE64_STANDARD.decode(base64).unwrap();
|
||
|
|
Ok(Keypair::from_protobuf_encoding(&vec).unwrap())
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
pub struct PartialNodeConfig {
|
||
|
|
pub secret: Option<Vec<u8>>,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(test)]
|
||
|
|
mod tests {
|
||
|
|
use libp2p::identity;
|
||
|
|
use super::*;
|
||
|
|
|
||
|
|
|
||
|
|
#[tokio::test]
|
||
|
|
async fn parse_ed25519() {
|
||
|
|
let keypair = identity::Keypair::generate_ed25519();
|
||
|
|
let config = NodeConfig {
|
||
|
|
secret: keypair.clone(),
|
||
|
|
};
|
||
|
|
let string = toml::to_string(&config).unwrap();
|
||
|
|
println!("Parsed config: {}", &string);
|
||
|
|
let parsed_config: NodeConfig = toml::from_str(&string).unwrap();
|
||
|
|
assert_eq!(keypair.public(), parsed_config.secret.public())
|
||
|
|
}
|
||
|
|
}
|