diff --git a/crates/nu-parser/src/parse_keywords.rs b/crates/nu-parser/src/parse_keywords.rs index ebb69b8e..d2025f84 100644 --- a/crates/nu-parser/src/parse_keywords.rs +++ b/crates/nu-parser/src/parse_keywords.rs @@ -331,7 +331,7 @@ pub fn parse_def( if let (Some(name), Some(mut signature), Some(block_id)) = (&name_expr.as_string(), sig.as_signature(), block.as_block()) { - if let Some(decl_id) = working_set.find_decl(name.as_bytes()) { + if let Some(decl_id) = working_set.find_predecl(name.as_bytes()) { let declaration = working_set.get_decl_mut(decl_id); signature.name = name.clone(); @@ -430,7 +430,7 @@ pub fn parse_extern( if let (Some(name_expr), Some(sig)) = (name_expr, sig) { if let (Some(name), Some(mut signature)) = (&name_expr.as_string(), sig.as_signature()) { - if let Some(decl_id) = working_set.find_decl(name.as_bytes()) { + if let Some(decl_id) = working_set.find_predecl(name.as_bytes()) { let declaration = working_set.get_decl_mut(decl_id); signature.name = name.clone(); diff --git a/crates/nu-protocol/src/engine/engine_state.rs b/crates/nu-protocol/src/engine/engine_state.rs index da5a0948..ed66f3ee 100644 --- a/crates/nu-protocol/src/engine/engine_state.rs +++ b/crates/nu-protocol/src/engine/engine_state.rs @@ -999,6 +999,16 @@ impl<'a> StateWorkingSet<'a> { self.delta.exit_scope(); } + pub fn find_predecl(&self, name: &[u8]) -> Option { + for scope in self.delta.scope.iter().rev() { + if let Some(decl_id) = scope.predecls.get(name) { + return Some(*decl_id); + } + } + + None + } + pub fn find_decl(&self, name: &[u8]) -> Option { let mut visibility: Visibility = Visibility::new();