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

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */

use std::path::Path;

use crate::generator::Generator;
use proc_macro2::TokenStream;

mod generated;
mod generator;
mod keywords;
mod parser;
mod ref_parser;
mod resolver;
mod sanitizer;
mod schema;

pub fn generate(path: &Path) -> String {
    generate_token_stream(path).to_string()
}

pub fn generate_token_stream(path: &Path) -> TokenStream {
    let mut generator = Generator::new();
    generator.add_file(path);
    generator.into()
}

#[cfg(test)]
mod lib_tests {
    use proc_macro2::TokenStream;

    use std::{
        fs,
        io::Write,
        path::Path,
        process::{Command, Stdio},
    };

    use crate::generator::Generator;

    #[test]
    fn test() {
        let mut generator = Generator::new();
        generator.add_file(Path::new("schemas/draft-04.json"));
        let tokens: TokenStream = generator.into();
        let actual = tokens.to_string();
        let expected = fs::read_to_string("schemas/draft-04.rs").unwrap();

        assert_eq!(format(actual), expected);
    }

    fn format(text: impl std::fmt::Display) -> String {
        let mut rustfmt = Command::new("rustfmt")
            .stdin(Stdio::piped())
            .stdout(Stdio::piped())
            .spawn()
            .unwrap();
        write!(rustfmt.stdin.take().unwrap(), "{}", text).unwrap();
        let output = rustfmt.wait_with_output().unwrap();
        String::from_utf8(output.stdout).unwrap()
    }
}