nushell/src/commands/cp.rs

81 lines
2.2 KiB
Rust
Raw Normal View History

2019-07-22 04:23:02 +02:00
use crate::errors::ShellError;
use crate::parser::hir::SyntaxType;
use crate::parser::registry::{CommandConfig, NamedType, PositionalType};
use crate::prelude::*;
use indexmap::IndexMap;
2019-08-07 19:49:11 +02:00
use std::path::{Path, PathBuf};
2019-07-22 04:23:02 +02:00
pub struct Copycp;
impl Command for Copycp {
fn run(&self, args: CommandArgs) -> Result<OutputStream, ShellError> {
cp(args)
}
fn name(&self) -> &str {
"cp"
}
fn config(&self) -> CommandConfig {
let mut named: IndexMap<String, NamedType> = IndexMap::new();
named.insert("recursive".to_string(), NamedType::Switch);
CommandConfig {
name: self.name().to_string(),
positional: vec![PositionalType::mandatory("file", SyntaxType::Path)],
rest_positional: false,
named,
is_sink: false,
is_filter: false,
}
}
}
pub fn cp(args: CommandArgs) -> Result<OutputStream, ShellError> {
2019-08-07 19:49:11 +02:00
let mut source = PathBuf::from(args.shell_manager.path());
let mut destination = PathBuf::from(args.shell_manager.path());
2019-07-22 04:23:02 +02:00
let mut dst = String::new();
match args
.nth(0)
.ok_or_else(|| ShellError::string(&format!("No file or directory specified")))?
.as_string()?
.as_str()
{
file => {
2019-07-22 04:23:02 +02:00
source.push(file);
}
}
match args
.nth(1)
.ok_or_else(|| ShellError::string(&format!("No file or directory specified")))?
.as_string()?
.as_str()
{
file => {
2019-07-22 04:23:02 +02:00
dst.push_str(file);
destination.push(file);
}
2019-07-22 04:23:02 +02:00
}
if destination.is_dir() {
if source.is_file() {
let file_name = source.file_name().expect("");
let file_name = file_name.to_str().expect("");
destination.push(Path::new(file_name));
} else if source.is_dir() {
return Err(ShellError::string(&format!(
"{:?} is a directory (not copied)",
source.to_string_lossy()
)));
2019-07-22 04:23:02 +02:00
}
}
match std::fs::copy(source, destination) {
Err(_error) => Err(ShellError::string("can not copy file")),
Ok(_) => Ok(OutputStream::empty()),
}
}