861dfb6497
Add documentation for those tools, and make it so that `gen_large_yaml` generates a predetermined set of files instead of outputting to its standard output.
38 lines
935 B
Rust
38 lines
935 B
Rust
use std::env;
|
|
use std::fs::File;
|
|
use std::io::prelude::*;
|
|
use yaml_rust2::{
|
|
parser::{MarkedEventReceiver, Parser},
|
|
scanner::Marker,
|
|
Event,
|
|
};
|
|
|
|
#[derive(Debug)]
|
|
struct EventSink {
|
|
events: Vec<(Event, Marker)>,
|
|
}
|
|
|
|
impl MarkedEventReceiver for EventSink {
|
|
fn on_event(&mut self, ev: Event, mark: Marker) {
|
|
eprintln!(" \x1B[;34m\u{21B3} {:?}\x1B[;m", &ev);
|
|
self.events.push((ev, mark));
|
|
}
|
|
}
|
|
|
|
fn str_to_events(yaml: &str) -> Vec<(Event, Marker)> {
|
|
let mut sink = EventSink { events: Vec::new() };
|
|
let mut parser = Parser::new(yaml.chars());
|
|
// Load events using our sink as the receiver.
|
|
parser.load(&mut sink, true).unwrap();
|
|
sink.events
|
|
}
|
|
|
|
fn main() {
|
|
let args: Vec<_> = env::args().collect();
|
|
let mut f = File::open(&args[1]).unwrap();
|
|
let mut s = String::new();
|
|
f.read_to_string(&mut s).unwrap();
|
|
|
|
// dbg!(str_to_events(&s));
|
|
str_to_events(&s);
|
|
}
|