mirror of https://git.sr.ht/~nixgoat/vento
79 lines
3.1 KiB
Rust
79 lines
3.1 KiB
Rust
/*
|
|
* Vento, a CLI inventory for your files.
|
|
* Copyright (C) 2022 Lux Aliaga
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
use super::common;
|
|
use anyhow::{bail, Context, Result};
|
|
use colored::Colorize;
|
|
use fs_extra::dir::{move_dir, CopyOptions};
|
|
use std::fs;
|
|
use std::path::{Path, PathBuf};
|
|
|
|
pub fn take(file: &String) -> Result<()> {
|
|
// Takes a file or directory
|
|
let active = &common::env_config()?[1];
|
|
|
|
let sourcepath: PathBuf = Path::new(&file).to_path_buf();
|
|
let destpath: PathBuf = [&active, &Path::new(file).to_path_buf()].iter().collect();
|
|
|
|
if Path::exists(&destpath) {
|
|
bail!(
|
|
"❌ {}",
|
|
format!("A file with the same name already exists in your inventory!").red()
|
|
);
|
|
} else if sourcepath.is_file() | sourcepath.is_symlink() {
|
|
fs::copy(&file, &destpath).expect("❌ Vento was unable to copy the file.");
|
|
fs::remove_file(&file).expect("❌ Vento was unable to remove the file.");
|
|
} else if sourcepath.is_dir() {
|
|
let options = CopyOptions::new();
|
|
move_dir(&file, &active, &options).expect("❌ Vento was unable to move the directory.");
|
|
} else {
|
|
println!("❌ {}", format!("No such file or directory.").red());
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
pub fn drop(file: &String, dest: PathBuf) -> Result<()> {
|
|
// Drops a file or directory
|
|
let active = &common::env_config()?[1];
|
|
|
|
let sourcepath: PathBuf = [&active, &Path::new(file).to_path_buf()].iter().collect();
|
|
let destpath: PathBuf = [
|
|
Path::new(&dest).to_path_buf(),
|
|
Path::new(file).to_path_buf(),
|
|
]
|
|
.iter()
|
|
.collect();
|
|
|
|
if Path::exists(&destpath) {
|
|
// HAHA YANDEREDEV MOMENT. This checks what method to use for the file/directory the user has picked
|
|
bail!("❌ {}", format!("A file with the same name already exists in the destination! Try renaming it or dropping this file somewhere else.").red());
|
|
} else if sourcepath.is_file() | sourcepath.is_symlink() {
|
|
fs::copy(&sourcepath, &destpath).context("❌ Vento was unable to copy the file.")?;
|
|
fs::remove_file(&sourcepath).context("❌ Vento was unable to remove the file.")?;
|
|
} else if sourcepath.is_dir() {
|
|
let destpath: PathBuf = Path::new(&dest).to_path_buf();
|
|
let options = CopyOptions::new();
|
|
move_dir(&sourcepath, &destpath, &options)
|
|
.expect("❌ Vento was unable to move the directory.");
|
|
} else {
|
|
bail!("❌ {}", format!("No such file or directory.").red());
|
|
}
|
|
Ok(())
|
|
}
|