1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
use serde::bytes::ByteBuf; use std::collections::BTreeMap; #[derive(PartialOrd, Eq, Ord, PartialEq, Clone, Debug, Serialize, Deserialize)] pub enum Bencode { ByteString(ByteBuf), Integer(i64), List(Vec<Bencode>), Dict(BTreeMap<Bencode, Bencode>) } impl From<i64> for Bencode { fn from(i: i64) -> Bencode { Bencode::Integer(i) } } impl From<String> for Bencode { fn from(s: String) -> Bencode { Bencode::ByteString(s.into_bytes().into()) } } impl<'a> From<&'a str> for Bencode { fn from(s: &str) -> Bencode { Bencode::ByteString(s.as_bytes().to_vec().into()) } } impl From<Vec<u8>> for Bencode { fn from(bs: Vec<u8>) -> Bencode { Bencode::ByteString(bs.into()) } } impl From<Vec<Bencode>> for Bencode { fn from(l: Vec<Bencode>) -> Bencode { Bencode::List(l.clone()) } } impl From<BTreeMap<Vec<u8>, Bencode>> for Bencode { fn from(m: BTreeMap<Vec<u8>, Bencode>) -> Bencode { let mut n: BTreeMap<Bencode, Bencode> = BTreeMap::new(); for (key, value) in m.iter() { n.insert(key.clone().into(), value.clone()); } Bencode::Dict(n) } }