diff options
author | Aspen Smith <root@gws.fyi> | 2024-07-06T00·29-0400 |
---|---|---|
committer | clbot <clbot@tvl.fyi> | 2024-07-06T15·03+0000 |
commit | dfe137786c98f7e610cf95dd7cbc1fc476a766fd (patch) | |
tree | 2b03183b7cac856f2c5dd509027e3df58fc5db5c /tvix/cli/src/main.rs | |
parent | d5964c1d548615aea1f3165c478ad91ad87822de (diff) |
refactor(tvix/eval): Builderize Evaluation r/8351
Make constructing of a new Evaluation use the builder pattern rather than setting public mutable fields. This is currently a pure refactor (no functionality has changed) but has a few advantages: - We've encapsulated the internals of the fields in Evaluation, meaning we can change them without too much breakage of clients - We have type safety that prevents us from ever changing the fields of an Evaluation after it's built (which matters more in a world where we reuse Evaluations). More importantly, this paves the road for doing different things with the construction of an Evaluation - notably, sharing certain things like the GlobalsMap across subsequent evaluations in eg the REPL. Fixes: b/262 Change-Id: I4a27116faac14cdd144fc7c992d14ae095a1aca4 Reviewed-on: https://cl.tvl.fyi/c/depot/+/11956 Tested-by: BuildkiteCI Autosubmit: aspen <root@gws.fyi> Reviewed-by: flokli <flokli@flokli.de>
Diffstat (limited to 'tvix/cli/src/main.rs')
-rw-r--r-- | tvix/cli/src/main.rs | 43 |
1 files changed, 22 insertions, 21 deletions
diff --git a/tvix/cli/src/main.rs b/tvix/cli/src/main.rs index c4c70d2a1777..2f7dc58a23aa 100644 --- a/tvix/cli/src/main.rs +++ b/tvix/cli/src/main.rs @@ -167,26 +167,25 @@ fn evaluate( span.pb_set_style(&tvix_tracing::PB_SPINNER_STYLE); span.pb_set_message("Setting up evaluator…"); - let mut eval = tvix_eval::Evaluation::new( - Box::new(TvixIO::new(tvix_store_io.clone() as Rc<dyn EvalIO>)) as Box<dyn EvalIO>, - true, - ); - eval.strict = args.strict; - eval.builtins.extend(impure_builtins()); - if let Some(env) = env { - eval.env = Some(env); - } - add_derivation_builtins(&mut eval, Rc::clone(&tvix_store_io)); - add_fetcher_builtins(&mut eval, Rc::clone(&tvix_store_io)); - add_import_builtins(&mut eval, tvix_store_io); - configure_nix_path(&mut eval, &args.nix_search_path); - - let source_map = eval.source_map(); + let mut eval_builder = tvix_eval::Evaluation::builder(Box::new(TvixIO::new( + tvix_store_io.clone() as Rc<dyn EvalIO>, + )) as Box<dyn EvalIO>) + .enable_import() + .with_strict(args.strict) + .add_builtins(impure_builtins()) + .env(env); + + eval_builder = add_derivation_builtins(eval_builder, Rc::clone(&tvix_store_io)); + eval_builder = add_fetcher_builtins(eval_builder, Rc::clone(&tvix_store_io)); + eval_builder = add_import_builtins(eval_builder, tvix_store_io); + eval_builder = configure_nix_path(eval_builder, &args.nix_search_path); + + let source_map = eval_builder.source_map().clone(); let result = { let mut compiler_observer = DisassemblingObserver::new(source_map.clone(), std::io::stderr()); if args.dump_bytecode { - eval.compiler_observer = Some(&mut compiler_observer); + eval_builder.set_compiler_observer(Some(&mut compiler_observer)); } let mut runtime_observer = TracingObserver::new(std::io::stderr()); @@ -194,10 +193,12 @@ fn evaluate( if args.trace_runtime_timing { runtime_observer.enable_timing() } - eval.runtime_observer = Some(&mut runtime_observer); + eval_builder.set_runtime_observer(Some(&mut runtime_observer)); } span.pb_set_message("Evaluating…"); + + let eval = eval_builder.build(); eval.evaluate(code, path) }; @@ -262,21 +263,21 @@ fn interpret( /// Interpret the given code snippet, but only run the Tvix compiler /// on it and return errors and warnings. fn lint(code: &str, path: Option<PathBuf>, args: &Args) -> bool { - let mut eval = tvix_eval::Evaluation::new_impure(); - eval.strict = args.strict; + let mut eval_builder = tvix_eval::Evaluation::builder_impure().with_strict(args.strict); - let source_map = eval.source_map(); + let source_map = eval_builder.source_map().clone(); let mut compiler_observer = DisassemblingObserver::new(source_map.clone(), std::io::stderr()); if args.dump_bytecode { - eval.compiler_observer = Some(&mut compiler_observer); + eval_builder.set_compiler_observer(Some(&mut compiler_observer)); } if args.trace_runtime { eprintln!("warning: --trace-runtime has no effect with --compile-only!"); } + let eval = eval_builder.build(); let result = eval.compile_only(code, path); if args.display_ast { |