forked from extern/nushell
78697bb8cf
related to - https://github.com/nushell/nushell/pull/9404 # Description to support our cratification effort and moving non-1.0 commands outside of the main focus, this PR - creates a new `nu-cmd-base` crate to hold the common structs, traits and functions used by all command-related crates - to start the transition, moves the `input_handler` module from `nu-command` to `nu-cmd-base` # User-Facing Changes ``` $nothing ``` # Tests + Formatting - 🟢 `toolkit fmt` - 🟢 `toolkit clippy` - ⚫ `toolkit test` - ⚫ `toolkit test stdlib` # After Submitting ``` $nothing ```
81 lines
2.4 KiB
Rust
81 lines
2.4 KiB
Rust
mod camel_case;
|
|
mod capitalize;
|
|
mod downcase;
|
|
mod kebab_case;
|
|
mod pascal_case;
|
|
mod screaming_snake_case;
|
|
mod snake_case;
|
|
mod str_;
|
|
mod title_case;
|
|
mod upcase;
|
|
|
|
pub use camel_case::SubCommand as StrCamelCase;
|
|
pub use capitalize::SubCommand as StrCapitalize;
|
|
pub use downcase::SubCommand as StrDowncase;
|
|
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;
|
|
pub use upcase::SubCommand as StrUpcase;
|
|
|
|
use nu_engine::CallExt;
|
|
|
|
use nu_cmd_base::input_handler::{operate as general_operate, CmdArgument};
|
|
use nu_protocol::ast::{Call, CellPath};
|
|
use nu_protocol::engine::{EngineState, Stack};
|
|
use nu_protocol::{PipelineData, ShellError, Span, Value};
|
|
|
|
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.ctrlc.clone())
|
|
}
|
|
|
|
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 {
|
|
val: case_operation(val),
|
|
span: head,
|
|
},
|
|
Value::Error { .. } => input.clone(),
|
|
_ => Value::Error {
|
|
error: Box::new(ShellError::OnlySupportsThisInputType {
|
|
exp_input_type: "string".into(),
|
|
wrong_type: input.get_type().to_string(),
|
|
dst_span: head,
|
|
src_span: input.expect_span(),
|
|
}),
|
|
},
|
|
}
|
|
}
|