rust-crafting-interpreters-part-1/src/main.rs

mod ast;
mod errors;
mod parser;
mod scanner;
mod tokens;

use std::{
    env, fs,
    io::{self, Write},
    process::ExitCode,
};

use ast::AstDumper;
use errors::ErrorHandler;
use parser::Parser;
use scanner::Scanner;

/// Execute a script.
fn run(source: String) -> ErrorHandler {
    let mut error_handler = ErrorHandler::default();

    let scanner = Scanner::new(source);
    let tokens = scanner.scan_tokens(&mut error_handler);

    for token in tokens.iter() {
        println!("{:#?}", token);
    }

    let parser = Parser::new(tokens);
    match parser.parse(&mut error_handler) {
        None => (),
        Some(ast) => {
            println!("AST generated ! {}", ast.dump())
        }
    }

    error_handler
}

/// Run the REPL.
fn run_prompt() {
    let mut buffer = String::new();
    let stdin = io::stdin();
    let mut stdout = io::stdout();
    loop {
        print!("slox> ");
        stdout.flush().unwrap();
        buffer.clear();
        let n_read = stdin
            .read_line(&mut buffer)
            .expect("Failed to read from stdin");
        let _ = match n_read {
            0 => return,
            _ => run(buffer.clone()),
        };
    }
}

/// Load a file and run the script it contains.
fn run_file(file: &str) -> ErrorHandler {
    let contents = fs::read_to_string(file)
        .unwrap_or_else(|_| panic!("Could not load {}", file));
    run(contents)
}

/// Main program. Either load a script from a file, or start the REPL.
fn main() -> Result<(), ExitCode> {
    let args: Vec<String> = env::args().skip(1).collect();
    let n_args = args.len();
    if n_args == 0 {
        run_prompt()
    } else if n_args == 1 {
        if run_file(&args[0]).had_error() {
            return Err(ExitCode::from(65));
        }
    } else {
        println!("Usage: slox [script]");
    }
    Ok(())
}