From 8bd7ced1fb0a5b598ec305a0b34d35f1efb1ce16 Mon Sep 17 00:00:00 2001 From: Florian Klink Date: Sun, 14 May 2023 19:55:17 +0300 Subject: feat(tvix/eval/io): allow &mut self in EvalIO It's okay if these calls mutate some internal state inside an implementation. Change-Id: I12bb11bde0310778c3da1275696bf7de058863a3 Reviewed-on: https://cl.tvl.fyi/c/depot/+/8571 Tested-by: BuildkiteCI Reviewed-by: tazjin --- tvix/eval/src/io.rs | 24 ++++++++++++------------ tvix/eval/src/nix_search_path.rs | 28 ++++++++++++++++------------ tvix/eval/src/vm/mod.rs | 2 +- 3 files changed, 29 insertions(+), 25 deletions(-) (limited to 'tvix/eval/src') diff --git a/tvix/eval/src/io.rs b/tvix/eval/src/io.rs index 8268b5990d57..789f83b2e8ed 100644 --- a/tvix/eval/src/io.rs +++ b/tvix/eval/src/io.rs @@ -34,14 +34,14 @@ pub enum FileType { /// Defines how filesystem interaction occurs inside of tvix-eval. pub trait EvalIO { /// Verify whether the file at the specified path exists. - fn path_exists(&self, path: PathBuf) -> Result; + fn path_exists(&mut self, path: PathBuf) -> Result; /// Read the file at the specified path to a string. - fn read_to_string(&self, path: PathBuf) -> Result; + fn read_to_string(&mut self, path: PathBuf) -> Result; /// Read the directory at the specified path and return the names /// of its entries associated with their [`FileType`]. - fn read_dir(&self, path: PathBuf) -> Result, ErrorKind>; + fn read_dir(&mut self, path: PathBuf) -> Result, ErrorKind>; /// Import the given path. What this means depends on the /// implementation, for example for a `std::io`-based @@ -50,7 +50,7 @@ pub trait EvalIO { /// /// This is primarily used in the context of things like coercing /// a local path to a string, or builtins like `path`. - fn import_path(&self, path: &Path) -> Result; + fn import_path(&mut self, path: &Path) -> Result; /// Returns the root of the store directory, if such a thing /// exists in the evaluation context. @@ -66,21 +66,21 @@ pub struct StdIO; #[cfg(feature = "impure")] impl EvalIO for StdIO { - fn path_exists(&self, path: PathBuf) -> Result { + fn path_exists(&mut self, path: PathBuf) -> Result { path.try_exists().map_err(|e| ErrorKind::IO { path: Some(path), error: Rc::new(e), }) } - fn read_to_string(&self, path: PathBuf) -> Result { + fn read_to_string(&mut self, path: PathBuf) -> Result { std::fs::read_to_string(&path).map_err(|e| ErrorKind::IO { path: Some(path), error: Rc::new(e), }) } - fn read_dir(&self, path: PathBuf) -> Result, ErrorKind> { + fn read_dir(&mut self, path: PathBuf) -> Result, ErrorKind> { let mut result = vec![]; let mk_err = |err| ErrorKind::IO { @@ -116,7 +116,7 @@ impl EvalIO for StdIO { // this is a no-op for `std::io`, as the user can already refer to // the path directly - fn import_path(&self, path: &Path) -> Result { + fn import_path(&mut self, path: &Path) -> Result { Ok(path.to_path_buf()) } } @@ -126,25 +126,25 @@ impl EvalIO for StdIO { pub struct DummyIO; impl EvalIO for DummyIO { - fn path_exists(&self, _: PathBuf) -> Result { + fn path_exists(&mut self, _: PathBuf) -> Result { Err(ErrorKind::NotImplemented( "I/O methods are not implemented in DummyIO", )) } - fn read_to_string(&self, _: PathBuf) -> Result { + fn read_to_string(&mut self, _: PathBuf) -> Result { Err(ErrorKind::NotImplemented( "I/O methods are not implemented in DummyIO", )) } - fn read_dir(&self, _: PathBuf) -> Result, ErrorKind> { + fn read_dir(&mut self, _: PathBuf) -> Result, ErrorKind> { Err(ErrorKind::NotImplemented( "I/O methods are not implemented in DummyIO", )) } - fn import_path(&self, _: &Path) -> Result { + fn import_path(&mut self, _: &Path) -> Result { Err(ErrorKind::NotImplemented( "I/O methods are not implemented in DummyIO", )) diff --git a/tvix/eval/src/nix_search_path.rs b/tvix/eval/src/nix_search_path.rs index 9e588d2242f0..5a650a55c7ba 100644 --- a/tvix/eval/src/nix_search_path.rs +++ b/tvix/eval/src/nix_search_path.rs @@ -63,7 +63,11 @@ impl NixSearchPathEntry { /// /// For prefixed path, an entry matches if the prefix does. // TODO(tazjin): verify these rules in the C++ impl, seems fishy. - fn resolve(&self, io: &dyn EvalIO, lookup_path: &Path) -> Result, ErrorKind> { + fn resolve( + &self, + io: &mut dyn EvalIO, + lookup_path: &Path, + ) -> Result, ErrorKind> { let path = match self { NixSearchPathEntry::Path(parent) => canonicalise(parent.join(lookup_path))?, @@ -112,7 +116,7 @@ pub struct NixSearchPath { impl NixSearchPath { /// Attempt to resolve the given `path` within this [`NixSearchPath`] using the /// path resolution rules for `<...>`-style paths - pub fn resolve

