Change parser cwd when running a file (#7134)

* Change parser cwd when running a file

* Add test

* Add missing file

crates/nu-cli/src/eval_file.rs

@@ -29,6 +29,8 @@ pub fn evaluate_file(
 
     let file = std::fs::read(&path).into_diagnostic()?;
 
+    engine_state.start_in_file(Some(&path));
+
     let mut working_set = StateWorkingSet::new(engine_state);
     trace!("parsing file: {}", path);
 

crates/nu-protocol/src/engine/engine_state.rs

@@ -92,6 +92,8 @@ pub struct EngineState {
     sig_quit: Option<Arc<AtomicBool>>,
     config_path: HashMap<String, PathBuf>,
     pub history_session_id: i64,
+    // If Nushell was started, e.g., with `nu spam.nu`, the file's parent is stored here
+    pub currently_parsed_cwd: Option<PathBuf>,
 }
 
 pub const NU_VARIABLE_ID: usize = 0;
@@ -134,6 +136,7 @@ impl EngineState {
             sig_quit: None,
             config_path: HashMap::new(),
             history_session_id: 0,
+            currently_parsed_cwd: None,
         }
     }
 
@@ -251,6 +254,15 @@ impl EngineState {
         Ok(())
     }
 
+    /// Mark a starting point if it is a script (e.g., nu spam.nu)
+    pub fn start_in_file(&mut self, file_path: Option<&str>) {
+        self.currently_parsed_cwd = if let Some(path) = file_path {
+            Path::new(path).parent().map(PathBuf::from)
+        } else {
+            None
+        };
+    }
+
     pub fn has_overlay(&self, name: &[u8]) -> bool {
         self.scope
             .overlays
@@ -987,7 +999,7 @@ impl<'a> StateWorkingSet<'a> {
             permanent_state,
             external_commands: vec![],
             type_scope: TypeScope::default(),
-            currently_parsed_cwd: None,
+            currently_parsed_cwd: permanent_state.currently_parsed_cwd.clone(),
             parsed_module_files: vec![],
         }
     }

tests/parsing/mod.rs

@@ -2,6 +2,15 @@ use nu_test_support::fs::Stub::FileWithContentToBeTrimmed;
 use nu_test_support::playground::Playground;
 use nu_test_support::{nu, pipeline};
 
+#[test]
+fn source_file_relative_to_file() {
+    let actual = nu!(cwd: "tests/parsing/samples", r#"
+        nu source_file_relative.nu
+        "#);
+
+    assert_eq!(actual.out, "5");
+}
+
 #[test]
 fn run_nu_script_single_line() {
     let actual = nu!(cwd: "tests/parsing/samples", r#"

tests/parsing/samples/source_file_relative.nu

@@ -0,0 +1 @@
+source single_line.nu