mdebray/tvl-depot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore(tvix/cli): re-add --raw argument

Browse source 
Change-Id: I82f469f7f186c754c40bd941587b30aa3fc48045
Reviewed-on: https://cl.tvl.fyi/c/depot/+/7548
Reviewed-by: grfn <grfn@gws.fyi>
Tested-by: BuildkiteCI
This commit is contained in:
Vincent Ambo 2022-12-09 13:29:08 +03:00 committed by tazjin
parent 32dba980eb
commit 36eaa9b3b3
1 changed files with 15 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
tvix/cli/src/main.rs
View file

@ -11,15 +11,16 @@ struct Args {
#[clap(long, short = 'E')]
expr: Option<String>,
// TODO: port these options here directly
// #[clap(flatten)]
// eval_options: tvix_eval::Options,
/// Print "raw" (unquoted) output.
#[clap(long)]
raw: bool,
}
/// Interprets the given code snippet, printing out warnings, errors
/// and the result itself. The return value indicates whether
/// evaluation succeeded.
fn interpret(code: &str, path: Option<PathBuf>) -> bool {
fn interpret(code: &str, path: Option<PathBuf>, args: &Args) -> bool {
let eval = tvix_eval::Evaluation::new(code, path);
let source_map = eval.source_map();
let result = eval.evaluate();
@ -33,7 +34,7 @@ fn interpret(code: &str, path: Option<PathBuf>) -> bool {
}
if let Some(value) = result.value.as_ref() {
println_result(value, /* TODO raw = */ false);
println_result(value, args.raw);
}
// inform the caller about any errors
@ -43,24 +44,24 @@ fn interpret(code: &str, path: Option<PathBuf>) -> bool {
fn main() {
let args = Args::parse();
if let Some(file) = args.script {
run_file(file /* TODO, args.eval_options*/)
} else if let Some(expr) = args.expr {
if !interpret(&expr, None) {
if let Some(file) = &args.script {
run_file(file.clone(), &args)
} else if let Some(expr) = &args.expr {
if !interpret(expr, None, &args) {
std::process::exit(1);
}
} else {
run_prompt(/* TODO args.eval_options */)
run_prompt(&args)
}
}
fn run_file(mut path: PathBuf /* TODO: , eval_options: tvix_eval::Options */) {
fn run_file(mut path: PathBuf, args: &Args) {
if path.is_dir() {
path.push("default.nix");
}
let contents = fs::read_to_string(&path).expect("failed to read the input file");
if !interpret(&contents, Some(path)) {
if !interpret(&contents, Some(path), args) {
std::process::exit(1);
}
}
@ -81,7 +82,7 @@ fn state_dir() -> Option<PathBuf> {
path
}
fn run_prompt(/* TODO eval_options: tvix_eval::Options */) {
fn run_prompt(args: &Args) {
let mut rl = Editor::<()>::new().expect("should be able to launch rustyline");
let history_path = match state_dir() {
@ -106,7 +107,7 @@ fn run_prompt(/* TODO eval_options: tvix_eval::Options */) {
}
rl.add_history_entry(&line);
interpret(&line, None);
interpret(&line, None, args);
}
Err(ReadlineError::Interrupted) | Err(ReadlineError::Eof) => break,