beerxml 0.1.8

beerxml/json/yaml/toml parser generator
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

// (c) 2017 Joost Yervante Damad <joost@damad.be>

use std::io::Read;
use std::fs::File;
use std::path::Path;
use std::collections::HashMap;

use serde_toml;

use data::*;
use error::*;

#[derive(Deserialize)]
struct PrivateDocument {
    document: String,
}

fn read_str(s: &str) -> Result<RecordSet> {
    let pd: PrivateDocument = serde_toml::from_str(s)?;
    let document = pd.document.as_str();
    let marker = format!("document = \"{}\"", document);
    let s2 = s.replacen(&marker, "", 1);
    match document {
        "Fermentables" => {
            let h: HashMap<String, Fermentable> = serde_toml::from_str(&s2)?;
            Ok(RecordSet::Fermentables(h))
        }
        "Hops" => {
            let h: HashMap<String, Hop> = serde_toml::from_str(&s2)?;
            Ok(RecordSet::Hops(h))
        }
        "Miscs" => {
            let h: HashMap<String, Misc> = serde_toml::from_str(&s2)?;
            Ok(RecordSet::Miscs(h))
        }
        "Waters" => {
            let h: HashMap<String, Water> = serde_toml::from_str(&s2)?;
            Ok(RecordSet::Waters(h))
        }
        "Yeasts" => {
            let h: HashMap<String, Yeast> = serde_toml::from_str(&s2)?;
            Ok(RecordSet::Yeasts(h))
        }
        "Recipes" => {
            let h: HashMap<String, Recipe> = serde_toml::from_str(&s2)?;
            Ok(RecordSet::Recipes(h))
        }
        e => Err(format!("Toml: Unimplemented document type {}", e).into()),
    }
}

/// try to read a `RecordSet` from a toml file
pub fn read_file(filename: &Path) -> Result<RecordSet> {
    let mut f = File::open(filename)?;
    let mut contents = String::new();
    f.read_to_string(&mut contents)?;
    read_str(&contents)
}