49 lines
1.7 KiB
Rust
49 lines
1.7 KiB
Rust
use super::*;
|
|
use pairing::Engine;
|
|
use ff::PrimeField;
|
|
use util::hash_to_fr;
|
|
use std::fmt;
|
|
|
|
#[derive(Clone, Serialize, Deserialize)]
|
|
#[serde(bound(serialize = "<E as ff::ScalarEngine>::Fr: serde::Serialize"))]
|
|
#[serde(bound(deserialize = "<E as ff::ScalarEngine>::Fr: serde::Deserialize<'de>"))]
|
|
pub struct Wallet<E: Engine> {
|
|
pub channelId: E::Fr,
|
|
pub wpk: E::Fr,
|
|
pub bc: i64,
|
|
pub bm: i64,
|
|
pub close: Option<E::Fr>,
|
|
}
|
|
|
|
impl<E: Engine> Wallet<E> {
|
|
pub fn as_fr_vec(&self) -> Vec<E::Fr> {
|
|
if self.close.is_some() {
|
|
vec!(self.channelId, self.wpk, E::Fr::from_str(&self.bc.to_string()).unwrap(), E::Fr::from_str(&self.bm.to_string()).unwrap(), self.close.unwrap())
|
|
} else {
|
|
vec!(self.channelId, self.wpk, E::Fr::from_str(&self.bc.to_string()).unwrap(), E::Fr::from_str(&self.bm.to_string()).unwrap())
|
|
}
|
|
}
|
|
|
|
pub fn without_close(&self) -> Vec<E::Fr> {
|
|
vec!(self.channelId, self.wpk, E::Fr::from_str(&self.bc.to_string()).unwrap(), E::Fr::from_str(&self.bm.to_string()).unwrap())
|
|
}
|
|
|
|
pub fn with_close(&mut self, msg: String) -> Vec<E::Fr> {
|
|
let m = hash_to_fr::<E>(msg.into_bytes() );
|
|
self.close = Some(m.clone());
|
|
return self.as_fr_vec();
|
|
}
|
|
}
|
|
|
|
impl<E: Engine> fmt::Display for Wallet<E> {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
if self.close.is_some() {
|
|
let close_str = self.close.unwrap();
|
|
write!(f, "Wallet : (\nchannelId={}\nwpk={}\nbc={}\nbm={}\nclose={}\n)", &self.channelId, &self.wpk, &self.bc, &self.bm, close_str)
|
|
} else {
|
|
write!(f, "Wallet : (\nchannelId={}\nwpk={}\nbc={}\nbm={}\nclose=None\n)", &self.channelId, &self.wpk, &self.bc, &self.bm)
|
|
}
|
|
}
|
|
}
|
|
|