rustlings/src/main.rs

142 lines
4.9 KiB
Rust
Raw Normal View History

use crate::exercise::{Exercise, ExerciseList};
2019-01-09 19:33:58 +00:00
use crate::run::run;
use crate::verify::verify;
use clap::{crate_version, App, Arg, SubCommand};
use notify::DebouncedEvent;
use notify::{RecommendedWatcher, RecursiveMode, Watcher};
2019-03-06 18:38:55 +00:00
use std::ffi::OsStr;
use std::fs;
use std::path::Path;
use std::sync::mpsc::channel;
use std::time::Duration;
2019-01-09 19:33:43 +00:00
mod exercise;
2019-01-09 19:33:43 +00:00
mod run;
2019-01-09 19:33:58 +00:00
mod verify;
2018-05-14 16:41:58 +00:00
2018-11-09 19:31:14 +00:00
fn main() {
let matches = App::new("rustlings")
2018-11-14 19:12:20 +00:00
.version(crate_version!())
2019-01-23 20:43:32 +00:00
.author("Olivia Hugger, Carol Nichols")
.about("Rustlings is a collection of small exercises to get you used to writing and reading Rust code")
.subcommand(SubCommand::with_name("verify").alias("v").about("Verifies all exercises according to the recommended order"))
.subcommand(SubCommand::with_name("watch").alias("w").about("Reruns `verify` when files were edited"))
2018-11-23 14:18:43 +00:00
.subcommand(
SubCommand::with_name("run")
.alias("r")
2019-01-23 20:43:32 +00:00
.about("Runs/Tests a single exercise")
.arg(Arg::with_name("name").required(true).index(1)),
2019-01-09 19:33:58 +00:00
)
.subcommand(
SubCommand::with_name("hint")
.alias("h")
.about("Returns a hint for the current exercise")
.arg(Arg::with_name("name").required(true).index(1)),
)
2019-01-09 19:33:58 +00:00
.get_matches();
2018-11-14 19:12:20 +00:00
2019-01-09 19:44:55 +00:00
if None == matches.subcommand_name() {
2019-03-11 14:09:20 +00:00
println!();
2019-01-09 19:44:55 +00:00
println!(r#" welcome to... "#);
println!(r#" _ _ _ "#);
println!(r#" _ __ _ _ ___| |_| (_)_ __ __ _ ___ "#);
println!(r#" | '__| | | / __| __| | | '_ \ / _` / __| "#);
println!(r#" | | | |_| \__ \ |_| | | | | | (_| \__ \ "#);
println!(r#" |_| \__,_|___/\__|_|_|_| |_|\__, |___/ "#);
println!(r#" |___/ "#);
2019-03-11 14:09:20 +00:00
println!();
2019-01-09 19:44:55 +00:00
}
2018-11-14 19:12:20 +00:00
if !Path::new("info.toml").exists() {
println!(
"{} must be run from the rustlings directory",
std::env::current_exe().unwrap().to_str().unwrap()
);
println!("Try `cd rustlings/`!");
std::process::exit(1);
}
let toml_str = &fs::read_to_string("info.toml").unwrap();
let exercises = toml::from_str::<ExerciseList>(toml_str).unwrap().exercises;
if let Some(ref matches) = matches.subcommand_matches("run") {
let name = matches.value_of("name").unwrap_or_else(|| {
println!("Please supply an exercise name!");
std::process::exit(1);
});
2019-04-12 21:24:13 +00:00
let matching_exercise = |e: &&Exercise| {
name == e.name
2019-04-12 21:24:13 +00:00
};
let exercise = exercises.iter().find(matching_exercise).unwrap_or_else(|| {
println!("No exercise found for your given name!");
2019-04-12 21:24:13 +00:00
std::process::exit(1)
});
run(&exercise).unwrap_or_else(|_| std::process::exit(1));
2018-11-23 14:18:43 +00:00
}
if let Some(ref matches) = matches.subcommand_matches("hint") {
let name = matches.value_of("name").unwrap_or_else(|| {
println!("Please supply an exercise name!");
std::process::exit(1);
});
let exercise = exercises.iter().find(|e| name == e.name).unwrap_or_else(|| {
println!("No exercise found for your given name!");
std::process::exit(1)
});
println!("{}", exercise.hint);
}
2019-03-11 14:09:20 +00:00
if matches.subcommand_matches("verify").is_some() {
verify(&exercises).unwrap_or_else(|_| std::process::exit(1));
}
2019-03-11 14:09:20 +00:00
if matches.subcommand_matches("watch").is_some() {
watch(&exercises).unwrap();
2018-11-14 19:12:20 +00:00
}
2019-03-11 14:09:20 +00:00
if matches.subcommand_name().is_none() {
2019-06-05 10:18:50 +00:00
let text = fs::read_to_string("default_out.txt").unwrap();
println!("{}", text);
}
2018-05-06 16:59:50 +00:00
}
fn watch(exercises: &[Exercise]) -> notify::Result<()> {
/* Clears the terminal with an ANSI escape code.
Works in UNIX and newer Windows terminals. */
fn clear_screen() {
println!("\x1Bc");
}
let (tx, rx) = channel();
let mut watcher: RecommendedWatcher = Watcher::new(tx, Duration::from_secs(2))?;
2019-03-27 09:58:56 +00:00
watcher.watch(Path::new("./exercises"), RecursiveMode::Recursive)?;
clear_screen();
let _ignored = verify(exercises.iter());
loop {
2019-03-15 23:01:45 +00:00
match rx.recv() {
2019-01-09 19:33:58 +00:00
Ok(event) => match event {
2019-03-06 18:38:55 +00:00
DebouncedEvent::Create(b) | DebouncedEvent::Chmod(b) | DebouncedEvent::Write(b) => {
if b.extension() == Some(OsStr::new("rs")) && b.exists() {
let filepath = b.as_path().canonicalize().unwrap();
let exercise = exercises
.iter()
.skip_while(|e| !filepath.ends_with(&e.path));
clear_screen();
let _ignored = verify(exercise);
2019-03-06 18:38:55 +00:00
}
}
2019-01-09 19:33:58 +00:00
_ => {}
},
Err(e) => println!("watch error: {:?}", e),
}
}
}