diff --git a/crates/nu-plugin-engine/src/interface/mod.rs b/crates/nu-plugin-engine/src/interface/mod.rs index 78ed966dd6..0fe4aee1eb 100644 --- a/crates/nu-plugin-engine/src/interface/mod.rs +++ b/crates/nu-plugin-engine/src/interface/mod.rs @@ -12,7 +12,7 @@ use nu_plugin_protocol::{ }; use nu_protocol::{ ast::Operator, engine::Sequence, CustomValue, IntoSpanned, PipelineData, PluginMetadata, - PluginSignature, ShellError, Signals, Span, Spanned, Value, + PluginSignature, ShellError, SignalAction, Signals, Span, Spanned, Value, }; use nu_utils::SharedCow; use std::{ @@ -665,8 +665,8 @@ impl PluginInterface { } /// Send the plugin a ctrl-c signal. - pub fn ctrlc(&self) -> Result<(), ShellError> { - self.write(PluginInput::Ctrlc)?; + pub fn signal(&self, action: SignalAction) -> Result<(), ShellError> { + self.write(PluginInput::Signal(action))?; self.flush() } diff --git a/crates/nu-plugin-engine/src/persistent.rs b/crates/nu-plugin-engine/src/persistent.rs index 69f5c7f7a8..943e4dad7d 100644 --- a/crates/nu-plugin-engine/src/persistent.rs +++ b/crates/nu-plugin-engine/src/persistent.rs @@ -6,8 +6,9 @@ use crate::{ use super::{PluginInterface, PluginSource}; use nu_plugin_core::CommunicationMode; use nu_protocol::{ - engine::{ctrlc, EngineState, Stack}, - PluginGcConfig, PluginIdentity, PluginMetadata, RegisteredPlugin, ShellError, + engine::{EngineState, Stack}, + HandlerGuard, Handlers, PluginGcConfig, PluginIdentity, PluginMetadata, RegisteredPlugin, + ShellError, }; use std::{ collections::HashMap, @@ -37,8 +38,8 @@ struct MutableState { preferred_mode: Option, /// Garbage collector config gc_config: PluginGcConfig, - /// RAII guard for this plugin's ctrl-c handler - ctrlc_guard: Option, + /// RAII guard for this plugin's signal handler + signal_guard: Option, } #[derive(Debug, Clone, Copy)] @@ -66,7 +67,7 @@ impl PersistentPlugin { metadata: None, preferred_mode: None, gc_config, - ctrlc_guard: None, + signal_guard: None, }), } } @@ -303,21 +304,18 @@ impl RegisteredPlugin for PersistentPlugin { self } - fn configure_ctrlc_handler( - self: Arc, - handlers: &ctrlc::Handlers, - ) -> Result<(), ShellError> { + fn configure_signal_handler(self: Arc, handlers: &Handlers) -> Result<(), ShellError> { let guard = { // We take a weakref to the plugin so that we don't create a cycle to the // RAII guard that will be stored on the plugin. let plugin = Arc::downgrade(&self); - handlers.register(Box::new(move || { + handlers.register(Box::new(move |action| { // write a Ctrl-C packet through the PluginInterface if the plugin is alive and // running if let Some(plugin) = plugin.upgrade() { if let Ok(mutable) = plugin.mutable.lock() { if let Some(ref running) = mutable.running { - let _ = running.interface.ctrlc(); + let _ = running.interface.signal(action); } } } @@ -325,7 +323,7 @@ impl RegisteredPlugin for PersistentPlugin { }; if let Ok(mut mutable) = self.mutable.lock() { - mutable.ctrlc_guard = Some(guard); + mutable.signal_guard = Some(guard); } Ok(()) diff --git a/crates/nu-plugin-protocol/src/lib.rs b/crates/nu-plugin-protocol/src/lib.rs index 1a2bbd99ff..7e607fdeab 100644 --- a/crates/nu-plugin-protocol/src/lib.rs +++ b/crates/nu-plugin-protocol/src/lib.rs @@ -23,7 +23,7 @@ pub mod test_util; use nu_protocol::{ ast::Operator, engine::Closure, ByteStreamType, Config, DeclId, LabeledError, PipelineData, - PluginMetadata, PluginSignature, ShellError, Span, Spanned, Value, + PluginMetadata, PluginSignature, ShellError, SignalAction, Span, Spanned, Value, }; use nu_utils::SharedCow; use serde::{Deserialize, Serialize}; @@ -208,8 +208,8 @@ pub enum PluginInput { Drop(StreamId), /// See [`StreamMessage::Ack`]. Ack(StreamId), - /// Signal a ctrlc event - Ctrlc, + /// Relay signals to the plugin + Signal(SignalAction), } impl TryFrom for StreamMessage { diff --git a/crates/nu-protocol/src/engine/engine_state.rs b/crates/nu-protocol/src/engine/engine_state.rs index 7c7cfb9bb5..d71b7aa170 100644 --- a/crates/nu-protocol/src/engine/engine_state.rs +++ b/crates/nu-protocol/src/engine/engine_state.rs @@ -2,14 +2,14 @@ use crate::{ ast::Block, debugger::{Debugger, NoopDebugger}, engine::{ - ctrlc, usage::{build_usage, Usage}, CachedFile, Command, CommandType, EnvVars, OverlayFrame, ScopeFrame, Stack, StateDelta, Variable, Visibility, DEFAULT_OVERLAY_NAME, }, eval_const::create_nu_constant, - BlockId, Category, Config, DeclId, FileId, GetSpan, HistoryConfig, Module, ModuleId, OverlayId, - ShellError, Signals, Signature, Span, SpanId, Type, Value, VarId, VirtualPathId, + BlockId, Category, Config, DeclId, FileId, GetSpan, Handlers, HistoryConfig, Module, ModuleId, + OverlayId, ShellError, SignalAction, Signals, Signature, Span, SpanId, Type, Value, VarId, + VirtualPathId, }; use fancy_regex::Regex; use lru::LruCache; @@ -86,8 +86,8 @@ pub struct EngineState { pub spans: Vec, usage: Usage, pub scope: ScopeFrame, - pub ctrlc_handlers: Option, signals: Signals, + pub signal_handlers: Option, pub env_vars: Arc, pub previous_env_vars: Arc>, pub config: Arc, @@ -147,7 +147,7 @@ impl EngineState { 0, false, ), - ctrlc_handlers: None, + signal_handlers: None, signals: Signals::empty(), env_vars: Arc::new( [(DEFAULT_OVERLAY_NAME.to_string(), HashMap::new())] @@ -186,7 +186,10 @@ impl EngineState { } pub fn reset_signals(&mut self) { - self.signals.reset() + self.signals.reset(); + if let Some(ref handlers) = self.signal_handlers { + handlers.run(SignalAction::Reset); + } } pub fn set_signals(&mut self, signals: Signals) { @@ -272,9 +275,9 @@ impl EngineState { #[cfg(feature = "plugin")] if !delta.plugins.is_empty() { for plugin in std::mem::take(&mut delta.plugins) { - // Connect plugins to the ctrlc handlers - if let Some(handlers) = &self.ctrlc_handlers { - plugin.clone().configure_ctrlc_handler(handlers)?; + // Connect plugins to the signal handlers + if let Some(handlers) = &self.signal_handlers { + plugin.clone().configure_signal_handler(handlers)?; } // Replace plugins that overlap in identity. diff --git a/crates/nu-protocol/src/engine/mod.rs b/crates/nu-protocol/src/engine/mod.rs index 24e4b15094..0cf45217ea 100644 --- a/crates/nu-protocol/src/engine/mod.rs +++ b/crates/nu-protocol/src/engine/mod.rs @@ -34,5 +34,3 @@ pub use stack_out_dest::*; pub use state_delta::*; pub use state_working_set::*; pub use variable::*; - -pub mod ctrlc; diff --git a/crates/nu-protocol/src/engine/ctrlc.rs b/crates/nu-protocol/src/pipeline/handlers.rs similarity index 89% rename from crates/nu-protocol/src/engine/ctrlc.rs rename to crates/nu-protocol/src/pipeline/handlers.rs index 11121f7c9c..5c979c5b02 100644 --- a/crates/nu-protocol/src/engine/ctrlc.rs +++ b/crates/nu-protocol/src/pipeline/handlers.rs @@ -1,10 +1,10 @@ use std::fmt::Debug; use std::sync::{Arc, Mutex}; -use crate::{engine::Sequence, ShellError}; +use crate::{engine::Sequence, ShellError, SignalAction}; /// Handler is a closure that can be sent across threads and shared. -pub type Handler = Box; +pub type Handler = Box; /// Manages a collection of handlers. #[derive(Clone)] @@ -23,16 +23,16 @@ impl Debug for Handlers { } } -/// Guard that unregisters a handler when dropped. +/// HandlerGuard that unregisters a handler when dropped. #[derive(Clone)] -pub struct Guard { +pub struct HandlerGuard { /// Unique ID of the handler. id: usize, /// Reference to the handlers list. handlers: Arc>>, } -impl Drop for Guard { +impl Drop for HandlerGuard { /// Drops the `Guard`, removing the associated handler from the list. fn drop(&mut self) { if let Ok(mut handlers) = self.handlers.lock() { @@ -41,7 +41,7 @@ impl Drop for Guard { } } -impl Debug for Guard { +impl Debug for HandlerGuard { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { f.debug_struct("Guard").field("id", &self.id).finish() } @@ -56,23 +56,23 @@ impl Handlers { /// Registers a new handler and returns an RAII guard which will unregister the handler when /// dropped. - pub fn register(&self, handler: Handler) -> Result { + pub fn register(&self, handler: Handler) -> Result { let id = self.next_id.next()?; if let Ok(mut handlers) = self.handlers.lock() { handlers.push((id, handler)); } - Ok(Guard { + Ok(HandlerGuard { id, handlers: Arc::clone(&self.handlers), }) } /// Runs all registered handlers. - pub fn run(&self) { + pub fn run(&self, action: SignalAction) { if let Ok(handlers) = self.handlers.lock() { for (_, handler) in handlers.iter() { - handler(); + handler(action); } } } diff --git a/crates/nu-protocol/src/pipeline/mod.rs b/crates/nu-protocol/src/pipeline/mod.rs index 525806425d..0d0931c396 100644 --- a/crates/nu-protocol/src/pipeline/mod.rs +++ b/crates/nu-protocol/src/pipeline/mod.rs @@ -1,4 +1,5 @@ pub mod byte_stream; +mod handlers; pub mod list_stream; mod metadata; mod out_dest; @@ -6,6 +7,7 @@ mod pipeline_data; mod signals; pub use byte_stream::*; +pub use handlers::*; pub use list_stream::*; pub use metadata::*; pub use out_dest::*; diff --git a/crates/nu-protocol/src/pipeline/signals.rs b/crates/nu-protocol/src/pipeline/signals.rs index 2057e27fe0..be3dffe4f0 100644 --- a/crates/nu-protocol/src/pipeline/signals.rs +++ b/crates/nu-protocol/src/pipeline/signals.rs @@ -4,6 +4,8 @@ use std::sync::{ Arc, }; +use serde::{Deserialize, Serialize}; + /// Used to check for signals to suspend or terminate the execution of Nushell code. /// /// For now, this struct only supports interruption (ctrl+c or SIGINT). @@ -81,3 +83,11 @@ impl Signals { } } } + +/// The types of things that can be signaled. Its anticipated this will change as we learn how we'd +/// like signals to be handled. +#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)] +pub enum SignalAction { + Interrupt, + Reset, +} diff --git a/crates/nu-protocol/src/plugin/registered.rs b/crates/nu-protocol/src/plugin/registered.rs index 8749c24af9..7f3906310a 100644 --- a/crates/nu-protocol/src/plugin/registered.rs +++ b/crates/nu-protocol/src/plugin/registered.rs @@ -1,6 +1,6 @@ use std::{any::Any, sync::Arc}; -use crate::{engine::ctrlc, PluginGcConfig, PluginIdentity, PluginMetadata, ShellError}; +use crate::{Handlers, PluginGcConfig, PluginIdentity, PluginMetadata, ShellError}; /// Trait for plugins registered in the [`EngineState`](crate::engine::EngineState). pub trait RegisteredPlugin: Send + Sync { @@ -36,10 +36,7 @@ pub trait RegisteredPlugin: Send + Sync { fn as_any(self: Arc) -> Arc; /// Give this plugin a chance to register for Ctrl-C signals. - fn configure_ctrlc_handler( - self: Arc, - _handler: &ctrlc::Handlers, - ) -> Result<(), ShellError> { + fn configure_signal_handler(self: Arc, _handler: &Handlers) -> Result<(), ShellError> { Ok(()) } } diff --git a/src/signals.rs b/src/signals.rs index 857f639843..ca22eef255 100644 --- a/src/signals.rs +++ b/src/signals.rs @@ -1,7 +1,4 @@ -use nu_protocol::{ - engine::{ctrlc::Handlers, EngineState}, - Signals, -}; +use nu_protocol::{engine::EngineState, Handlers, SignalAction, Signals}; use std::sync::{ atomic::{AtomicBool, Ordering}, Arc, @@ -11,12 +8,12 @@ pub(crate) fn ctrlc_protection(engine_state: &mut EngineState) { let interrupt = Arc::new(AtomicBool::new(false)); engine_state.set_signals(Signals::new(interrupt.clone())); - let ctrlc_handlers = Handlers::new(); - engine_state.ctrlc_handlers = Some(ctrlc_handlers.clone()); + let signal_handlers = Handlers::new(); + engine_state.signal_handlers = Some(signal_handlers.clone()); ctrlc::set_handler(move || { interrupt.store(true, Ordering::Relaxed); - ctrlc_handlers.run(); + signal_handlers.run(SignalAction::Interrupt); }) .expect("Error setting Ctrl-C handler"); }