rustlings/src/main.rs

91 lines
2.8 KiB
Rust
Raw Normal View History

2019-01-09 13:33:58 -06:00
use crate::run::run;
use crate::verify::verify;
use clap::{crate_version, App, Arg, SubCommand};
use notify::DebouncedEvent;
use notify::{RecommendedWatcher, RecursiveMode, Watcher};
use std::io::BufRead;
use std::sync::mpsc::channel;
use std::time::Duration;
2019-01-09 13:33:58 -06:00
use syntect::easy::HighlightFile;
use syntect::highlighting::{Style, ThemeSet};
use syntect::parsing::SyntaxSet;
use syntect::util::as_24_bit_terminal_escaped;
2019-01-09 13:33:43 -06:00
mod run;
mod util;
2019-01-09 13:33:58 -06:00
mod verify;
2018-05-14 11:41:58 -05:00
2018-11-09 13:31:14 -06:00
fn main() {
let matches = App::new("rustlings")
2018-11-14 13:12:20 -06:00
.version(crate_version!())
.author("Olivia Hugger")
.about("Test")
.subcommand(SubCommand::with_name("verify").alias("v"))
.subcommand(SubCommand::with_name("watch").alias("w"))
2018-11-23 08:18:43 -06:00
.subcommand(
SubCommand::with_name("run")
.alias("r")
.arg(Arg::with_name("file").required(true).index(1)),
2019-01-09 13:33:58 -06:00
)
.get_matches();
2018-11-14 13:12:20 -06:00
let ss = SyntaxSet::load_defaults_newlines();
let ts = ThemeSet::load_defaults();
2019-01-09 13:33:58 -06:00
2018-11-14 13:12:20 -06:00
println!(r#" _ _ _ "#);
println!(r#" _ __ _ _ ___| |_| (_)_ __ __ _ ___ "#);
println!(r#" | '__| | | / __| __| | | '_ \ / _` / __| "#);
println!(r#" | | | |_| \__ \ |_| | | | | | (_| \__ \ "#);
println!(r#" |_| \__,_|___/\__|_|_|_| |_|\__, |___/ "#);
println!(r#" |___/ "#);
println!("");
2018-11-23 08:18:43 -06:00
if let Some(matches) = matches.subcommand_matches("run") {
2019-01-09 13:33:43 -06:00
run(matches.clone());
2018-11-23 08:18:43 -06:00
}
2018-11-14 13:12:20 -06:00
if let Some(_) = matches.subcommand_matches("verify") {
match verify() {
Ok(_) => {}
Err(_) => std::process::exit(1),
}
}
if let Some(_) = matches.subcommand_matches("watch") {
watch().unwrap();
2018-11-14 13:12:20 -06:00
}
if let None = matches.subcommand_name() {
2019-01-09 13:33:58 -06:00
let mut highlighter =
HighlightFile::new("default_out.md", &ss, &ts.themes["base16-eighties.dark"]).unwrap();
for maybe_line in highlighter.reader.lines() {
let line = maybe_line.unwrap();
let regions: Vec<(Style, &str)> = highlighter.highlight_lines.highlight(&line, &ss);
println!("{}", as_24_bit_terminal_escaped(&regions[..], true));
}
}
println!("\x1b[0m");
2018-05-06 11:59:50 -05:00
}
fn watch() -> notify::Result<()> {
let (tx, rx) = channel();
let mut watcher: RecommendedWatcher = Watcher::new(tx, Duration::from_secs(2))?;
watcher.watch("./exercises", RecursiveMode::Recursive)?;
let _ignored = verify();
loop {
match rx.recv() {
2019-01-09 13:33:58 -06:00
Ok(event) => match event {
DebouncedEvent::Chmod(_) | DebouncedEvent::Write(_) => {
let _ignored = verify();
}
2019-01-09 13:33:58 -06:00
_ => {}
},
Err(e) => println!("watch error: {:?}", e),
}
}
}