rustlings/src/run.rs

use anyhow::{bail, Result};
use std::io::{self, stdout, Write};
use std::time::Duration;

use crate::embedded::{WriteStrategy, EMBEDDED_FILES};
use crate::exercise::{Exercise, Mode};
use crate::verify::test;
use indicatif::ProgressBar;

// Invoke the rust compiler on the path of the given exercise,
// and run the ensuing binary.
// The verbose argument helps determine whether or not to show
// the output from the test harnesses (if the mode of the exercise is test)
pub fn run(exercise: &Exercise, verbose: bool) -> Result<()> {
    match exercise.mode {
        Mode::Test => test(exercise, verbose),
        Mode::Compile | Mode::Clippy => compile_and_run(exercise),
    }
}

// Resets the exercise by stashing the changes.
pub fn reset(exercise: &Exercise) -> io::Result<()> {
    EMBEDDED_FILES.write_exercise_to_disk(&exercise.path, WriteStrategy::Overwrite)
}

// Invoke the rust compiler on the path of the given exercise
// and run the ensuing binary.
// This is strictly for non-test binaries, so output is displayed
fn compile_and_run(exercise: &Exercise) -> Result<()> {
    let progress_bar = ProgressBar::new_spinner();
    progress_bar.set_message(format!("Running {exercise}..."));
    progress_bar.enable_steady_tick(Duration::from_millis(100));

    let output = exercise.run()?;
    progress_bar.finish_and_clear();

    stdout().write_all(&output.stdout)?;
    if !output.status.success() {
        stdout().write_all(&output.stderr)?;
        warn!("Ran {} with errors", exercise);
        bail!("TODO");
    }

    success!("Successfully ran {}", exercise);
    Ok(())
}