7928122fce
I’ve been wanting to do this for a while, but always procrastinated on it. We’ve been using Clap since the 2.0 rewrite, but Clap is known to be a fairly heavy library. Since Rustlings is usually peoples’ first contact with a Rust compilation, I think it’s in our best interests that this complation is as fast as possible. In effect, replacing Clap with the smaller, structopt-style `argh` reduces the amount of crates needing to be compiled from 82 to 60. I also think this makes the code way easier to read, we don’t need to use Clap’s methods anymore, but can switch over to using pure Rust methods, e.g., switches are booleans, options are Option<String>s or the like, and subcommands are just structs.
361 lines
14 KiB
Rust
361 lines
14 KiB
Rust
use crate::exercise::{Exercise, ExerciseList};
|
|
use crate::run::run;
|
|
use crate::verify::verify;
|
|
use argh::FromArgs;
|
|
use console::Emoji;
|
|
use notify::DebouncedEvent;
|
|
use notify::{RecommendedWatcher, RecursiveMode, Watcher};
|
|
use std::ffi::OsStr;
|
|
use std::fs;
|
|
use std::io::{self, prelude::*};
|
|
use std::path::Path;
|
|
use std::process::{Command, Stdio};
|
|
use std::sync::mpsc::channel;
|
|
use std::sync::{Arc, Mutex};
|
|
use std::thread;
|
|
use std::time::Duration;
|
|
|
|
#[macro_use]
|
|
mod ui;
|
|
|
|
mod exercise;
|
|
mod run;
|
|
mod verify;
|
|
|
|
// In sync with crate version
|
|
const VERSION: &str = "4.3.0";
|
|
|
|
#[derive(FromArgs, PartialEq, Debug)]
|
|
/// Rustlings is a collection of small exercises to get you used to writing and reading Rust code
|
|
struct Args {
|
|
/// show outputs from the test exercises
|
|
#[argh(switch)]
|
|
nocapture: bool,
|
|
/// show the executable version
|
|
#[argh(switch, short = 'v')]
|
|
version: bool,
|
|
#[argh(subcommand)]
|
|
nested: Option<Subcommands>,
|
|
}
|
|
|
|
#[derive(FromArgs, PartialEq, Debug)]
|
|
#[argh(subcommand)]
|
|
enum Subcommands {
|
|
Verify(VerifyArgs),
|
|
Watch(WatchArgs),
|
|
Run(RunArgs),
|
|
Hint(HintArgs),
|
|
List(ListArgs),
|
|
}
|
|
|
|
#[derive(FromArgs, PartialEq, Debug)]
|
|
#[argh(subcommand, name = "verify")]
|
|
/// Verifies all exercises according to the recommended order
|
|
struct VerifyArgs {}
|
|
|
|
#[derive(FromArgs, PartialEq, Debug)]
|
|
#[argh(subcommand, name = "watch")]
|
|
/// Reruns `verify` when files were edited
|
|
struct WatchArgs {}
|
|
|
|
#[derive(FromArgs, PartialEq, Debug)]
|
|
#[argh(subcommand, name = "run")]
|
|
/// Runs/Tests a single exercise
|
|
struct RunArgs {
|
|
#[argh(positional)]
|
|
/// the name of the exercise
|
|
name: String,
|
|
}
|
|
|
|
#[derive(FromArgs, PartialEq, Debug)]
|
|
#[argh(subcommand, name = "hint")]
|
|
/// Returns a hint for the given exercise
|
|
struct HintArgs {
|
|
#[argh(positional)]
|
|
/// the name of the exercise
|
|
name: String,
|
|
}
|
|
|
|
#[derive(FromArgs, PartialEq, Debug)]
|
|
#[argh(subcommand, name = "list")]
|
|
/// Lists the exercises available in Rustlings
|
|
struct ListArgs {
|
|
#[argh(switch, short = 'p')]
|
|
/// show only the paths of the exercises
|
|
paths: bool,
|
|
#[argh(switch, short = 'n')]
|
|
/// show only the names of the exercises
|
|
names: bool,
|
|
#[argh(option, short = 'f')]
|
|
/// provide a string to match exercise names
|
|
/// comma separated patterns are acceptable
|
|
filter: Option<String>,
|
|
#[argh(switch, short = 'u')]
|
|
/// display only exercises not yet solved
|
|
unsolved: bool,
|
|
#[argh(switch, short = 's')]
|
|
/// display only exercises that have been solved
|
|
solved: bool,
|
|
}
|
|
|
|
fn main() {
|
|
let args: Args = argh::from_env();
|
|
|
|
if args.version {
|
|
println!("v{}", VERSION);
|
|
std::process::exit(0);
|
|
}
|
|
|
|
if args.nested.is_none() {
|
|
println!();
|
|
println!(r#" welcome to... "#);
|
|
println!(r#" _ _ _ "#);
|
|
println!(r#" _ __ _ _ ___| |_| (_)_ __ __ _ ___ "#);
|
|
println!(r#" | '__| | | / __| __| | | '_ \ / _` / __| "#);
|
|
println!(r#" | | | |_| \__ \ |_| | | | | | (_| \__ \ "#);
|
|
println!(r#" |_| \__,_|___/\__|_|_|_| |_|\__, |___/ "#);
|
|
println!(r#" |___/ "#);
|
|
println!();
|
|
}
|
|
|
|
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);
|
|
}
|
|
|
|
if !rustc_exists() {
|
|
println!("We cannot find `rustc`.");
|
|
println!("Try running `rustc --version` to diagnose your problem.");
|
|
println!("For instructions on how to install Rust, check the README.");
|
|
std::process::exit(1);
|
|
}
|
|
|
|
let toml_str = &fs::read_to_string("info.toml").unwrap();
|
|
let exercises = toml::from_str::<ExerciseList>(toml_str).unwrap().exercises;
|
|
let verbose = args.nocapture;
|
|
|
|
let command = args.nested.unwrap_or_else(|| {
|
|
let text = fs::read_to_string("default_out.txt").unwrap();
|
|
println!("{}", text);
|
|
std::process::exit(0);
|
|
});
|
|
match command {
|
|
Subcommands::List(subargs) => {
|
|
if !subargs.paths && !subargs.names {
|
|
println!("{:<17}\t{:<46}\t{:<7}", "Name", "Path", "Status");
|
|
}
|
|
let mut exercises_done: u16 = 0;
|
|
let filters = subargs.filter.clone().unwrap_or_default().to_lowercase();
|
|
exercises.iter().for_each(|e| {
|
|
let fname = format!("{}", e.path.display());
|
|
let filter_cond = filters
|
|
.split(',')
|
|
.filter(|f| !f.trim().is_empty())
|
|
.any(|f| e.name.contains(&f) || fname.contains(&f));
|
|
let status = if e.looks_done() {
|
|
exercises_done += 1;
|
|
"Done"
|
|
} else {
|
|
"Pending"
|
|
};
|
|
let solve_cond = {
|
|
(e.looks_done() && subargs.solved)
|
|
|| (!e.looks_done() && subargs.unsolved)
|
|
|| (!subargs.solved && !subargs.unsolved)
|
|
};
|
|
if solve_cond && (filter_cond || subargs.filter.is_none()) {
|
|
let line = if subargs.paths {
|
|
format!("{}\n", fname)
|
|
} else if subargs.names {
|
|
format!("{}\n", e.name)
|
|
} else {
|
|
format!("{:<17}\t{:<46}\t{:<7}\n", e.name, fname, status)
|
|
};
|
|
// Somehow using println! leads to the binary panicking
|
|
// when its output is piped.
|
|
// So, we're handling a Broken Pipe error and exiting with 0 anyway
|
|
let stdout = std::io::stdout();
|
|
{
|
|
let mut handle = stdout.lock();
|
|
handle.write_all(line.as_bytes()).unwrap_or_else(|e| {
|
|
match e.kind() {
|
|
std::io::ErrorKind::BrokenPipe => std::process::exit(0),
|
|
_ => std::process::exit(1),
|
|
};
|
|
});
|
|
}
|
|
}
|
|
});
|
|
let percentage_progress = exercises_done as f32 / exercises.len() as f32 * 100.0;
|
|
println!(
|
|
"Progress: You completed {} / {} exercises ({:.2} %).",
|
|
exercises_done,
|
|
exercises.len(),
|
|
percentage_progress
|
|
);
|
|
std::process::exit(0);
|
|
}
|
|
|
|
Subcommands::Run(subargs) => {
|
|
let exercise = find_exercise(subargs.name, exercises);
|
|
|
|
run(&exercise, verbose).unwrap_or_else(|_| std::process::exit(1));
|
|
}
|
|
|
|
Subcommands::Hint(subargs) => {
|
|
let exercise = find_exercise(subargs.name, exercises);
|
|
|
|
println!("{}", exercise.hint);
|
|
}
|
|
|
|
Subcommands::Verify(_subargs) => {
|
|
verify(&exercises, verbose).unwrap_or_else(|_| std::process::exit(1));
|
|
}
|
|
|
|
Subcommands::Watch(_subargs) => {
|
|
if let Err(e) = watch(&exercises, verbose) {
|
|
println!(
|
|
"Error: Could not watch your progress. Error message was {:?}.",
|
|
e
|
|
);
|
|
println!("Most likely you've run out of disk space or your 'inotify limit' has been reached.");
|
|
std::process::exit(1);
|
|
}
|
|
println!(
|
|
"{emoji} All exercises completed! {emoji}",
|
|
emoji = Emoji("🎉", "★")
|
|
);
|
|
println!();
|
|
println!("+----------------------------------------------------+");
|
|
println!("| You made it to the Fe-nish line! |");
|
|
println!("+-------------------------- ------------------------+");
|
|
println!(" \\/ ");
|
|
println!(" ▒▒ ▒▒▒▒▒▒▒▒ ▒▒▒▒▒▒▒▒ ▒▒ ");
|
|
println!(" ▒▒▒▒ ▒▒ ▒▒ ▒▒ ▒▒ ▒▒ ▒▒ ▒▒▒▒ ");
|
|
println!(" ▒▒▒▒ ▒▒ ▒▒ ▒▒ ▒▒ ▒▒ ▒▒▒▒ ");
|
|
println!(" ░░▒▒▒▒░░▒▒ ▒▒ ▒▒ ▒▒ ▒▒░░▒▒▒▒ ");
|
|
println!(" ▓▓▓▓▓▓▓▓ ▓▓ ▓▓██ ▓▓ ▓▓██ ▓▓ ▓▓▓▓▓▓▓▓ ");
|
|
println!(" ▒▒▒▒ ▒▒ ████ ▒▒ ████ ▒▒░░ ▒▒▒▒ ");
|
|
println!(" ▒▒ ▒▒▒▒▒▒ ▒▒▒▒▒▒ ▒▒▒▒▒▒ ▒▒ ");
|
|
println!(" ▒▒▒▒▒▒▒▒▒▒▓▓▓▓▓▓▒▒▒▒▒▒▒▒▓▓▒▒▓▓▒▒▒▒▒▒▒▒ ");
|
|
println!(" ▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒ ");
|
|
println!(" ▒▒▒▒▒▒▒▒▒▒██▒▒▒▒▒▒██▒▒▒▒▒▒▒▒▒▒ ");
|
|
println!(" ▒▒ ▒▒▒▒▒▒▒▒▒▒██████▒▒▒▒▒▒▒▒▒▒ ▒▒ ");
|
|
println!(" ▒▒ ▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒ ▒▒ ");
|
|
println!(" ▒▒ ▒▒ ▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒ ▒▒ ▒▒ ");
|
|
println!(" ▒▒ ▒▒ ▒▒ ▒▒ ▒▒ ▒▒ ");
|
|
println!(" ▒▒ ▒▒ ▒▒ ▒▒ ");
|
|
println!();
|
|
println!("We hope you enjoyed learning about the various aspects of Rust!");
|
|
println!(
|
|
"If you noticed any issues, please don't hesitate to report them to our repo."
|
|
);
|
|
println!("You can also contribute your own exercises to help the greater community!");
|
|
println!();
|
|
println!("Before reporting an issue or contributing, please read our guidelines:");
|
|
println!("https://github.com/rust-lang/rustlings/blob/main/CONTRIBUTING.md");
|
|
}
|
|
}
|
|
}
|
|
|
|
fn spawn_watch_shell(failed_exercise_hint: &Arc<Mutex<Option<String>>>) {
|
|
let failed_exercise_hint = Arc::clone(failed_exercise_hint);
|
|
println!("Type 'hint' to get help or 'clear' to clear the screen");
|
|
thread::spawn(move || loop {
|
|
let mut input = String::new();
|
|
match io::stdin().read_line(&mut input) {
|
|
Ok(_) => {
|
|
let input = input.trim();
|
|
if input.eq("hint") {
|
|
if let Some(hint) = &*failed_exercise_hint.lock().unwrap() {
|
|
println!("{}", hint);
|
|
}
|
|
} else if input.eq("clear") {
|
|
println!("\x1B[2J\x1B[1;1H");
|
|
} else {
|
|
println!("unknown command: {}", input);
|
|
}
|
|
}
|
|
Err(error) => println!("error reading command: {}", error),
|
|
}
|
|
});
|
|
}
|
|
|
|
fn find_exercise(name: String, exercises: Vec<Exercise>) -> Exercise {
|
|
let matching_exercise = |e: &Exercise| name == e.name;
|
|
|
|
exercises
|
|
.into_iter()
|
|
.find(matching_exercise)
|
|
.unwrap_or_else(|| {
|
|
println!("No exercise found for your given name!");
|
|
std::process::exit(1)
|
|
})
|
|
}
|
|
|
|
fn watch(exercises: &[Exercise], verbose: bool) -> 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))?;
|
|
watcher.watch(Path::new("./exercises"), RecursiveMode::Recursive)?;
|
|
|
|
clear_screen();
|
|
|
|
let to_owned_hint = |t: &Exercise| t.hint.to_owned();
|
|
let failed_exercise_hint = match verify(exercises.iter(), verbose) {
|
|
Ok(_) => return Ok(()),
|
|
Err(exercise) => Arc::new(Mutex::new(Some(to_owned_hint(exercise)))),
|
|
};
|
|
spawn_watch_shell(&failed_exercise_hint);
|
|
loop {
|
|
match rx.recv() {
|
|
Ok(event) => match event {
|
|
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 pending_exercises = exercises
|
|
.iter()
|
|
.skip_while(|e| !filepath.ends_with(&e.path))
|
|
// .filter(|e| filepath.ends_with(&e.path))
|
|
.chain(
|
|
exercises
|
|
.iter()
|
|
.filter(|e| !e.looks_done() && !filepath.ends_with(&e.path)),
|
|
);
|
|
clear_screen();
|
|
match verify(pending_exercises, verbose) {
|
|
Ok(_) => return Ok(()),
|
|
Err(exercise) => {
|
|
let mut failed_exercise_hint = failed_exercise_hint.lock().unwrap();
|
|
*failed_exercise_hint = Some(to_owned_hint(exercise));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
_ => {}
|
|
},
|
|
Err(e) => println!("watch error: {:?}", e),
|
|
}
|
|
}
|
|
}
|
|
|
|
fn rustc_exists() -> bool {
|
|
Command::new("rustc")
|
|
.args(&["--version"])
|
|
.stdout(Stdio::null())
|
|
.spawn()
|
|
.and_then(|mut child| child.wait())
|
|
.map(|status| status.success())
|
|
.unwrap_or(false)
|
|
}
|