1
0
Fork 0
mirror of https://github.com/notohh/rustlings.git synced 2024-12-21 16:08:09 -05:00
rustlings/src/app_state.rs

431 lines
14 KiB
Rust
Raw Normal View History

use anyhow::{bail, Context, Result};
2024-04-29 19:41:08 -04:00
use crossterm::style::Stylize;
2024-04-27 11:31:51 -04:00
use serde::Deserialize;
2024-04-13 23:13:27 -04:00
use std::{
fs::{self, File},
io::{Read, StdoutLock, Write},
path::{Path, PathBuf},
2024-04-18 06:41:17 -04:00
process::{Command, Stdio},
2024-04-13 23:13:27 -04:00
};
2024-04-24 19:56:01 -04:00
use crate::{
2024-04-29 19:41:08 -04:00
clear_terminal,
2024-04-24 19:56:01 -04:00
embedded::EMBEDDED_FILES,
exercise::{Exercise, OUTPUT_CAPACITY},
info_file::ExerciseInfo,
DEBUG_PROFILE,
};
2024-04-13 23:13:27 -04:00
const STATE_FILE_NAME: &str = ".rustlings-state.txt";
2024-04-13 19:15:43 -04:00
const BAD_INDEX_ERR: &str = "The current exercise index is higher than the number of exercises";
2024-04-12 14:06:56 -04:00
#[must_use]
pub enum ExercisesProgress {
AllDone,
Pending,
}
2024-04-14 10:03:49 -04:00
pub enum StateFileStatus {
Read,
NotRead,
}
2024-04-29 11:01:47 -04:00
// Parses parts of the output of `cargo metadata`.
2024-04-27 11:31:51 -04:00
#[derive(Deserialize)]
struct CargoMetadata {
target_directory: PathBuf,
}
pub struct AppState {
2024-04-13 19:15:43 -04:00
current_exercise_ind: usize,
exercises: Vec<Exercise>,
2024-04-29 11:01:47 -04:00
// Caches the number of done exercises to avoid iterating over all exercises every time.
n_done: u16,
2024-04-13 19:15:43 -04:00
final_message: String,
2024-04-29 11:01:47 -04:00
// Preallocated buffer for reading and writing the state file.
2024-04-13 23:13:27 -04:00
file_buf: Vec<u8>,
2024-04-18 06:41:17 -04:00
official_exercises: bool,
2024-04-29 11:01:47 -04:00
// Cargo's target directory.
target_dir: PathBuf,
}
impl AppState {
2024-04-29 11:01:47 -04:00
// Update the app state from the state file.
2024-04-14 10:03:49 -04:00
fn update_from_file(&mut self) -> StateFileStatus {
2024-04-13 23:13:27 -04:00
self.file_buf.clear();
self.n_done = 0;
if File::open(STATE_FILE_NAME)
.and_then(|mut file| file.read_to_end(&mut self.file_buf))
2024-04-14 10:03:49 -04:00
.is_err()
2024-04-13 23:13:27 -04:00
{
2024-04-14 10:03:49 -04:00
return StateFileStatus::NotRead;
}
2024-04-13 23:13:27 -04:00
2024-04-14 10:03:49 -04:00
// See `Self::write` for more information about the file format.
2024-04-21 17:39:44 -04:00
let mut lines = self.file_buf.split(|c| *c == b'\n').skip(2);
2024-04-14 10:03:49 -04:00
let Some(current_exercise_name) = lines.next() else {
return StateFileStatus::NotRead;
};
if current_exercise_name.is_empty() || lines.next().is_none() {
return StateFileStatus::NotRead;
}
2024-04-13 23:13:27 -04:00
2024-04-14 10:03:49 -04:00
let mut done_exercises = hashbrown::HashSet::with_capacity(self.exercises.len());
2024-04-13 23:13:27 -04:00
2024-04-14 10:03:49 -04:00
for done_exerise_name in lines {
if done_exerise_name.is_empty() {
break;
2024-04-13 23:13:27 -04:00
}
2024-04-14 10:03:49 -04:00
done_exercises.insert(done_exerise_name);
}
2024-04-13 23:13:27 -04:00
2024-04-14 10:03:49 -04:00
for (ind, exercise) in self.exercises.iter_mut().enumerate() {
if done_exercises.contains(exercise.name.as_bytes()) {
exercise.done = true;
self.n_done += 1;
}
2024-04-13 23:13:27 -04:00
2024-04-14 10:03:49 -04:00
if exercise.name.as_bytes() == current_exercise_name {
self.current_exercise_ind = ind;
2024-04-13 23:13:27 -04:00
}
}
2024-04-14 10:03:49 -04:00
StateFileStatus::Read
2024-04-13 23:13:27 -04:00
}
2024-04-14 10:03:49 -04:00
pub fn new(
exercise_infos: Vec<ExerciseInfo>,
final_message: String,
2024-04-27 11:31:51 -04:00
) -> Result<(Self, StateFileStatus)> {
2024-04-29 11:01:47 -04:00
// Get the target directory from Cargo.
2024-04-27 11:31:51 -04:00
let metadata_output = Command::new("cargo")
.arg("metadata")
.arg("-q")
.arg("--format-version")
.arg("1")
.arg("--no-deps")
.stdin(Stdio::null())
.stderr(Stdio::inherit())
.output()
.context(CARGO_METADATA_ERR)?
.stdout;
let target_dir = serde_json::de::from_slice::<CargoMetadata>(&metadata_output)
.context(
"Failed to read the field `target_directory` from the `cargo metadata` output",
)?
.target_directory;
2024-04-14 10:03:49 -04:00
let exercises = exercise_infos
2024-04-13 19:15:43 -04:00
.into_iter()
.map(Exercise::from)
2024-04-13 19:15:43 -04:00
.collect::<Vec<_>>();
2024-04-13 23:13:27 -04:00
let mut slf = Self {
current_exercise_ind: 0,
exercises,
2024-04-13 23:13:27 -04:00
n_done: 0,
2024-04-14 10:03:49 -04:00
final_message,
2024-04-13 23:13:27 -04:00
file_buf: Vec::with_capacity(2048),
2024-04-18 06:41:17 -04:00
official_exercises: !Path::new("info.toml").exists(),
target_dir,
2024-04-13 23:13:27 -04:00
};
2024-04-14 10:03:49 -04:00
let state_file_status = slf.update_from_file();
2024-04-13 23:13:27 -04:00
2024-04-27 11:31:51 -04:00
Ok((slf, state_file_status))
}
#[inline]
pub fn current_exercise_ind(&self) -> usize {
2024-04-13 19:15:43 -04:00
self.current_exercise_ind
}
#[inline]
2024-04-13 19:15:43 -04:00
pub fn exercises(&self) -> &[Exercise] {
&self.exercises
}
#[inline]
pub fn n_done(&self) -> u16 {
self.n_done
}
#[inline]
2024-04-13 19:15:43 -04:00
pub fn current_exercise(&self) -> &Exercise {
&self.exercises[self.current_exercise_ind]
}
#[inline]
pub fn target_dir(&self) -> &Path {
&self.target_dir
}
2024-04-29 11:01:47 -04:00
// Write the state file.
// The file's format is very simple:
// - The first line is a comment.
// - The second line is an empty line.
// - The third line is the name of the current exercise. It must end with `\n` even if there
// are no done exercises.
// - The fourth line is an empty line.
// - All remaining lines are the names of done exercises.
fn write(&mut self) -> Result<()> {
self.file_buf.clear();
self.file_buf
.extend_from_slice(b"DON'T EDIT THIS FILE!\n\n");
self.file_buf
.extend_from_slice(self.current_exercise().name.as_bytes());
self.file_buf.push(b'\n');
for exercise in &self.exercises {
if exercise.done {
self.file_buf.push(b'\n');
self.file_buf.extend_from_slice(exercise.name.as_bytes());
}
}
fs::write(STATE_FILE_NAME, &self.file_buf)
.with_context(|| format!("Failed to write the state file {STATE_FILE_NAME}"))?;
Ok(())
}
pub fn set_current_exercise_ind(&mut self, ind: usize) -> Result<()> {
if ind >= self.exercises.len() {
bail!(BAD_INDEX_ERR);
}
2024-04-13 19:15:43 -04:00
self.current_exercise_ind = ind;
2024-04-13 23:13:27 -04:00
self.write()
}
pub fn set_current_exercise_by_name(&mut self, name: &str) -> Result<()> {
2024-04-13 19:15:43 -04:00
// O(N) is fine since this method is used only once until the program exits.
// Building a hashmap would have more overhead.
self.current_exercise_ind = self
.exercises
.iter()
2024-04-13 19:15:43 -04:00
.position(|exercise| exercise.name == name)
.with_context(|| format!("No exercise found for '{name}'!"))?;
2024-04-13 23:13:27 -04:00
self.write()
}
pub fn set_pending(&mut self, ind: usize) -> Result<()> {
2024-04-13 19:15:43 -04:00
let exercise = self.exercises.get_mut(ind).context(BAD_INDEX_ERR)?;
if exercise.done {
exercise.done = false;
self.n_done -= 1;
2024-04-13 23:13:27 -04:00
self.write()?;
}
Ok(())
}
2024-04-29 11:01:47 -04:00
// Official exercises: Dump the original file from the binary.
// Third-party exercises: Reset the exercise file with `git stash`.
fn reset(&self, ind: usize, dir_name: Option<&str>, path: &str) -> Result<()> {
2024-04-18 06:41:17 -04:00
if self.official_exercises {
return EMBEDDED_FILES
.write_exercise_to_disk(
ind,
dir_name.context(
"Official exercises must be nested in the `exercises` directory",
)?,
path,
)
2024-04-18 06:41:17 -04:00
.with_context(|| format!("Failed to reset the exercise {path}"));
}
let output = Command::new("git")
.arg("stash")
.arg("push")
.arg("--")
.arg(path)
.stdin(Stdio::null())
.stdout(Stdio::null())
.output()
.with_context(|| format!("Failed to run `git stash push -- {path}`"))?;
if !output.status.success() {
bail!(
"`git stash push -- {path}` didn't run successfully: {}",
String::from_utf8_lossy(&output.stderr),
);
}
Ok(())
}
pub fn reset_current_exercise(&mut self) -> Result<&'static str> {
self.set_pending(self.current_exercise_ind)?;
let exercise = self.current_exercise();
self.reset(self.current_exercise_ind, exercise.dir, exercise.path)?;
2024-04-18 06:41:17 -04:00
Ok(exercise.path)
2024-04-18 06:41:17 -04:00
}
pub fn reset_exercise_by_ind(&mut self, exercise_ind: usize) -> Result<&'static str> {
if exercise_ind >= self.exercises.len() {
bail!(BAD_INDEX_ERR);
}
self.set_pending(exercise_ind)?;
let exercise = &self.exercises[exercise_ind];
self.reset(exercise_ind, exercise.dir, exercise.path)?;
2024-04-18 06:41:17 -04:00
Ok(exercise.path)
2024-04-18 06:41:17 -04:00
}
2024-04-29 11:01:47 -04:00
// Return the index of the next pending exercise or `None` if all exercises are done.
fn next_pending_exercise_ind(&self) -> Option<usize> {
2024-04-13 19:15:43 -04:00
if self.current_exercise_ind == self.exercises.len() - 1 {
// The last exercise is done.
// Search for exercises not done from the start.
2024-04-13 19:15:43 -04:00
return self.exercises[..self.current_exercise_ind]
.iter()
2024-04-13 19:15:43 -04:00
.position(|exercise| !exercise.done);
}
// The done exercise isn't the last one.
// Search for a pending exercise after the current one and then from the start.
2024-04-13 19:15:43 -04:00
match self.exercises[self.current_exercise_ind + 1..]
.iter()
2024-04-13 19:15:43 -04:00
.position(|exercise| !exercise.done)
{
2024-04-13 19:15:43 -04:00
Some(ind) => Some(self.current_exercise_ind + 1 + ind),
None => self.exercises[..self.current_exercise_ind]
.iter()
2024-04-13 19:15:43 -04:00
.position(|exercise| !exercise.done),
}
}
2024-04-29 11:01:47 -04:00
// Official exercises: Dump the solution file form the binary and return its path.
// Third-party exercises: Check if a solution file exists and return its path in that case.
2024-04-23 20:52:30 -04:00
pub fn current_solution_path(&self) -> Result<Option<String>> {
if DEBUG_PROFILE {
return Ok(None);
}
2024-04-23 20:52:30 -04:00
let current_exercise = self.current_exercise();
if self.official_exercises {
let dir_name = current_exercise
.dir
.context("Official exercises must be nested in the `exercises` directory")?;
let solution_path = format!("solutions/{dir_name}/{}.rs", current_exercise.name);
2024-04-23 18:47:46 -04:00
EMBEDDED_FILES.write_solution_to_disk(
self.current_exercise_ind,
2024-04-23 20:52:30 -04:00
dir_name,
&solution_path,
2024-04-23 18:47:46 -04:00
)?;
2024-04-23 20:52:30 -04:00
Ok(Some(solution_path))
} else {
let solution_path = if let Some(dir) = current_exercise.dir {
format!("solutions/{dir}/{}.rs", current_exercise.name)
} else {
format!("solutions/{}.rs", current_exercise.name)
};
if Path::new(&solution_path).exists() {
return Ok(Some(solution_path));
}
Ok(None)
}
}
2024-04-29 11:01:47 -04:00
// Mark the current exercise as done and move on to the next pending exercise if one exists.
// If all exercises are marked as done, run all of them to make sure that they are actually
// done. If an exercise which is marked as done fails, mark it as pending and continue on it.
2024-04-23 20:52:30 -04:00
pub fn done_current_exercise(&mut self, writer: &mut StdoutLock) -> Result<ExercisesProgress> {
let exercise = &mut self.exercises[self.current_exercise_ind];
if !exercise.done {
exercise.done = true;
self.n_done += 1;
2024-04-23 18:47:46 -04:00
}
2024-04-29 11:01:47 -04:00
if let Some(ind) = self.next_pending_exercise_ind() {
self.set_current_exercise_ind(ind)?;
2024-04-29 11:01:47 -04:00
return Ok(ExercisesProgress::Pending);
}
2024-04-12 13:30:29 -04:00
2024-04-29 11:01:47 -04:00
writer.write_all(RERUNNING_ALL_EXERCISES_MSG)?;
2024-04-29 11:01:47 -04:00
let mut output = Vec::with_capacity(OUTPUT_CAPACITY);
for (exercise_ind, exercise) in self.exercises().iter().enumerate() {
write!(writer, "Running {exercise} ... ")?;
writer.flush()?;
2024-04-29 11:01:47 -04:00
let success = exercise.run(&mut output, &self.target_dir)?;
if !success {
writeln!(writer, "{}\n", "FAILED".red())?;
2024-04-29 11:01:47 -04:00
self.current_exercise_ind = exercise_ind;
2024-04-29 11:01:47 -04:00
// No check if the exercise is done before setting it to pending
// because no pending exercise was found.
self.exercises[exercise_ind].done = false;
self.n_done -= 1;
2024-04-29 11:01:47 -04:00
self.write()?;
2024-04-14 10:04:05 -04:00
2024-04-29 11:01:47 -04:00
return Ok(ExercisesProgress::Pending);
2024-04-14 10:04:05 -04:00
}
2024-04-29 11:01:47 -04:00
writeln!(writer, "{}", "ok".green())?;
}
2024-04-13 23:13:27 -04:00
2024-04-29 19:41:08 -04:00
clear_terminal(writer)?;
2024-04-29 11:01:47 -04:00
writer.write_all(FENISH_LINE.as_bytes())?;
2024-04-13 23:13:27 -04:00
2024-04-29 11:01:47 -04:00
let final_message = self.final_message.trim();
if !final_message.is_empty() {
writer.write_all(final_message.as_bytes())?;
writer.write_all(b"\n")?;
2024-04-13 23:13:27 -04:00
}
2024-04-29 11:01:47 -04:00
Ok(ExercisesProgress::AllDone)
2024-04-13 23:13:27 -04:00
}
}
2024-04-27 11:31:51 -04:00
const CARGO_METADATA_ERR: &str = "Failed to run the command `cargo metadata …`
Did you already install Rust?
Try running `cargo --version` to diagnose the problem.";
const RERUNNING_ALL_EXERCISES_MSG: &[u8] = b"
All exercises seem to be done.
Recompiling and running all exercises to make sure that all of them are actually done.
";
2024-04-18 05:21:39 -04:00
const FENISH_LINE: &str = "+----------------------------------------------------+
| You made it to the Fe-nish line! |
+-------------------------- ------------------------+
\\/\x1b[31m
\x1b[0m
";