(&self, io: &dyn EvalIO, path: P) -> Result + pub fn resolve

(&self, io: &mut dyn EvalIO, path: P) -> Result where P: AsRef, { @@ -188,16 +192,16 @@ mod tests { #[test] fn simple_dir() { let nix_search_path = NixSearchPath::from_str("./.").unwrap(); - let io = StdIO {}; - let res = nix_search_path.resolve(&io, "src").unwrap(); + let mut io = StdIO {}; + let res = nix_search_path.resolve(&mut io, "src").unwrap(); assert_eq!(res, current_dir().unwrap().join("src").clean()); } #[test] fn failed_resolution() { let nix_search_path = NixSearchPath::from_str("./.").unwrap(); - let io = StdIO {}; - let err = nix_search_path.resolve(&io, "nope").unwrap_err(); + let mut io = StdIO {}; + let err = nix_search_path.resolve(&mut io, "nope").unwrap_err(); assert!( matches!(err, ErrorKind::NixPathResolution(..)), "err = {err:?}" @@ -207,24 +211,24 @@ mod tests { #[test] fn second_in_path() { let nix_search_path = NixSearchPath::from_str("./.:/").unwrap(); - let io = StdIO {}; - let res = nix_search_path.resolve(&io, "etc").unwrap(); + let mut io = StdIO {}; + let res = nix_search_path.resolve(&mut io, "etc").unwrap(); assert_eq!(res, Path::new("/etc")); } #[test] fn prefix() { let nix_search_path = NixSearchPath::from_str("/:tvix=.").unwrap(); - let io = StdIO {}; - let res = nix_search_path.resolve(&io, "tvix/src").unwrap(); + let mut io = StdIO {}; + let res = nix_search_path.resolve(&mut io, "tvix/src").unwrap(); assert_eq!(res, current_dir().unwrap().join("src").clean()); } #[test] fn matching_prefix() { let nix_search_path = NixSearchPath::from_str("/:tvix=.").unwrap(); - let io = StdIO {}; - let res = nix_search_path.resolve(&io, "tvix").unwrap(); + let mut io = StdIO {}; + let res = nix_search_path.resolve(&mut io, "tvix").unwrap(); assert_eq!(res, current_dir().unwrap().clean()); } } diff --git a/tvix/eval/src/vm/mod.rs b/tvix/eval/src/vm/mod.rs index 81c9ef39d94d..4af23a72d73b 100644 --- a/tvix/eval/src/vm/mod.rs +++ b/tvix/eval/src/vm/mod.rs @@ -834,7 +834,7 @@ impl<'o> VM<'o> { Value::UnresolvedPath(path) => { let resolved = self .nix_search_path - .resolve(&*self.io_handle, *path) + .resolve(&mut *self.io_handle, *path) .with_span(&frame, self)?; self.stack.push(resolved.into()); } -- cgit 1.4.1