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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
use contract::Error;
use core::Address;
use glob::glob;
use serde_json;
use std::fs::File;
use std::path::{Path, PathBuf};
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct ContractStorage {
dir: PathBuf,
}
impl ContractStorage {
pub fn new(dir: PathBuf) -> ContractStorage {
ContractStorage { dir }
}
pub fn validate(&self, contract: &serde_json::Value) -> Result<(), Error> {
if !contract.is_object() {
return Err(Error::InvalidContract("Invalid data format".to_string()));
}
let addr = match contract.get("address") {
Some(addr) => addr,
None => return Err(Error::InvalidContract("Missing address".to_string())),
};
if !addr.is_string() {
return Err(Error::InvalidContract("Invalid address format".to_string()));
}
match addr.as_str().unwrap().parse::<Address>() {
Ok(_) => {}
Err(_) => return Err(Error::InvalidContract("Can't parse address".to_string())),
}
Ok(())
}
pub fn add(&self, contract: &serde_json::Value) -> Result<(), Error> {
self.validate(contract)?;
let addr = contract
.get("address")
.expect("Expect address for a contract")
.as_str()
.expect("Expect address be convertible to a string");
let mut filename: PathBuf = self.dir.clone();
filename.push(format!("{}.json", addr));
let mut f = File::create(filename.as_path())?;
match serde_json::to_writer_pretty(&mut f, contract) {
Ok(_) => Ok(()),
Err(_) => Err(Error::IO(format!(
"Can't write contract for address {}",
addr
))),
}
}
pub fn list(&self) -> Vec<serde_json::Value> {
let files = glob(&format!("{}/*.json", &self.dir.to_str().unwrap())).unwrap();
files
.filter(|x| x.is_ok())
.map(|x| ContractStorage::read_json(x.unwrap().as_path()))
.filter(|x| x.is_ok())
.map(|x| x.unwrap())
.collect()
}
fn read_json(path: &Path) -> Result<serde_json::Value, Error> {
match File::open(path) {
Ok(f) => serde_json::from_reader(f)
.or_else(|_| Err(Error::IO("Can't read contract file".to_string()))),
Err(_) => Err(Error::IO("Can't open contract file".to_string())),
}
}
}