mirror of
https://github.com/nushell/nushell.git
synced 2025-05-01 00:24:29 +02:00
restrict plugin file name (#6479)
This commit is contained in:
parent
65327e0e7e
commit
a6ba58ec41
@ -3197,13 +3197,28 @@ pub fn parse_register(
|
|||||||
nu_engine::env::env_to_strings(working_set.permanent_state, &stack).unwrap_or_default();
|
nu_engine::env::env_to_strings(working_set.permanent_state, &stack).unwrap_or_default();
|
||||||
let error = match signature {
|
let error = match signature {
|
||||||
Some(signature) => arguments.and_then(|(path, encoding)| {
|
Some(signature) => arguments.and_then(|(path, encoding)| {
|
||||||
|
// restrict plugin file name starts with `nu_plugin_`
|
||||||
|
let f_name = path
|
||||||
|
.file_name()
|
||||||
|
.map(|s| s.to_string_lossy().starts_with("nu_plugin_"));
|
||||||
|
|
||||||
|
if let Some(true) = f_name {
|
||||||
signature.map(|signature| {
|
signature.map(|signature| {
|
||||||
let plugin_decl = PluginDeclaration::new(path, signature, encoding, shell);
|
let plugin_decl = PluginDeclaration::new(path, signature, encoding, shell);
|
||||||
working_set.add_decl(Box::new(plugin_decl));
|
working_set.add_decl(Box::new(plugin_decl));
|
||||||
working_set.mark_plugins_file_dirty();
|
working_set.mark_plugins_file_dirty();
|
||||||
})
|
})
|
||||||
|
} else {
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
}),
|
}),
|
||||||
None => arguments.and_then(|(path, encoding)| {
|
None => arguments.and_then(|(path, encoding)| {
|
||||||
|
// restrict plugin file name starts with `nu_plugin_`
|
||||||
|
let f_name = path
|
||||||
|
.file_name()
|
||||||
|
.map(|s| s.to_string_lossy().starts_with("nu_plugin_"));
|
||||||
|
|
||||||
|
if let Some(true) = f_name {
|
||||||
get_signature(path.as_path(), &encoding, &shell, ¤t_envs)
|
get_signature(path.as_path(), &encoding, &shell, ¤t_envs)
|
||||||
.map_err(|err| {
|
.map_err(|err| {
|
||||||
ParseError::LabeledError(
|
ParseError::LabeledError(
|
||||||
@ -3228,6 +3243,9 @@ pub fn parse_register(
|
|||||||
|
|
||||||
working_set.mark_plugins_file_dirty();
|
working_set.mark_plugins_file_dirty();
|
||||||
})
|
})
|
||||||
|
} else {
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
}),
|
}),
|
||||||
}
|
}
|
||||||
.err();
|
.err();
|
||||||
|
Loading…
Reference in New Issue
Block a user