use std::{ fs, path::{Path, PathBuf}, }; use clap::Parser; use rustyline::{error::ReadlineError, Editor}; #[derive(Parser)] struct Args { /// Path to a script to evaluate script: Option<PathBuf>, #[clap(flatten)] eval_options: tvix_eval::Options, } fn main() { let args = Args::parse(); if let Some(file) = &args.script { run_file(file, args.eval_options) } else { run_prompt(args.eval_options) } } fn run_file(file: &Path, eval_options: tvix_eval::Options) { let contents = fs::read_to_string(file).expect("failed to read the input file"); let path = Path::new(file).to_owned(); match tvix_eval::interpret(&contents, Some(path), eval_options) { Ok(result) => println!("=> {} :: {}", result, result.type_of()), Err(err) => eprintln!("{}", err), } } fn state_dir() -> Option<PathBuf> { let mut path = dirs::data_dir(); if let Some(p) = path.as_mut() { p.push("tvix") } path } fn run_prompt(eval_options: tvix_eval::Options) { let mut rl = Editor::<()>::new().expect("should be able to launch rustyline"); let history_path = match state_dir() { // Attempt to set up these paths, but do not hard fail if it // doesn't work. Some(mut path) => { let _ = std::fs::create_dir_all(&path); path.push("history.txt"); let _ = rl.load_history(&path); Some(path) } None => None, }; loop { let readline = rl.readline("tvix-repl> "); match readline { Ok(line) => { if line.is_empty() { continue; } rl.add_history_entry(&line); match tvix_eval::interpret(&line, None, eval_options) { Ok(result) => { println!("=> {} :: {}", result, result.type_of()); } Err(err) => println!("{}", err), } } Err(ReadlineError::Interrupted) | Err(ReadlineError::Eof) => break, Err(err) => { eprintln!("error: {}", err); break; } } } if let Some(path) = history_path { rl.save_history(&path).unwrap(); } }