399a7c8836
# Description This PR introduces a new `Signals` struct to replace our adhoc passing around of `ctrlc: Option<Arc<AtomicBool>>`. Doing so has a few benefits: - We can better enforce when/where resetting or triggering an interrupt is allowed. - Consolidates `nu_utils::ctrl_c::was_pressed` and other ad-hoc re-implementations into a single place: `Signals::check`. - This allows us to add other types of signals later if we want. E.g., exiting or suspension. - Similarly, we can more easily change the underlying implementation if we need to in the future. - Places that used to have a `ctrlc` of `None` now use `Signals::empty()`, so we can double check these usages for correctness in the future.
69 lines
2.0 KiB
Rust
69 lines
2.0 KiB
Rust
mod camel_case;
|
|
mod kebab_case;
|
|
mod pascal_case;
|
|
mod screaming_snake_case;
|
|
mod snake_case;
|
|
mod str_;
|
|
mod title_case;
|
|
|
|
pub use camel_case::SubCommand as StrCamelCase;
|
|
pub use kebab_case::SubCommand as StrKebabCase;
|
|
pub use pascal_case::SubCommand as StrPascalCase;
|
|
pub use screaming_snake_case::SubCommand as StrScreamingSnakeCase;
|
|
pub use snake_case::SubCommand as StrSnakeCase;
|
|
pub use str_::Str;
|
|
pub use title_case::SubCommand as StrTitleCase;
|
|
|
|
use nu_cmd_base::input_handler::{operate as general_operate, CmdArgument};
|
|
use nu_engine::command_prelude::*;
|
|
|
|
struct Arguments<F: Fn(&str) -> String + Send + Sync + 'static> {
|
|
case_operation: &'static F,
|
|
cell_paths: Option<Vec<CellPath>>,
|
|
}
|
|
|
|
impl<F: Fn(&str) -> String + Send + Sync + 'static> CmdArgument for Arguments<F> {
|
|
fn take_cell_paths(&mut self) -> Option<Vec<CellPath>> {
|
|
self.cell_paths.take()
|
|
}
|
|
}
|
|
|
|
pub fn operate<F>(
|
|
engine_state: &EngineState,
|
|
stack: &mut Stack,
|
|
call: &Call,
|
|
input: PipelineData,
|
|
case_operation: &'static F,
|
|
) -> Result<PipelineData, ShellError>
|
|
where
|
|
F: Fn(&str) -> String + Send + Sync + 'static,
|
|
{
|
|
let cell_paths: Vec<CellPath> = call.rest(engine_state, stack, 0)?;
|
|
let cell_paths = (!cell_paths.is_empty()).then_some(cell_paths);
|
|
let args = Arguments {
|
|
case_operation,
|
|
cell_paths,
|
|
};
|
|
general_operate(action, args, input, call.head, engine_state.signals())
|
|
}
|
|
|
|
fn action<F>(input: &Value, args: &Arguments<F>, head: Span) -> Value
|
|
where
|
|
F: Fn(&str) -> String + Send + Sync + 'static,
|
|
{
|
|
let case_operation = args.case_operation;
|
|
match input {
|
|
Value::String { val, .. } => Value::string(case_operation(val), head),
|
|
Value::Error { .. } => input.clone(),
|
|
_ => Value::error(
|
|
ShellError::OnlySupportsThisInputType {
|
|
exp_input_type: "string".into(),
|
|
wrong_type: input.get_type().to_string(),
|
|
dst_span: head,
|
|
src_span: input.span(),
|
|
},
|
|
head,
|
|
),
|
|
}
|
|
}
|