48 lines
1.1 KiB
Rust
48 lines
1.1 KiB
Rust
|
extern crate serde;
|
||
|
|
extern crate serde_json;
|
||
|
|
|
||
|
|
use serde::{Deserialize, Serialize};
|
||
|
|
|
||
|
|
#[derive(Debug, Serialize, Deserialize)]
|
||
|
|
pub enum Message {
|
||
|
|
Error,
|
||
|
|
Hand { chain: String, version: u32 },
|
||
|
|
Shake { ok: bool, height: u64 },
|
||
|
|
Ping { height: u64 },
|
||
|
|
Pong { height: u64 },
|
||
|
|
GetPeers,
|
||
|
|
Peers,
|
||
|
|
GetBlock { index: u64 },
|
||
|
|
Block { index: u64, block: String },
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Message {
|
||
|
|
pub fn from_bytes(bytes: Vec<u8>) -> Result<Self, ()> {
|
||
|
|
let text = String::from_utf8(bytes).unwrap_or(String::from("Error{}"));
|
||
|
|
match serde_json::from_str(&text) {
|
||
|
|
Ok(cmd) => Ok(cmd),
|
||
|
|
Err(_) => Err(())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn hand(chain: &str, version: u32) -> Self {
|
||
|
|
Message::Hand { chain: chain.to_owned(), version }
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn shake(ok: bool, height: u64) -> Self {
|
||
|
|
Message::Shake { ok, height }
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn ping(height: u64) -> Self {
|
||
|
|
Message::Ping { height }
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn pong(height: u64) -> Self {
|
||
|
|
Message::Pong { height }
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn block(height: u64, str: String) -> Self {
|
||
|
|
Message::Block { index: height, block: str }
|
||
|
|
}
|
||
|
|
}
|