nushell/crates/nu-command/src/filesystem/mkdir.rs

79 lines
2.2 KiB
Rust
Raw Normal View History

2021-10-08 00:20:23 +02:00
use std::collections::VecDeque;
use std::env::current_dir;
use nu_engine::CallExt;
use nu_protocol::ast::Call;
2021-10-25 18:58:58 +02:00
use nu_protocol::engine::{Command, EngineState, Stack};
2021-10-28 06:13:10 +02:00
use nu_protocol::{
IntoInterruptiblePipelineData, PipelineData, ShellError, Signature, SyntaxShape, Value,
};
2021-10-08 00:20:23 +02:00
2021-10-25 06:01:02 +02:00
#[derive(Clone)]
2021-10-08 00:20:23 +02:00
pub struct Mkdir;
impl Command for Mkdir {
fn name(&self) -> &str {
"mkdir"
}
fn signature(&self) -> Signature {
Signature::build("mkdir")
.rest(
"rest",
SyntaxShape::Filepath,
"the name(s) of the path(s) to create",
)
.switch("show-created-paths", "show the path(s) created.", Some('s'))
}
fn usage(&self) -> &str {
"Make directories, creates intermediary directories as required."
}
fn run(
&self,
2021-10-25 08:31:39 +02:00
engine_state: &EngineState,
stack: &mut Stack,
2021-10-08 00:20:23 +02:00
call: &Call,
2021-10-25 06:01:02 +02:00
_input: PipelineData,
) -> Result<PipelineData, ShellError> {
2021-10-08 00:20:23 +02:00
let path = current_dir()?;
let mut directories = call
2021-10-25 08:31:39 +02:00
.rest::<String>(engine_state, stack, 0)?
2021-10-08 00:20:23 +02:00
.into_iter()
.map(|dir| path.join(dir))
.peekable();
let show_created_paths = call.has_flag("show-created-paths");
let mut stream: VecDeque<Value> = VecDeque::new();
if directories.peek().is_none() {
return Err(ShellError::MissingParameter(
"requires directory paths".to_string(),
call.head,
));
}
for (i, dir) in directories.enumerate() {
let span = call.positional[i].span;
let dir_res = std::fs::create_dir_all(&dir);
if let Err(reason) = dir_res {
return Err(ShellError::CreateNotPossible(
format!("failed to create directory: {}", reason),
call.positional[i].span,
));
}
if show_created_paths {
let val = format!("{:}", dir.to_string_lossy());
stream.push_back(Value::String { val, span });
}
}
2021-10-28 06:13:10 +02:00
Ok(stream
.into_iter()
.into_pipeline_data(engine_state.ctrlc.clone()))
2021-10-08 00:20:23 +02:00
}
}